Skip to content

Commit 03d5db0

Browse files
authored
Merge pull request #21 from NousResearch/ollama-base-url-1
add `ollama_base_url` setting
2 parents 9ade305 + 66a88e6 commit 03d5db0

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/funcchain/backend/settings.py

+5-2
Original file line numberDiff line numberDiff line change
@@ -25,11 +25,12 @@ class FuncchainSettings(BaseSettings):
2525
retry_parse: int = 3
2626
retry_parse_sleep: float = 0.1
2727

28-
# KEYS
28+
# KEYS / URLS
2929
openai_api_key: Optional[str] = None
3030
azure_api_key: Optional[str] = None
3131
anthropic_api_key: Optional[str] = None
3232
google_api_key: Optional[str] = None
33+
ollama_base_url: str = "http://localhost:11434"
3334

3435
# MODEL KWARGS
3536
verbose: bool = False
@@ -58,7 +59,9 @@ def openai_kwargs(self) -> dict:
5859
}
5960

6061
def ollama_kwargs(self) -> dict:
61-
return {}
62+
return {
63+
"base_url": self.ollama_base_url
64+
}
6265

6366
def llamacpp_kwargs(self) -> dict:
6467
return {

0 commit comments

Comments
 (0)