update agent_api_web_demo.py

examples/agent_api_web_demo.py

@@ -2,7 +2,7 @@ import copy
 import os
 from typing import List
 import streamlit as st
-from lagent.actions import ArxivSearch,WeatherQuery
+from lagent.actions import ArxivSearch
 from lagent.prompts.parsers import PluginParser
 from lagent.agents.stream import INTERPRETER_CN, META_CN, PLUGIN_CN, AgentForInternLM, get_plugin_prompt
 from lagent.llms import GPTAPI
@@ -17,7 +17,6 @@ class SessionState:
         # 初始化插件列表
         action_list = [
             ArxivSearch(),
-            WeatherQuery(),
         ]
         st.session_state['plugin_map'] = {action.name: action for action in action_list}
         st.session_state['model_map'] = {}  # 存储模型实例
@@ -46,11 +45,11 @@ class StreamlitUI:
 
     def init_streamlit(self):
         """初始化 Streamlit 的 UI 设置。"""
-        # st.set_page_config(
-        #     layout='wide',
-        #     page_title='lagent-web',
-        #     page_icon='./docs/imgs/lagent_icon.png'
-        # )
+        st.set_page_config(
+            layout='wide',
+            page_title='lagent-web',
+            page_icon='./docs/imgs/lagent_icon.png'
+        )
         st.header(':robot_face: :blue[Lagent] Web Demo ', divider='rainbow')
 
     def setup_sidebar(self):
@@ -128,16 +127,16 @@ class StreamlitUI:
 
 def main():
     """主函数，运行 Streamlit 应用。"""
-    if 'ui' not in F.session_state:
+    if 'ui' not in st.session_state:
         session_state = SessionState()
         session_state.init_state()
         st.session_state['ui'] = StreamlitUI(session_state)
     else:
-        # st.set_page_config(
-        #     layout='wide',
-        #     page_title='lagent-web',
-        #     page_icon='./docs/imgs/lagent_icon.png'
-        # )
+        st.set_page_config(
+            layout='wide',
+            page_title='lagent-web',
+            page_icon='./docs/imgs/lagent_icon.png'
+        )
         st.header(':robot_face: :blue[Lagent] Web Demo ', divider='rainbow')
 
     # 设置侧边栏并获取模型和插件信息