FdddlmZddlmZddlmZmZmZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZmZdd lmZdd lmZmZmZd d lT d+d,dZd-d Z d.d/d&Zd0d*ZdS)1) annotations)asdict)ListLiteralOptional)Fore)Config)logger) ApiManager)ChatModelResponse ChatSequenceMessage)openai)OPEN_AI_CHAT_MODELSOpenAIFunctionCallOpenAIFunctionSpec)*Nfunctionstrargslist descriptionmodel Optional[str]configOptional[Config]returnc ||j}d|D}d|}tj|t dd|d|dt d|g}t |d | jS) aCall an AI function This is a magic function that can do anything with no-code. See https://github.com/Torantulino/AI-Functions for more info. Args: function (str): The function to call args (list): The arguments to pass to the function description (str): The description of the function model (str, optional): The model to use. Defaults to None. Returns: str: The response from the function Nc4g|]}|t|ndS)NNone)r).0args NC:\Users\Administrator\Downloads\Auto-GPT-master\autogpt\llm\utils\__init__.py z$call_ai_function...s& D D DCHHHV D D Dz, systemz0You are now the following python function: ```#  z+``` Only respond with your `return` value.userr)prompt temperaturer)smart_llm_modeljoinr for_modelrcreate_chat_completioncontent)rrrrrarg_strr+s r%call_ai_functionr3s* }& D Dt D D DD99T??G  # M;MMMMM   FG $ $    F "Qv N N N VVr'r+r r,Optional[float]max_output_tokens Optional[int]c||j}||j}|jrd||i}nd|i}t jdd|i||||jd}tjd||j dj S)N deployment_idrr+)r, max_tokensapi_key Response: r) fast_llm_modelr, use_azure!get_azure_deployment_id_for_modeliopenaicreate_text_completionopenai_api_keyr debugchoicestext)r+rrr,r5kwargsresponses r%rArA@s }%(  "!6#K#KE#R#RS5!-  $% H L(h(()))  A  ##r'r functions"Optional[List[OpenAIFunctionSpec]]r9r c | |jj}||j}tjt jd|d|d|t j|||d}|jD]J}|j dd| i|r'|j dd| i|}||cSK|j |d<|j r|||d<|r,d |D|d <tjd |d tjdd| i|} tjd | t#| d r-tj| jt'| j| jdj} | d} | d} |jD],}|s|| } -t3t4|| | S)aCreate a chat completion using the OpenAI API Args: messages (List[Message]): The messages to send to the chat completion model (str, optional): The model to use. Defaults to None. temperature (float, optional): The temperature to use. Defaults to 0.9. max_tokens (int, optional): The max tokens to use. Defaults to None. Returns: str: The response from the chat completion Nz$Creating chat completion with model z, temperature z , max_tokens )rr,r9messagesr:r8cg|] }|j Sr<)__dict__)r#rs r%r&z*create_chat_completion..s(/ / / "*H / / / r'rHzFunction dicts: r;errorrr1 function_call) model_infor1rOr<)rnamer,r rCrGREENRESETpluginscan_handle_chat_completionrawhandle_chat_completionrBr>r?r@r0hasattrrN RuntimeErrorrDmessagegetcan_handle_on_response on_responser r) r+rrHrr,r9chat_completion_kwargspluginrZrG first_messager1rOs r%r0r0^s( } !(  L : C C5 C CP[ C Cjt CvzwA C C"  .   ,6 ,  ZZ\\ $  4f3(G"(.(=9% <  4 4U ; ;   O/ / .7/ / / {+  M(>{(KMMNNN- H L(h(()))x!!+ X^$$$8>***$Q'/M'++I66G/zcheck_model..s, 1 1Q:4 1 1 1 1 1 1r'z WARNING: zYou do not have access to z . Setting z to gpt-3.5-turbo.z gpt-3.5-turbo)r get_modelsanyr typewriter_logrYELLOW)rarb api_managermodelss` r% check_modelros,,K  # # % %F 1 1 1 1& 1 1 111   Z  :    ?r')NN) rrrrrrrrrrrr) r+rrr rrr,r4r5r6rr)NNNN)r+rrr rHrIrrr,r4r9r6rr )rarrbrcrr) __future__r dataclassesrtypingrrrcoloramarautogpt.configr autogpt.logsr rmr baser rr providersrr@providers.openairrr token_counterr3rAr0ror<r'r%rzs""""""**********!!!!!!$$$$$$;;;;;;;;;;))))))  # 'W'W'W'W'WT$$$$B59#' $ PPPPPfr'