.h|ddlmZddlmZddlmZddlmZmZddl m Z Gdd Z Gd d eZ d S) ) annotations) AsyncOpenAI) UserError)Model ModelProvider)OpenAIProvidercHeZdZdZdZddZdd Zdd Zdd ZddZ ddZ dS)MultiProviderMapz/A map of model name prefixes to ModelProviders.ci|_dS)N_mappingselfs Y/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/agents/models/multi_provider.py__init__zMultiProviderMap.__init__ s 24 prefixstrreturnboolc||jvS)z3Returns True if the given prefix is in the mapping.rrrs r has_prefixzMultiProviderMap.has_prefixs&&rdict[str, ModelProvider]c4|jS)z>Returns a copy of the current prefix -> ModelProvider mapping.)rcopyrs r get_mappingzMultiProviderMap.get_mappings}!!###rmappingc||_dS)z.Overwrites the current mapping with a new one.Nr)rr s r set_mappingzMultiProviderMap.set_mappings  rModelProvider | Nonec6|j|S)zReturns the ModelProvider for the given prefix. Args: prefix: The prefix of the model name e.g. "openai" or "my_prefix". )rgetrs r get_providerzMultiProviderMap.get_providers }  (((rproviderr c||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's r add_providerzMultiProviderMap.add_provider$s!) frc|j|=dS)zRemoves the mapping for the given prefix. Args: prefix: The prefix of the model name e.g. "openai" or "my_prefix". Nrrs rremove_providerz MultiProviderMap.remove_provider-s M& ! ! !rN)rrrr)rr)r r)rrrr#)rrr'r )rr) __name__ __module__ __qualname____doc__rrrr"r&r)r+rrr r s99555''''$$$$    ))))))))""""""rr cLeZdZdZddddddddddZddZddZddZd dZdS)! 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_responsesr3MultiProviderMap | Noner4 str | Noner5r6AsyncOpenAI | Noner7r8r9 bool | NonerNonecV||_t|||||||_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_urlr6 organizationproject use_responsesN)r3r openai_provider_fallback_providers)rr3r4r5r6r7r8r9s rrzMultiProvider.__init__?sF6)-"$',".    >@   r model_nametuple[str | None, str | None]cT|dSd|vr|dd\}}||fSd|fS)N)NN/r)split)rrGrs r_get_prefix_and_model_namez(MultiProvider._get_prefix_and_model_namefsH  : J  !+!1!1#q!9!9 FJ:% %# #rrrr cR|dkrddlm}|Std|)Nlitellmr)LitellmProviderzUnknown prefix: )"extensions.models.litellm_providerrOr)rrrOs r_create_fallback_providerz'MultiProvider._create_fallback_providerosD Y   L L L L L L"?$$ $7v7788 8rc||dkr|jS||jvr |j|S|||j|<|j|S)Nopenai)rErFrQrs r_get_fallback_providerz$MultiProvider._get_fallback_providerws_ >Vx//' ' t/ / /+F3 3/3/M/Mf/U/UD $V ,+F3 3rrc||\}}|r8|jr1|j|x}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. )rLr3r& get_modelrT)rrGrr's rrVzMultiProvider.get_models"<)rGr;rrH)rrrr )rr;rr )rGr;rr) r,r-r.r/rrLrQrTrVr0rrr2r26s15%)&*,0*.%),0%@%@%@%@%@%@N$$$$99994444MMMMMMrr2N) __future__rrSr exceptionsr interfacerr rEr r r2r0rrrZs""""""""""""++++++++++++++)")")")")")")")"XZMZMZMZMZMMZMZMZMZMZMr