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

2026-06-02 03:55:55 +00:00 · 2025-08-20 10:25:33 +08:00
parent ff7bebb782
commit 37fa318258
1 changed files with 2 additions and 2 deletions
--- a/pkg/provider/modelmgr/requesters/ollamachat.py
+++ b/pkg/provider/modelmgr/requesters/ollamachat.py
@@ -139,8 +139,8 @@ class OllamaChatCompletions(requester.ProviderAPIRequester):
        input_text: list[str],
        extra_args: dict[str, typing.Any] = {},
    ) -> list[list[float]]:
-        return await self.client.embed(
+        return (await self.client.embed(
            model=model.model_entity.name,
            input=input_text,
            **extra_args,
-        )
+        )).embeddings