Skip to content

Commit 332eeab

Browse files
committed
feat: add base_url to ChatOllama to read from remote server
1 parent 319a6c8 commit 332eeab

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/utils/utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,7 @@ def get_llm_model(provider: str, **kwargs):
8989
model=kwargs.get("model_name", "qwen2.5:7b"),
9090
temperature=kwargs.get("temperature", 0.0),
9191
num_ctx=128000,
92+
base_url=kwargs.get("base_url", "http://localhost:11434"),
9293
)
9394
elif provider == "azure_openai":
9495
if not kwargs.get("base_url", ""):

0 commit comments

Comments
 (0)