Update agent.py

agent.py

@@ -62,7 +62,7 @@ def build_graph(tools: list, agent: AgentExecutor) -> StateGraph:
     
     return workflow.compile()
 
-class AIAgent:
+class AdvancedAIAgent:
     def __init__(self, model_name: str = "gpt-4-turbo"):
         self.tools = [wikipedia_search, web_search, calculate]
         self.llm = ChatOpenAI(model=model_name, temperature=0.7)
@@ -101,7 +101,7 @@ class AIAgent:
         return steps
 
 # Example usage
-if __name__ == "__main__":
-    agent = AIAgent()
-    response = agent("What's the capital of France?")
-    print(response)
+#if __name__ == "__main__":
+#    agent = AdvancedAIAgent()
+#    response = agent("What's the capital of France?")
+#    print(response)