diff --git a/packages/dbgpt-core/src/dbgpt/model/cluster/embedding/remote_embedding.py b/packages/dbgpt-core/src/dbgpt/model/cluster/embedding/remote_embedding.py index 5eda65912d..cac009a1b5 100644 --- a/packages/dbgpt-core/src/dbgpt/model/cluster/embedding/remote_embedding.py +++ b/packages/dbgpt-core/src/dbgpt/model/cluster/embedding/remote_embedding.py @@ -26,7 +26,8 @@ async def aembed_documents(self, texts: List[str]) -> List[List[float]]: 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):