From 7aed5cf1ed94830e738181f376b03fa0890b0d88 Mon Sep 17 00:00:00 2001
From: Junyan Qin <rockchinq@gmail.com>
Date: Thu, 24 Jul 2025 10:36:32 +0800
Subject: [PATCH] feat: ollama embeddings models

---
 pkg/provider/modelmgr/requesters/ollamachat.py   | 12 ++++++++++++
 pkg/provider/modelmgr/requesters/ollamachat.yaml |  1 +
 2 files changed, 13 insertions(+)

diff --git a/pkg/provider/modelmgr/requesters/ollamachat.py b/pkg/provider/modelmgr/requesters/ollamachat.py
index 1456515f..9e6f5a77 100644
--- a/pkg/provider/modelmgr/requesters/ollamachat.py
+++ b/pkg/provider/modelmgr/requesters/ollamachat.py
@@ -129,3 +129,15 @@ class OllamaChatCompletions(requester.ProviderAPIRequester):
             )
         except asyncio.TimeoutError:
             raise errors.RequesterError('请求超时')
+
+    async def invoke_embedding(
+        self,
+        model: requester.RuntimeEmbeddingModel,
+        input_text: list[str],
+        extra_args: dict[str, typing.Any] = {},
+    ) -> list[list[float]]:
+        return await self.client.embed(
+            model=model.model_entity.name,
+            input=input_text,
+            **extra_args,
+        )
diff --git a/pkg/provider/modelmgr/requesters/ollamachat.yaml b/pkg/provider/modelmgr/requesters/ollamachat.yaml
index f4c4bf5a..f7cdeeba 100644
--- a/pkg/provider/modelmgr/requesters/ollamachat.yaml
+++ b/pkg/provider/modelmgr/requesters/ollamachat.yaml
@@ -24,6 +24,7 @@ spec:
       default: 120
   support_type:
     - llm
+    - text-embedding
 execution:
   python:
     path: ./ollamachat.py