o h(@sUddlmZddlZddlmZmZddlmZddlm Z ddl m Z m Z ddl mZdd lmZd Zd ed <dad ed<dddZGddde ZdS)) annotationsN) AsyncOpenAIDefaultAsyncHttpxClient)_openai_shared)get_default_model)Model ModelProvider)OpenAIChatCompletionsModel)OpenAIResponsesModelzgpt-4ostr DEFAULT_MODELzhttpx.AsyncClient | None _http_clientreturnhttpx.AsyncClientcCstdurtatS)N)rrrr^/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/models/openai_provider.pyshared_http_clientsrc@s:eZdZddddddddddZdddZdddZdS)OpenAIProviderN)api_keybase_url openai_client organizationproject use_responsesr str | NonerrAsyncOpenAI | Nonerrr bool | NonerNonecCsf|dur|dur |dusJd||_nd|_||_||_||_||_|dur,||_dSt|_dS)aCreate a new OpenAI provider. Args: api_key: The API key to use for the OpenAI client. If not provided, we will use the default API key. base_url: The base URL to use for the OpenAI client. 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. organization: The organization to use for the OpenAI client. project: The project to use for the OpenAI client. use_responses: Whether to use the OpenAI responses API. Nz>Don't provide api_key or base_url if you provide openai_client)_client_stored_api_key_stored_base_url_stored_organization_stored_project_use_responsesrget_use_responses_by_default)selfrrrrrrrrr__init__s zOpenAIProvider.__init__rcCs>|jdurtpt|jpt|j|j|jt d|_|jS)N)rrrr http_client) rrget_default_openai_clientrr get_default_openai_keyr!r"r#r)r&rrr _get_clientGs   zOpenAIProvider._get_client model_namercCs4|durt}|}|jrt||dSt||dS)N)modelr)rr+r$r r )r&r,clientrrr get_modelSs  zOpenAIProvider.get_model)rrrrrrrrrrrrrr)rr)r,rrr)__name__ __module__ __qualname__r'r+r/rrrrrs * r)rr) __future__rhttpxopenairrrdefault_modelsr interfacerr openai_chatcompletionsr openai_responsesr r __annotations__rrrrrrrs