fix run duration error (#182)

Signed-off-by: kerthcet <kerthcet@gmail.com>

Author: kerthcet

alphatrion/server/graphql/resolvers.py

@@ -188,6 +188,7 @@ def list_runs(
                 experiment_id=r.experiment_id,
                 meta=r.meta,
                 status=GraphQLStatusEnum[Status(r.status).name],
+                duration=r.duration,
                 created_at=r.created_at,
             )
             for r in runs
@@ -205,6 +206,7 @@ def get_run(id: strawberry.ID) -> Run | None:
                 experiment_id=run.experiment_id,
                 meta=run.meta,
                 status=GraphQLStatusEnum[Status(run.status).name],
+                duration=run.duration,
                 created_at=run.created_at,
             )
         return None

alphatrion/server/graphql/types.py

@@ -155,6 +155,7 @@ class Run:
     user_id: strawberry.ID
     experiment_id: strawberry.ID
     meta: JSON | None
+    duration: float
     status: GraphQLStatusEnum
     created_at: datetime
 

dashboard/src/lib/graphql-client.ts

@@ -206,6 +206,7 @@ export const queries = {
         userId
         experimentId
         meta
+        duration
         status
         createdAt
       }
@@ -220,6 +221,7 @@ export const queries = {
         userId
         experimentId
         meta
+        duration
         status
         createdAt
         aggregatedTokens {

dashboard/src/pages/experiments/index.tsx

@@ -270,11 +270,10 @@ export function ExperimentsPage() {
               <Table>
                 <TableHeader>
                   <TableRow className="hover:bg-transparent border-b">
-                    <TableHead className="h-11 text-xs font-semibold uppercase tracking-wider text-muted-foreground bg-muted/50">UUID</TableHead>
+                    <TableHead className="h-11 text-xs font-semibold uppercase tracking-wider text-muted-foreground bg-muted/50">ID</TableHead>
                     <TableHead className="h-11 text-xs font-semibold uppercase tracking-wider text-muted-foreground bg-muted/50">Name</TableHead>
                     <TableHead className="h-11 text-xs font-semibold uppercase tracking-wider text-muted-foreground bg-muted/50">Labels</TableHead>
                     <TableHead className="h-11 text-xs font-semibold uppercase tracking-wider text-muted-foreground bg-muted/50">Status</TableHead>
-                    <TableHead className="h-11 text-xs font-semibold uppercase tracking-wider text-muted-foreground bg-muted/50 text-right">Duration</TableHead>
                     <TableHead className="h-11 text-xs font-semibold uppercase tracking-wider text-muted-foreground bg-muted/50 text-right">Created</TableHead>
                   </TableRow>
                 </TableHeader>
@@ -320,11 +319,6 @@ export function ExperimentsPage() {
                           {experiment.status}
                         </Badge>
                       </TableCell>
-                      <TableCell className="py-3 text-sm text-foreground tabular-nums text-right">
-                        {experiment.duration > 0
-                          ? `${experiment.duration.toFixed(2)}s`
-                          : '-'}
-                      </TableCell>
                       <TableCell className="py-3 text-sm text-muted-foreground text-right">
                         {formatDistanceToNow(new Date(experiment.createdAt), {
                           addSuffix: true,

dashboard/src/pages/runs/[id].tsx

@@ -179,6 +179,16 @@ export function RunDetailPage() {
                 )}
               </dd>
             </div>
+            <div>
+              <dt className="text-xs font-medium uppercase tracking-wide text-muted-foreground">Duration</dt>
+              <dd className="mt-1.5 text-foreground font-mono text-sm">
+                {run.duration !== undefined && run.duration > 0 ? (
+                  `${run.duration.toFixed(2)}s`
+                ) : (
+                  <span className="text-muted-foreground">-</span>
+                )}
+              </dd>
+            </div>
             <div>
               <dt className="text-xs font-medium uppercase tracking-wide text-muted-foreground">Created</dt>
               <dd className="mt-1.5 text-foreground text-sm">

dashboard/src/pages/runs/index.tsx

@@ -151,7 +151,7 @@ export function RunsPage() {
               <Table>
                 <TableHeader>
                   <TableRow className="hover:bg-transparent border-b">
-                    <TableHead className="h-11 text-xs font-semibold uppercase tracking-wider text-muted-foreground bg-muted/50">UUID</TableHead>
+                    <TableHead className="h-11 text-xs font-semibold uppercase tracking-wider text-muted-foreground bg-muted/50">ID</TableHead>
                     <TableHead className="h-11 text-xs font-semibold uppercase tracking-wider text-muted-foreground bg-muted/50">Experiment ID</TableHead>
                     <TableHead className="h-11 text-xs font-semibold uppercase tracking-wider text-muted-foreground bg-muted/50">Status</TableHead>
                     <TableHead className="h-11 text-xs font-semibold uppercase tracking-wider text-muted-foreground bg-muted/50 text-right">Created</TableHead>

dashboard/src/types/index.ts

@@ -72,6 +72,7 @@ export interface Run {
   userId: string;
   experimentId: string;
   meta: Record<string, unknown> | null;
+  duration: number;
   status: Status;
   createdAt: string;
   aggregatedTokens: TokenStats;