zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-17 13:17:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(embedding): do not append /v1 to Ollama native embedding base_url (#16974)

Browse Source
This commit is contained in:
iwasaki
2026-07-16 14:21:55 +09:00
committed by GitHub
parent d2f4304ff4
commit d577b49a00
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
rag/llm/embedding_model.py
View File

@@ -469,7 +469,10 @@ class OllamaEmbed(Base):
_special_tokens = ["<|endoftext|>"]
def __init__(self, key, model_name, **kwargs):
self.base_url = ensure_v1(kwargs["base_url"])
# Native ollama.Client builds paths like "/api/embed" directly off this
# host; unlike the OpenAI-compatible providers below, it must NOT go
# through ensure_v1 (which appends "/v1") or every request 404s.
self.base_url = kwargs["base_url"].rstrip("/")
self.client = Client(host=self.base_url) if not key or key == "x" else Client(host=self.base_url, headers={"Authorization": f"Bearer {key}"})
self.model_name = model_name
self.keep_alive = kwargs.get("ollama_keep_alive", int(os.environ.get("OLLAMA_KEEP_ALIVE", -1)))