Ricky_fan commited on
Commit
4c7b2d9
·
1 Parent(s): c8aba9a
examples/agent_api_web_demo.py CHANGED
@@ -2,7 +2,7 @@ import copy
2
  import os
3
  from typing import List
4
  import streamlit as st
5
- from actions import ArxivSearch, WeatherQuery #0214 add weather
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
 
2
  import os
3
  from typing import List
4
  import streamlit as st
5
+ from lagent.actions import ArxivSearch, WeatherQuery #0214 add weather
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
examples/multi_agents_api_web_demo.py CHANGED
@@ -5,7 +5,7 @@ import re
5
  import requests
6
  import streamlit as st
7
 
8
- from agents import Agent
9
  from lagent.prompts.parsers import PluginParser
10
  from agents.stream import PLUGIN_CN, get_plugin_prompt
11
  from lagent.schema import AgentMessage
 
5
  import requests
6
  import streamlit as st
7
 
8
+ from lagent.agents import Agent
9
  from lagent.prompts.parsers import PluginParser
10
  from agents.stream import PLUGIN_CN, get_plugin_prompt
11
  from lagent.schema import AgentMessage