Only Push to gdirve correct timestamp

agbenchmark/reports/ReportManager.py

@@ -63,7 +63,7 @@ def end_info_report(self, config: Dict[str, Any]) -> None:
             "command": command.split(os.sep)[-1],
             "benchmark_git_commit_sha": BENCHMARK_GIT_COMMIT_SHA,
             "agent_git_commit_sha": AGENT_GIT_COMMIT_SHA,
-            "completion_time": datetime.now(timezone.utc).isoformat(),
+            "completion_time": datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%S+00:00"),
             "benchmark_start_time": BENCHMARK_START_TIME,
             "metrics": {
                 "run_time": str(round(time.time() - self.start_time, 2)) + " seconds",

agbenchmark/start_benchmark.py

@@ -18,7 +18,7 @@
 )
 
 CURRENT_DIRECTORY = Path(__file__).resolve().parent
-BENCHMARK_START_TIME = datetime.now(timezone.utc).isoformat()
+BENCHMARK_START_TIME = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%S+00:00")
 if os.environ.get("HELICONE_API_KEY"):
     HeliconeLockManager.write_custom_property(
         "benchmark_start_time", BENCHMARK_START_TIME

send_to_googledrive.py

@@ -7,6 +7,8 @@
 from dotenv import load_dotenv
 from oauth2client.service_account import ServiceAccountCredentials
 
+import re
+
 # Load environment variables from .env file
 load_dotenv()
 
@@ -109,7 +111,12 @@ def process_test(
                     # Load the JSON data from the file
                     with open(report_path, "r") as f:
                         data = json.load(f)
+                    benchmark_start_time = data.get("benchmark_start_time", "")
 
+                    # Check if benchmark_start_time complies with the required format
+                    pattern = re.compile(r'\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\+00:00')
+                    if not pattern.fullmatch(benchmark_start_time):
+                        continue  # Skip processing this report if the date is not in the correct format
                     # Loop through each test
                     for test_name, test_info in data["tests"].items():
                         process_test(test_name, test_info, agent_dir, data)