修复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
+7 -2
View File
@@ -95,8 +95,13 @@ 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:
if "/api" in url:
url = f"{url}/embeddings"
else:
if "/v1" in url:
url = url[:url.index("/v1")]
url = f"{url}/api/embeddings" url = f"{url}/api/embeddings"
data = { data = {
+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',