alihan/browser-use-web-ui
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #230 from carl-shen/xs_fix_ollama_base_url

Browse Source 
fix: ollama provider not respecting OLLAMA_ENDPOINT env var
This commit is contained in:
warmshao
2025-02-04 20:03:12 +08:00
committed by GitHub
parent 6e90106766 f96d83b4c9
commit c4b0da5c8f
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/utils/utils.py
View File

@@ -106,13 +106,13 @@ def get_llm_model(provider: str, **kwargs):
base_url = os.getenv("OLLAMA_ENDPOINT", "http://localhost:11434")
else:
base_url = kwargs.get("base_url")
if "deepseek-r1" in kwargs.get("model_name", "qwen2.5:7b"):
return DeepSeekR1ChatOllama(
model=kwargs.get("model_name", "deepseek-r1:14b"),
temperature=kwargs.get("temperature", 0.0),
num_ctx=kwargs.get("num_ctx", 32000),
base_url=kwargs.get("base_url", base_url),
base_url=base_url,
)
else:
return ChatOllama(
@@ -120,7 +120,7 @@ def get_llm_model(provider: str, **kwargs):
temperature=kwargs.get("temperature", 0.0),
num_ctx=kwargs.get("num_ctx", 32000),
num_predict=kwargs.get("num_predict", 1024),
base_url=kwargs.get("base_url", base_url),
base_url=base_url,
)
elif provider == "azure_openai":
if not kwargs.get("base_url", ""):