update embedding endpoint for no wrapper.

opea-project · Nov 3, 2024 · 466f965 · 466f965
1 parent ccfd127
commit 466f965
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/evals/evaluation/rag_eval/evaluator.py b/evals/evaluation/rag_eval/evaluator.py
@@ -155,11 +155,14 @@ def post_process(self, result):
 
     def get_retrieved_documents(self, data, arguments):
         query = self.get_query(data)
-        data = {"text": query}
+        data = {"inputs": query}
         headers = {"Content-Type": "application/json"}
-        response = requests.post(arguments.embedding_endpoint, data=json.dumps(data), headers=headers)
+        response = requests.post(arguments.tei_embedding_endpoint + "/embed",
+            data=json.dumps(data),
+            headers=headers
+        )
         if response.ok:
-            embedding = response.json()["embedding"]
+            embedding = response.json()[0]
         else:
             print(f"Request for embedding failed due to {response.text}.")
             return []

diff --git a/evals/evaluation/rag_eval/examples/eval_multihop.py b/evals/evaluation/rag_eval/examples/eval_multihop.py
@@ -42,11 +42,14 @@ def get_reranked_documents(self, query, docs, arguments):
             return []
 
     def get_retrieved_documents(self, query, arguments):
-        data = {"text": query}
+        data = {"inputs": query}
         headers = {"Content-Type": "application/json"}
-        response = requests.post(arguments.embedding_endpoint, data=json.dumps(data), headers=headers)
+        response = requests.post(arguments.tei_embedding_endpoint + "/embed",
+            data=json.dumps(data),
+            headers=headers
+        )
         if response.ok:
-            embedding = response.json()["embedding"]
+            embedding = response.json()[0]
         else:
             print(f"Request for embedding failed due to {response.text}.")
             return []