修复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
"""
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,
+2
View File
@@ -7,6 +7,8 @@ hiddenimports = ['typing_extensions',
'langchain-openai',
'langgraph',
'openai',
'google-genai',
'google',
'nltk',
'sentence_transformers',
'scikit-learn',