Merge pull request #570 from valentimarco/develop

Fix Fastembed bug
cheshire-cat-ai · Nov 15, 2023 · 11b86cf · 11b86cf
2 parents bb44da4 + c5c9356
commit 11b86cf
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/core/cat/factory/custom_embedder.py b/core/cat/factory/custom_embedder.py
@@ -75,14 +75,12 @@ def __init__(self, url, model,max_length) -> None:
 
 
     def embed_documents(self, texts: List[str]):
-        payload = json.dumps({"document": texts})
-        ret = httpx.post(f"{self.url}/embeddings/document", data=payload, timeout=None)
+        ret = httpx.post(f"{self.url}/embeddings/document", json={"document": texts}, timeout=None)
         ret.raise_for_status()
         return ret.json()
 
     def embed_query(self, text: str) -> List[float]:
-        payload = json.dumps({"prompt": text})
-        ret = httpx.post(f"{self.url}/embeddings/prompt", data=payload, timeout=None)
+        ret = httpx.post(f"{self.url}/embeddings/prompt", json={"prompt": text}, timeout=None)
         ret.raise_for_status()
         return ret.json()