github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-23 12:21:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: aembed_query type error (#2441)

Browse Source
This commit is contained in:
utopia2077 2025-03-12 10:54:39 +08:00 committed by GitHub
parent 150f33bd26
commit dc5b1f79aa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
packages/dbgpt-core/src/dbgpt/model/cluster/embedding/remote_embedding.py
View File

@ -26,7 +26,8 @@ class RemoteEmbeddings(Embeddings):
async def aembed_query(self, text: str) -> List[float]:
"""Asynchronous Embed query text."""
return await self.aembed_documents([text])[0]
result = await self.aembed_documents([text])
return result[0]
class RemoteRerankEmbeddings(RerankEmbeddings):