chore add datetime in prompt

pandasai/ee/agents/judge_agent/pipeline/judge_prompt_generation.py

@@ -1,3 +1,4 @@
+import datetime
 from typing import Any
 
 from pandasai.ee.agents.judge_agent.prompts.judge_agent_prompt import JudgeAgentPrompt
@@ -30,8 +31,14 @@ def execute(self, input_data: JudgePipelineInput, **kwargs) -> Any:
         self.context = kwargs.get("context")
         self.logger: Logger = kwargs.get("logger")
 
+        now = datetime.datetime.now()
+        human_readable_datetime = now.strftime("%A, %B %d, %Y %I:%M %p")
+
         prompt = JudgeAgentPrompt(
-            query=input_data.query, code=input_data.code, context=self.context
+            query=input_data.query,
+            code=input_data.code,
+            context=self.context,
+            date=human_readable_datetime,
         )
         self.logger.log(f"Using prompt: {prompt}")
 

pandasai/ee/agents/judge_agent/prompts/templates/judge_agent_prompt.tmpl

@@ -1,3 +1,4 @@
+Today is {{date}}
 ### QUERY
 {{query}}
 ### GENERATED CODE

tests/unit_tests/ee/judge_agent/test_judge_prompt_gen.py

@@ -1,3 +1,4 @@
+import re
 from typing import Optional
 from unittest.mock import patch
 
@@ -155,9 +156,16 @@ def test_validate_input_semantic_prompt(self, sample_df, context, logger):
             input_data=input_data, context=context, logger=logger
         )
 
+        match = re.search(
+            r"Today is ([A-Za-z]+, [A-Za-z]+ \d{1,2}, \d{4} \d{2}:\d{2} [APM]{2})",
+            response.output.to_string(),
+        )
+        datetime_str = match.group(1)
+
         assert (
             response.output.to_string()
-            == """### QUERY
+            == f"""Today is {datetime_str}
+### QUERY
 What is test?
 ### GENERATED CODE
 print('Code Data')