Update app.py

app.py

@@ -18,6 +18,8 @@ import os
 hf_token = os.environ['HF_TOKEN']
 serp_token = os.environ['SERP_TOKEN']
 repo_id = "sentence-transformers/all-mpnet-base-v2"
+from datetime import datetime
+from langchain.tools import Tool
 
 
 
@@ -119,7 +121,13 @@ def main():
             name="search",
             func=search,
             description="use this tool to answer real time or current search related questions."
-)
+        )
+        # Define a new tool that returns the current datetime
+        datetime_tool = Tool(
+        name="Datetime",
+        func=lambda x: datetime.now().isoformat(),
+        description="Returns the current datetime",
+        )
         # tool for asking human
         human_ask_tool = CustomAskHumanTool()
         # agent prompt
@@ -129,7 +137,7 @@ def main():
         # initialize agent
         agent = initialize_agent(
             agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
-                    tools=[db_search_tool,search_tool],
+                    tools=[db_search_tool,search_tool,datetime_tool],
             llm=llm,
             verbose=True,
             max_iterations=5,