o RTh@sddlZddlZddlZddlZddlmZddlmZddlZej dZ e dZ gdZddZGd d d eZd d Zd dZddZddZddZddZ     d!ddZddZddZdefdd ZdS)"N)Enum)Optional OPENAI_LOGopenai)log_info log_debuglog_warnlogfmtcCs(|tjtjfvrdd|iSd|iS)N AuthorizationzBearer zapi-key)ApiTypeOPEN_AIAZURE_AD)apikeyrN/home/air/segue/gemini/backup/venv/lib/python3.10/site-packages/openai/util.pys rc@s$eZdZdZdZdZeddZdS)r cCsB|dkr tjS|dvrtjS|dvrtjStjd)Nazure)azure_adazuread)open_airznThe API type provided in invalid. Please select one of the supported API types: 'azure', 'azure_ad', 'open_ai')lowerr AZUREr r rerrorInvalidAPIType)labelrrrfrom_str!s   zApiType.from_strN)__name__ __module__ __qualname__rr r staticmethodrrrrrr s r cCs tjdvrtjStdvrtSdS)Ndebuginfo)rlogrrrrr_console_log_level/s r(cKs<ttdd|i|}tdkrt|tjdt|dS)Nmessager%filer)r dictr(printsysstderrloggerr%r)paramsmsgrrrr8 rcKs<ttdd|i|}tdvrt|tjdt|dS)Nr)r$r*r)r r,r(r-r.r/r0r&r1rrrr?r4rcKs2ttdd|i|}t|tjdt|dS)Nr)r*r)r r,r-r.r/r0warnr1rrrrFsrcs(dddfddt|DS)NcSs\t|dr |d}t|tst|}td|rt|}td|r't|}dj||dS)Ndecodezutf-8z\sz {key}={val}rval)hasattrr6 isinstancestrresearchreprformatr7rrrfmtMs     zlogfmt..fmt csg|] \}}||qSrr).0rr8r@rr [zlogfmt..)joinsorteditems)propsrrCrr Ls r cCsddlm}|S)NrOBJECT_CLASSES)openai.object_classesrKrJrrrget_object_classes^s rMFc sd}t|tjjr|j|j}|j}|r|St|tr'fdd|DSt|trZt|tj j sZ| }| d}t|t rKt |tj j }ntj j }|j||dS|S)Nc sg|] }t|dqS))engine)convert_to_openai_objectrBiapi_key api_versionrN organizationrrrDxs  z,convert_to_openai_object..object)rSrTrU response_msrN)r:ropenai_responseOpenAIResponserUrWdatalistr, openai_object OpenAIObjectcopygetr;rMconstruct_from) resprSrTrUrNplain_old_datarW klass_nameklassrrRrrOes<      rOcCs8t|tr dd|DSt|trdd|DS|S)zConverts a OpenAIObject back to a regular dict. Nested OpenAIObjects are also converted back to regular dicts. :param obj: The OpenAIObject to convert. :returns: The OpenAIObject as a dict. cSsg|]}t|qSrconvert_to_dictrPrrrrDsz#convert_to_dict..cSsi|] \}}|t|qSrre)rBkvrrr rEz#convert_to_dict..)r:r[r,rH)objrrrrfs  rfcCs|}|||S)N)r^update)xyzrrr merge_dictss roreturncCs~tjr1ttjd}|}|dstdtjd|WdS1s*wYdStjdur9tjStj d)Nrtzsk-zMalformed API key in .aNo API key provided. You can set your API key in code using 'openai.api_key = ', or you can set the environment variable OPENAI_API_KEY=). If your API key is stored in a file, you can point the openai module at it with 'openai.api_key_path = '. You can generate API keys in the OpenAI web interface. See https://platform.openai.com/account/api-keys for details.) r api_key_pathopenreadstrip startswith ValueErrorrSrAuthenticationError)rgrSrrrdefault_api_keys  $ rz)NNNNF)loggingosr<r.enumrtypingrrenvironr_r getLoggerr0__all__api_key_to_headerr r(rrrr rMrOrfror;rzrrrrs8       1