fix: restore model in model block detail ui

Signed-off-by: Nick Mitchell <[email protected]>

Author: starpit

pdl-live-react/src/helpers.ts

@@ -7,6 +7,7 @@ import type {
   ArgsBlock,
   CodeBlock,
   PdlModelInput,
+  LocalizedExpression,
 } from "./pdl_ast"
 
 /** Re-export for convenience */
@@ -16,10 +17,7 @@ type MakeNonNullable<T> = {
   [K in keyof T]-?: NonNullable<T[K]>
 }
 
-export type ModelBlock = Extract<
-  PdlBlock,
-  LitellmModelBlock | GraniteioModelBlock
->
+export type ModelBlock = LitellmModelBlock | GraniteioModelBlock
 
 export type ModelBlockWithUsage = ModelBlock & {
   pdl__usage: Required<MakeNonNullable<import("./pdl_ast").PdlUsage>>
@@ -317,3 +315,17 @@ export function extractCode({ code }: CodeBlock): string {
 
   return String(code)
 }
+
+function isExpr(e: unknown): e is LocalizedExpression {
+  return (
+    e !== null && typeof e === "object" && "expr" in (e as LocalizedExpression)
+  )
+}
+
+export function extractModel({ model }: ModelBlock): string {
+  return typeof model === "string"
+    ? model
+    : isExpr(model)
+      ? String(model.pdl__result)
+      : "unknown"
+}

pdl-live-react/src/view/detail/kind/Model.tsx

@@ -4,12 +4,14 @@ import Group from "../Group"
 import Result from "../../Result"
 import {
   capitalizeAndUnSnakeCase,
+  extractModel,
   extractStructuredModelResponse,
   ModelBlock,
 } from "../../../helpers"
 
 export default function ModelItems({ block }: { block: ModelBlock }) {
-  const { platform, model, input } = block
+  const { platform, input } = block
+  const model = extractModel(block)
   const { meta } = extractStructuredModelResponse(block)
 
   const metaForDisplay = meta?.map(([k, v], idx) => (