From 0fcecc589d3f66889ac26399fdce1438b8608b82 Mon Sep 17 00:00:00 2001 From: utopia2077 Date: Tue, 11 Mar 2025 23:17:21 +0800 Subject: [PATCH] fix: aembed_query type error --- .../src/dbgpt/model/cluster/embedding/remote_embedding.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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):