if the user sets arch-function base_url we should honor it

This commit is contained in:
Salman Paracha 2025-11-16 20:44:41 -08:00
parent d253a29d38
commit 014eca8795

View file

@ -13,10 +13,10 @@ SUPPORTED_PROVIDERS_WITH_BASE_URL = [
"ollama", "ollama",
"qwen", "qwen",
"amazon_bedrock", "amazon_bedrock",
"arch",
] ]
SUPPORTED_PROVIDERS_WITHOUT_BASE_URL = [ SUPPORTED_PROVIDERS_WITHOUT_BASE_URL = [
"arch",
"deepseek", "deepseek",
"groq", "groq",
"mistral", "mistral",
@ -304,14 +304,16 @@ def validate_and_render_schema():
} }
) )
# Always add arch-function model provider # Always add arch-function model provider if not already defined
updated_model_providers.append( if "arch-function" not in model_provider_name_set:
{ updated_model_providers.append(
"name": "arch-function", {
"provider_interface": "arch", "name": "arch-function",
"model": "Arch-Function", "provider_interface": "arch",
} "model": "Arch-Function",
) }
)
config_yaml["model_providers"] = deepcopy(updated_model_providers) config_yaml["model_providers"] = deepcopy(updated_model_providers)
listeners_with_provider = 0 listeners_with_provider = 0