From a2dabbffac264a8be00f0b0d5012c79a7670ae10 Mon Sep 17 00:00:00 2001 From: Philipp Heyken Soares Date: Tue, 18 Nov 2025 17:08:16 +0100 Subject: [PATCH] fix cohere default url setting to None --- rag/llm/rerank_model.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/rag/llm/rerank_model.py b/rag/llm/rerank_model.py index f295266b4..2ae626642 100644 --- a/rag/llm/rerank_model.py +++ b/rag/llm/rerank_model.py @@ -234,7 +234,11 @@ class CoHereRerank(Base): def __init__(self, key, model_name, base_url=None): from cohere import Client - self.client = Client(api_key=key, base_url=base_url) + # Only pass base_url if it's a non-empty string, otherwise use default Cohere API endpoint + client_kwargs = {"api_key": key} + if base_url and base_url.strip(): + client_kwargs["base_url"] = base_url + self.client = Client(**client_kwargs) self.model_name = model_name.split("___")[0] def similarity(self, query: str, texts: list):