mirror of
https://github.com/katanemo/plano.git
synced 2026-07-02 15:51:02 +02:00
if the user sets arch-function base_url we should honor it
This commit is contained in:
parent
d253a29d38
commit
014eca8795
1 changed files with 11 additions and 9 deletions
|
|
@ -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
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue