Fd"dZddlmZmZmZmZmZmZmZddl m Z edZ GddeZ Gdde Z d S) zHandles loading of plugins.)AnyDictListOptionalTuple TypedDictTypeVar)AutoGPTPluginTemplatePromptGeneratorc$eZdZUeed<eed<dS)MessagerolecontentN)__name__ __module__ __qualname__str__annotations__VC:\Users\Administrator\Downloads\Auto-GPT-master\autogpt\models\base_open_ai_plugin.pyr r s" III LLLLLrr c NeZdZdZdefdZdefdZdedefdZ defdZ d e de fd Z defd Z d e d eedeefd ZdefdZdedefdZdefdZd eedeefdZdefdZd eedeefdZdefdZdedefdZdefdZdedeeefdeeeeefffdZdefdZdededefdZ d eeefdede!de"def dZ#d eedede!de"def d Z$d!edefd"Z%d!ede&fd#Z'd$edefd%Z(d$edefd&Z)defd'Z*d(edd)fd*Z+d)S)+BaseOpenAIPluginzK This is a BaseOpenAIPlugin class for generating Auto-GPT plugins. manifests_specs_clientsc|dd|_|dd|_|dd|_|d|_|d|_|d|_dS)Nmanifestname_for_modelschema_versiondescription_for_modelclient openapi_spec)_name_version _description_client _manifest _openapi_spec)selfrs r__init__zBaseOpenAIPlugin.__init__sd,Z89IJ / ;rcdS)zThis method is called to check that the plugin can handle the post_instruction method. Returns: bool: True if the plugin can handle the post_instruction method.Frr,s rcan_handle_post_instructionz,BaseOpenAIPlugin.can_handle_post_instructionwr.rc|S)zThis method is called after the instruction chat is done. Args: response (str): The response. Returns: str: The resulting response. rrBs rpost_instructionz!BaseOpenAIPlugin.post_instruction~rDrcdS)zThis method is called to check that the plugin can handle the pre_command method. Returns: bool: True if the plugin can handle the pre_command method.Frr,s rcan_handle_pre_commandz'BaseOpenAIPlugin.can_handle_pre_commandr.r command_name argumentsc ||fS)aThis method is called before the command is executed. Args: command_name (str): The command name. arguments (Dict[str, Any]): The arguments. Returns: Tuple[str, Dict[str, Any]]: The command name and the arguments. r)r(rTrUs r pre_commandzBaseOpenAIPlugin.pre_commandsY&&rcdS)zThis method is called to check that the plugin can handle the post_command method. Returns: bool: True if the plugin can handle the post_command method.Frr,s rcan_handle_post_commandz(BaseOpenAIPlugin.can_handle_post_commandr.rc|S)zThis method is called after the command is executed. Args: command_name (str): The command name. response (str): The response. Returns: str: The resulting response. r)r(rTr/s r post_commandzBaseOpenAIPlugin.post_commands rmodel temperature max_tokenscdS)aThis method is called to check that the plugin can handle the chat_completion method. Args: messages (List[Message]): The messages. model (str): The model name. temperature (float): The temperature. max_tokens (int): The max tokens. Returns: bool: True if the plugin can handle the chat_completion method.Frr(r;r\r]r^s rcan_handle_chat_completionz+BaseOpenAIPlugin.can_handle_chat_completions urcdS)aEThis method is called when the chat completion is done. Args: messages (List[Message]): The messages. model (str): The model name. temperature (float): The temperature. max_tokens (int): The max tokens. Returns: str: The resulting response. Nrr`s rhandle_chat_completionz'BaseOpenAIPlugin.handle_chat_completionr>rtextcdS)a This method is called to check that the plugin can handle the text_embedding method. Args: text (str): The text to be convert to embedding. Returns: bool: True if the plugin can handle the text_embedding method.Frr(rds rcan_handle_text_embeddingz*BaseOpenAIPlugin.can_handle_text_embeddings urcdS)zThis method is called when the chat completion is done. Args: text (str): The text to be convert to embedding. Returns: list: The text embedding. Nrrfs rhandle_text_embeddingz&BaseOpenAIPlugin.handle_text_embeddingr>r user_inputcdS)zThis method is called to check that the plugin can handle the user_input method. Args: user_input (str): The user input. Returns: bool: True if the plugin can handle the user_input method.Frr(rjs rcan_handle_user_inputz&BaseOpenAIPlugin.can_handle_user_inputs urcdS)zThis method is called to request user input to the user. Args: user_input (str): The question or prompt to ask the user. Returns: str: The user input. Nrrls rrjzBaseOpenAIPlugin.user_inputr>rcdS)zThis method is called to check that the plugin can handle the report method. Returns: bool: True if the plugin can handle the report method.Frr,s rcan_handle_reportz"BaseOpenAIPlugin.can_handle_reports urmessageNcdS)zThis method is called to report a message to the user. Args: message (str): The message to report. Nr)r(rqs rreportzBaseOpenAIPlugin.reportr>r),rrr__doc__dictr)boolr-rr3r5r r8r:rr rr=r@rCrFrIrKrMrOrQrSrrrrWrYr[floatintrarcrglistrirmrjrprsrrrrrsEEEEECS/o % 15g #    $ccDW $w-4 tG} #    T ' ',0cN ' sDcN" # ' ' ' ' S#X /2 AF TW        W   .1  @E  SV       cd # $           S S    4 c d      rrN)rttypingrrrrrrr auto_gpt_plugin_templater r r rrrrr|s!!GGGGGGGGGGGGGGGGGG::::::'+,,i k k k k k ,k k k k k r