progress ai forgot to push

src/ai_client.py

@@ -49,6 +49,19 @@ _history_trunc_limit: int = 8000
 # Global event emitter for API lifecycle events
 events: EventEmitter = EventEmitter()
 
+_ai_proxy = None
+
+def _get_proxy():
+    global _ai_proxy
+    if _ai_proxy is None and os.environ.get("AI_SERVER_ENABLED"):
+        try:
+            from src.ai_client_proxy import AIProxyClient
+            _ai_proxy = AIProxyClient()
+            _ai_proxy.start_server()
+        except Exception:
+            _ai_proxy = None
+    return _ai_proxy
+
 def set_model_params(temp: float, max_tok: int, trunc_limit: int = 8000, top_p: float = 1.0) -> None:
  """
  Sets global generation parameters like temperature and max tokens.
@@ -532,21 +545,26 @@ def get_gemini_cache_stats() -> dict[str, Any]:
  }
 
 def list_models(provider: str) -> list[str]:
- """
- [C: src/app_controller.py:AppController.do_fetch, tests/test_agent_capabilities.py:test_agent_capabilities_listing, tests/test_ai_client_list_models.py:test_list_models_gemini_cli, tests/test_deepseek_infra.py:test_deepseek_model_listing, tests/test_minimax_provider.py:test_minimax_list_models]
- """
- creds = _load_credentials()
- if provider == "gemini":
-  return _list_gemini_models(creds["gemini"]["api_key"])
- elif provider == "anthropic":
-  return _list_anthropic_models()
- elif provider == "deepseek":
-  return _list_deepseek_models(creds["deepseek"]["api_key"])
- elif provider == "gemini_cli":
-  return _list_gemini_cli_models()
- elif provider == "minimax":
-  return _list_minimax_models(creds["minimax"]["api_key"])
- return []
+    """
+    [C: src/app_controller.py:AppController.do_fetch, tests/test_agent_capabilities.py:test_agent_capabilities_listing, tests/test_ai_client_list_models.py:test_list_models_gemini_cli, tests/test_deepseek_infra.py:test_deepseek_model_listing, tests/test_minimax_provider.py:test_minimax_list_models]
+    """
+    proxy = _get_proxy()
+    if proxy and proxy.status == "ready":
+        result = proxy.send_command("list_models", {"provider": provider})
+        if "result" in result:
+            return result["result"].get("models", [])
+    creds = _load_credentials()
+    if provider == "gemini":
+        return _list_gemini_models(creds["gemini"]["api_key"])
+    elif provider == "anthropic":
+        return _list_anthropic_models()
+    elif provider == "deepseek":
+        return _list_deepseek_models(creds["deepseek"]["api_key"])
+    elif provider == "gemini_cli":
+        return _list_gemini_cli_models()
+    elif provider == "minimax":
+        return _list_minimax_models(creds["minimax"]["api_key"])
+    return []
 
 def _list_gemini_cli_models() -> list[str]:
  return [