Update app.py

app.py

@@ -85,6 +85,10 @@ def method_timer(method):
 class InferenceEndpointTextEmbedder:
     @component.output_types(embedding=List[float])
     def run(self, text: str):
+        return self.request(text)
+
+    @method_timer
+    def request(self, text: str):
         payload = {"text": text, "inputs": ""}
         response = post(EMBEDDER_URL, payload)
 
@@ -98,6 +102,10 @@ class InferenceEndpointTextEmbedder:
 class InferenceEndpointDocumentEmbedder:
     @component.output_types(documents=List[Document])
     def run(self, documents: List[Document]):
+        return self.request(documents)
+
+    @method_timer
+    def request(self, documents: List[Document]):
         documents = [d.to_dict() for d in documents]
 
         payload = {"documents": documents, "inputs": ""}
@@ -116,6 +124,10 @@ class InferenceEndpointRanker:
 
     @component.output_types(documents=List[Document])
     def run(self, query: str, documents: List[Document]):
+        return self.request(query, documents)
+
+    @method_timer
+    def request(self, query: str, documents: List[Document]):
         documents = [d.to_dict() for d in documents]
 
         payload = {
@@ -145,9 +157,8 @@ if os.path.exists("data/qdrant"):
     except Exception:
         shutil.rmtree("data/qdrant", ignore_errors=True)
 
-if document_store is None:
-    shutil.rmtree("data/qdrant", ignore_errors=True)
 
+if document_store is None:
     document_store = QdrantDocumentStore(
         path="./data/qdrant",
         return_embedding=True,