Update agent.py

agent.py

@@ -47,12 +47,18 @@ sys_msg = SystemMessage(content=system_prompt)
 
 # Loading the assistant
 #chat = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
-chat = ChatGoogleGenerativeAI(
-    model="gemini-2.5-flash-preview-04-17",
-    temperature=0,
-    max_retries=1,
-    rate_limiter=InMemoryRateLimiter(
-        requests_per_second=1/60,
+# chat = ChatGoogleGenerativeAI(
+#     model="gemini-2.5-flash-preview-04-17",
+#     temperature=0,
+#     max_retries=1,
+#     rate_limiter=InMemoryRateLimiter(
+#         requests_per_second=1/60,
+#     ),
+# )
+chat = ChatGroq(model_name="llama3-70b-8192",
+        temperature=0, 
+        rate_limiter=InMemoryRateLimiter(
+            requests_per_second=1/60,
     ),
 )
 
@@ -82,30 +88,13 @@ def simple_graph():
     def assistant(state: MessagesState):
         """Assistant node"""
         return {"messages": [chat_with_tools.invoke([sys_msg] + state["messages"])]}
-
-    # def retriever(state: MessagesState):
-    #     """Retriever node"""
-    #     similar_question = vector_store.similarity_search(state["messages"][0].content)
-
-    #     if similar_question:  # Check if the list is not empty
-    #         example_msg = HumanMessage(
-    #             content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
-    #         )
-    #         return {"messages": [sys_msg] + state["messages"] + [example_msg]}
-    #     else:
-    #         # Handle the case when no similar questions are found
-    #         return {"messages": [sys_msg] + state["messages"]}
-
     
     # Build graph / nodes
     builder = StateGraph(MessagesState)
-    #builder.add_node("retriever", retriever) # Retriever
     builder.add_node("assistant", assistant) # Assistant
     builder.add_node("tools", ToolNode(tools)) # Tools
     
     # Logic / edges
-    # builder.add_edge(START, "retriever")
-    # builder.add_edge("retriever", "assistant")
     builder.add_edge(START, "assistant")
     builder.add_conditional_edges("assistant", tools_condition)
     builder.add_edge("tools", "assistant")