Update agent.py

agent.py

@@ -33,6 +33,8 @@ HUGGINGFACEHUB_API_TOKEN = getenv("HUGGINGFACEHUB_API_TOKEN")
 SUPABASE_URL = os.environ.get("SUPABASE_URL")
 SUPABASE_SERVICE_ROLE_KEY = os.environ.get("SUPABASE_SERVICE_ROLE_KEY")
 TAVILY_API_KEY = os.environ.get("TAVILY_API_KEY")
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
+
 
 # load the system prompt from the file
 with open("prompt.txt", "r", encoding="utf-8") as f:
@@ -44,8 +46,7 @@ sys_msg = SystemMessage(content=system_prompt)
 
 # Loading the assistant
 #chat = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
-chat = ChatGoogleGenerativeAI(model="gemini-1.5-pro", temperature=0)
-
+chat = ChatGroq(model_name="mixtral-8x7b-32768", temperature=0)
 
 tools = [add, 
          substract, 
@@ -72,7 +73,6 @@ def simple_graph():
     ## Defining our nodes
     def assistant(state: MessagesState):
         """Assistant node"""
-        time.sleep(45)
         return {"messages": [chat_with_tools.invoke([sys_msg] + state["messages"])]}
 
     # def retriever(state: MessagesState):