Spaces:
Running
Running
File size: 2,495 Bytes
4ef118d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 | """
Provider adapter factory.
Creates the appropriate adapter based on provider name.
"""
from .base import BaseProviderAdapter
from .openai import OpenAIAdapter
from .other_providers import (
GeminiAdapter,
GLMAdapter,
KimiAdapter,
MinimaxAdapter,
ModelScopeAdapter,
NvidiaAdapter,
SiliconFlowAdapter,
)
# Cache adapter instances for reuse
_adapter_cache: dict[str, BaseProviderAdapter] = {}
def get_provider_adapter(
provider: str,
) -> BaseProviderAdapter:
"""
Get provider adapter instance.
Args:
provider: Provider name (openai, gemini, siliconflow, etc.)
Returns:
Provider adapter instance
"""
# Return cached instance if available
if provider in _adapter_cache:
return _adapter_cache[provider]
# Create new adapter instance
adapter: BaseProviderAdapter
match provider:
case "openai" | "openai_compatibility":
adapter = OpenAIAdapter()
case "siliconflow":
adapter = SiliconFlowAdapter()
case "kimi":
adapter = KimiAdapter()
case "glm":
adapter = GLMAdapter()
case "deepseek":
adapter = GLMAdapter()
case "volcengine":
adapter = GLMAdapter()
case "modelscope":
adapter = ModelScopeAdapter()
case "gemini":
adapter = GeminiAdapter()
case "nvidia":
adapter = NvidiaAdapter()
case "minimax":
adapter = MinimaxAdapter()
case _:
# Fallback to OpenAI adapter for unknown providers
# (assumes OpenAI-compatible API)
import warnings
warnings.warn(f"Unknown provider: {provider}, using OpenAI adapter as fallback")
adapter = OpenAIAdapter()
# Cache for future use
_adapter_cache[provider] = adapter
return adapter
def is_provider_supported(provider: str) -> bool:
"""Check if provider is supported."""
return provider in [
"openai",
"openai_compatibility",
"siliconflow",
"kimi",
"glm",
"deepseek",
"volcengine",
"modelscope",
"gemini",
"nvidia",
"minimax",
]
# Export all supported providers
SUPPORTED_PROVIDERS: list[str] = [
"openai",
"openai_compatibility",
"siliconflow",
"kimi",
"glm",
"deepseek",
"volcengine",
"modelscope",
"gemini",
"nvidia",
"minimax",
]
|