From 30bb1a68396ad34bba5b5e3df29b9acec1406d05 Mon Sep 17 00:00:00 2001 From: yujj128 Date: Thu, 6 Nov 2025 16:32:32 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- service/cus_vanna_srevice.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/service/cus_vanna_srevice.py b/service/cus_vanna_srevice.py index 2da472d..7b5a218 100644 --- a/service/cus_vanna_srevice.py +++ b/service/cus_vanna_srevice.py @@ -489,20 +489,20 @@ class CustomQdrant_VectorStore(Qdrant_VectorStore): return str(e) return "Unknown error" - # request_body = { - # "model": self.embedding_model_name, - # "encoding_format": "float", - # "input": [data], - # } request_body = { "model": self.embedding_model_name, "encoding_format": "float", - "input": [{"type":"text","text":data}], + "input": [data], } + # request_body = { + # "model": self.embedding_model_name, + # "encoding_format": "float", + # "input": [{"type":"text","text":data}], + # } request_body.update(kwargs) response = requests.post( - url=f"{self.embedding_api_base}", + url=f"{self.embedding_api_base}/v1/embeddings", json=request_body, headers={"Authorization": f"Bearer {self.embedding_api_key}", 'Content-Type': 'application/json'}, ) @@ -512,7 +512,7 @@ class CustomQdrant_VectorStore(Qdrant_VectorStore): ) result = response.json() # embeddings = result['embeddings'] - embeddings = result['data']['embedding'] + embeddings = result['data'][0]['embedding'] # return embeddings[0] return embeddings