o h@sbddlmZddlmZddlmZddlmZmZddl m Z Gdd d Z Gd d d eZ d S) ) annotations) AsyncOpenAI) UserError)Model ModelProvider)OpenAIProviderc@sTeZdZdZddZddd Zdd d ZdddZdddZdddZ dddZ dS) MultiProviderMapz/A map of model name prefixes to ModelProviders.cCs i|_dS)N_mappingselfr]/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/models/multi_provider.py__init__ s zMultiProviderMap.__init__prefixstrreturnboolcCs ||jvS)z3Returns True if the given prefix is in the mapping.r rrrrr has_prefix zMultiProviderMap.has_prefixdict[str, ModelProvider]cCs |jS)z>Returns a copy of the current prefix -> ModelProvider mapping.)r copyr rrr get_mappingrzMultiProviderMap.get_mappingmappingcCs ||_dS)z.Overwrites the current mapping with a new one.Nr )rrrrr set_mappingrzMultiProviderMap.set_mappingModelProvider | NonecCs |j|S)zReturns the ModelProvider for the given prefix. Args: prefix: The prefix of the model name e.g. "openai" or "my_prefix". )r getrrrr get_provider zMultiProviderMap.get_providerproviderrcCs||j|<dS)zAdds a new prefix -> ModelProvider mapping. Args: prefix: The prefix of the model name e.g. "openai" or "my_prefix". provider: The ModelProvider to use for the given prefix. Nr )rrr"rrr add_provider$szMultiProviderMap.add_providercCs |j|=dS)zRemoves the mapping for the given prefix. Args: prefix: The prefix of the model name e.g. "openai" or "my_prefix". Nr rrrrremove_provider-r!z MultiProviderMap.remove_providerN)rrrr)rr)rr)rrrr)rrr"r)rr) __name__ __module__ __qualname____doc__rrrrr r#r$rrrrr s      r c@sTeZdZdZddddddddd!ddZd"ddZd#ddZd$ddZd%dd ZdS)& MultiProvideraAThis ModelProvider maps to a Model based on the prefix of the model name. By default, the mapping is: - "openai/" prefix or no prefix -> OpenAIProvider. e.g. "openai/gpt-4.1", "gpt-4.1" - "litellm/" prefix -> LitellmProvider. e.g. "litellm/openai/gpt-4.1" You can override or customize this mapping. N) provider_mapopenai_api_keyopenai_base_url openai_clientopenai_organizationopenai_projectopenai_use_responsesr*MultiProviderMap | Noner+ str | Noner,r-AsyncOpenAI | Noner.r/r0 bool | NonerNonecCs&||_t||||||d|_i|_dS)aCreate a new OpenAI provider. Args: provider_map: A MultiProviderMap that maps prefixes to ModelProviders. If not provided, we will use a default mapping. See the documentation for this class to see the default mapping. openai_api_key: The API key to use for the OpenAI provider. If not provided, we will use the default API key. openai_base_url: The base URL to use for the OpenAI provider. If not provided, we will use the default base URL. openai_client: An optional OpenAI client to use. If not provided, we will create a new OpenAI client using the api_key and base_url. openai_organization: The organization to use for the OpenAI provider. openai_project: The project to use for the OpenAI provider. openai_use_responses: Whether to use the OpenAI responses API. )api_keybase_urlr- organizationproject use_responsesN)r*r openai_provider_fallback_providers)rr*r+r,r-r.r/r0rrrr?s zMultiProvider.__init__ model_nametuple[str | None, str | None]cCs4|durdSd|vr|dd\}}||fSd|fS)N)NN/r)split)rr=rrrr_get_prefix_and_model_namefs z(MultiProvider._get_prefix_and_model_namerrrcCs(|dkr ddlm}|Std|)Nlitellmr)LitellmProviderzUnknown prefix: )"extensions.models.litellm_providerrCr)rrrCrrr_create_fallback_provideros z'MultiProvider._create_fallback_providercCsD|dus|dkr |jS||jvr|j|S|||j|<|j|S)Nopenai)r;r<rErrrr_get_fallback_providerws    z$MultiProvider._get_fallback_providerrcCsB||\}}|r|jr|j|}r||S|||S)aKReturns a Model based on the model name. The model name can have a prefix, ending with a "/", which will be used to look up the ModelProvider. If there is no prefix, we will use the OpenAI provider. Args: model_name: The name of the model to get. Returns: A Model. )rAr*r get_modelrG)rr=rr"rrrrHs  zMultiProvider.get_model)r*r1r+r2r,r2r-r3r.r2r/r2r0r4rr5)r=r2rr>)rrrr)rr2rr)r=r2rr) r%r&r'r(rrArErGrHrrrrr)6s  '  r)N) __future__rrFr exceptionsr interfacerrr;r r r)rrrrs   ,