Spaces:
Sleeping
Sleeping
File size: 2,023 Bytes
dc9e27a f3f614f dc9e27a f3f614f dc9e27a f3f614f dc9e27a f3f614f dc9e27a f3f614f dc9e27a f3f614f dc9e27a f3f614f dc9e27a f3f614f dc9e27a f3f614f dc9e27a f3f614f dc9e27a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
import os
import sys
from datetime import datetime
from lagent.actions import WebBrowser
from lagent.agents.stream import get_plugin_prompt
from lagent.llms import INTERNLM2_META, LMDeployServer
from lagent.prompts import InterpreterParser, PluginParser
from mindsearch.agent.mindsearch_agent import MindSearchAgent
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"
date = datetime.now().strftime("The current date is %Y-%m-%d.")
llm = LMDeployServer(
path="internlm/internlm2_5-7b-chat",
model_name="internlm2",
meta_template=INTERNLM2_META,
top_p=0.8,
top_k=1,
temperature=1.0,
max_new_tokens=8192,
repetition_penalty=1.02,
stop_words=["<|im_end|>", "<|action_end|>"],
)
plugins = [WebBrowser(searcher_type="BingSearch", topk=6)]
agent = MindSearchAgent(
llm=llm,
template=date,
output_format=InterpreterParser(template=GRAPH_PROMPT_CN if lang == "cn" else GRAPH_PROMPT_EN),
searcher_cfg=dict(
llm=llm,
plugins=plugins,
template=date,
output_format=PluginParser(
template=searcher_system_prompt_cn if lang == "cn" else searcher_system_prompt_en,
tool_info=get_plugin_prompt(plugins),
),
user_input_template=searcher_input_template_cn
if lang == "cn"
else searcher_input_template_en,
user_context_template=searcher_context_template_cn
if lang == "cn"
else searcher_context_template_en,
),
summary_prompt=FINAL_RESPONSE_CN if lang == "cn" else FINAL_RESPONSE_EN,
max_turn=10,
)
for agent_return in agent("上海今天适合穿什么衣服"):
pass
print(agent_return.sender)
print(agent_return.content)
print(agent_return.formatted["ref2url"])
|