fixed #82

修复ollama的格式检查
2025-02-11 16:16:21 +08:00
parent 7d401c6ee6
commit d54871b2c2
2 changed files with 10 additions and 3 deletions
@@ -95,9 +95,14 @@ class OllamaEmbeddingAdapter(BaseEmbeddingAdapter):
        """
        调用 Ollama 本地服务 /api/embeddings 接口，获取文本 embedding
        """
-        url = self.base_url
-        if "api/embeddings" not in url:
-            url = f"{url}/api/embeddings"
+        url = self.base_url.rstrip("/")
+        if "/api/embeddings" not in url:
+            if "/api" in url:
+                url = f"{url}/embeddings"
+            else:
+                if "/v1" in url:
+                    url = url[:url.index("/v1")]
+                url = f"{url}/api/embeddings"

        data = {
            "model": self.model_name,