update

app.py

@@ -9,7 +9,7 @@ import datetime
 from dotenv import load_dotenv
 
 from pydantic import Field, BaseModel
-from vectara_agent.agent import Agent, AgentType, AgentStatusType
+from vectara_agent.agent import Agent, AgentStatusType
 from vectara_agent.tools import ToolsFactory
 from vectara_agent.tools_catalog import rephrase_text
 
@@ -19,7 +19,7 @@ languages = {'English': 'en', 'Spanish': 'es', 'French': 'fr', 'German': 'de', '
              'Hebrew': 'he', 'Hindi': 'hi', 'Italian': 'it', 'Japanese': 'ja', 'Korean': 'ko', 'Portuguese': 'pt'}
 initial_prompt = "How can I help you today?"
 
-load_dotenv()
+load_dotenv(override=True)
 
 def create_tools(cfg):        
 
@@ -78,7 +78,7 @@ def create_tools(cfg):
             [query_tool]
     )
 
-def initialize_agent(agent_type: AgentType, _cfg):
+def initialize_agent(_cfg):
     date = datetime.datetime.now().strftime("%Y-%m-%d")
     bot_instructions = f"""
     - You are a helpful teacher assistant, with expertise in education in various teaching styles.
@@ -95,7 +95,6 @@ def initialize_agent(agent_type: AgentType, _cfg):
         st.session_state.log_messages.append(output)
 
     agent = Agent(
-        agent_type=agent_type,
         tools=create_tools(_cfg),
         topic="An educator with expertise in philosophy",
         custom_instructions=bot_instructions,
@@ -103,12 +102,12 @@ def initialize_agent(agent_type: AgentType, _cfg):
     )
     return agent
 
-def launch_bot(agent_type: AgentType):
+def launch_bot():
     def reset():
         cfg = st.session_state.cfg
         st.session_state.messages = [{"role": "assistant", "content": initial_prompt, "avatar": "🦖"}]
         st.session_state.thinking_message = "Agent at work..."
-        st.session_state.agent = initialize_agent(agent_type, cfg)
+        st.session_state.agent = initialize_agent(cfg)
         st.session_state.log_messages = []
         st.session_state.show_logs = False
 
@@ -213,5 +212,5 @@ def launch_bot(agent_type: AgentType):
     
  
 if __name__ == "__main__":
-    launch_bot(agent_type=AgentType.OPENAI)
+    launch_bot()