From dc5b1f79aaf1cae3c7faa5da97c6858fac10ebdd Mon Sep 17 00:00:00 2001
From: utopia2077 <78017255+utopia2077@users.noreply.github.com>
Date: Wed, 12 Mar 2025 10:54:39 +0800
Subject: [PATCH] fix: aembed_query type error (#2441)

---
 .../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 5eda65912..cac009a1b 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 @@ 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):