From 37fa318258c02da33df120991e7fdbdd4337cef8 Mon Sep 17 00:00:00 2001
From: Ljzd_PRO <mcdhacg@foxmail.com>
Date: Wed, 20 Aug 2025 10:25:33 +0800
Subject: [PATCH] fix: update invoke_embedding to return only embeddings from
 client.embed (#1619)

---
 pkg/provider/modelmgr/requesters/ollamachat.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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