a f g@srddlZddlmZddlZddlZddlmZejddddZddd d Zddd d Z ddd dZ dS)N)NamedTemporaryFile)utilfunction)ZscopeccsXtj}zDtddd}|jt_|VWdn1s80YW|t_n|t_0dS)Nzopenai-api-keywt)prefixmode)openaiZ api_key_pathrname)Z saved_pathtmpr Y/var/www/html/llm_bihealth/app/venv/lib/python3.9/site-packages/openai/tests/test_util.py api_key_file s &r )returncCs(td|d|tdks$JdS)Nzsk-foofile)printflushrdefault_api_keyr r r r test_openai_api_key_paths rcCsNtd|d|tjtddtWdn1s@0YdS)Nzmalformed-api-keyrzMalformed API key)match)rrpytestZraises ValueErrorrrrr r r +test_openai_api_key_path_with_malformed_keys rc Cs^ddddddddd d d d d dgd}t|}ttt|t|ksZJdS)Nz&chatcmpl-7NaPEA6sgX7LnNPyKPbRlsyqLbr5Vzchat.completionid|dzgpt-3.5-turbo-03019(a)Z prompt_tokensZcompletion_tokensZ total_tokensZ assistantzThe 2020 World Series was played at Globe Life Field in Arlington, Texas. It was the first time that the World Series was played at a neutral site because of the COVID-19 pandemic.)rolecontentstopr)messageZ finish_reasonindex)idobjectcreatedmodelusagechoices)rZconvert_to_openai_objectlistjsonloadsstrkeys)Zsample_responseZ oai_objectr r r &test_key_order_openai_object_rendering"s  r-) r)tempfilerrrrZfixturer rrr-r r r r s