o g^/@sddlZddlZddlZddlmZddlmZddlmZmZddl m Z ddl m Z ddl mZejddd lmZmZdd lmZeeZGd d d Zed krhedddZeZeedSdS)N)datetime) load_dotenv) AgentExecutorcreate_tool_calling_agent)ChatPromptTemplate)tool) ChatOpenAI.)OPENAI_MODEL_4OOPENAI_MODEL_MINI)PerplexityResearchc @seZdZdZeededdZeede ddZ edZ dZ dZ d ZiZ d!d ed efd dZddZddZdedefddZeed"ded efddZeed"deded efddZeed"ded efddZeed"ded efdd Zd S)# Researcherzresearcher classOPENAI_API_KEYg?)api_keymodel temperatureg?PERPLEXITY_KEYiNparent_questionpersonacCsD||_|r|nd|_||_|r|nd|_ttds dt_dSdS)zinitialize the classNresearch_text_combined)initial_questionprimary_research persona_input build_personapersona_extendedhasattrthread_local_datar)selfrrrr!:/var/www/html/XCapMarket/utils/researcher/researcher_v2.py__init__*s  zResearcher.__init__cCs4tddd|jdfdg}||jB}|ijS)N)systemz]You are a skillful LLM user and understand how to create personas to the best of your abilityhumanzBuild a persona for z1. This should include a system prompt for an LLM.)r$z!Only return the persona as string)r from_messagesrllm4oinvokecontent)r promptchainr!r!r"r7s  zResearcher.build_personacCs|jr dd|jfgng}|ddddddd d d d dd tdfdd|jdfdg 7}|j|j|j|jg}t ||j t |d}t ||dd}|id}||jtj}tjd|jd|7_tjS)Nr$zYour persona is: )r$zfYou have access to tools which allow you to search the web, verify research, and fill out the research)r$zGYou should use the tools and your own knowledge to answer the question.)r$zDFor complex questions, you should use the create research plan tool.)r$zJIf you use the web search tool, then you must use the verify research tool)r$zMIf the research is not complete, then you must use the fill out research tool)r$z/Make sure to return the citations when provided)r$zB*** Before you provide an answer, you must verify the research ***)r%zQYou are not allowed to exit the chain without providing the research and answers.)r$zJWhen you pass in the persona, you should make it at least 100 tokens long.)r$zWhen you use the web search tool, make your questions as specific as you can. Get the most relevant information and up to date information.zToday's date is: z%Y-%m-%dr%z)The question you are going to answer is: z/. You must use the tools to answer the question)r$z{agent_scratchpad})toolsllmr*T)agentr,verboseoutputz # Final Output for:  )rrnowstrftimer web_searchverify_researchfill_out_researchcreate_research_planrllm4rr&rr(summarize_researchrr)r messagesr,r.agent_executorresearchr!r!r"conduct_researchHsP /zResearcher.conduct_researchrr<cCsBddg}|d|dt|}|tjB}|||djS)Create a research plan. Args: question (str): The question to create a research plan for. persona (str, optional): The persona to use. Defaults to None. Returns: str: The research plan. )r$z\You are going to be provided with a question and associated research to answer the question.)r$zhYou need to understand the research and then answer how the question can be answered using the research.)r%z9The question you are trying to answer: {initial_question})r%z+The research you have access to: {research}rr<appendrr&r r'r(r))r rr<r:r*r+r!r!r"r9s     zResearcher.summarize_researchquestionc Csz8t}dg}|r|d|f|d|f||}||}ttds)dt_tjd|d|7_|WStyO}z t|WYd}~d Sd}~ww) a5Search the web for information. Args: question (str): The question to search for. persona (str, optional): The persona to use. Defaults to None. Returns: answer (dict): The answer to the question. answer: str, citations: list(url) )r$z!You are a helpful research agent.r$r%rrz # r1Nz?This search was unsuccessful, you can try a different question.) r rAask_perplexity pretty_printrrr Exceptionprint)rBrperplexity_researchr:responseresponse_prettyer!r!r"r4s&    zResearcher.web_searchcCshgd}|r |d|f|dd|f|dd|ft|}|tjB}|||djS)asVerify the research. Args: initial_question (str): The question to verify the research for. research (str): The research to verify. persona (str, optional): The persona to use. Defaults to None. Returns: str: The verification of the research. True or False and if False, an explanation as to why. ))r$zJYou are going to be tasked with verifying a piece of research is complete.)r$z?Your goal is to determine if the research answers the question.)r$zGYou will respond with True or False and if False, you will explain why.)r$ziIf you think the question can be answered more completely, then you should respond False and explain why.r$r%z#The question you are answering is: z#The research you are verifying is: r?r@)rr<rr:r*r+r!r!r"r5s   zResearcher.verify_researchlacking_researchcCsRgd}|r |d|f|dd|ft|}|tjB}|d|ijS)a#fill out the research This tool takes in the research that is currently lacking and fills it out with additional information. The tool should respond with a list of questions that can be asked to an LLM with sufficient information to get the correct answer. ))r$zPYou will be provided with what the research is missing and you will fill it out.)r$zaYou should respond in such a way that another LLM can use the information to answer the question.r$z{You will respond with a list of questions that can be asked to an LLM with sufficient information to get the correct answerr$r%z;The current research is missing the following information: rKr@)rKrr:r*r+r!r!r"r6s   zResearcher.fill_out_researchcCs^gd}|r |d|f|dd|ft|}|tjB}|d|ij}d|dS)r>))r$zhYou are going to be provided with a question and you will create a research plan to answer the question.)r$zSYou should respond with a step by step plan to comprehensively answer the question.rLr$r%zThe question is: rBzThe research plan is: zG. YOu are now ready to use the web research tool to answer the questionr@)rBrr:r*r+ chain_outputr!r!r"r7"s    zResearcher.create_research_plan)NNN)N)__name__ __module__ __qualname____doc__rosgetenvr r'r r8perplexity_keymax_number_of_questionssummary_token_limitmax_follow_up_levellinks_researchedstrr#rr=r9 staticmethodrr4r5r6r7r!r!r!r"r sD     L*!r __main__z=What is the TAM for the Contract Lifecycle Management market?zIndustry analyst)rr)rRsys threadingrdotenvrlangchain.agentsrrlangchain.promptsrlangchain.toolsrlangchain_openairpathrAconfigs.configr r utils.researcher.perplexityr localrr rNrr=resultsrFr!r!r!r"s2       1