gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-28 16:07:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: return 1D embedding if create_embedding receive a string input (#670)

Browse Source 
* return single dimension embedding for string input of create_embedding

* Update base.py

* fix

* fix

* fix

* fix

---------

Co-authored-by: Xu <v-xuminrui@microsoft.com>
This commit is contained in:
Roland Minrui
2025-03-12 16:10:33 +08:00
committed by GitHub
parent 20778a9b87
commit cf86d8feeb
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/oai/backend/base.py
+3 -1
View File
@@ -273,7 +273,7 @@ class APIBackend(ABC):
logger.log_object({"system": system_prompt, "user": user_prompt, "resp": resp}, tag="debug_llm")
return resp
def create_embedding(self, input_content: str | list[str], *args, **kwargs) -> list[list[float]]: # type: ignore[no-untyped-def]
def create_embedding(self, input_content: str | list[str], *args, **kwargs) -> list[float] | list[list[float]]: # type: ignore[no-untyped-def]
input_content_list = [input_content] if isinstance(input_content, str) else input_content
resp = self._try_create_chat_completion_or_embedding( # type: ignore[misc]
input_content_list=input_content_list,
@@ -281,6 +281,8 @@ class APIBackend(ABC):
*args,
**kwargs,
)
if isinstance(input_content, str):
return resp[0] # type: ignore[return-value]
return resp # type: ignore[return-value]
def build_messages_and_calculate_token(