@@ -48,68 +48,74 @@ def model_profile(self, model_name: str) -> ModelProfile | None:
4848return None # pragma: no cover
4949
5050
51- def infer_provider (provider :str )-> Provider [Any ]:# noqa: C901
52- """Infer the provider from the provider name."""
51+ def infer_provider_class (provider :str )-> type [ Provider [Any ] ]:# noqa: C901
52+ """Infers the provider class from the provider name."""
5353if provider == 'openai' :
5454from .openai import OpenAIProvider
5555
56- return OpenAIProvider ()
56+ return OpenAIProvider
5757elif provider == 'deepseek' :
5858from .deepseek import DeepSeekProvider
5959
60- return DeepSeekProvider ()
60+ return DeepSeekProvider
6161elif provider == 'openrouter' :
6262from .openrouter import OpenRouterProvider
6363
64- return OpenRouterProvider ()
64+ return OpenRouterProvider
6565elif provider == 'azure' :
6666from .azure import AzureProvider
6767
68- return AzureProvider ()
68+ return AzureProvider
6969elif provider == 'google-vertex' :
7070from .google_vertex import GoogleVertexProvider
7171
72- return GoogleVertexProvider ()
72+ return GoogleVertexProvider
7373elif provider == 'google-gla' :
7474from .google_gla import GoogleGLAProvider
7575
76- return GoogleGLAProvider ()
76+ return GoogleGLAProvider
7777# NOTE: We don't test because there are many ways the `boto3.client` can retrieve the credentials.
7878elif provider == 'bedrock' :
7979from .bedrock import BedrockProvider
8080
81- return BedrockProvider ()
81+ return BedrockProvider
8282elif provider == 'groq' :
8383from .groq import GroqProvider
8484
85- return GroqProvider ()
85+ return GroqProvider
8686elif provider == 'anthropic' :
8787from .anthropic import AnthropicProvider
8888
89- return AnthropicProvider ()
89+ return AnthropicProvider
9090elif provider == 'mistral' :
9191from .mistral import MistralProvider
9292
93- return MistralProvider ()
93+ return MistralProvider
9494elif provider == 'cohere' :
9595from .cohere import CohereProvider
9696
97- return CohereProvider ()
97+ return CohereProvider
9898elif provider == 'grok' :
9999from .grok import GrokProvider
100100
101- return GrokProvider ()
101+ return GrokProvider
102102elif provider == 'fireworks' :
103103from .fireworks import FireworksProvider
104104
105- return FireworksProvider ()
105+ return FireworksProvider
106106elif provider == 'together' :
107107from .together import TogetherProvider
108108
109- return TogetherProvider ()
109+ return TogetherProvider
110110elif provider == 'heroku' :
111111from .heroku import HerokuProvider
112112
113- return HerokuProvider ()
113+ return HerokuProvider
114114else :# pragma: no cover
115115raise ValueError (f'Unknown provider:{ provider } )
116+
117+
118+ def infer_provider (provider :str )-> Provider [Any ]:
119+ """Infer the provider from the provider name."""
120+ provider_class = infer_provider_class (provider )
121+ return provider_class ()