FdcdZddlZddlmZmZmZmZmZmZddl m Z ddl m Z erddl mZGddZd e d efd ZdS) z/ A module for generating custom prompt strings.N) TYPE_CHECKINGAnyCallableDictListOptional)Config)llm_response_schema)CommandRegistryc eZdZdZddZdeddfdZ ddeded eeddfd Z d e ee fdefd Z d eddfdZ deddfdZddee defdZdedefdZdS)PromptGeneratorz A class for generating custom prompt strings based on constraints, commands, resources, and performance evaluations. returnNcvg|_g|_g|_g|_g|_d|_d|_d|_dS)z Initialize the PromptGenerator object with empty lists of constraints, commands, resources, and performance evaluations. NBobAI) constraintscommands resourcesperformance_evaluationgoalscommand_registrynamerole)selfs MC:\Users\Administrator\Downloads\Auto-GPT-master\autogpt\prompts\generator.py__init__zPromptGenerator.__init__sC  &(# 8<  constraintc:|j|dS)z Add a constraint to the constraints list. Args: constraint (str): The constraint to be added. N)rappend)rrs radd_constraintzPromptGenerator.add_constraint s!  +++++r command_label command_namefunctionc|i}d|D}||||d}|j|dS)a Add a command to the commands list with a label, name, and optional arguments. Args: command_label (str): The label of the command. command_name (str): The name of the command. args (dict, optional): A dictionary containing argument names and their values. Defaults to None. function (callable, optional): A callable function to be called when the command is executed. Defaults to None. Nci|]\}}|| Sr').0arg_key arg_values r z/PromptGenerator.add_command..>sRRR/Aw RRRr)labelrargsr$)itemsrr )rr"r#r-r$ command_argscommands r add_commandzPromptGenerator.add_command)sa$ <DRRTZZ\\RRR #      W%%%%%rr0cdd|dD}|dd|dd|S)z Generate a formatted string representation of a command. Args: command (dict): A dictionary containing command information. Returns: str: The formatted command string. z, c3.K|]\}}d|d|dVdS)"z": "Nr')r(keyvalues r z;PromptGenerator._generate_command_string..SsK  &0c5 ! ! ! ! ! !      rr-r,z: "rz ", args: )joinr.)rr0 args_strings r_generate_command_stringz(PromptGenerator._generate_command_stringIsjii  4;FO4I4I4K4K     '"NNwvNNNNNrresourcec:|j|dS)z| Add a resource to the resources list. Args: resource (str): The resource to be added. N)rr )rr;s r add_resourcezPromptGenerator.add_resourceXs  h'''''r evaluationc:|j|dS)z Add a performance evaluation item to the performance_evaluation list. Args: evaluation (str): The evaluation item to be added. N)rr )rr>s radd_performance_evaluationz*PromptGenerator.add_performance_evaluationas! #**:66666rlistr.cJ|dkrqg}jr+|djjDz }|fd|Dz }ddt |DSddt |DS)aF Generate a numbered list from given items based on the item_type. Args: items (list): A list of items to be numbered. item_type (str, optional): The type of items in the list. Defaults to 'list'. Returns: str: The formatted numbered list. r0c:g|]}|j t|Sr')enabledstr)r(items r z;PromptGenerator._generate_numbered_list..ys6$$$|$II$$$rc:g|]}|Sr')r:)r(rFrs rrGz;PromptGenerator._generate_numbered_list..s'VVV = =d C CVVVr c30K|]\}}|dzd|VdSz. Nr'r(irFs rr7z:PromptGenerator._generate_numbered_list..s8XX'!T!--t--XXXXXXrc30K|]\}}|dzd|VdSrKr'rMs rr7z:PromptGenerator._generate_numbered_list..s8NN'!T!--t--NNNNNNr)rrvaluesr8 enumerate)rr. item_typecommand_stringss` r_generate_numbered_listz'PromptGenerator._generate_numbered_listjs  ! ! O$ $$ $ 5 > E E G G$$$ VVVVPUVVV VO99XXY=W=WXXXXX X99NNYu=M=MNNNNN Nrconfigcd||jdt||d||jd||jdt jt|d S)z Generate a prompt string based on the constraints, commands, resources, and performance evaluations. Returns: str: The generated prompt string. z Constraints:  z Resources: z Performance Evaluation: zD Respond with only valid JSON conforming to the following schema: rI)rTrgenerate_commandsrrjsondumpsr rrUs rgenerate_prompt_stringz&PromptGenerator.generate_prompt_strings ;T99$:JKK ; ; v.. ; ;77GG ; ;++D,GHH  ; ; z-f5566  ; ; ; r)rN)NN)rA)__name__ __module__ __qualname____doc__rrEr!rrr1rrr:r=r@rrTr r\r'rrr r si    ,,,,,,'+ &&&& 8$ &  &&&&@ OS#X O3 O O O O(S(T((((7S7T7777OOT#YOSOOOO4 V       rr rUrcT|jrdSd||jddS)z Generate a prompt string based on the constraints, commands, resources, and performance evaluations. Returns: str: The generated prompt string. z Commands: r0)rRrW)openai_functionsrTrr[s rrXrXsIr R  ' '  ' K K R R Rr)r`rYtypingrrrrrrautogpt.configr autogpt.json_utils.utilitiesr autogpt.models.command_registryr r rErXr'rrrhs55 EEEEEEEEEEEEEEEE!!!!!!<<<<<<@??????H H H H H H H H V F s      r