From 014eca879581b47e815ec0ba9cd54ff7729dd547 Mon Sep 17 00:00:00 2001
From: Salman Paracha <salmanparacha@MacBook-Pro-288.local>
Date: Sun, 16 Nov 2025 20:44:41 -0800
Subject: [PATCH] if the user sets arch-function base_url we should honor it

---
 arch/tools/cli/config_generator.py | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/arch/tools/cli/config_generator.py b/arch/tools/cli/config_generator.py
index 6d63f1ed..fd306c60 100644
--- a/arch/tools/cli/config_generator.py
+++ b/arch/tools/cli/config_generator.py
@@ -13,10 +13,10 @@ SUPPORTED_PROVIDERS_WITH_BASE_URL = [
     "ollama",
     "qwen",
     "amazon_bedrock",
+    "arch",
 ]
 
 SUPPORTED_PROVIDERS_WITHOUT_BASE_URL = [
-    "arch",
     "deepseek",
     "groq",
     "mistral",
@@ -304,14 +304,16 @@ def validate_and_render_schema():
                 }
             )
 
-    # Always add arch-function model provider
-    updated_model_providers.append(
-        {
-            "name": "arch-function",
-            "provider_interface": "arch",
-            "model": "Arch-Function",
-        }
-    )
+    # Always add arch-function model provider if not already defined
+    if "arch-function" not in model_provider_name_set:
+        updated_model_providers.append(
+            {
+                "name": "arch-function",
+                "provider_interface": "arch",
+                "model": "Arch-Function",
+            }
+        )
+
     config_yaml["model_providers"] = deepcopy(updated_model_providers)
 
     listeners_with_provider = 0