o }!g @s<ddlZddlZddlmZmZmZmZmZddlm Z ddl m Z m Z m Z mZmZmZmZddlmZddlZddlmZddlmZdZd Zd ZeeZeeefZ  dde de!dee"de"de"deegeff ddZ#   ddee$dee$dee$deee$e$e$ffddZ%ddee$ddfddZ&dS) N)AnyCallableOptionalTupleUnion)get_from_param_or_env)before_sleep_logretryretry_if_exception_typestop_after_attemptstop_after_delaywait_exponentialwait_random_exponential) stop_base)ChatCompletionMessageToolCall)ChoiceDeltaToolCallzhttps://api.openai.com/v1zNo API key found for OpenAI. Please set either the OPENAI_API_KEY environment variable or openai.api_key prior to initialization. API keys can be found or created at https://platform.openai.com/account/api-keys F max_retriesrandom_exponentialstop_after_delay_seconds min_seconds max_secondsreturnc Csh|rt||dntd||d}t|}|dur|t|B}td||ttjtjtj tj ft t t jdS)N)minmax) multiplierrrT)reraisestopwaitr before_sleep)rr r r r r openaiAPIConnectionErrorAPITimeoutErrorRateLimitErrorInternalServerErrorrloggerloggingWARNING)rrrrr wait_strategy stop_strategyr-e/mnt/skqttb/ctump_chatbot/chatbot/lib/python3.10/site-packages/llama_index/embeddings/openai/utils.pycreate_retry_decorator$s(    r/api_keyapi_base api_versioncCsbtd|dd}td|dd}td|dd}|ptjpd}|p"tjp"t}|p)tjp)t}|t||fS)z "Resolve OpenAI credentials. The order of precedence is: 1. param 2. env 3. openai module 4. default r0OPENAI_API_KEYrr1OPENAI_API_BASEr2OPENAI_API_VERSION)rr#r0base_urlDEFAULT_OPENAI_API_BASEr2DEFAULT_OPENAI_API_VERSIONstr)r0r1r2 final_api_keyfinal_api_basefinal_api_versionr-r-r.resolve_openai_credentialsGsr=cCs"|ptjdd}|sttdS)Nr3r)osenvironget ValueErrorMISSING_API_KEY_ERROR_MESSAGE)r0openai_api_keyr-r-r.validate_openai_api_keycsrD)FNrr)NNN)N)'r)r>typingrrrrr(llama_index.core.base.llms.generic_utilsrtenacityrr r r r r r tenacity.stoprr#openai.types.chatr'openai.types.chat.chat_completion_chunkrr7r8rB getLogger__name__r(OpenAIToolCallintboolfloatr/r9r=rDr-r-r-r.sX $       $