@'hi<UddlZddlZddlmZmZmZmZmZmZm Z m Z m Z ddl Z ddl m Z ddlmZmZddlmZddlmZddlmZddlmZdd lmZmZmZmZmZmZm Z dd l!m"Z"dd l#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,dd l-m.Z.dZ/dZ0dZ1dddddddddddddddZ2ee3e4fe5d<hdZ6iddddddddddddddddd dd!dd"dd#dd$dd%dd&dd'dd(did)dd*dd+dd,dd-dd.dd/dd0dd1dd2dd3d4d5d4d6d4d7d4d8d4d9d4Z7ee3e4fe5d:<ddd;d<d;d;d;d;ddKZ?ejeAZBe eefZC ddLe4dMeDdNeeEdOeEdPeEdQeegeff dRZFdSe3dQe4fdTZGdUe3dQeDfdVZHdUe3dQeDfdWZI ddXe&dYeDdUee3dQefdZZJ ddXe&dYeDdUee3dQe ee3efeee3efffd[ZK dd\ee&dYeDdUee3d]eDdQeef d^ZLd_ed`ee3dQe&fdaZMdbedQee(fdcZNddeedQeee(fdeZOdfee3eEfdQee(fdgZPdhedQeee(fdiZQdjeed`ee3dQee&fdkZRdleSdQe&fdmZTdneeSdQee&fdoZUe dpdqe e.dQee3effdrZV ddqe e.dsee3dQee3effdtZW dduee3dvee3dwee3dQe ee3e3e3ffdxZXdduee3dQdfdyZYddze e3eSfdQe e3eSffd{ZZd|eed}eeedQeefd~Z[y)N) AnyCallableDictListOptionalSequenceTupleTypeUnion) deprecated)ChatCompletionMessageParamChatCompletionMessageToolCall)ChoiceDeltaToolCall)ChatCompletionMessage)ChatCompletionTokenLogprob)Logprobs)before_sleep_logretryretry_if_exception_typestop_after_attemptstop_after_delaywait_exponentialwait_random_exponential) stop_base)get_from_param_or_env) ChatMessage ImageBlockLogProb MessageRole TextBlock AudioBlock DocumentBlock) BaseModelopen_aizhttps://api.openai.com/v1i@ i)o1z o1-2024-12-17zo1-prozo1-pro-2025-03-19 o1-previewo1-preview-2024-09-12o1-minio1-mini-2024-09-12o3-minizo3-mini-2025-01-31o3z o3-2025-04-16zo4-minizo4-mini-2025-04-16 O1_MODELS>r)r'r*r(zgpt-4i z gpt-4-32kizgpt-4-1106-previewzgpt-4-0125-previewzgpt-4-turbo-previewzgpt-4-vision-previewzgpt-4-1106-vision-previewzgpt-4-turbo-2024-04-09z gpt-4-turbogpt-4ozgpt-4o-audio-previewzgpt-4o-audio-preview-2024-12-17zgpt-4o-audio-preview-2024-10-01zgpt-4o-mini-audio-previewz$gpt-4o-mini-audio-preview-2024-12-17zgpt-4o-2024-05-13zgpt-4o-2024-08-06zgpt-4o-2024-11-20zgpt-4.5-previewzgpt-4.5-preview-2025-02-27zchatgpt-4o-latest gpt-4o-minizgpt-4o-mini-2024-07-18z gpt-4-0613zgpt-4-32k-0613z gpt-4-0314zgpt-4-32k-0314zgpt-4.1iz gpt-4.1-miniz gpt-4.1-nanozgpt-4.1-2025-04-14zgpt-4.1-mini-2025-04-14zgpt-4.1-nano-2025-04-14 GPT4_MODELSi@i)r.r/zgpt-35-turbo-16kz gpt-35-turbozgpt-35-turbo-0125zgpt-35-turbo-1106zgpt-35-turbo-0613zgpt-35-turbo-16k-0613AZURE_TURBO_MODELS)z gpt-3.5-turbozgpt-3.5-turbo-16kzgpt-3.5-turbo-0125zgpt-3.5-turbo-1106zgpt-3.5-turbo-0613zgpt-3.5-turbo-16k-0613zgpt-3.5-turbo-0301 TURBO_MODELSi)ztext-davinci-003ztext-davinci-002zgpt-3.5-turbo-instruct GPT3_5_MODELSii)z text-ada-001ztext-babbage-001ztext-curie-001adababbagecuriedavinci GPT3_MODELSiAi)zcode-davinci-002zcode-davinci-001zcode-cushman-002zcode-cushman-001zNo 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 max_retriesrandom_exponentialstop_after_delay_seconds min_seconds max_secondsreturnc T|r t||n td||}t|}||t|z}t d||t t jt jt jt jftttjS)N)minmax) multiplierr@rAT)reraisestopwaitr before_sleep)rrrrrropenaiAPIConnectionErrorAPITimeoutErrorRateLimitErrorInternalServerErrorrloggerloggingWARNING)r9r:r;r<r= wait_strategy stop_strategys d/home/kushmeetdev/Regenta/Chatbot/venv/lib/python3.12/site-packages/llama_index/llms/openai/utils.pycreate_retry_decoratorrSs  K[A   M 2+>M+%(89Q(RR    #--**))..  &fgoo>  modelnamec 2|jdr|jdd}nd|vr|jdd}|tvrtd|d|tvr2td|d d j tj t|S) a Calculate the maximum number of tokens possible to generate for a model. Args: modelname: The modelname we want to know the context size for. Returns: The maximum context size Example: .. code-block:: python max_tokens = openai.modelname_to_contextsize("text-davinci-003") Modified from: https://github.com/hwchase17/langchain/blob/master/langchain/llms/openai.py ft::rBz:ft-rz OpenAI model z4 has been discontinued. Please choose another model.zUnknown model z/. Please provide a valid OpenAI model name in: z, ) startswithsplitDISCONTINUED_MODELS ValueErrorALL_AVAILABLE_MODELSjoinkeys)rUs rRopenai_modelname_to_contextsizer`s&E"OOC(+ 9 OOC(+ ''I;'+ +  ,,YM* .33567 9   **rTmodelc|tvSN) CHAT_MODELS)ras rR is_chat_modelre s K rTc|tvry|jdr|jdd}n%|jdr|jdd}t|}d|vxsd|v}|tv}|xr| xr| S) NTzft-rXrrWrB03140301)r]rYrZre"O1_MODELS_WITHOUT_FUNCTION_CALLING)rais_chat_model_is_old is_o1_betas rRis_function_calling_modelrms ((   C #   %  C #"5)N u_ /%F<z)to_openai_message_dict..Pz%3Prcontentrr developer)%blocksrr ASSISTANTadditional_kwargsgetrr appendrsr"rresolve_document_get_b64_stringdecodedocument_mimetypetitlerrstrr resolve_imagereadimage_mimetyper! resolve_audiorru__name__r\valueallr-riupdateitemspop)rnrorar content_txtrrr file_buffer b64_stringmimetype img_bytesimg_str audio_bytes audio_strmsg message_dictkeyr null_keyss rRto_openai_message_dictr#s GK9" <<;00 0!(!:!:!>!>$d" " eY ' NNFEJJ? @ 5:: %K } -::#446 "22{2K "ZZ..w7 272I2I2Uu..[lH NNLekkZ_`h_iiqr|q}X~ A z *yy +#&uyy>&+ll&|j&dD cgc]$} t| t,r| n| j/&} } | S|j j(|j j(dvst1d|jDr|n|d} ||t2vr|t4vr | ddk(rd| d<| j7D cgc] \} }| |  }} }|r|D]} | j9| | Scc} wcc}} w)rqr% input_textrt input_imager~)rur}rTrrwryrzrrrNr tool_call_idcall_idzJtool_call_id or call_id is required in additional_kwargs for tool messagesfunction_call_output)ruoutputr)rrc3<K|]}t|tywrcrrs rRrz3to_openai_responses_message_dict..rrrrrr)rrr rrsrrrrrrrrrurr\rrrrrrdict model_dumprr-rirr)rnrorarrrrrrrr tool_call message_dictsrrrs rR to_openai_responses_message_dictrs GK" eY ' NNL%**E F 5:: %K z *yy -%(^"',,"8&"//$/?DDF #**73 -',U-A-A,B(7)%T"',,"8&5T%[5I5I4JKCS/ !3"@ "  LLK11 1 w88 8w888 ||V#++// G5599)D  ?\  +!  22 2%66|D $It4I):N:N:P P  LL&&<<%%)@@PPP    Y  ; ;  8 +#.L '3'9'9';Meu}MIM "C   S ! " A 4Ns3)K! K,Kmessagesis_responses_apic|rLg}|D]C}t||d}t|tr|j|3|j |E|S|Dcgc]}t |||c}Scc}w)z1Convert generic messages to OpenAI message dicts.r+)rora)rrlistextendrr)rrorarfinal_message_dictsrnrs rRto_openai_message_dictsr s  :G<#M -.#**=9#**=9 :#"$   ##    sA+openai_message modalitiesc|j}d|vr(|jrt|jxsdg}ng}i}|jr|j}|j ||j rQd|vrM|j j }|j j}||d<|jt|dt||| S) /Convert openai message dict to generic message.rsr%rs)rrrrmp3)rr)rrr) rrr rrrrrrr!r)rrrrrrrr audio_datas rRfrom_openai_messager,s   D  6 6!7!7!=2>?(*  :H:S:S   J 7: 5+1144#)).. 2D./ jz%@A DCT UUrTopenai_token_logprobc g}|jrJ |jDcgc]2}t|j|j|jxsg4}}|S|Scc}w#t $r t |wxYw)zBConvert a single openai token logprob to generic list of logprobs.tokenlogprobbytes) top_logprobsrrrr Exceptionprint)rresultels rRfrom_openai_token_logprobrEsF(( /;;bhh "((.bQF M6M  & '  sA 7AA A A6openai_token_logprobscVg}|D]!}t|x}s|j|#|S)z9Convert openai token logprobs to generic list of LogProb.)rr)rr token_logproblogprobss rRfrom_openai_token_logprobsrVs;F.$ 0? ?8 ? MM( #$ MrTopenai_completion_logprobc l|jDcgc]\}}t||gc}}Scc}}w)>Convert openai completion logprobs to generic list of LogProb.r)rr)rtvs rRfrom_openai_completion_logprobras: .335  Aq a"-  s0openai_completion_logprobscpg}|jr"|jDcgc] }t|}}|Scc}w)r)rr)rrcompletion_logprobs rRfrom_openai_completion_logprobsrksJF!..'A&M&M " ++= >   M  s3openai_messagesc@|Dcgc]}t||c}Scc}wz1Convert openai message dicts to generic messages.)r)rrrns rRfrom_openai_messagesrxs"ET T  4 TT TsrcH|d}|jd}g}t|tr|D]}|jd}|dk(r+|jt |jdD|dk(r[|dd}|dd}|j d r|jt || |jt || d |}t|d }|j} | jd| jdd t||| |S)rrrrursrr}rrry)imagerrzUnsupported message type: N)rrrr) rrrrr rYrr\copyrr) rrrrelemrimgrrrs rRfrom_openai_message_dictrs#  Dy)G F'4  &D AF{ iTXXf-=>?k!;'.k*84>>'*MM*3v"FGMM*V"DE21#6 o% &$))+&!)T*  76GPV rTrc>|Dcgc] }t|c}Scc}wr)r)rrs rRfrom_openai_message_dictsrsGT U| $\ 2 UU UszFDeprecated in favor of `to_openai_tool`, which should be used instead.pydantic_classct|dS)zb Deprecated in favor of `to_openai_tool`. Convert pydantic class to OpenAI function. N) description)to_openai_tool)rs rRto_openai_functionrs .d ;;rTrc|j}|jddxs|}|d||jd}d|dS)z&Convert pydantic class to OpenAI tool.rNr)namer parametersfunctionrur )schemar)rrr schema_descriptionr s rRrrsR " " $FM48GKw)$++-H H 55rTapi_keyapi_base api_versionctd|dd}td|dd}td|dd}|xstjxsd}|xstjxst}|xstj xst }|t||fS)z "Resolve OpenAI credentials. The order of precedence is: 1. param 2. env 3. openai module 4. default r OPENAI_API_KEYr%rOPENAI_API_BASErOPENAI_API_VERSION)rrHr base_urlDEFAULT_OPENAI_API_BASErDEFAULT_OPENAI_API_VERSIONr)r rr final_api_keyfinal_api_basefinal_api_versions rRresolve_openai_credentialsrs$Iw8H"MG$Z;LbQH'{$8"K 3v~~3MKK4KN#Wv'9'9W=W #n-/@ @@rTcn|xs tjjdd}|stty)Nrr%)osenvironrr\MISSING_API_KEY_ERROR_MESSAGE)r openai_api_keys rRvalidate_openai_api_keyr s0D /? DN 677 rT tool_choicec<t|tr |dvrdd|idS|S)zj Resolve tool choice. If tool_choice is a function name string, return the appropriate dict. )noner~requiredr rr )rr)r!s rRresolve_tool_choicer%s, +s# ;W(W"0EFF rTrtool_calls_deltac|t|dk(r|S|d}t|dk(r|j||S|d}|j|jk7r|j||S|jJ|jJ|jjd|j_|jj d|j_|j d|_|jxj|jjxsdz c_|jxj |jj xsdz c_|xj |j xsdz c_|S)ac Use the tool_calls_delta objects received from openai stream chunks to update the running tool_calls object. Args: tool_calls (List[ChoiceDeltaToolCall]): the list of tool calls tool_calls_delta (ChoiceDeltaToolCall): the delta to update tool_calls Returns: List[ChoiceDeltaToolCall]: the updated tool calls rr%)lenrindexr  argumentsrr)rr&tc_deltars rRupdate_tool_callsr-sM 3'7#8A#="H :!(#> 5 rN 77hnn $   h '. %::) ))$$0 00 zz##+') $zz&"$ tt| JJ H$5$5$?$?$E2 E JJOOx0055; ;O DDHKK%2 %D rT)FN<)FN)FNFrc)NNN)r~)\rNrtypingrrrrrrr r r rHr openai.types.chatr r'openai.types.chat.chat_completion_chunkr)openai.types.chat.chat_completion_messager/openai.types.chat.chat_completion_token_logprobropenai.types.completion_choicertenacityrrrrrrr tenacity.stopr(llama_index.core.base.llms.generic_utilsr llama_index.core.base.llms.typesrrrrr r!r" llama_index.core.bridge.pydanticr#DEFAULT_OPENAI_API_TYPErrr-rint__annotations__rir0r1r2r3r8r]rdr[r getLoggerrrMOpenAIToolCallboolfloatrSr`rermrrrrrrrrrrrrrrrr r%r-rTrRrCsr TTT !WGKV3$J7#5 #    4S>"&", T ,  ,&,&,6,F, ,f,6, f, F!,"&v#,$&v%,& ',(+F),*+,,-,./,0v1,2!&3,67,869,:f;,@$A,BeC,F$G,HeI,LwM,NGO,PGQ,R'S,TwU,VwW, T#s(^,^" &DcN "#!  d38n(" ! tCH~   T#s(^        !   8 $46IIJ %04   'uo     seSj F"+s"+s"+J   4S#X/ /0 gX"  {#   C=    $%  @V)V7;CyVV24 ']"#$>? $w-#CJ/ ']  (  $w- U34UBFs)U +U4K@VXd^V[@QV  TU<V<CG 6O 62:3- 6 #s(^ 6""!%A c]AsmA#A 8C=#s "# A:8Xc]8d8 U39%5 5dCS 5()5t$7895 5rT