xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-07-18 02:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: update invoke_embedding to return only embeddings from client.embed (#1619)

Browse Source
This commit is contained in:
Ljzd_PRO
2025-08-20 10:25:33 +08:00
committed by GitHub
parent ff7bebb782
commit 37fa318258
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/provider/modelmgr/requesters/ollamachat.py
+2 -2
View File
@@ -139,8 +139,8 @@ class OllamaChatCompletions(requester.ProviderAPIRequester):
input_text: list[str], input_text: list[str],
extra_args: dict[str, typing.Any] = {}, extra_args: dict[str, typing.Any] = {},
) -> list[list[float]]: ) -> list[list[float]]:
return await self.client.embed( return (await self.client.embed(
model=model.model_entity.name, model=model.model_entity.name,
input=input_text, input=input_text,
**extra_args, **extra_args,
) )).embeddings