a dgz@sddlZddlmZddlmZddlmZddlmZddl m Z eddd Z Gd d d eZ edd d Z GdddeZeddgededZdS)N)dedent)List)BaseOutputParser)PromptTemplate)ThoughtValidityreturncCstdddgtddS)z4Get the prompt for the Chain of Thought (CoT) chain.jinja2problem_descriptionthoughtsa You are an intelligent agent that is generating one thought at a time in a tree of thoughts setting. PROBLEM {{problem_description}} {% if thoughts %} THOUGHTS {% for thought in thoughts %} {{ thought }} {% endfor %} {% endif %} Let's think step by step. )template_formatinput_variablestemplate)rrstriprrp/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_experimental/tot/prompts.pyget_cot_prompt src@s6eZdZdZeedddZeeedddZdS) JSONListOutputParserz.Parse the output of a PROPOSE_PROMPT response.rcCsdS)NZ json_listrselfrrr_type+szJSONListOutputParser._typetextrcCsF|ddd}z t|WStjy@gYS0dS)z Parse the output of an LLM call.z```jsonz```N)splitrjsonloadsJSONDecodeError)rrZ json_stringrrrparse/s  zJSONListOutputParser.parseN) __name__ __module__ __qualname____doc__propertystrrrrrrrrr(srcCstdgdttddS)z,Get the prompt for the PROPOSE_PROMPT chain.r )r r na- You are an intelligent agent that is generating thoughts in a tree of thoughts setting. The output should be a markdown code snippet formatted as a JSON list of strings, including the leading and trailing "```json" and "```": ```json [ "", "", "" ] ``` PROBLEM {{ problem_description }} {% if thoughts %} VALID THOUGHTS {% for thought in thoughts %} {{ thought }} {% endfor %} Possible next {{ n }} valid thoughts based on the last valid thought: {% else %} Possible next {{ n }} valid thoughts based on the PROBLEM: {%- endif -%} )r r output_parserr)rrrrrrrrget_propose_prompt9sr'c@s2eZdZdZeedddZeedddZdS) CheckerOutputParserz1Parse and check the output of the language model.rcCs<|}d|vrtjSd|vr$tjSd|vr2tjStjSdS)z'Parse the output of the language model.INVALIDZ INTERMEDIATEZVALIDN)upperrr)ZVALID_INTERMEDIATEZ VALID_FINAL)rrrrrrhszCheckerOutputParser.parsercCsdS)NZtot_llm_checker_outputrrrrrrtszCheckerOutputParser._typeN) rr r!r"r$rrr#rrrrrr(es r(r r a$ You are an intelligent agent, validating thoughts of another intelligent agent. PROBLEM {problem_description} THOUGHTS {thoughts} Evaluate the thoughts and respond with one word. - Respond VALID if the last thought is a valid final solution to the problem. - Respond INVALID if the last thought is invalid. - Respond INTERMEDIATE if the last thought is valid but not the final solution to the problem. This chain of thoughts is)r rr&)rtextwraprtypingrZlangchain_core.output_parsersrZlangchain_core.promptsrZ"langchain_experimental.tot.thoughtrrrr'r(rZCHECKER_PROMPTrrrrs      ,