Fd' UddlmZddlZddlZddlmZddlmZmZm Z ddl m Z ddl Z ddl mcmcmZddlmZmZddlmZmZmZddlmZerdd lmZdd lmZmZmZm Z m!Z!dd l"m#Z#d ed dddeddddeddddeddddeddddeddddeddddfDZ$d dddd Z%e%&D]$\Z'Z(edKie$e(j)Z*e'e*_+e*e$e'<%d!e d"d#d#d$fDZ,d%ed&d'd(d)*fDZ-ie$e,e-Z.d+e/d,<d-Z0 dLdMd7Z1e0e1dNd<Z2e0e1dOd?Z3e0e1dPdBZ4eGdCdDZ5eGdEdFZ6dQdJZ7dS)R) annotationsN) dataclass) TYPE_CHECKINGListOptional)patch)ForeStyle)APIErrorRateLimitErrorTimeout) OpenAIObject)Agent) ChatModelInfoEmbeddingModelInfo MessageDict TextModelInfoTText)loggerci|] }|j| Sname.0infos PC:\Users\Administrator\Downloads\Auto-GPT-master\autogpt\llm\providers\openai.py rs,...  It...zgpt-3.5-turbo-0301g~jtX?gMb`?i)rprompt_token_costcompletion_token_cost max_tokenszgpt-3.5-turbo-0613zgpt-3.5-turbo-16k-0613g~jth?gMbp?i@z gpt-4-0314gQ?gQ?i z gpt-4-0613zgpt-4-32k-0314gQ?izgpt-4-32k-0613)z gpt-3.5-turbozgpt-3.5-turbo-16kzgpt-4z gpt-4-32kci|] }|j| Srrrs rrrVs,     It   rztext-davinci-003g{Gz?ici|] }|j| Srrrs rrrbs,     It   rztext-embedding-ada-002g-C6?ii)rr r"embedding_dimensionsz=dict[str, ChatModelInfo | EmbeddingModelInfo | TextModelInfo]OPEN_AI_MODELScvddlm}|tjjdfd fdfd}|S) zAAdds ApiManager metering to functions which make OpenAI API callsr) ApiManagerresponserc: |j}tjd|jd||jjd|vr |jjnd|jdS#t$r/}tjd|j j d|Yd}~dSd}~wwxYw)Nz"Reported usage from call to model z: completion_tokensrzFailed to update API costs: ) usagerdebugmodel update_cost prompt_tokensr+ Exceptionwarn __class____name__)r)r,err api_managers rupdate_usage_with_responsez-meter_api..update_usage_with_response}s XNE LWhnWWPUWW X X X  # #,4G54P4P00VW       X X X KVs}7MVVQTVV W W W W W W W W W XsAA!! B+$BBc`|i|}t|trd|vr ||S)Nr,) isinstancer)argskwargs openai_objopenai_obj_processorr7s rmetering_wrapperz#meter_api..metering_wrappersK))4:6:: j, / / 3Gz4I4I & &z 2 2 2rctjtjd5|i|cdddS#1swxYwYdS)Nconvert_to_openai_object) side_effect)robjectengine_api_resourceutil)r:r;funcr>s r metered_funczmeter_api..metered_funcs \  $ &(    ) ) 4(((  ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s 8<<)r)r)autogpt.llm.api_managerr(openairDr@)rEr(rFr6r>r=r7s` @@@@r meter_apirIus222222*,,K!;? X X X X X X )))))) r @T num_retriesint backoff_basefloat warn_userboolctjdtjdtjtjzdtjdtjdtjtjdtjfd}|S)zRetry an OpenAI API call. Args: num_retries int: Number of retries. Defaults to 10. backoff_base float: Base for exponential backoff. Defaults to 2. warn_user bool: Whether to warn the user. Defaults to True. z%Error: Reached rate limit, passing...z*Please double check that you have setup a PAIDz- OpenAI API Account. You can read more here: z.https://docs.agpt.co/setup/#getting-an-api-keyz4Error: API Bad gateway. Waiting {backoff} seconds...cTtjfd}|S)Nc } dz}td|dzD]}  |i|cS#t$r6||krtj |stjd}Yn-t t f$r}|jdvs||krYd}~nd}~wwxYw|dzz}tj |tj |dS)NT)iii)backoff) ranger rr- double_checkr r http_statusformattimesleep)r:r; user_warned num_attemptsattempterXapi_key_error_msgrN backoff_msgrErLretry_limit_msgrPs r_wrappedz-retry_api.._wrapper.._wrappeds@'-K&?L L1$455 $ $4000000%+++,..L111&++,=>>>&*  '* _<<<//0//// '7Q;7 [///@@AAA 7####+ $ $s*=B)B9BB) functoolswraps)rErfrcrNrdrLrerPs` r_wrapperzretry_api.._wrappersW    $ $ $ $ $ $ $ $ $ $   $4r)r REDRESETCYANr BRIGHT RESET_ALL)rLrNrPrircrdres``` @@@r retry_apirosXXDJXXO a 9u| # a a). a a9 a aTXT^ a a 8WW4:WW< OrmessagesList[MessageDict]returnrctjjdd|i|}t|dst jd||S)aCreate a chat completion using the OpenAI API Args: messages: A list of messages to feed to the chatbot. kwargs: Other arguments to pass to the OpenAI API chat completion call. Returns: OpenAIObject: The ChatCompletion response from OpenAI rperrorz Response: r)rHChatCompletioncreatehasattrrr-)rp_r; completions rcreate_chat_completionrzsa &4;      J :w ' '0 .*../// rpromptstrc2tjjdd|i|S)aCreate a text completion using the OpenAI API Args: prompt: A text prompt to feed to the LLM kwargs: Other arguments to pass to the OpenAI API text completion call. Returns: OpenAIObject: The Completion response from OpenAI r{r)rH Completionrv)r{rxr;s rcreate_text_completionrs1   #      rinput%str | TText | List[str] | List[TText]c2tjjdd|i|S)zCreate an embedding using the OpenAI API Args: input: The text to embed. kwargs: Other arguments to pass to the OpenAI API embedding call. Returns: OpenAIObject: The Embedding response from OpenAI rr)rH Embeddingrv)rrxr;s rcreate_embeddingrs1   "      rc(eZdZUdZded<ded<dS)OpenAIFunctionCallzRepresents a function call as generated by an OpenAI model Attributes: name: the name of the function that the LLM wants to call arguments: a stringified JSON object (unverified) containing `arg: value` pairs r|r argumentsN)r4 __module__ __qualname____doc____annotations__rrrrrs.IIINNNNNrrcreZdZUdZded<ded<ded<eGddZed Zd S) OpenAIFunctionSpeczKRepresents a "function" in OpenAI, which is mapped to a Command in Auto-GPTr|r descriptionzdict[str, ParameterSpec] parametersc<eZdZUded<ded<ded<dZded<d S) OpenAIFunctionSpec.ParameterSpecr|rtypez Optional[str]rFrQrequiredN)r4rrrrrrr ParameterSpecr+sA  """"rrc|j|jdd|jDd|jDddS)z2Output an OpenAI-consumable function specificationrBc8i|]}|j|j|jdS))rr)rrrrparams rrz/OpenAIFunctionSpec.__dict__..:sA  J % ','8!!rc*g|]}|j |jSr)rrrs r z/OpenAIFunctionSpec.__dict__..As1#(Jr)r propertiesrrrr)rrrvalues)selfs r__dict__zOpenAIFunctionSpec.__dict__2sI+  "&!7!7!9!9 ,0O,B,B,D,D     rN) r4rrrrrrpropertyrrrrrr#sUU III((((Y   X   rragentrlist[OpenAIFunctionSpec]cn|jjsgSd|jjDS)zGet OpenAI-consumable function specs for the agent's available commands. see https://platform.openai.com/docs/guides/gpt/function-calling cdg|]-}t|j|jd|jD.S)c ~i|]:}|jt|j|j|j|j;S))rrrr)rrrrrrrs rrz7get_openai_command_specs...SsY  .<<"^ % 1 =rr)rrrr)rcommands rrz,get_openai_command_specs..Ose     +%/    r)configopenai_functionscommand_registrycommandsr)rs rget_openai_command_specsrHsK < (   -6==??   rr)rJrKT)rLrMrNrOrPrQ)rprqrrr)r{r|rrr)rrrrr)rrrrr)8 __future__rrgr] dataclassesrtypingrrr unittest.mockrrH1openai.api_resources.abstract.engine_api_resource api_resourcesabstractrCcoloramar r openai.errorr r r openai.openai_objectrautogpt.agent.agentrautogpt.llm.baserrrrr autogpt.logsrOPEN_AI_CHAT_MODELSchat_model_mappingitemsaliastargetr alias_inforOPEN_AI_TEXT_MODELSOPEN_AI_EMBEDDING_MODELSr&rrIrorzrrrrrrrrrs""""""" !!!!!!0000000000 OOOOOOOOOOOO ::::::::::------*)))))) ..  %$"'     %$"'     )#"'     ""&     ""&     !""&     !""&    K+...b*1 !  (--//,,ME6FF!4V!