修复ollama的格式检查
This commit is contained in:
YILING0013
2025-02-11 16:16:21 +08:00
parent 7d401c6ee6
commit d54871b2c2
2 changed files with 10 additions and 3 deletions
+8 -3
View File
@@ -95,9 +95,14 @@ class OllamaEmbeddingAdapter(BaseEmbeddingAdapter):
""" """
调用 Ollama 本地服务 /api/embeddings 接口,获取文本 embedding 调用 Ollama 本地服务 /api/embeddings 接口,获取文本 embedding
""" """
url = self.base_url url = self.base_url.rstrip("/")
if "api/embeddings" not in url: if "/api/embeddings" not in url:
url = f"{url}/api/embeddings" if "/api" in url:
url = f"{url}/embeddings"
else:
if "/v1" in url:
url = url[:url.index("/v1")]
url = f"{url}/api/embeddings"
data = { data = {
"model": self.model_name, "model": self.model_name,
+2
View File
@@ -7,6 +7,8 @@ hiddenimports = ['typing_extensions',
'langchain-openai', 'langchain-openai',
'langgraph', 'langgraph',
'openai', 'openai',
'google-genai',
'google',
'nltk', 'nltk',
'sentence_transformers', 'sentence_transformers',
'scikit-learn', 'scikit-learn',