| from datetime import datetime | |
| from lagent.actions import ActionExecutor, BingBrowser | |
| from lagent.llms import INTERNLM2_META, LMDeployServer | |
| from mindsearch.agent.mindsearch_agent import (MindSearchAgent, | |
| MindSearchProtocol) | |
| from mindsearch.agent.mindsearch_prompt import ( | |
| FINAL_RESPONSE_CN, FINAL_RESPONSE_EN, GRAPH_PROMPT_CN, GRAPH_PROMPT_EN, | |
| searcher_context_template_cn, searcher_context_template_en, | |
| searcher_input_template_cn, searcher_input_template_en, | |
| searcher_system_prompt_cn, searcher_system_prompt_en) | |
| lang = 'cn' | |
| llm = LMDeployServer(path='internlm/internlm2_5-7b-chat', | |
| model_name='internlm2', | |
| meta_template=INTERNLM2_META, | |
| top_p=0.8, | |
| top_k=1, | |
| temperature=0, | |
| max_new_tokens=8192, | |
| repetition_penalty=1.02, | |
| stop_words=['<|im_end|>']) | |
| agent = MindSearchAgent( | |
| llm=llm, | |
| protocol=MindSearchProtocol( | |
| meta_prompt=datetime.now().strftime('The current date is %Y-%m-%d.'), | |
| interpreter_prompt=GRAPH_PROMPT_CN | |
| if lang == 'cn' else GRAPH_PROMPT_EN, | |
| response_prompt=FINAL_RESPONSE_CN | |
| if lang == 'cn' else FINAL_RESPONSE_EN), | |
| searcher_cfg=dict( | |
| llm=llm, | |
| plugin_executor=ActionExecutor( | |
| BingBrowser(searcher_type='DuckDuckGoSearch', topk=6)), | |
| protocol=MindSearchProtocol( | |
| meta_prompt=datetime.now().strftime( | |
| 'The current date is %Y-%m-%d.'), | |
| plugin_prompt=searcher_system_prompt_cn | |
| if lang == 'cn' else searcher_system_prompt_en, | |
| ), | |
| template=dict(input=searcher_input_template_cn | |
| if lang == 'cn' else searcher_input_template_en, | |
| context=searcher_context_template_cn | |
| if lang == 'cn' else searcher_context_template_en)), | |
| max_turn=10) | |
| for agent_return in agent.stream_chat('上海今天适合穿什么衣服'): | |
| pass | |