1 files changed, 48 insertions, 0 deletions
diff --git a/benchmark/agbenchmark/reports/processing/report_types_v2.py b/benchmark/agbenchmark/reports/processing/report_types_v2.py
new file mode 100644
index 000000000..b26adaa6d
--- /dev/null
+++ b/benchmark/agbenchmark/reports/processing/report_types_v2.py
@@ -0,0 +1,48 @@
+"""Model definitions for use in the API"""
+
+from pydantic import BaseModel, constr
+
+datetime_format = r"^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\+00:00$"
+
+
+class TaskInfo(BaseModel):
+    data_path: str
+    is_regression: bool | None
+    answer: str
+    description: str
+    category: list[str]
+    task: str
+
+
+class RepositoryInfo(BaseModel):
+    repo_url: str | None = None
+    team_name: str | None = None
+    agent_git_commit_sha: str | None = None
+    benchmark_git_commit_sha: str | None = None
+
+
+class Metrics(BaseModel):
+    cost: float | None = None
+    success: bool
+    attempted: bool
+    difficulty: str | None = None
+    run_time: str | None = None
+    fail_reason: str | None = None
+    success_percentage: float | None = None
+
+
+class RunDetails(BaseModel):
+    test_name: str
+    run_id: str | None = None
+    command: str
+    completion_time: str | None = None
+    benchmark_start_time: constr(regex=datetime_format)
+
+
+class BenchmarkRun(BaseModel):
+    repository_info: RepositoryInfo
+    run_details: RunDetails
+    task_info: TaskInfo
+    metrics: Metrics
+    reached_cutoff: bool | None = None
+    config: dict[str, str | dict[str, str]]