Update agent.py

agent.py

@@ -46,8 +46,8 @@ import torch.nn.functional as F
 from langchain.agents import initialize_agent, AgentType
 from langchain_community.chat_models import ChatOpenAI
 from langchain_community.tools import Tool
-
 import time
+from langchain_mistralai.chat_models import ChatMistralAI
 
 load_dotenv()
 
@@ -449,7 +449,8 @@ wikiq_tool = wikidata_query
 tools = [wiki_tool, calc_tool, file_tool, web_tool, arvix_tool, youtube_tool, video_tool, analyze_tool, wikiq_tool]
 
 # Define the LLM before using it
-llm = ChatOpenAI(temperature=0, model="gpt-4") # or "gpt-3.5-turbo"
+#llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo") # or "gpt-3.5-turbo" "gpt-4"
+llm = ChatMistralAI(model="mistral-7b-instruct-v0.1")
 
 # Create an agent using the planner, task classifier, and decision logic
 agent = initialize_agent(
@@ -461,6 +462,7 @@ agent = initialize_agent(
 
 
 
+
 # -------------------------------
 # Step 8: Use the Planner, Classifier, and Decision Logic
 # -------------------------------