Spaces:
Sleeping
Sleeping
sunxiaokang
commited on
Update app.py
Browse files暂时删除自定义action,找不到设置lagent的action的地方
app.py
CHANGED
@@ -2,7 +2,7 @@ import copy
|
|
2 |
import os
|
3 |
from typing import List
|
4 |
import streamlit as st
|
5 |
-
from lagent.actions import ArxivSearch
|
6 |
from lagent.prompts.parsers import PluginParser
|
7 |
from lagent.agents.stream import INTERPRETER_CN, META_CN, PLUGIN_CN, AgentForInternLM, get_plugin_prompt
|
8 |
from lagent.llms import GPTAPI
|
@@ -17,7 +17,6 @@ class SessionState:
|
|
17 |
# 初始化插件列表
|
18 |
action_list = [
|
19 |
ArxivSearch(),
|
20 |
-
WeatherQuery(),
|
21 |
]
|
22 |
st.session_state['plugin_map'] = {action.name: action for action in action_list}
|
23 |
st.session_state['model_map'] = {} # 存储模型实例
|
|
|
2 |
import os
|
3 |
from typing import List
|
4 |
import streamlit as st
|
5 |
+
from lagent.actions import ArxivSearch
|
6 |
from lagent.prompts.parsers import PluginParser
|
7 |
from lagent.agents.stream import INTERPRETER_CN, META_CN, PLUGIN_CN, AgentForInternLM, get_plugin_prompt
|
8 |
from lagent.llms import GPTAPI
|
|
|
17 |
# 初始化插件列表
|
18 |
action_list = [
|
19 |
ArxivSearch(),
|
|
|
20 |
]
|
21 |
st.session_state['plugin_map'] = {action.name: action for action in action_list}
|
22 |
st.session_state['model_map'] = {} # 存储模型实例
|