gfdZddlmZmZddlZddlmZmZmZm Z ddl m Z ddl m Z Gd d e Zy) z| Handler file for calls to Azure OpenAI's o1 family of models Written separately to handle faking streaming for o1 models. )OptionalUnionN)AsyncAzureOpenAI AsyncOpenAI AzureOpenAIOpenAI)OpenAIChatCompletion)get_azure_openai_clientceZdZdddejddddfdedeedeedeedee ejfdee d eed eee e e efd eee e e effd Zy) AzureOpenAIO1ChatCompletionNr is_asyncapi_keyapi_base api_versiontimeout max_retries organizationclientreturnc lt|tst|trd}t||||||||S)N)rrrrrrr _is_async) isinstancerrr ) selfrrrrrrrrs W/var/www/openai/venv/lib/python3.12/site-packages/litellm/llms/azure/chat/o1_handler.py_get_openai_clientz.AzureOpenAIO1ChatCompletion._get_openai_clientsA, ff %FK)HF&#%#  )__name__ __module__ __qualname__httpxTimeoutboolrstrrfloatintrrrrrrrrrs"&"&%)/r.s+ # EE12# "6# r