This commit is contained in:
yujj128
2025-11-06 16:32:32 +08:00
parent a7043e4ec3
commit 30bb1a6839

View File

@@ -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