Update agent.py

agent.py

@@ -1,4 +1,6 @@
 # agent.py
+import os
+from dotenv import load_dotenv
 from typing import TypedDict, Annotated, Sequence
 from langchain_core.messages import BaseMessage, HumanMessage
 from langchain_core.tools import tool
@@ -12,6 +14,8 @@ from langchain.agents import AgentExecutor
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 import operator
 
+load_dotenv()
+
 class AgentState(TypedDict):
     messages: Annotated[Sequence[BaseMessage], operator.add]
     sender: str
@@ -101,7 +105,7 @@ class AdvancedAIAgent:
         return steps
 
 # Example usage
-#if __name__ == "__main__":
-#    agent = AdvancedAIAgent()
-#    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)