Update agent.py

agent.py

@@ -1,3 +1,7 @@
+import os
+from dotenv import load_dotenv
+load_dotenv()
+
 from langchain_core.tools import tool
 from langgraph.graph import StateGraph, START, MessagesState
 from langgraph.prebuilt import tools_condition, ToolNode
@@ -5,6 +9,9 @@ from langchain_groq import ChatGroq
 from langchain_core.messages import HumanMessage, SystemMessage
 import math
 
+groq_api_key = os.getenv("GROQ_API_KEY")
+serpapi_api_key = os.getenv("SERPAPI_API_KEY")
+
 # -------------------------
 # Tools
 # -------------------------
@@ -82,7 +89,7 @@ sys_msg = SystemMessage(content=system_prompt)
 # Build LangGraph Agent
 # -------------------------
 def build_deepseek_graph():
-    llm = ChatGroq(model="deepseek-llm-67b", temperature=0.3)
+    llm = ChatGroq(model="deepseek-r1-distill-llama-70b", groq_api_key=groq_api_key)
 
     all_tools = [
         add, subtract, multiply, divide, modulus,
@@ -105,14 +112,3 @@ def build_deepseek_graph():
 
     ninu = builder.compile()  
     return ninu
-
-# -------------------------
-# Example Run
-# -------------------------
-if __name__ == "__main__":
-    ninu = build_deepseek_graph()  
-    user_question = "ترجم لي الجملة: Artificial intelligence is transforming education."
-    messages = [sys_msg, HumanMessage(content=user_question)]
-    result = ninu.invoke({"messages": messages})
-    for msg in result["messages"]:
-        print("\n", msg.content)