Update agent.py

agent.py

@@ -9,6 +9,7 @@ from langgraph.graph.message import add_messages
 from langchain_groq import ChatGroq
 from langgraph.graph import StateGraph, START, END
 from langgraph.prebuilt import ToolNode, tools_condition
+import os
 
 @tool
 def read_file(directory:str)->str:
@@ -130,8 +131,10 @@ tools = [read_file, wikipedia_fetcher, web_search, add, subtract, divide, multip
 class AgentState(TypedDict):
     messages: Annotated[List[AnyMessage], add_messages]
 
+api_key = os.getenv('GROQ_API_KEY')
+
 def call_model(state: AgentState):
-    llm = ChatGroq(model="qwen/qwen3-32b", temperature=0, api_key='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')
+    llm = ChatGroq(model="qwen/qwen3-32b", temperature=0, api_key=api_key)
     llm_with_tools = llm.bind_tools(tools)
     response = llm_with_tools.invoke(state['messages'])
     return {'messages': [response]}