diff --git a/embedding_adapters.py b/embedding_adapters.py index 4304644..8e9cf6f 100644 --- a/embedding_adapters.py +++ b/embedding_adapters.py @@ -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, diff --git a/main.spec b/main.spec index 50a88dd..5935660 100644 --- a/main.spec +++ b/main.spec @@ -7,6 +7,8 @@ hiddenimports = ['typing_extensions', 'langchain-openai', 'langgraph', 'openai', + 'google-genai', + 'google', 'nltk', 'sentence_transformers', 'scikit-learn',