FddZddlmZddlZddlZddlmZddlmZm Z ddl Z ddl Z er ddl m Z ddlmZeeejdz ZGd d ZdS) zR A module that contains the AIConfig class object that contains the configuration ) annotationsN)Path) TYPE_CHECKINGOptional)CommandRegistry)PromptGeneratorzai_settings.yamlcXeZdZdZ dddZeefddZefddZ dddZ dS)AIConfigal A class object that contains the configuration information for the AI Attributes: ai_name (str): The name of the AI. ai_role (str): The description of the AI's role. ai_goals (list): The list of objectives the AI is supposed to complete. api_budget (float): The maximum dollar value for API calls (0.0 means infinite) Nai_namestrai_roleai_goals list | None api_budgetfloatreturnNonecb|g}||_||_||_||_d|_d|_dS)a Initialize a class instance Parameters: ai_name (str): The name of the AI. ai_role (str): The description of the AI's role. ai_goals (list): The list of objectives the AI is supposed to complete. api_budget (float): The maximum dollar value for API calls (0.0 means infinite) Returns: None N)r rrrprompt_generatorcommand_registry)selfr rrrs LC:\Users\Administrator\Downloads\Auto-GPT-master\autogpt\config\ai_config.py__init__zAIConfig.__init__"sA$  H    $8<8<ai_settings_file 'AIConfig'c t|d5}tj|tjpi}dddn #1swxYwYn#t$ri}YnwxYw|dd}|dd}d|d gD}|d d }t ||||S) a Returns class object with parameters (ai_name, ai_role, ai_goals, api_budget) loaded from yaml file if yaml file exists, else returns class with no parameters. Parameters: ai_settings_file (int): The path to the config yaml file. DEFAULT: "../ai_settings.yaml" Returns: cls (object): An instance of given cls object utf-8encoding)LoaderNr r rcg|]p}t|trJt|dddddnt|qS)z{}'r ") isinstancedictrstripreplace).0goals r z!AIConfig.load..Tsy   $%% CIIOOD ! ! ) )#r 2 2 : :3 C C CT   rrrr )openyamlload FullLoaderFileNotFoundErrorgetr )rfile config_paramsr rrrs rr0z AIConfig.load=sD &999 NT $ $t G G G M2  N N N N N N N N N N N N N N N    MMM  ##Ir22##Ir22  &))*b99     #&&|S99 (J???s3A#A AAA A A AAc|j|j|j|jd}t |dd5}t j||dddddS#1swxYwYdS)a  Saves the class parameters to the specified file yaml file path as a yaml file. Parameters: ai_settings_file(str): The path to the config yaml file. DEFAULT: "../ai_settings.yaml" Returns: None )r rrrwr r!T) allow_unicodeN)r rrrr.r/dump)rrconfigr4s rsavez AIConfig.save^s|| /    "C' : : : 8d Ifd$ 7 7 7 7 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8sAAArOptional[PromptGenerator]cd}ddlm}| ||}|j|_|j|_|j|_|j|_|j D],}| s| |}-|j rKtj}|dkrtj dntjd}|d |z }d |jd |jd |d }t!|jD]\} } || dzd| d z }|jdkr|d|jdz }||_|d||z }|S)a5 Returns a prompt to the user with the class information in an organized fashion. Parameters: None Returns: full_prompt (str): A string containing the initial prompt for the user including the ai_name, ai_role, ai_goals, and api_budget. zYour decisions must always be made independently without seeking user assistance. Play to your strengths as an LLM and pursue simple strategies with no legal complications.r)build_default_prompt_generatorNLinuxT)terse)prettyz The OS you are running on is: zYou are z,  z GOALS: z. r z4 It takes money to let you run. Your API budget is $z.3fz )autogpt.prompts.promptr>rgoalsr namerrolerpluginscan_handle_post_prompt post_promptexecute_local_commandsplatformsystemdistro enumeraterrgenerate_prompt_string) rr:r prompt_startr>pluginos_nameos_info full_promptir,s rconstruct_full_promptzAIConfig.construct_full_promptss   JIIIII  #==fEE !% $  $ ,0,A)n D DF0022 %112BCC    ( Io''Gg%%!----[---  HwHH HLp!1!6oo:J:OooS_ooo  // . .GAt ac--T--- -KK ?S hSWSbhhh hK 0O.EEfMMOOO r)r r Nr ) r rrrrrrrrr)rrrr)rrrr)N)rr<rr) __name__ __module__ __qualname____doc__r staticmethod SAVE_FILEr0r;rWrrr r s $ =====6%.@@@@\@@,588888,EI5555555rr )r[ __future__rosrLpathlibrtypingrrrNr/autogpt.models.command_registryrautogpt.prompts.generatorrrgetcwdr]r r^rrrfs#""""" ******** :??????999999 CYRY[[!!$66 7 7 QQQQQQQQQQr