diff --git a/pkg/provider/modelmgr/requesters/ollamachat.py b/pkg/provider/modelmgr/requesters/ollamachat.py index 42203650..e993cab2 100644 --- 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