sync: mnogo

agentui/providers/adapters/registry.py

@@ -0,0 +1,32 @@
+from __future__ import annotations
+
+from typing import Optional
+
+from agentui.providers.adapters.base import ProviderAdapter, default_base_url_for as _default_base_url_for
+from agentui.providers.adapters.openai import OpenAIAdapter
+try:
+    from agentui.providers.adapters.gemini import GeminiAdapter, GeminiImageAdapter
+except Exception:
+    GeminiAdapter = None  # type: ignore
+    GeminiImageAdapter = None  # type: ignore
+try:
+    from agentui.providers.adapters.claude import ClaudeAdapter
+except Exception:
+    ClaudeAdapter = None  # type: ignore
+
+
+def get_adapter(provider: str) -> Optional[ProviderAdapter]:
+    p = (provider or "").strip().lower()
+    if p == "openai":
+        return OpenAIAdapter()
+    if p == "gemini" and GeminiAdapter:
+        return GeminiAdapter()  # type: ignore[operator]
+    if p == "gemini_image" and GeminiImageAdapter:
+        return GeminiImageAdapter()  # type: ignore[operator]
+    if p == "claude" and ClaudeAdapter:
+        return ClaudeAdapter()  # type: ignore[operator]
+    return None
+
+
+def default_base_url_for(vendor: str) -> Optional[str]:
+    return _default_base_url_for(vendor)