Add current datetime

Author: BenConstable9

text_2_sql/autogen/agents/llm_agents/sql_query_correction_agent.yaml

@@ -20,3 +20,4 @@ system_message:
 tools:
   - sql_get_entity_schemas_tool
   - sql_query_execution_tool
+  - current_datetime_tool

text_2_sql/autogen/agents/llm_agents/sql_query_generation_agent.yaml

@@ -39,3 +39,4 @@ tools:
   - sql_query_execution_tool
   - sql_get_entity_schemas_tool
   - sql_query_validation_tool
+  - current_datetime_tool

text_2_sql/autogen/utils/llm_agent_creator.py

@@ -6,6 +6,7 @@
 from utils.sql import query_execution, get_entity_schemas, query_validation
 from utils.models import MINI_MODEL
 from jinja2 import Template
+from datetime import datetime
 
 
 class LLMAgentCreator:
@@ -40,6 +41,11 @@ def get_tool(cls, tool_name):
                 query_validation,
                 description="Validates the SQL query to ensure that it is syntactically correct for the target database engine. Use this BEFORE executing any SQL statement.",
             )
+        elif tool_name == "current_datetime_tool":
+            return FunctionTool(
+                lambda: datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+                description="Gets the current date and time.",
+            )
         else:
             raise ValueError(f"Tool {tool_name} not found")