tunacode-cli 0.1.21__py3-none-any.whl → 0.1.24__py3-none-any.whl

tunacode/constants.py

@@ -11,7 +11,7 @@ KB = 1024
 MB = KB * 1024
 
 APP_NAME = "TunaCode"
-APP_VERSION = "0.1.21"
+APP_VERSION = "0.1.24"
 
 
 GUIDE_FILE_NAME = "AGENTS.md"

tunacode/core/agents/agent_components/agent_config.py

@@ -15,6 +15,7 @@ from pydantic_ai.providers.openai import OpenAIProvider
 from pydantic_ai.retries import AsyncTenacityTransport, RetryConfig, wait_retry_after
 from tenacity import retry_if_exception_type, stop_after_attempt
 
+from tunacode.configuration.models import load_models_registry
 from tunacode.constants import ENV_OPENAI_BASE_URL, SETTINGS_BASE_URL, UI_THINKING_MESSAGE
 from tunacode.core.agents.delegation_tools import create_research_codebase_tool
 from tunacode.core.prompting import (
@@ -265,6 +266,22 @@ def load_tunacode_context() -> str:
         return ""
 
 
+def _get_provider_config_from_registry(provider_name: str) -> dict:
+    """Get provider config from models registry.
+
+    Returns dict with 'api' (base_url) and 'env' (list of env var names).
+    Returns empty dict if provider not found.
+    """
+    registry = load_models_registry()
+    provider_data = registry.get(provider_name, {})
+    return {
+        "api": provider_data.get("api"),
+        "env": provider_data.get("env", []),
+    }
+
+
+
+
 def _create_model_with_retry(
     model_string: str, http_client: AsyncClient, state_manager: StateManager
 ):
@@ -272,69 +289,51 @@ def _create_model_with_retry(
 
     Parses model string in format 'provider:model_name' and creates
     appropriate provider and model instances with the retry-enabled HTTP client.
+
+    Uses models_registry.json for provider configuration (api URL, env vars).
+    Only 'anthropic' provider uses AnthropicModel, all others use OpenAI.
     """
-    # Extract environment config
     env = state_manager.session.user_config.get("env", {})
-
     settings = state_manager.session.user_config.get("settings", {})
+
+    # User overrides for base_url
     env_base_url_raw = env.get(ENV_OPENAI_BASE_URL)
     settings_base_url_raw = settings.get(SETTINGS_BASE_URL)
     env_base_url = _coerce_optional_str(env_base_url_raw, ENV_OPENAI_BASE_URL)
     settings_base_url = _coerce_optional_str(settings_base_url_raw, SETTINGS_BASE_URL)
     base_url_override = _resolve_base_url_override(env_base_url, settings_base_url)
 
-    # Provider configuration: API key names and base URLs
-    PROVIDER_CONFIG = {
-        "anthropic": {"api_key_name": "ANTHROPIC_API_KEY", "base_url": None},
-        "openai": {"api_key_name": "OPENAI_API_KEY", "base_url": None},
-        "openrouter": {
-            "api_key_name": "OPENROUTER_API_KEY",
-            "base_url": "https://openrouter.ai/api/v1",
-        },
-        "azure": {
-            "api_key_name": "AZURE_OPENAI_API_KEY",
-            "base_url": env.get("AZURE_OPENAI_ENDPOINT"),
-        },
-        "deepseek": {"api_key_name": "DEEPSEEK_API_KEY", "base_url": None},
-        "cerebras": {
-            "api_key_name": "CEREBRAS_API_KEY",
-            "base_url": "https://api.cerebras.ai/v1",
-        },
-    }
-
     # Parse model string
     if ":" in model_string:
         provider_name, model_name = model_string.split(":", 1)
     else:
-        # Auto-detect provider from model name
         model_name = model_string
         if model_name.startswith("claude"):
             provider_name = "anthropic"
         elif model_name.startswith(("gpt", "o1", "o3")):
             provider_name = "openai"
         else:
-            # Default to treating as model string (pydantic-ai will auto-detect)
             return model_string
 
-    # Create provider with api_key + base_url + http_client
+    # Get config from registry
+    registry_config = _get_provider_config_from_registry(provider_name)
+    api_url = registry_config.get("api")
+    env_vars = registry_config.get("env", [])
+    api_key_name = env_vars[0] if env_vars else f"{provider_name.upper()}_API_KEY"
+
+    # Use user override if set, otherwise registry URL
+    base_url = base_url_override if base_url_override else api_url
+
+    # Get API key
+    api_key = env.get(api_key_name)
+
+    # Only anthropic provider uses AnthropicModel, all others use OpenAI
     if provider_name == "anthropic":
-        api_key = env.get("ANTHROPIC_API_KEY")
-        provider = AnthropicProvider(api_key=api_key, http_client=http_client)
+        provider = AnthropicProvider(api_key=api_key, base_url=base_url, http_client=http_client)
         return AnthropicModel(model_name, provider=provider)
-    elif provider_name in ("openai", "openrouter", "azure", "deepseek", "cerebras"):
-        # OpenAI-compatible providers all use OpenAIChatModel
-        config = PROVIDER_CONFIG.get(provider_name, {})
-        api_key_name = config.get("api_key_name")
-        api_key = env.get(api_key_name) if api_key_name else None
-        base_url = config.get("base_url")
-        if base_url is None and provider_name != "azure":
-            base_url = base_url_override
+    else:
         provider = OpenAIProvider(api_key=api_key, base_url=base_url, http_client=http_client)
         return OpenAIChatModel(model_name, provider=provider)
-    else:
-        # Unsupported provider, return string and let pydantic-ai handle it
-        # (won't have retry support but won't break)
-        return model_string
 
 
 def get_or_create_agent(model: ModelName, state_manager: StateManager) -> PydanticAgent:

{tunacode_cli-0.1.21.dist-info → tunacode_cli-0.1.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tunacode-cli
-Version: 0.1.21
+Version: 0.1.24
 Summary: Your agentic CLI developer.
 Project-URL: Homepage, https://tunacode.xyz/
 Project-URL: Repository, https://github.com/alchemiststudiosDOTai/tunacode

{tunacode_cli-0.1.21.dist-info → tunacode_cli-0.1.24.dist-info}/RECORD

@@ -1,12 +1,12 @@
 tunacode/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tunacode/constants.py,sha256=BoRbr8KF6D0LJYEVFFNRJHAQbMipCWi3DuIJ6UQhtTg,6221
+tunacode/constants.py,sha256=HCO7azR502kN1Ku4SgbMxb4eu0b-pWlVn0zCJddC3j0,6221
 tunacode/exceptions.py,sha256=9QgK4pc57lzjsJ4vaSWYB9ruOUMFtJcK6A5y7DTi4ik,10011
 tunacode/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tunacode/cli/textual_repl.tcss,sha256=Zi4Bo8YVv8nWYemOrf_zPIjBovA-CfHoncZqjS7fPrI,6095
 tunacode/configuration/__init__.py,sha256=MbVXy8bGu0yKehzgdgZ_mfWlYGvIdb1dY2Ly75nfuPE,17
 tunacode/configuration/defaults.py,sha256=uI8w4lGz8Sh9ZeYuvSnSb_TNmy3AZuzX_s9vf-cnqeg,1644
 tunacode/configuration/models.py,sha256=uk-fRhS6wezlhy7chT3IQareNqV0T_humtxPzLadAq8,4222
-tunacode/configuration/models_registry.json,sha256=P-ZmwLf2EFc8-pz65KUlkVxquhV433pl9r3fZj6WK40,848632
+tunacode/configuration/models_registry.json,sha256=Na-uddJEAYbRyGM8sQM_xW6XmZKWurLTCvYNwtBNnrg,861521
 tunacode/configuration/pricing.py,sha256=lsmDRcL-cLR1SEURsUyJfhqOMh_MQRV2aRBsAF-_LEE,2252
 tunacode/configuration/settings.py,sha256=2Au6U9DusgVSBCPP0i5B5sXtVRC5SeApXkze713gtFY,1026
 tunacode/core/__init__.py,sha256=X_qrgtH_2D3d7XyTFQHAL74uOu3Ud8uynt1ikYKCIi4,167
@@ -18,7 +18,7 @@ tunacode/core/agents/main.py,sha256=WRdv5BuTyUm4u1UDGQWRbQeIrfHlXqOIKDhyMSj8ubU,
 tunacode/core/agents/prompts.py,sha256=Eps3qQm8zdgPnyYOvRYwINMGVa71ttuqGlgtH17-3ag,2357
 tunacode/core/agents/research_agent.py,sha256=7vno4jVUCvZV5nqI_VCg74Ooi6chj3w1_2CQQayrUZw,8395
 tunacode/core/agents/agent_components/__init__.py,sha256=FDw5nF_t6RmVwkY99e3Jopgc9OyRGLuJkuLA2JQhSTg,1461
-tunacode/core/agents/agent_components/agent_config.py,sha256=g_zAmj-5aEoIqjIFdGFVA7AQ7dXGGx_QUmy10tu-iPI,17038
+tunacode/core/agents/agent_components/agent_config.py,sha256=iavdeNXQxEyulVtZzDNArAzB2xV40VPtlgZaQprWkOA,16661
 tunacode/core/agents/agent_components/agent_helpers.py,sha256=KVD9VgUwI2L6fWRlAP9_RuDgdF6XZKCZpRXsUXFCa3A,10500
 tunacode/core/agents/agent_components/message_handler.py,sha256=RswLyhtA2Zx7XxtbdqZGXsNSDMPw9FPJwXxgdaPqPXI,3624
 tunacode/core/agents/agent_components/node_processor.py,sha256=lik0iSjRNbOILpb3y6mbtHas3MMBD4ln_TznwBO9VpQ,18597
@@ -167,8 +167,8 @@ tunacode/utils/system/paths.py,sha256=NqrVZAtiuayLjkHpOXHhoFK3tvlCaCcaoYGG_kjkbk
 tunacode/utils/ui/__init__.py,sha256=3weQWW9YYZDhZz-f6wM_ekhJiIjQcMigXeAZAqiiIAM,208
 tunacode/utils/ui/file_filter.py,sha256=048k1WmufkzKKSOSxT519_j9dcTBxS1yv-Qr4nkbZWM,3972
 tunacode/utils/ui/helpers.py,sha256=ec5WYbhx-P6Zh47LL6ltyaHPJK9abA5UsWFmY5CF2pA,615
-tunacode_cli-0.1.21.dist-info/METADATA,sha256=XuHZ1nMveMt2_YbjjbMVktLvXOZmoUbbG725GQbeHpI,5604
-tunacode_cli-0.1.21.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-tunacode_cli-0.1.21.dist-info/entry_points.txt,sha256=pBNNa1Ikbt8pXk39eaj78hUIGx6noRGkOvpWuJ58szQ,50
-tunacode_cli-0.1.21.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
-tunacode_cli-0.1.21.dist-info/RECORD,,
+tunacode_cli-0.1.24.dist-info/METADATA,sha256=oFM3qoGeeCilkWcMyW1-b1ZiZrnzzrk9ObZjr_8BpWk,5604
+tunacode_cli-0.1.24.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+tunacode_cli-0.1.24.dist-info/entry_points.txt,sha256=pBNNa1Ikbt8pXk39eaj78hUIGx6noRGkOvpWuJ58szQ,50
+tunacode_cli-0.1.24.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
+tunacode_cli-0.1.24.dist-info/RECORD,,