feat: cached call ui (#288)

lunary-ai · May 8, 2024 · bfaa939 · bfaa939
1 parent 1d0d4ce
commit bfaa939
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 5 deletions.
diff --git a/ops b/ops
diff --git a/packages/backend/src/api/v1/runs/index.ts b/packages/backend/src/api/v1/runs/index.ts
@@ -333,7 +333,7 @@ runs.get("/:id/related", checkAccess("logs", "read"), async (ctx) => {
       INNER JOIN related_runs rr ON rr.id = r2.parent_run_id
   )
   SELECT rr.created_at, rr.tags, rr.project_id, rr.id, rr.status, rr.name, rr.ended_at, rr.error, rr.input, rr.output, 
-        rr.params, rr.type, rr.parent_run_id, rr.completion_tokens, rr.prompt_tokens, rr.feedback
+        rr.params, rr.type, rr.parent_run_id, rr.completion_tokens, rr.prompt_tokens, rr.feedback, rr.metadata
   FROM related_runs rr;
   `
 

diff --git a/packages/frontend/components/blocks/DurationBadge.tsx b/packages/frontend/components/blocks/DurationBadge.tsx
@@ -1,11 +1,35 @@
 import { Badge, ThemeIcon } from "@mantine/core"
-import { IconClock } from "@tabler/icons-react"
+import { IconBolt, IconClock } from "@tabler/icons-react"
 
-export default function DurationBadge({ createdAt, endedAt, minimal = false }) {
+export default function DurationBadge({
+  cached = false,
+  createdAt,
+  endedAt,
+  minimal = false,
+}) {
   const duration = endedAt
     ? new Date(endedAt).getTime() - new Date(createdAt).getTime()
     : NaN
 
+  if (cached) {
+    return (
+      <Badge
+        variant="light"
+        color="yellow"
+        pl={0}
+        pr={5}
+        leftSection={
+          <ThemeIcon variant="subtle" size="sm" color="light">
+            <IconBolt size="12" />
+          </ThemeIcon>
+        }
+        tt="none"
+      >
+        Cached ({(duration / 1000).toFixed(2)}s)
+      </Badge>
+    )
+  }
+
   return (
     <Badge
       variant="light"

diff --git a/packages/frontend/pages/traces/[id].tsx b/packages/frontend/pages/traces/[id].tsx
@@ -165,7 +165,11 @@ const TraceTree = ({
           )}
 
           {run.endedAt && (
-            <DurationBadge createdAt={run.createdAt} endedAt={run.endedAt} />
+            <DurationBadge
+              cached={run.metadata?.cache}
+              createdAt={run.createdAt}
+              endedAt={run.endedAt}
+            />
           )}
 
           {/* {timeAfterFirst > 0 && (