apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

fixed docs and added ollama as a first-class LLM provider

Browse source
This commit is contained in:
Salman Paracha 2025-09-18 23:10:21 -07:00
parent 8d0b468345
commit f7c9d04da9
16 changed files with 1612 additions and 149 deletions
Download patch file Download diff file Expand all files Collapse all files

9
arch/tools/cli/config_generator.py
View file

@ -17,6 +17,7 @@ SUPPORTED_PROVIDERS = [
"together_ai",
"azure_openai",
"xai",
"ollama",
]
@ -124,10 +125,12 @@ def validate_and_render_schema():
f"Invalid model name {model_name}. Please provide model name in the format <provider>/<model_id>."
)
provider = model_name_tokens[0]
# Validate azure_openai provider requires base_url
if provider == "azure_openai" and llm_provider.get("base_url") is None:
# Validate azure_openai and ollama provider requires base_url
if (provider == "azure_openai" or provider == "ollama") and llm_provider.get(
"base_url"
) is None:
raise Exception(
f"Provider 'azure_openai' requires 'base_url' to be set for model {model_name}"
f"Provider '{provider}' requires 'base_url' to be set for model {model_name}"
)
model_id = "/".join(model_name_tokens[1:])