opensource/Agent-S
3
0
Fork 0
mirror of https://github.com/simular-ai/Agent-S.git synced 2026-08-29 02:03:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: restore OLLAMA_HOST env var support

Browse Source
This commit is contained in:
majiayu000
2026-01-13 23:16:29 +08:00
parent 02a9da5591
commit 234450bb0c
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
gui_agents/s3/core/mllm.py
+8 -1
View File
@@ -39,7 +39,14 @@ class LMMAgent:
elif engine_type == "ollama":
# Reuse LMMEngineOpenAI for Ollama, defaulting to localhost if not specified
if "base_url" not in engine_params:
engine_params["base_url"] = "http://localhost:11434/v1"
import os
base_url = os.getenv("OLLAMA_HOST")
if base_url:
if not base_url.endswith("/v1"):
base_url = base_url.rstrip("/") + "/v1"
engine_params["base_url"] = base_url
else:
engine_params["base_url"] = "http://localhost:11434/v1"
if "api_key" not in engine_params:
engine_params["api_key"] = "ollama"
self.engine = LMMEngineOpenAI(**engine_params)