Update agent.py

agent.py

@@ -15,6 +15,7 @@ from prompt import system_prompt
 
 # gpt-4.1-nano (cheaper for debugging) with temperature 0 for less randomness
 # for benchmarking use 04-mini (better reasoning) or gpt-4.1-mini
+
 llm = ChatOpenAI(
     model="gpt-4.1-nano",
     api_key=os.getenv("OPENAI_API_KEY"),
@@ -95,7 +96,7 @@ builder.add_edge("environment", "llm_call")  # after running the tools go back t
 
 gaia_agent = builder.compile()  # converts my builder into a runnable agent by using gaia_agent.invoke()
 
-# Call the Agent
+# Wrapper class to initialize and call the LangGraph agent with a user question
 class LangGraphAgent:
     def __init__(self):
         print("LangGraphAgent initialized.")