diff --git a/aobench/scenario-server/src/scenario_server/endpoints.py b/aobench/scenario-server/src/scenario_server/endpoints.py index 56f6d4be..1f7a7c19 100644 --- a/aobench/scenario-server/src/scenario_server/endpoints.py +++ b/aobench/scenario-server/src/scenario_server/endpoints.py @@ -1,5 +1,6 @@ import asyncio import logging +import os import uuid import mlflow @@ -73,6 +74,10 @@ class Answer(BaseModel): ) +class BuildDate(BaseModel): + build_date: str + + class TrackingContext(BaseModel): uri: str = TRACKING_URI experiment_id: str @@ -86,6 +91,11 @@ class Submission(BaseModel): tracking_context: TrackingContext | None = None +@get("/build-date") +async def build_date() -> BuildDate: + return BuildDate(build_date=os.getenv("BUILD_DATE", "unknown")) + + @post("/scenario-set/{scenario_set_id: str}/deferred-grading") async def deferred_grading( scenario_set_id: str, data: Submission, state: State @@ -307,6 +317,7 @@ async def scenario_types() -> list[ScenarioType]: ) ROUTE_HANDLERS: list[HTTPRouteHandler] = [ + build_date, health, scenario_types, fetch_scenario,