+
-
- {isSubmitting ? (
-
+ {submitDisabled && isInterruptible ? (
+
+ {isInterrupting ? (
+
+ ) : (
+
+ {isSubmitting ? (
+
+ )}
);
}
diff --git a/ui/app/components/autopilot/EventStream.stories.tsx b/ui/app/components/autopilot/EventStream.stories.tsx
index 493c7db2710..f1a28ca05ac 100644
--- a/ui/app/components/autopilot/EventStream.stories.tsx
+++ b/ui/app/components/autopilot/EventStream.stories.tsx
@@ -1,6 +1,7 @@
import type { Meta, StoryObj } from "@storybook/react-vite";
import EventStream from "./EventStream";
import type { GatewayEvent } from "~/types/tensorzero";
+import { GlobalToastProvider } from "~/providers/global-toast-provider";
const baseTime = new Date("2026-04-12T10:00:00Z").getTime();
const sessionId = "d1a0b0c0-0000-0000-0000-000000000001";
@@ -280,6 +281,95 @@ const longText =
"Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. " +
"Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.";
+const markdownText = `# Project Overview
+
+This is a **markdown** example with various formatting options.
+
+## Features
+- Bullet point one
+- Bullet point two
+- Bullet point three
+
+### Code Example
+\`\`\`javascript
+function hello() {
+ console.log("Hello, world!");
+}
+\`\`\`
+
+### Links and Formatting
+Visit [TensorZero](https://tensorzero.com) for more information.
+
+*Italic text* and **bold text** are supported.
+
+> This is a blockquote with important information.
+
+1. Ordered list item
+2. Another item
+3. Final item`;
+
+const markdownEvents: GatewayEvent[] = [
+ buildEvent(
+ {
+ id: "md-user-1",
+ session_id: sessionId,
+ created_at: "",
+ payload: {
+ type: "message",
+ role: "user",
+ content: [
+ { type: "text", text: "Can you give me a project overview?" },
+ ],
+ },
+ },
+ 0,
+ ),
+ buildEvent(
+ {
+ id: "md-assistant-1",
+ session_id: sessionId,
+ created_at: "",
+ payload: {
+ type: "message",
+ role: "assistant",
+ content: [{ type: "text", text: markdownText }],
+ },
+ },
+ 1,
+ ),
+ buildEvent(
+ {
+ id: "md-user-2",
+ session_id: sessionId,
+ created_at: "",
+ payload: {
+ type: "message",
+ role: "user",
+ content: [{ type: "text", text: "Thanks! Can you show inline code?" }],
+ },
+ },
+ 2,
+ ),
+ buildEvent(
+ {
+ id: "md-assistant-2",
+ session_id: sessionId,
+ created_at: "",
+ payload: {
+ type: "message",
+ role: "assistant",
+ content: [
+ {
+ type: "text",
+ text: "Sure! You can use `console.log()` to print values, or run `npm install` to install dependencies.",
+ },
+ ],
+ },
+ },
+ 3,
+ ),
+];
+
const longToolArguments = JSON.stringify({
query: longText,
filters: {
@@ -372,6 +462,13 @@ const longFormEvents: GatewayEvent[] = [
const meta = {
title: "Autopilot/EventStream",
component: EventStream,
+ decorators: [
+ (Story) => (
+
+
+ ),
+ ],
render: (args) => (
@@ -411,3 +508,9 @@ export const LongForm: Story = {
events: longFormEvents,
},
};
+
+export const MarkdownContent: Story = {
+ args: {
+ events: markdownEvents,
+ },
+};
diff --git a/ui/app/components/autopilot/EventStream.tsx b/ui/app/components/autopilot/EventStream.tsx
index 0e1f13505d0..cae56857f92 100644
--- a/ui/app/components/autopilot/EventStream.tsx
+++ b/ui/app/components/autopilot/EventStream.tsx
@@ -1,6 +1,12 @@
-import { AlertTriangle, ChevronRight, Loader2 } from "lucide-react";
-import { type RefObject, useState } from "react";
+import { AlertCircle, AlertTriangle, ChevronRight } from "lucide-react";
+import { Component, type RefObject, useState } from "react";
+import {
+ AnimatedEllipsis,
+ EllipsisMode,
+} from "~/components/ui/AnimatedEllipsis";
+import { Markdown, ReadOnlyCodeBlock } from "~/components/ui/markdown";
import { Skeleton } from "~/components/ui/skeleton";
+import { logger } from "~/utils/logger";
import { TableItemTime } from "~/components/ui/TableItems";
import {
Tooltip,
@@ -141,10 +147,9 @@ function summarizeEvent(event: GatewayEvent): EventSummary {
return {
description: payload.status_update.text,
};
- case "tool_call":
- return {
- description: JSON.stringify(payload.arguments, null, 2),
- };
+ case "tool_call": {
+ return { description: JSON.stringify(payload.arguments, null, 2) };
+ }
case "tool_call_authorization":
return {
description:
@@ -186,7 +191,7 @@ function renderEventTitle(event: GatewayEvent) {
: payload.role === "assistant"
? "Assistant"
: "Message";
- return `${roleLabel} Message`;
+ return roleLabel;
}
case "status_update":
return "Status Update";
@@ -318,6 +323,58 @@ function renderEventTitle(event: GatewayEvent) {
}
}
+/**
+ * Error boundary for individual event items.
+ * Prevents a single malformed event from crashing the entire chat.
+ */
+interface EventErrorBoundaryState {
+ hasError: boolean;
+}
+
+interface EventErrorBoundaryProps {
+ eventId: string;
+ children: React.ReactNode;
+}
+
+class EventErrorBoundary extends Component<
+ EventErrorBoundaryProps,
+ EventErrorBoundaryState
+> {
+ constructor(props: EventErrorBoundaryProps) {
+ super(props);
+ this.state = { hasError: false };
+ }
+
+ static getDerivedStateFromError(): EventErrorBoundaryState {
+ return { hasError: true };
+ }
+
+ componentDidCatch(error: Error, errorInfo: React.ErrorInfo) {
+ logger.error(
+ `Event ${this.props.eventId} failed to render:`,
+ error,
+ errorInfo,
+ );
+ }
+
+ render() {
+ if (this.state.hasError) {
+ return (
+
+
+
+
+ Failed to display event. The event data may be corrupted.
+
+
+
+ );
+ }
+
+ return this.props.children;
+ }
+}
+
function EventItem({
event,
isPending = false,
@@ -382,18 +439,25 @@ function EventItem({
+ {event.payload.type === "message" &&
+ event.payload.role === "assistant" ? (
+ {summary.description}
+ ) : event.payload.type === "tool_call" ? (
+
+
);
}
+function SessionStartDivider() {
+ return
Start ;
+}
+
function OptimisticMessageItem({ message }: { message: OptimisticMessage }) {
// Optimistic messages are shown after POST succeeds, so the message is saved.
// We display it like a regular user message, with a skeleton for the timestamp.
return (
- User Message
+ User
@@ -444,42 +512,36 @@ function OptimisticMessageItem({ message }: { message: OptimisticMessage }) {
);
}
-function getStatusLabel(status: AutopilotStatus): string {
+function getStatusLabel(status: AutopilotStatus): {
+ text: string;
+ showEllipsis: boolean;
+} {
switch (status.status) {
case "idle":
- return "Ready";
+ return { text: "Ready", showEllipsis: false };
case "server_side_processing":
- return "Thinking...";
+ return { text: "Thinking", showEllipsis: true };
case "waiting_for_tool_call_authorization":
- return "Waiting";
+ return { text: "Waiting", showEllipsis: false };
case "waiting_for_tool_execution":
- return "Executing tool...";
+ return { text: "Executing tool", showEllipsis: true };
case "waiting_for_retry":
- return "Something went wrong. Retrying...";
+ return { text: "Something went wrong. Retrying", showEllipsis: true };
case "failed":
- return "Something went wrong. Please try again.";
+ return {
+ text: "Something went wrong. Please try again.",
+ showEllipsis: false,
+ };
}
}
-function isLoadingStatus(status: AutopilotStatus): boolean {
- return (
- status.status === "server_side_processing" ||
- status.status === "waiting_for_tool_execution" ||
- status.status === "waiting_for_retry"
- );
-}
-
function StatusIndicator({ status }: { status: AutopilotStatus }) {
- const showSpinner = isLoadingStatus(status);
+ const { text, showEllipsis } = getStatusLabel(status);
return (
-
-
-
- {getStatusLabel(status)}
- {showSpinner &&
-
-
+
+ {text}
+ {showEllipsis &&
);
}
@@ -495,10 +557,10 @@ export default function EventStream({
}: EventStreamProps) {
return (
- {/* Session started indicator, or sentinel for loading more */}
+ {/* Session start indicator, or sentinel for loading more */}
{/* Show divider when we've reached the start OR when there are optimistic messages (new session) */}
{(hasReachedStart || optimisticMessages.length > 0) && !isLoadingOlder ? (
-
+
) : (
)}
@@ -507,11 +569,12 @@ export default function EventStream({
{isLoadingOlder &&
}
{events.map((event) => (
-
+
+
))}
{/* Optimistic messages at the end */}
diff --git a/ui/app/components/autopilot/PendingToolCallCard.tsx b/ui/app/components/autopilot/PendingToolCallCard.tsx
index 831ad49c57a..73ececd0912 100644
--- a/ui/app/components/autopilot/PendingToolCallCard.tsx
+++ b/ui/app/components/autopilot/PendingToolCallCard.tsx
@@ -12,6 +12,7 @@ type PendingToolCallCardProps = {
onAuthorize: (approved: boolean) => void;
additionalCount: number;
isInCooldown?: boolean;
+ className?: string;
};
export function PendingToolCallCard({
@@ -21,6 +22,7 @@ export function PendingToolCallCard({
onAuthorize,
additionalCount,
isInCooldown = false,
+ className,
}: PendingToolCallCardProps) {
const [isExpanded, setIsExpanded] = useState(false);
const [confirmReject, setConfirmReject] = useState(false);
@@ -60,6 +62,7 @@ export function PendingToolCallCard({
"flex flex-col gap-2 rounded-md border border-blue-300 bg-blue-50 px-4 py-3 dark:border-blue-700 dark:bg-blue-950/30",
isInCooldown &&
"animate-in fade-in zoom-in-95 duration-1000 ease-in-out",
+ className,
)}
>
diff --git a/ui/app/components/layout/app.sidebar.tsx b/ui/app/components/layout/app.sidebar.tsx
index 2b86f55e1ee..2ee9706038c 100644
--- a/ui/app/components/layout/app.sidebar.tsx
+++ b/ui/app/components/layout/app.sidebar.tsx
@@ -13,7 +13,7 @@ import {
SidebarCollapse,
SidebarExpand,
} from "~/components/icons/Icons";
-import { KeyRound, LayoutGrid } from "lucide-react";
+import { KeyRound, LayoutGrid, Plus } from "lucide-react";
import {
Sidebar,
SidebarContent,
@@ -162,7 +162,7 @@ export function AppSidebar({ ...props }: React.ComponentProps
) {
{autopilotAvailable && (
-
+
) {
+ {state === "expanded" && (
+
)}
@@ -190,11 +199,7 @@ export function AppSidebar({ ...props }: React.ComponentProps) {
tooltip={state === "collapsed" ? item.title : undefined}
isActive={activePathUtils.isActive(item.url)}
>
-
{item.title}
diff --git a/ui/app/components/ui/AnimatedEllipsis.tsx b/ui/app/components/ui/AnimatedEllipsis.tsx
new file mode 100644
index 00000000000..68d3c88cad6
--- /dev/null
+++ b/ui/app/components/ui/AnimatedEllipsis.tsx
@@ -0,0 +1,66 @@
+import { useState, useEffect } from "react";
+import { cn } from "~/utils/common";
+
+type ReservedWidthProps = {
+ /** The text whose width to reserve (rendered invisibly) */
+ children: string;
+};
+
+/** Reserves the exact width of the given text without rendering it visibly */
+export function ReservedWidth({ children }: ReservedWidthProps) {
+ return {children} ;
+}
+
+/** Layout mode for AnimatedEllipsis */
+export const EllipsisMode = {
+ /** Dynamic width - shifts layout as dots change */
+ Dynamic: "dynamic",
+ /** Fixed width - reserves space for "..." to prevent layout shift */
+ FixedWidth: "fixed-width",
+ /** Absolute - positioned outside layout flow */
+ Absolute: "absolute",
+} as const;
+
+export type EllipsisMode = (typeof EllipsisMode)[keyof typeof EllipsisMode];
+
+type AnimatedEllipsisProps = {
+ /** Animation interval in ms */
+ interval?: number;
+ /** Layout mode */
+ mode?: EllipsisMode;
+ className?: string;
+};
+
+export function AnimatedEllipsis({
+ interval = 400,
+ mode = EllipsisMode.FixedWidth,
+ className,
+}: AnimatedEllipsisProps) {
+ const [dots, setDots] = useState(0);
+
+ useEffect(() => {
+ const timer = setInterval(() => {
+ setDots((prev) => (prev + 1) % 4);
+ }, interval);
+ return () => clearInterval(timer);
+ }, [interval]);
+
+ const dotsContent = ".".repeat(dots);
+
+ if (mode === EllipsisMode.Absolute) {
+ return (
+ {dotsContent}
+ );
+ }
+
+ if (mode === EllipsisMode.FixedWidth) {
+ return (
+
+ ...
+ {dotsContent}
+
+ );
+ }
+
+ return {dotsContent} ;
+}
diff --git a/ui/app/components/ui/FadeGradient.tsx b/ui/app/components/ui/FadeGradient.tsx
new file mode 100644
index 00000000000..a7e391540bf
--- /dev/null
+++ b/ui/app/components/ui/FadeGradient.tsx
@@ -0,0 +1,56 @@
+import { cn } from "~/utils/common";
+
+/** Direction the gradient fades toward */
+export const FadeDirection = {
+ /** Fades from solid at top to transparent at bottom */
+ Top: "top",
+ /** Fades from solid at bottom to transparent at top */
+ Bottom: "bottom",
+} as const;
+
+export type FadeDirection = (typeof FadeDirection)[keyof typeof FadeDirection];
+
+/** Common surface colors for backgrounds and gradients */
+export const SurfaceColor = {
+ Primary: "primary",
+ Secondary: "secondary",
+ Tertiary: "tertiary",
+} as const;
+
+export type SurfaceColor = (typeof SurfaceColor)[keyof typeof SurfaceColor];
+
+const surfaceColorClasses: Record = {
+ [SurfaceColor.Primary]: "from-bg-primary",
+ [SurfaceColor.Secondary]: "from-bg-secondary",
+ [SurfaceColor.Tertiary]: "from-bg-tertiary",
+};
+
+type FadeGradientProps = {
+ direction: FadeDirection;
+ visible: boolean;
+ color?: SurfaceColor;
+ className?: string;
+};
+
+export function FadeGradient({
+ direction,
+ visible,
+ color = SurfaceColor.Secondary,
+ className,
+}: FadeGradientProps) {
+ return (
+
+ );
+}
diff --git a/ui/app/components/ui/code-editor.tsx b/ui/app/components/ui/code-editor.tsx
index 096d6a33b71..3246c3d4185 100644
--- a/ui/app/components/ui/code-editor.tsx
+++ b/ui/app/components/ui/code-editor.tsx
@@ -94,7 +94,7 @@ const CUSTOM_EDITOR_THEME = EditorView.theme({
},
});
-// Pre-computed theme variants (only 2 needed: mono vs sans font)
+// Editor theme with monospace font
const THEME_MONO = githubLightInit({
settings: {
fontFamily: "var(--font-mono)",
@@ -103,14 +103,6 @@ const THEME_MONO = githubLightInit({
background: "transparent",
},
});
-const THEME_SANS = githubLightInit({
- settings: {
- fontFamily: "var(--font-sans)",
- fontSize: "var(--text-xs)",
- gutterBorder: "transparent",
- background: "transparent",
- },
-});
// Cache for extension combinations (max 16 combinations: 4 languages × 2 wordWrap × 2 readOnly)
const extensionCache = new Map();
@@ -321,8 +313,7 @@ export const CodeEditor: React.FC = ({
const buttonClassName =
"flex h-6 w-6 cursor-pointer items-center justify-center p-3 text-xs";
- // Use pre-computed theme (only 2 variants exist)
- const theme = language === "text" ? THEME_SANS : THEME_MONO;
+ const theme = THEME_MONO;
// Use cached basicSetup (shared across all CodeEditor instances)
const basicSetup = getBasicSetup(showLineNumbers, readOnly);
diff --git a/ui/app/components/ui/markdown.stories.tsx b/ui/app/components/ui/markdown.stories.tsx
new file mode 100644
index 00000000000..8bda46f9080
--- /dev/null
+++ b/ui/app/components/ui/markdown.stories.tsx
@@ -0,0 +1,151 @@
+import type { Meta, StoryObj } from "@storybook/react-vite";
+import { Markdown } from "./markdown";
+import { GlobalToastProvider } from "~/providers/global-toast-provider";
+
+const meta = {
+ title: "UI/Markdown",
+ component: Markdown,
+ parameters: {
+ layout: "padded",
+ },
+ decorators: [
+ (Story) => (
+
+
+ ),
+ ],
+ render: (args) => (
+
+
+ ),
+} satisfies Meta;
+
+export default meta;
+type Story = StoryObj;
+
+export const Simple: Story = {
+ args: {
+ children: "This is a simple paragraph with **bold** and *italic* text.",
+ },
+};
+
+export const Headings: Story = {
+ args: {
+ children: `# Heading 1
+## Heading 2
+### Heading 3
+#### Heading 4
+##### Heading 5
+###### Heading 6`,
+ },
+};
+
+export const Lists: Story = {
+ args: {
+ children: `## Unordered List
+- Item one
+- Item two
+- Item three
+
+## Ordered List
+1. First item
+2. Second item
+3. Third item`,
+ },
+};
+
+export const CodeBlocks: Story = {
+ args: {
+ children: `Here is some inline code: \`console.log("hello")\`
+
+And here is a code block:
+
+\`\`\`javascript
+function greet(name) {
+ return \`Hello, \${name}!\`;
+}
+
+greet("World");
+\`\`\`
+
+And another in Python:
+
+\`\`\`python
+def greet(name):
+ return f"Hello, {name}!"
+
+greet("World")
+\`\`\``,
+ },
+};
+
+export const Blockquotes: Story = {
+ args: {
+ children: `> This is a blockquote with some important information.
+> It can span multiple lines.
+
+Regular text after the blockquote.`,
+ },
+};
+
+export const Links: Story = {
+ args: {
+ children: `Check out [TensorZero](https://tensorzero.com) for more information.
+
+You can also visit [GitHub](https://github.com) or [Google](https://google.com).`,
+ },
+};
+
+export const Tables: Story = {
+ args: {
+ children: `| Feature | Status | Notes |
+|---------|--------|-------|
+| Markdown | Done | Basic support |
+| Tables | Done | With headers |
+| Code | Done | Inline and blocks |`,
+ },
+};
+
+export const FullDocument: Story = {
+ args: {
+ children: `# Project Overview
+
+This is a **markdown** example with various formatting options.
+
+## Features
+- Bullet point one
+- Bullet point two
+- Bullet point three
+
+### Code Example
+\`\`\`javascript
+function hello() {
+ console.log("Hello, world!");
+}
+\`\`\`
+
+### Links and Formatting
+Visit [TensorZero](https://tensorzero.com) for more information.
+
+*Italic text* and **bold text** are supported.
+
+> This is a blockquote with important information.
+
+1. Ordered list item
+2. Another item
+3. Final item
+
+---
+
+## Summary
+
+| Feature | Supported |
+|---------|-----------|
+| Headings | Yes |
+| Lists | Yes |
+| Code | Yes |
+| Tables | Yes |`,
+ },
+};
diff --git a/ui/app/components/ui/markdown.tsx b/ui/app/components/ui/markdown.tsx
new file mode 100644
index 00000000000..163565c0b33
--- /dev/null
+++ b/ui/app/components/ui/markdown.tsx
@@ -0,0 +1,223 @@
+import { Children, isValidElement } from "react";
+import type { Components } from "react-markdown";
+import ReactMarkdown from "react-markdown";
+import remarkGfm from "remark-gfm";
+import {
+ CodeEditor,
+ type CodeEditorProps,
+ type Language,
+} from "~/components/ui/code-editor";
+import { cn } from "~/utils/common";
+
+// Common spacing for block elements
+const BLOCK_SPACING = "mb-3 last:mb-0";
+
+// Common table cell styling
+const TABLE_CELL_BASE = "border-border border px-3 py-2";
+
+/**
+ * Map markdown language identifiers to CodeEditor languages.
+ * Falls back to "text" for unsupported languages (no syntax highlighting).
+ */
+function mapLanguage(lang: string | undefined): Language {
+ if (!lang) return "text";
+ const normalized = lang.toLowerCase();
+ switch (normalized) {
+ case "json":
+ case "jsonc":
+ return "json";
+ case "md":
+ case "markdown":
+ return "markdown";
+ case "jinja":
+ case "jinja2":
+ return "jinja2";
+ default:
+ return "text"; // No syntax highlighting for unknown languages
+ }
+}
+
+/**
+ * Recursively extract text from React children.
+ */
+function getTextContent(node: React.ReactNode): string {
+ if (node == null) return "";
+ if (typeof node === "string" || typeof node === "number") return String(node);
+ if (Array.isArray(node)) return node.map(getTextContent).join("");
+ if (isValidElement(node)) {
+ const props = node.props as { children?: React.ReactNode };
+ return getTextContent(props.children);
+ }
+ return "";
+}
+
+/**
+ * Extract text content and language from pre > code children.
+ * react-markdown wraps code blocks as ... {
+ code: string;
+ language?: Language;
+}
+
+export function ReadOnlyCodeBlock({
+ code,
+ language = "text",
+ maxHeight,
+ className,
+}: ReadOnlyCodeBlockProps) {
+ return (
+
+
+ );
+}
+
+const components: Components = {
+ // Headings
+ h1: ({ children }) => (
+ {children}
+ ),
+ h2: ({ children }) => (
+ {children}
+ ),
+ h3: ({ children }) => (
+ {children}
+ ),
+ h4: ({ children }) => (
+ {children}
+ ),
+ h5: ({ children }) => (
+ {children}
+ ),
+ h6: ({ children }) => (
+ {children}
+ ),
+
+ // Paragraphs
+ p: ({ children }) => {children}
,
+
+ // Lists - symmetric vertical margin for consistent spacing
+ ul: ({ children }) => (
+
+ ),
+ ol: ({ children }) => (
+ {children}
+ ),
+ li: ({ children }) => {children} ,
+
+ // Inline code - slightly smaller than body text for visual balance
+ code: ({ children }) => (
+
+ {children}
+
+ ),
+
+ // Code blocks - use CodeEditor for consistent UX
+ pre: MarkdownCodeBlock,
+
+ // Blockquotes
+ blockquote: ({ children }) => (
+
+ {children}
+
+ ),
+
+ // Links
+ a: ({ href, children }) => (
+
+ {children}
+
+ ),
+
+ // Strong and emphasis
+ strong: ({ children }) => (
+ {children}
+ ),
+ em: ({ children }) => {children} ,
+
+ // Horizontal rule
+ hr: () => {children} ,
+ tbody: ({ children }) => {children} ,
+ tr: ({ children }) => {children} ,
+ th: ({ children }) => (
+
+ {children}
+
+ ),
+ td: ({ children }) => {children} ,
+};
+
+interface MarkdownProps {
+ children: string;
+ className?: string;
+}
+
+export function Markdown({ children, className }: MarkdownProps) {
+ return (
+
+
+ {children}
+
+
+ );
+}
diff --git a/ui/app/hooks/useElementHeight.ts b/ui/app/hooks/useElementHeight.ts
new file mode 100644
index 00000000000..f79662cacee
--- /dev/null
+++ b/ui/app/hooks/useElementHeight.ts
@@ -0,0 +1,31 @@
+import { useEffect, useRef, useState } from "react";
+
+/**
+ * Hook to measure an element's height dynamically using ResizeObserver.
+ * Returns a ref to attach to the element and the current height.
+ */
+export function useElementHeight(initialHeight: number = 0) {
+ const ref = useRef(null);
+ const [height, setHeight] = useState(initialHeight);
+
+ useEffect(() => {
+ const element = ref.current;
+ if (!element) return;
+
+ const measureHeight = () => {
+ const newHeight = element.offsetHeight;
+ if (newHeight > 0) {
+ setHeight((prev) => (prev !== newHeight ? newHeight : prev));
+ }
+ };
+
+ measureHeight();
+
+ const resizeObserver = new ResizeObserver(measureHeight);
+ resizeObserver.observe(element);
+
+ return () => resizeObserver.disconnect();
+ }, []);
+
+ return [ref, height] as const;
+}
diff --git a/ui/app/routes.ts b/ui/app/routes.ts
index 964971c06a6..3a9fdaa6f3b 100644
--- a/ui/app/routes.ts
+++ b/ui/app/routes.ts
@@ -69,6 +69,10 @@ export default [
"autopilot/sessions/:session_id/events/message",
"routes/api/autopilot/sessions/$session_id/events/message.route.ts",
),
+ route(
+ "autopilot/sessions/:session_id/actions/interrupt",
+ "routes/api/autopilot/sessions/$session_id/actions/interrupt.route.ts",
+ ),
]),
// Datasets
diff --git a/ui/app/routes/api/autopilot/sessions/$session_id/actions/interrupt.route.ts b/ui/app/routes/api/autopilot/sessions/$session_id/actions/interrupt.route.ts
new file mode 100644
index 00000000000..ccfba01993f
--- /dev/null
+++ b/ui/app/routes/api/autopilot/sessions/$session_id/actions/interrupt.route.ts
@@ -0,0 +1,37 @@
+import type { ActionFunctionArgs } from "react-router";
+import { getAutopilotClient } from "~/utils/tensorzero.server";
+import { logger } from "~/utils/logger";
+
+/**
+ * API route for interrupting an autopilot session.
+ *
+ * Route: POST /api/autopilot/sessions/:session_id/actions/interrupt
+ */
+export async function action({ params, request }: ActionFunctionArgs) {
+ const sessionId = params.session_id;
+ if (!sessionId) {
+ return Response.json(
+ { success: false, error: "Session ID is required" },
+ { status: 400 },
+ );
+ }
+
+ if (request.method !== "POST") {
+ return Response.json(
+ { success: false, error: "Method not allowed" },
+ { status: 405 },
+ );
+ }
+
+ const client = getAutopilotClient();
+
+ try {
+ await client.interruptAutopilotSession(sessionId);
+ return Response.json({ success: true });
+ } catch (error) {
+ logger.error("Failed to interrupt session:", error);
+ const message =
+ error instanceof Error ? error.message : "Failed to interrupt session";
+ return Response.json({ success: false, error: message }, { status: 500 });
+ }
+}
diff --git a/ui/app/routes/autopilot/sessions/$session_id/route.tsx b/ui/app/routes/autopilot/sessions/$session_id/route.tsx
index f93c3c9c7c7..d820e48b37d 100644
--- a/ui/app/routes/autopilot/sessions/$session_id/route.tsx
+++ b/ui/app/routes/autopilot/sessions/$session_id/route.tsx
@@ -1,7 +1,6 @@
import type { Route } from "./+types/route";
import {
Suspense,
- use,
useCallback,
useEffect,
useMemo,
@@ -9,24 +8,31 @@ import {
useState,
} from "react";
import {
+ Await,
data,
isRouteErrorResponse,
- Link,
+ useAsyncError,
+ useFetcher,
useNavigate,
type RouteHandle,
} from "react-router";
-import { Loader2, Plus } from "lucide-react";
-import { PageHeader, Breadcrumbs } from "~/components/layout/PageLayout";
+import debounce from "lodash-es/debounce";
+import { AlertCircle, Loader2 } from "lucide-react";
+import { Breadcrumbs } from "~/components/layout/PageLayout";
import EventStream, {
type OptimisticMessage,
} from "~/components/autopilot/EventStream";
import { PendingToolCallCard } from "~/components/autopilot/PendingToolCallCard";
import { ChatInput } from "~/components/autopilot/ChatInput";
+import { FadeDirection, FadeGradient } from "~/components/ui/FadeGradient";
import { logger } from "~/utils/logger";
import { getAutopilotClient } from "~/utils/tensorzero.server";
import { useAutopilotEventStream } from "~/hooks/useAutopilotEventStream";
+import { useElementHeight } from "~/hooks/useElementHeight";
import type { AutopilotStatus, GatewayEvent } from "~/types/tensorzero";
import { useToast } from "~/hooks/use-toast";
+import { SectionErrorNotice } from "~/components/ui/error/ErrorContentPrimitives";
+import { getFeatureFlags } from "~/utils/feature_flags";
// Nil UUID for creating new sessions
const NIL_UUID = "00000000-0000-0000-0000-000000000000";
@@ -103,33 +109,50 @@ export async function loader({ params }: Route.LoaderArgs) {
};
}
-// Simple debounce helper
-function debounce) => void>(
- fn: T,
- delay: number,
-): (...args: Parameters) => void {
- let timeoutId: NodeJS.Timeout | null = null;
- return (...args: Parameters) => {
- if (timeoutId) {
- clearTimeout(timeoutId);
- }
- timeoutId = setTimeout(() => {
- fn(...args);
- timeoutId = null;
- }, delay);
- };
-}
-
// Skeleton shown while events are loading
function EventStreamSkeleton() {
return (
-
+
);
}
-// Main content component that resolves the promise and renders the event stream with SSE
+// Warning banner for transient errors
+function ErrorBanner({ children }: { children: React.ReactNode }) {
+ return (
+
+ {children}
+
+ );
+}
+
+/**
+ * Error state shown when initial event stream load fails.
+ * Preserves the chat container layout so the page doesn't completely break.
+ */
+function EventStreamLoadError({ onError }: { onError: () => void }) {
+ const error = useAsyncError();
+ const message =
+ error instanceof Error ? error.message : "Failed to load session events";
+
+ // Notify parent that we're in error state (disables ChatInput)
+ useEffect(() => {
+ onError();
+ }, [onError]);
+
+ return (
+
+
+ );
+}
+
+// Main content component that renders the event stream with SSE
function EventStreamContent({
sessionId,
eventsData,
@@ -139,23 +162,26 @@ function EventStreamContent({
scrollContainerRef,
onLoaded,
onStatusChange,
+ onPendingToolCallsChange,
+ onErrorChange,
}: {
sessionId: string;
- eventsData: EventsData | Promise
;
+ eventsData: EventsData;
isNewSession: boolean;
optimisticMessages: OptimisticMessage[];
onOptimisticMessagesChange: (messages: OptimisticMessage[]) => void;
scrollContainerRef: React.RefObject;
onLoaded: () => void;
onStatusChange: (status: AutopilotStatus) => void;
+ onPendingToolCallsChange: (pendingToolCalls: GatewayEvent[]) => void;
+ onErrorChange: (error: string | null, isRetrying: boolean) => void;
}) {
- // Resolve promise (or use direct data for new session)
const {
events: initialEvents,
hasMoreEvents: initialHasMore,
pendingToolCalls: initialPendingToolCalls,
status: initialStatus,
- } = eventsData instanceof Promise ? use(eventsData) : eventsData;
+ } = eventsData;
// Signal that loading is complete (this runs after promise resolves)
useEffect(() => {
@@ -177,94 +203,25 @@ function EventStreamContent({
onStatusChange(status);
}, [status, onStatusChange]);
+ // Notify parent of pending tool calls changes
+ useEffect(() => {
+ onPendingToolCallsChange(pendingToolCalls);
+ }, [pendingToolCalls, onPendingToolCallsChange]);
+
+ // Notify parent of error state changes
+ useEffect(() => {
+ onErrorChange(error, isRetrying);
+ }, [error, isRetrying, onErrorChange]);
+
// State for pagination
const [isLoadingOlder, setIsLoadingOlder] = useState(false);
const [hasReachedStart, setHasReachedStart] = useState(!initialHasMore);
- // State for tool call authorization loading
- const [authLoadingStates, setAuthLoadingStates] = useState<
- Map
- >(new Map());
-
- const { toast } = useToast();
-
- // Derive values for queue-based approval UI
+ // Derive pending tool call IDs for highlighting in the event stream
const pendingToolCallIds = useMemo(
() => new Set(pendingToolCalls.map((e) => e.id)),
[pendingToolCalls],
);
- const oldestPendingToolCall = pendingToolCalls[0] ?? null;
-
- // Cooldown animation: triggers when the queue top changes due to SSE (not user action).
- // Covers both directions: new item jumping to top, or top item removed by external approval.
- // Does NOT trigger when queue was empty and first item arrives (no accidental click risk).
- const prevQueueTopRef = useRef(null);
- const userActionRef = useRef(false);
- const [isInCooldown, setIsInCooldown] = useState(false);
-
- useEffect(() => {
- const currentTopId = oldestPendingToolCall?.id ?? null;
- const prevTopId = prevQueueTopRef.current;
-
- prevQueueTopRef.current = currentTopId;
- const wasUserAction = userActionRef.current;
- userActionRef.current = false;
-
- if (currentTopId !== prevTopId && prevTopId !== null && !wasUserAction) {
- setIsInCooldown(true);
- const timer = setTimeout(() => setIsInCooldown(false), 1000);
- return () => clearTimeout(timer);
- }
- return undefined;
- }, [oldestPendingToolCall?.id]);
-
- // Handle tool call authorization
- const handleAuthorize = useCallback(
- async (eventId: string, approved: boolean) => {
- userActionRef.current = true;
-
- setAuthLoadingStates((prev) =>
- new Map(prev).set(eventId, approved ? "approving" : "rejecting"),
- );
-
- try {
- const response = await fetch(
- `/api/autopilot/sessions/${encodeURIComponent(sessionId)}/events/authorize`,
- {
- method: "POST",
- headers: { "Content-Type": "application/json" },
- body: JSON.stringify({
- tool_call_event_id: eventId,
- status: approved
- ? { type: "approved" }
- : {
- type: "rejected",
- reason: "The user rejected the tool call.",
- },
- }),
- },
- );
-
- if (!response.ok) {
- throw new Error("Authorization failed");
- }
- } catch (err) {
- logger.error("Failed to authorize tool call:", err);
- toast.error({
- title: "Authorization failed",
- description:
- "Failed to submit tool call authorization. Please try again.",
- });
- } finally {
- setAuthLoadingStates((prev) => {
- const next = new Map(prev);
- next.delete(eventId);
- return next;
- });
- }
- },
- [sessionId, toast],
- );
/*
* SCROLL BEHAVIOR SPEC:
@@ -331,13 +288,23 @@ function EventStreamContent({
}
}, [events, isLoadingOlder, scrollToBottom, scrollContainerRef]);
+ // Listen to scroll events from the parent-provided scroll container
+ useEffect(() => {
+ const container = scrollContainerRef.current;
+ if (!container) return;
+
+ container.addEventListener("scroll", handleScroll);
+ return () => container.removeEventListener("scroll", handleScroll);
+ }, [scrollContainerRef, handleScroll]);
+
// Initial scroll to bottom on page load (once)
useEffect(() => {
if (!hasInitiallyScrolledRef.current && scrollContainerRef.current) {
scrollToBottom();
+ handleScroll();
hasInitiallyScrolledRef.current = true;
}
- }, [scrollToBottom, scrollContainerRef]);
+ }, [scrollToBottom, scrollContainerRef, handleScroll]);
// Load older events
const loadOlderEvents = useCallback(async () => {
@@ -467,52 +434,15 @@ function EventStreamContent({
);
return (
- <>
- {error && isRetrying && (
-
- Failed to fetch events. Retrying...
-
- )}
- {
- // Update parent's ref to point to the actual scrollable container
- if (scrollContainerRef) {
- (
- scrollContainerRef as React.MutableRefObject
- ).current = el;
- }
- }}
- onScroll={handleScroll}
- className="border-border mt-8 min-h-0 flex-1 overflow-y-auto rounded-lg border p-4"
- >
-
-
- {/* Pinned approval card - outside scroll container */}
- {oldestPendingToolCall && (
-
-
- handleAuthorize(oldestPendingToolCall.id, approved)
- }
- additionalCount={pendingToolCalls.length - 1}
- isInCooldown={isInCooldown}
- />
- (null);
// Lift optimistic messages state to parent so ChatInput can work outside Suspense
const [optimisticMessages, setOptimisticMessages] = useState<
OptimisticMessage[]
>([]);
- // Clear optimistic messages when session changes to prevent cross-session leakage
- useEffect(() => {
- setOptimisticMessages([]);
- }, [sessionId]);
-
- // Track if events are still loading (for disabling chat input)
- // New sessions have direct data (not a promise), so they're not loading
- const [isEventsLoading, setIsEventsLoading] = useState(
- !isNewSession && eventsData instanceof Promise,
- );
-
- // Reset loading state when session changes (useState initial value only applies on first mount)
- useEffect(() => {
- setIsEventsLoading(!isNewSession && eventsData instanceof Promise);
- }, [sessionId, isNewSession, eventsData]);
-
// Track autopilot status for disabling submit
const [autopilotStatus, setAutopilotStatus] = useState({
status: "idle",
});
- // Reset status when session changes
- useEffect(() => {
- setAutopilotStatus({ status: "idle" });
- }, [sessionId]);
-
const handleStatusChange = useCallback((status: AutopilotStatus) => {
setAutopilotStatus(status);
}, []);
+ // Pending tool calls state - lifted from EventStreamContent for footer rendering
+ const [pendingToolCalls, setPendingToolCalls] = useState([]);
+
+ const handlePendingToolCallsChange = useCallback(
+ (toolCalls: GatewayEvent[]) => {
+ setPendingToolCalls(toolCalls);
+ },
+ [],
+ );
+
+ // Derived values for queue-based approval UI
+ const oldestPendingToolCall = pendingToolCalls[0] ?? null;
+
+ // State for tool call authorization loading
+ const [authLoadingStates, setAuthLoadingStates] = useState<
+ Map
+ >(new Map());
+
+ // State for SSE connection error
+ const [sseError, setSseError] = useState<{
+ error: string | null;
+ isRetrying: boolean;
+ }>({ error: null, isRetrying: false });
+
+ const handleErrorChange = useCallback(
+ (error: string | null, isRetrying: boolean) => {
+ setSseError({ error, isRetrying });
+ },
+ [],
+ );
+
+ // Track loading/error state for ChatInput - disabled until events resolve
+ // For existing sessions, start loading until EventStreamContent calls onLoaded
+ const [isEventsLoading, setIsEventsLoading] = useState(!isNewSession);
+ const [hasLoadError, setHasLoadError] = useState(false);
+
+ // Cooldown animation: triggers when the queue top changes due to SSE (not user action).
+ // Covers both directions: new item jumping to top, or top item removed by external approval.
+ // Does NOT trigger when queue was empty and first item arrives (no accidental click risk).
+ const prevQueueTopRef = useRef(null);
+ const userActionRef = useRef(false);
+ const [isInCooldown, setIsInCooldown] = useState(false);
+
+ // Reset loading/error state when navigating to a different session
+ // Note: key={sessionId} on Suspense remounts EventStreamContent, which will call onLoaded
+ useEffect(() => {
+ setOptimisticMessages([]);
+ setIsEventsLoading(!isNewSession);
+ setHasLoadError(false);
+ setAutopilotStatus({ status: "idle" });
+ setPendingToolCalls([]);
+ setAuthLoadingStates(new Map());
+ setSseError({ error: null, isRetrying: false });
+ prevQueueTopRef.current = null;
+ }, [sessionId, isNewSession]);
+
+ useEffect(() => {
+ const currentTopId = oldestPendingToolCall?.id ?? null;
+ const prevTopId = prevQueueTopRef.current;
+
+ prevQueueTopRef.current = currentTopId;
+ const wasUserAction = userActionRef.current;
+ userActionRef.current = false;
+
+ if (currentTopId !== prevTopId && prevTopId !== null && !wasUserAction) {
+ setIsInCooldown(true);
+ const timer = setTimeout(() => setIsInCooldown(false), 1000);
+ return () => clearTimeout(timer);
+ }
+ return undefined;
+ }, [oldestPendingToolCall?.id]);
+
+ // Handle tool call authorization
+ const handleAuthorize = useCallback(
+ async (eventId: string, approved: boolean) => {
+ userActionRef.current = true;
+
+ setAuthLoadingStates((prev) =>
+ new Map(prev).set(eventId, approved ? "approving" : "rejecting"),
+ );
+
+ try {
+ const response = await fetch(
+ `/api/autopilot/sessions/${encodeURIComponent(sessionId)}/events/authorize`,
+ {
+ method: "POST",
+ headers: { "Content-Type": "application/json" },
+ body: JSON.stringify({
+ tool_call_event_id: eventId,
+ status: approved
+ ? { type: "approved" }
+ : {
+ type: "rejected",
+ reason: "The user rejected the tool call.",
+ },
+ }),
+ },
+ );
+
+ if (!response.ok) {
+ throw new Error("Authorization failed");
+ }
+ } catch (err) {
+ logger.error("Failed to authorize tool call:", err);
+ toast.error({
+ title: "Authorization failed",
+ description:
+ "Failed to submit tool call authorization. Please try again.",
+ });
+ } finally {
+ setAuthLoadingStates((prev) => {
+ const next = new Map(prev);
+ next.delete(eventId);
+ return next;
+ });
+ }
+ },
+ [sessionId, toast],
+ );
+
+ // Handle interrupt session
+ const handleInterruptSession = useCallback(() => {
+ interruptedSessionRef.current = sessionId;
+ interruptFetcher.submit(null, {
+ method: "POST",
+ action: `/api/autopilot/sessions/${encodeURIComponent(sessionId)}/actions/interrupt`,
+ });
+ }, [interruptFetcher, sessionId]);
+
+ // Show toast on interrupt result (only if still on the same session)
+ useEffect(() => {
+ if (interruptFetcher.state === "idle" && interruptFetcher.data) {
+ // Only show toast if we're still on the session that was interrupted
+ if (interruptedSessionRef.current !== sessionId) {
+ return;
+ }
+ const data = interruptFetcher.data as {
+ success: boolean;
+ error?: string;
+ };
+ if (data.success) {
+ toast.success({
+ title: "Session interrupted",
+ description: "The autopilot session has been interrupted.",
+ });
+ } else if (data.error) {
+ toast.error({
+ title: "Failed to interrupt session",
+ description: data.error,
+ });
+ }
+ }
+ }, [interruptFetcher.state, interruptFetcher.data, toast, sessionId]);
+
+ // Interruptible when actively processing (not idle or failed) and feature flag is enabled
+ const { FF_INTERRUPT_SESSION } = getFeatureFlags();
+ const isInterruptible =
+ FF_INTERRUPT_SESSION &&
+ autopilotStatus.status !== "idle" &&
+ autopilotStatus.status !== "failed";
+
// Disable submit unless status is idle or failed
const submitDisabled =
autopilotStatus.status !== "idle" && autopilotStatus.status !== "failed";
+ const handleEventsLoaded = useCallback(() => {
+ setIsEventsLoading(false);
+ setHasLoadError(false);
+ }, []);
+
+ const handleLoadError = useCallback(() => {
+ setIsEventsLoading(false);
+ setHasLoadError(true);
+ }, []);
+
// Ref for scroll container - shared between parent and EventStreamContent
const scrollContainerRef = useRef(null);
+ // Measure header/footer heights dynamically
+ const [headerRef, headerHeight] = useElementHeight(56);
+ const [footerRef, footerHeight] = useElementHeight(120);
+
+ // State for fade overlays (both start false, updated on scroll)
+ const [showTopFade, setShowTopFade] = useState(false);
+ const [showBottomFade, setShowBottomFade] = useState(false);
+
+ // Track previous footer height for scroll adjustment (null = initial mount)
+ const prevFooterHeightRef = useRef(null);
+
+ // Reset footer height ref on session change to avoid cross-session scroll jumps
+ useEffect(() => {
+ prevFooterHeightRef.current = null;
+ }, [sessionId]);
+
+ // Adjust scroll position when footer height changes (e.g., tool card appears)
+ // Only adjust if user is near bottom - don't disrupt users reading older messages
+ useEffect(() => {
+ const container = scrollContainerRef.current;
+
+ // Skip initial mount - just record the value
+ if (prevFooterHeightRef.current === null) {
+ prevFooterHeightRef.current = footerHeight;
+ return;
+ }
+
+ const delta = footerHeight - prevFooterHeightRef.current;
+ prevFooterHeightRef.current = footerHeight;
+
+ // Only adjust scroll when footer grows - shrinking doesn't need adjustment
+ if (container && delta > 0) {
+ const distanceFromBottom =
+ container.scrollHeight - container.scrollTop - container.clientHeight;
+ // When footer grows, distanceFromBottom increases by delta, so we need to
+ // subtract it to get the user's original position before the change
+ const originalDistance = distanceFromBottom - delta;
+ const wasNearBottom = originalDistance < 100;
+
+ if (wasNearBottom) {
+ container.scrollTop += delta;
+ }
+ }
+ }, [footerHeight]);
+
+ // Update fade overlay visibility based on scroll position
+ const handleScroll = useCallback((e: React.UIEvent) => {
+ const target = e.currentTarget;
+ setShowTopFade(target.scrollTop > 20);
+ const distanceFromBottom =
+ target.scrollHeight - target.scrollTop - target.clientHeight;
+ setShowBottomFade(distanceFromBottom > 20);
+ }, []);
+
const handleNavigateToSession = useCallback(
(newSessionId: string) => {
navigate(`/autopilot/sessions/${newSessionId}`);
@@ -614,89 +752,115 @@ export default function AutopilotSessionEventsPage({
);
return (
-
-
+ {/* Fixed header with breadcrumbs and fade gradient */}
+
+
+ {/* Header background - matches message width with slight outset */}
+
+
+
+ {sseError.error && sseError.isRetrying && (
+
Failed to fetch events. Retrying...
+ )}
+
+
- }
- name={isNewSession ? "New Session" : sessionId}
- tag={
- !isNewSession ? (
-
-
- New Session
-
- ) : undefined
- }
- />
+
+
setIsEventsLoading(false)}
- onStatusChange={handleStatusChange}
- />
-
-
- {/* Chat input - always visible outside Suspense, disabled while loading */}
-
+
+
}>
+
}
+ >
+ {(resolvedData) => (
+
+ )}
+
+
+
+
;
- isNewSession: boolean;
- optimisticMessages: OptimisticMessage[];
- onOptimisticMessagesChange: (messages: OptimisticMessage[]) => void;
- scrollContainerRef: React.RefObject;
- onLoaded: () => void;
- onStatusChange: (status: AutopilotStatus) => void;
-}) {
- return (
-
-
+ {/* Fixed footer with tool approval card and chat input */}
+
+
+
+ {/* Footer background - matches message width with slight outset */}
+
+
+ {oldestPendingToolCall && (
+
+ handleAuthorize(oldestPendingToolCall.id, approved)
+ }
+ additionalCount={pendingToolCalls.length - 1}
+ isInCooldown={isInCooldown}
+ />
+ )}
+
+
+
+
{
+ const endpoint = `/internal/autopilot/v1/sessions/${encodeURIComponent(sessionId)}/actions/interrupt`;
+ const response = await this.fetch(endpoint, { method: "POST" });
+ if (!response.ok) {
+ const message = await this.getErrorText(response);
+ this.handleHttpError({ message, response });
+ }
+ }
+
/**
* Lists events for an autopilot session with optional pagination.
*/
diff --git a/ui/fixtures/s3-fixtures/.gitkeep b/ui/fixtures/large-fixtures/.gitkeep
similarity index 100%
rename from ui/fixtures/s3-fixtures/.gitkeep
rename to ui/fixtures/large-fixtures/.gitkeep
diff --git a/ui/fixtures/load_fixtures_postgres.sh b/ui/fixtures/load_fixtures_postgres.sh
index e3c1c09e008..ab7bf455ab7 100755
--- a/ui/fixtures/load_fixtures_postgres.sh
+++ b/ui/fixtures/load_fixtures_postgres.sh
@@ -1,8 +1,8 @@
#!/bin/bash
set -euo pipefail
-# Load small fixtures into Postgres inference tables.
-# Only loads chat_inferences and json_inferences (Step 1 tables).
+# Load small fixtures into Postgres tables.
+# Loads inference tables (Step 1) and feedback tables (Step 2).
#
# Usage:
# ./ui/fixtures/load_fixtures_postgres.sh
@@ -54,14 +54,22 @@ if [ "${TENSORZERO_SKIP_TRUNCATE:-}" = "1" ]; then
else
echo "Truncating tables..."
psql -q "$POSTGRES_URL" <>'id')::uuid,
+ (j->>'target_id')::uuid,
+ j->>'metric_name',
+ (j->>'value')::boolean,
+ COALESCE(j->'tags', '{}')::jsonb,
+ tensorzero.uuid_v7_to_timestamp((j->>'id')::uuid)
+FROM tmp_jsonl, LATERAL (SELECT data::jsonb AS j) AS parsed
+ON CONFLICT (id) DO NOTHING;
+"
+
+# Float Metric Feedback
+# created_at is derived from the UUIDv7 id using tensorzero.uuid_v7_to_timestamp()
+load_jsonl "small-fixtures/float_metric_feedback_examples.jsonl" "tensorzero.float_metric_feedback" "
+INSERT INTO tensorzero.float_metric_feedback (
+ id, target_id, metric_name, value, tags, created_at
+)
+SELECT
+ (j->>'id')::uuid,
+ (j->>'target_id')::uuid,
+ j->>'metric_name',
+ (j->>'value')::double precision,
+ COALESCE(j->'tags', '{}')::jsonb,
+ tensorzero.uuid_v7_to_timestamp((j->>'id')::uuid)
+FROM tmp_jsonl, LATERAL (SELECT data::jsonb AS j) AS parsed
+ON CONFLICT (id) DO NOTHING;
+"
+
+# Also load jaro_winkler_similarity_feedback into float_metric_feedback
+load_jsonl "small-fixtures/jaro_winkler_similarity_feedback.jsonl" "tensorzero.float_metric_feedback" "
+INSERT INTO tensorzero.float_metric_feedback (
+ id, target_id, metric_name, value, tags, created_at
+)
+SELECT
+ (j->>'id')::uuid,
+ (j->>'target_id')::uuid,
+ j->>'metric_name',
+ (j->>'value')::double precision,
+ COALESCE(j->'tags', '{}')::jsonb,
+ tensorzero.uuid_v7_to_timestamp((j->>'id')::uuid)
+FROM tmp_jsonl, LATERAL (SELECT data::jsonb AS j) AS parsed
+ON CONFLICT (id) DO NOTHING;
+"
+
+# Comment Feedback
+# created_at is derived from the UUIDv7 id using tensorzero.uuid_v7_to_timestamp()
+load_jsonl "small-fixtures/comment_feedback_examples.jsonl" "tensorzero.comment_feedback" "
+INSERT INTO tensorzero.comment_feedback (
+ id, target_id, target_type, value, tags, created_at
+)
+SELECT
+ (j->>'id')::uuid,
+ (j->>'target_id')::uuid,
+ j->>'target_type',
+ j->>'value',
+ COALESCE(j->'tags', '{}')::jsonb,
+ tensorzero.uuid_v7_to_timestamp((j->>'id')::uuid)
+FROM tmp_jsonl, LATERAL (SELECT data::jsonb AS j) AS parsed
+ON CONFLICT (id) DO NOTHING;
+"
+
+# Demonstration Feedback
+# Note: demonstration_feedback uses inference_id instead of target_id
+# Note: value is JSONB (stores JsonInferenceOutput or Vec)
+# created_at is derived from the UUIDv7 id using tensorzero.uuid_v7_to_timestamp()
+load_jsonl "small-fixtures/demonstration_feedback_examples.jsonl" "tensorzero.demonstration_feedback" "
+INSERT INTO tensorzero.demonstration_feedback (
+ id, inference_id, value, tags, created_at
+)
+SELECT
+ (j->>'id')::uuid,
+ (j->>'inference_id')::uuid,
+ j->'value',
+ COALESCE(j->'tags', '{}')::jsonb,
+ tensorzero.uuid_v7_to_timestamp((j->>'id')::uuid)
+FROM tmp_jsonl, LATERAL (SELECT data::jsonb AS j) AS parsed
+ON CONFLICT (id) DO NOTHING;
+"
+
echo ""
echo "All fixtures loaded successfully!"
@@ -142,5 +239,13 @@ psql -q "$POSTGRES_URL" <= '3.11' \
- --hash=sha256:035796aaaddfe2f9664b9a9372f089cfc88bd795a67bd1bfe15e6e770934cf64 \
- --hash=sha256:043885b4f7e6e232d7df4f51ffdef8c36320ee9d5f227b380ea636722c7ed12e \
- --hash=sha256:04a69abe45b49c5955923cf2c407843d1c85013b424ae8a560bba16c92fe44a0 \
- --hash=sha256:0f2bcc76f1e05e5ab58893407c63d90b2029908fa41f9f1cc51eecce936c3365 \
- --hash=sha256:13b9062e4f5c7ee5c7e5be96f29ba71bc5a37fed3d1d77c37390ae00724d296d \
- --hash=sha256:15eea9f306b98e0be91eb344a94c0e630689ef302e10c2ce5f7e11905c704f9c \
- --hash=sha256:15fb27364ed84114438fff8aaf998c9e19adbeba08c0b75409f8c452a8692c52 \
- --hash=sha256:1b219560ae2c1de48ead517d085bc2d05b9433f8e49d0955c82e8cd37bd7bf36 \
- --hash=sha256:22758999b256b595cf0b1d102b133bb61866ba5ceecf15f759623b64c020c9ec \
- --hash=sha256:2ec646892819370cf3558f518797f16597b4e4669894a2ba712caccc9da53f1f \
- --hash=sha256:3634093d0b428e6c32c3a69b78e554f0cd20ee420dcad5a9f3b2a63762ce4197 \
- --hash=sha256:36dc13af226aeab72b7abad501d370d606326a0029b9f435eacb3b8c94b8a8b7 \
- --hash=sha256:3da3491cee49cf16157e70f607c03a217ea6647b1cea4819c4f48e53d49139b9 \
- --hash=sha256:40cc556d5abbc54aabe2b1ae287042d7bdb80c08edede19f0c0afb36ae586f37 \
- --hash=sha256:4121c5beb58a7f9e6dfdee612cb24f4df5cd4db6e8261d7f4d7450a997a65d6a \
- --hash=sha256:4635239814149e06e2cb9db3dd584b2fa64316c96f10656983b8026a82e6e4db \
- --hash=sha256:4c01835e718bcebe80394fd0ac66c07cbb90147ebbdad3dcecd3f25de2ae7e2c \
- --hash=sha256:4ee6a571d1e4f0ea6d5f22d6e5fbd6ed1dc2b18542848e1e7301bd190500c9d7 \
- --hash=sha256:56209416e81a7893036eea03abcb91c130643eb14233b2515c90dcac963fe99d \
- --hash=sha256:5e199c087e2aa71c8f9ce1cb7a8e10677dc12457e7cc1be4798632da37c3e86e \
- --hash=sha256:62b2198c438058a20b6704351b35a1d7db881812d8512d67a69c9de1f18ca05f \
- --hash=sha256:64c5825affc76942973a70acf438a8ab618dbd692b84cd5ec40a0a0509edc09a \
- --hash=sha256:65611ecbb00ac9846efe04db15cbe6186f562f6bb7e5e05f077e53a599225d16 \
- --hash=sha256:6d34ed9db9e6395bb6cd33286035f73a59b058169733a9db9f85e650b88df37e \
- --hash=sha256:6d9cd732068e8288dbe2717177320723ccec4fb064123f0caf9bbd90ab5be868 \
- --hash=sha256:6e274603039f924c0fe5cb73438fa9246699c78a6df1bd3decef9ae592ae1c05 \
- --hash=sha256:77b84453f3adcb994ddbd0d1c5d11db2d6bda1a2b7fd5ac5bd4649d6f5dc682e \
- --hash=sha256:7c26b0b2bf58009ed1f38a641f3db4be8d960a417ca96d14e5b06df1506d41ff \
- --hash=sha256:7fd09cc5d65bda1e79432859c40978010622112e9194e581e3415a3eccc7f43f \
- --hash=sha256:817e719a868f0dacde4abdfc5c1910b301877970195db9ab6a5e2c4bd5b121f7 \
- --hash=sha256:81b3a59793523e552c4a96109dde028aa4448ae06ccac5a76ff6532a85558a7f \
- --hash=sha256:81c3e6d8c97295a7360d367f9f8553973651b76907988bb6066376bc2252f24e \
- --hash=sha256:838f045478638b26c375ee96ea89464d38428c69170360b23a1a50fa4baa3562 \
- --hash=sha256:84f01a4d18b2cc4ade1814a08e5f3c907b079c847051d720fad15ce37aa930b6 \
- --hash=sha256:85597b2d25ddf655495e2363fe044b0ae999b75bc4d630dc0d886484b03a5eb0 \
- --hash=sha256:85d9fb2d8cd998c84d13a79a09cc0c1091648e848e4e6249b0ccd7f6b487fa26 \
- --hash=sha256:85e071da78d92a214212cacea81c6da557cab307f2c34b5f85b628e94803f9c0 \
- --hash=sha256:863e3b5f4d9915aaf1b8ec79ae560ad21f0b8d5e3adc31e73126491bb86dee1d \
- --hash=sha256:86966db35c4040fdca64f0816a1c1dd8dbd027d90fca5a57e00e1ca4cd41b879 \
- --hash=sha256:8ab1c5f5ee40d6e01cbe96de5863e39b215a4d24e7d007cad56c7184fdf4aeef \
- --hash=sha256:8b5a9a39c45d852b62693d9b3f3e0fe052541f804296ff401a72a1b60edafb29 \
- --hash=sha256:8dc20bde86802df2ed8397a08d793da0ad7a5fd4ea3ac85d757bf5dd4ad7c252 \
- --hash=sha256:957e92defe6c08211eb77902253b14fe5b480ebc5112bc741fd5e9cd0608f847 \
- --hash=sha256:962064de37b9aef801d33bc579690f8bfe6c5e70e29b61783f60bcba838a14d6 \
- --hash=sha256:985f1e46358f06c2a09921e8921e2c98168ed4ae12ccd6e5e87a4f1857923f32 \
- --hash=sha256:9984bd645a8db6ca15d850ff996856d8762c51a2239225288f08f9050ca240a0 \
- --hash=sha256:9cb177bc55b010b19798dc5497d540dea67fd13a8d9e882b2dae71de0cf09eb3 \
- --hash=sha256:9d729d60f8d53a7361707f4b68a9663c968882dd4f09e0d58c044c8bf5faee7b \
- --hash=sha256:a13fc473b6db0be619e45f11f9e81260f7302f8d180c49a22b6e6120022596b3 \
- --hash=sha256:a49d797192a8d950ca59ee2d0337a4d804f713bb5c3c50e8db26d49666e351dc \
- --hash=sha256:a700a4031bc0fd6936e78a752eefb79092cecad2599ea9c8039c548bc097f9bc \
- --hash=sha256:a7b2f9a18b5ff9824a6af80de4f37f4ec3c2aab05ef08f51c77a093f5b89adda \
- --hash=sha256:a7d018bfedb375a8d979ac758b120ba846a7fe764911a64465fd87b8729f4a6a \
- --hash=sha256:b6c231c9c2fadbae4011ca5e7e83e12dc4a5072f1a1d85a0a7b3ed754d145a40 \
- --hash=sha256:bafa7d87d4c99752d07815ed7a2c0964f8ab311eb8168f41b910bd01d15b6032 \
- --hash=sha256:bd0c630cf256b0a7fd9d0a11c9413b42fef5101219ce6ed5a09624f5a65392c7 \
- --hash=sha256:c090d4860032b857d94144d1a9976b8e36709e40386db289aaf6672de2a81966 \
- --hash=sha256:c2f91f496a87235c6aaf6d3f3d89b17dba64996abadccb289f48456cff931ca9 \
- --hash=sha256:d149aee5c72176d9ddbc6803aef9c0f6d2ceeea7626574fc68518da5476fa346 \
- --hash=sha256:d5e081bc082825f8b139f9e9fe42942cb4054524598aaeb177ff476cc76d09d2 \
- --hash=sha256:d7315ed1dab0286adca467377c8381cd748f3dc92235f22a7dfc42745644a96a \
- --hash=sha256:dabc42f9c6577bcc13001b8810d300fe814b4cfbe8a92c873f269484594f9786 \
- --hash=sha256:e1708fac43ef8b419c975926ce1eaf793b0c13b7356cfab6ab0dc34c0a02ac0f \
- --hash=sha256:e73d63fd04e3a9d6bc187f5455d81abfad05660b212c8804bf3b407e984cd2bc \
- --hash=sha256:e78aecd2800b32e8347ce49316d3eaf04aed849cd5b38e0af39f829a4e59f5eb \
- --hash=sha256:e8370eb6925bb8c1c4264fec52b0384b44f675f191df91cbe0140ec9f0955646 \
- --hash=sha256:ecb63014bb7f4ce653f8be7f1df8cbc6093a5a2811211770f6606cc92b5a78fd \
- --hash=sha256:ed759bf7a70342f7817d88376eb7142fab9fef8320d6019ef87fae05a99874e1 \
- --hash=sha256:ef1b5a3e808bc40827b5fa2c8196151a4c5abe110e1726949d7abddfe5c7ae11 \
- --hash=sha256:f77e5b3d3da652b474cc80a14084927a5e86a5eccf54ca8ca5cbd697bf7f2667 \
- --hash=sha256:faba246fb30ea2a526c2e9645f61612341de1a83fb1e0c5edf4ddda5a9c10996 \
- --hash=sha256:fc8a63918b04b8571789688b2780ab2b4a33ab44bfe8ccea36d3eba51228c953 \
- --hash=sha256:fdebe771ca06bb8d6abce84e51dca9f7921fe6ad34a0c914541b063e9a68928b \
- --hash=sha256:fea80f4f4cf83b54c3a051f2f727870ee51e22f0248d3114b8e755d160b38cfb
- # via pandas
-pandas==2.2.3 \
- --hash=sha256:062309c1b9ea12a50e8ce661145c6aab431b1e99530d3cd60640e255778bd43a \
- --hash=sha256:15c0e1e02e93116177d29ff83e8b1619c93ddc9c49083f237d4312337a61165d \
- --hash=sha256:1948ddde24197a0f7add2bdc4ca83bf2b1ef84a1bc8ccffd95eda17fd836ecb5 \
- --hash=sha256:1db71525a1538b30142094edb9adc10be3f3e176748cd7acc2240c2f2e5aa3a4 \
- --hash=sha256:22a9d949bfc9a502d320aa04e5d02feab689d61da4e7764b62c30b991c42c5f0 \
- --hash=sha256:29401dbfa9ad77319367d36940cd8a0b3a11aba16063e39632d98b0e931ddf32 \
- --hash=sha256:3508d914817e153ad359d7e069d752cdd736a247c322d932eb89e6bc84217f28 \
- --hash=sha256:37e0aced3e8f539eccf2e099f65cdb9c8aa85109b0be6e93e2baff94264bdc6f \
- --hash=sha256:381175499d3802cde0eabbaf6324cce0c4f5d52ca6f8c377c29ad442f50f6348 \
- --hash=sha256:38cf8125c40dae9d5acc10fa66af8ea6fdf760b2714ee482ca691fc66e6fcb18 \
- --hash=sha256:3b71f27954685ee685317063bf13c7709a7ba74fc996b84fc6821c59b0f06468 \
- --hash=sha256:3fc6873a41186404dad67245896a6e440baacc92f5b716ccd1bc9ed2995ab2c5 \
- --hash=sha256:4f18ba62b61d7e192368b84517265a99b4d7ee8912f8708660fb4a366cc82667 \
- --hash=sha256:56534ce0746a58afaf7942ba4863e0ef81c9c50d3f0ae93e9497d6a41a057645 \
- --hash=sha256:59ef3764d0fe818125a5097d2ae867ca3fa64df032331b7e0917cf5d7bf66b13 \
- --hash=sha256:5de54125a92bb4d1c051c0659e6fcb75256bf799a732a87184e5ea503965bce3 \
- --hash=sha256:61c5ad4043f791b61dd4752191d9f07f0ae412515d59ba8f005832a532f8736d \
- --hash=sha256:6374c452ff3ec675a8f46fd9ab25c4ad0ba590b71cf0656f8b6daa5202bca3fb \
- --hash=sha256:63cc132e40a2e084cf01adf0775b15ac515ba905d7dcca47e9a251819c575ef3 \
- --hash=sha256:66108071e1b935240e74525006034333f98bcdb87ea116de573a6a0dccb6c039 \
- --hash=sha256:6dfcb5ee8d4d50c06a51c2fffa6cff6272098ad6540aed1a76d15fb9318194d8 \
- --hash=sha256:7c2875855b0ff77b2a64a0365e24455d9990730d6431b9e0ee18ad8acee13dbd \
- --hash=sha256:800250ecdadb6d9c78eae4990da62743b857b470883fa27f652db8bdde7f6659 \
- --hash=sha256:86976a1c5b25ae3f8ccae3a5306e443569ee3c3faf444dfd0f41cda24667ad57 \
- --hash=sha256:a5a1595fe639f5988ba6a8e5bc9649af3baf26df3998a0abe56c02609392e0a4 \
- --hash=sha256:ad5b65698ab28ed8d7f18790a0dc58005c7629f227be9ecc1072aa74c0c1d43a \
- --hash=sha256:b1d432e8d08679a40e2a6d8b2f9770a5c21793a6f9f47fdd52c5ce1948a5a8a9 \
- --hash=sha256:b8661b0238a69d7aafe156b7fa86c44b881387509653fdf857bebc5e4008ad42 \
- --hash=sha256:ba96630bc17c875161df3818780af30e43be9b166ce51c9a18c1feae342906c2 \
- --hash=sha256:c124333816c3a9b03fbeef3a9f230ba9a737e9e5bb4060aa2107a86cc0a497fc \
- --hash=sha256:cd8d0c3be0515c12fed0bdbae072551c8b54b7192c7b1fda0ba56059a0179698 \
- --hash=sha256:d9c45366def9a3dd85a6454c0e7908f2b3b8e9c138f5dc38fed7ce720d8453ed \
- --hash=sha256:f00d1345d84d8c86a63e476bb4955e46458b304b9575dcf71102b5c705320015 \
- --hash=sha256:f3a255b2c19987fbbe62a9dfd6cff7ff2aa9ccab3fc75218fd4b7530f01efa24 \
- --hash=sha256:fffb8ae78d8af97f849404f21411c95062db1496aeb3e56f146f0355c9989319
- # via
- # fixtures
- # parquet-tools
-parquet-tools==0.2.16 \
- --hash=sha256:7bad2ad38f2381afde9d109577199d953946a904b0d74b2f074e47040673f60a \
- --hash=sha256:a360eec233123c2c66f38867ec6dd5c79d48b8f757af8162bf8825d0190dd3d5
- # via fixtures
-pyarrow==21.0.0 \
- --hash=sha256:072116f65604b822a7f22945a7a6e581cfa28e3454fdcc6939d4ff6090126623 \
- --hash=sha256:186aa00bca62139f75b7de8420f745f2af12941595bbbfa7ed3870ff63e25636 \
- --hash=sha256:1e005378c4a2c6db3ada3ad4c217b381f6c886f0a80d6a316fe586b90f77efd7 \
- --hash=sha256:222c39e2c70113543982c6b34f3077962b44fca38c0bd9e68bb6781534425c10 \
- --hash=sha256:26bfd95f6bff443ceae63c65dc7e048670b7e98bc892210acba7e4995d3d4b51 \
- --hash=sha256:3a302f0e0963db37e0a24a70c56cf91a4faa0bca51c23812279ca2e23481fccd \
- --hash=sha256:3a81486adc665c7eb1a2bde0224cfca6ceaba344a82a971ef059678417880eb8 \
- --hash=sha256:40ebfcb54a4f11bcde86bc586cbd0272bac0d516cfa539c799c2453768477569 \
- --hash=sha256:479ee41399fcddc46159a551705b89c05f11e8b8cb8e968f7fec64f62d91985e \
- --hash=sha256:5051f2dccf0e283ff56335760cbc8622cf52264d67e359d5569541ac11b6d5bc \
- --hash=sha256:555ca6935b2cbca2c0e932bedd853e9bc523098c39636de9ad4693b5b1df86d6 \
- --hash=sha256:585e7224f21124dd57836b1530ac8f2df2afc43c861d7bf3d58a4870c42ae36c \
- --hash=sha256:58c30a1729f82d201627c173d91bd431db88ea74dcaa3885855bc6203e433b82 \
- --hash=sha256:6299449adf89df38537837487a4f8d3bd91ec94354fdd2a7d30bc11c48ef6e79 \
- --hash=sha256:65f8e85f79031449ec8706b74504a316805217b35b6099155dd7e227eef0d4b6 \
- --hash=sha256:689f448066781856237eca8d1975b98cace19b8dd2ab6145bf49475478bcaa10 \
- --hash=sha256:69cbbdf0631396e9925e048cfa5bce4e8c3d3b41562bbd70c685a8eb53a91e61 \
- --hash=sha256:731c7022587006b755d0bdb27626a1a3bb004bb56b11fb30d98b6c1b4718579d \
- --hash=sha256:7be45519b830f7c24b21d630a31d48bcebfd5d4d7f9d3bdb49da9cdf6d764edb \
- --hash=sha256:8d58d8497814274d3d20214fbb24abcad2f7e351474357d552a8d53bce70c70e \
- --hash=sha256:9b0b14b49ac10654332a805aedfc0147fb3469cbf8ea951b3d040dab12372594 \
- --hash=sha256:9d9f8bcb4c3be7738add259738abdeddc363de1b80e3310e04067aa1ca596634 \
- --hash=sha256:a7a102574faa3f421141a64c10216e078df467ab9576684d5cd696952546e2da \
- --hash=sha256:b6b27cf01e243871390474a211a7922bfbe3bda21e39bc9160daf0da3fe48876 \
- --hash=sha256:b7ae0bbdc8c6674259b25bef5d2a1d6af5d39d7200c819cf99e07f7dfef1c51e \
- --hash=sha256:bd04ec08f7f8bd113c55868bd3fc442a9db67c27af098c5f814a3091e71cc61a \
- --hash=sha256:c077f48aab61738c237802836fc3844f85409a46015635198761b0d6a688f87b \
- --hash=sha256:cf56ec8b0a5c8c9d7021d6fd754e688104f9ebebf1bf4449613c9531f5346a18 \
- --hash=sha256:d2fe8e7f3ce329a71b7ddd7498b3cfac0eeb200c2789bd840234f0dc271a8efe \
- --hash=sha256:dc56bc708f2d8ac71bd1dcb927e458c93cec10b98eb4120206a4091db7b67b99 \
- --hash=sha256:e563271e2c5ff4d4a4cbeb2c83d5cf0d4938b891518e676025f7268c6fe5fe26 \
- --hash=sha256:e72a8ec6b868e258a2cd2672d91f2860ad532d590ce94cdf7d5e7ec674ccf03d \
- --hash=sha256:e99310a4ebd4479bcd1964dff9e14af33746300cb014aa4a3781738ac63baf4a \
- --hash=sha256:f522e5709379d72fb3da7785aa489ff0bb87448a9dc5a75f45763a795a089ebd \
- --hash=sha256:fc0d2f88b81dcf3ccf9a6ae17f89183762c8a94a5bdcfa09e05cfe413acf0503 \
- --hash=sha256:fee33b0ca46f4c85443d6c450357101e47d53e6c3f008d658c27a2d020d44c79
- # via parquet-tools
-python-dateutil==2.9.0.post0 \
- --hash=sha256:37dd54208da7e1cd875388217d5e00ebd4179249f90fb72437e91a35459a0ad3 \
- --hash=sha256:a8b2bc7bffae282281c8140a97d3aa9c14da0b136dfe83f850eea9a5f7470427
- # via
- # botocore
- # pandas
-pytz==2025.2 \
- --hash=sha256:360b9e3dbb49a209c21ad61809c7fb453643e048b38924c765813546746e81c3 \
- --hash=sha256:5ddf76296dd8c44c26eb8f4b6f35488f3ccbf6fbbd7adee0b7262d43f0ec2f00
- # via pandas
-s3transfer==0.14.0 \
- --hash=sha256:ea3b790c7077558ed1f02a3072fb3cb992bbbd253392f4b6e9e8976941c7d456 \
- --hash=sha256:eff12264e7c8b4985074ccce27a3b38a485bb7f7422cc8046fee9be4983e4125
- # via boto3
-six==1.17.0 \
- --hash=sha256:4721f391ed90541fddacab5acf947aa0d3dc7d27b2e1e8eda2be8970586c3274 \
- --hash=sha256:ff70335d468e7eb6ec65b95b99d3a2836546063f63acc5171de367e834932a81
- # via
- # halo
- # python-dateutil
- # thrift
-spinners==0.0.24 \
- --hash=sha256:1eb6aeb4781d72ab42ed8a01dcf20f3002bf50740d7154d12fb8c9769bf9e27f \
- --hash=sha256:2fa30d0b72c9650ad12bbe031c9943b8d441e41b4f5602b0ec977a19f3290e98
- # via halo
-tabulate==0.9.0 \
- --hash=sha256:0095b12bf5966de529c0feb1fa08671671b3368eec77d7ef7ab114be2c068b3c \
- --hash=sha256:024ca478df22e9340661486f85298cff5f6dcdba14f3813e8830015b9ed1948f
- # via parquet-tools
-termcolor==3.1.0 \
- --hash=sha256:591dd26b5c2ce03b9e43f391264626557873ce1d379019786f99b0c2bee140aa \
- --hash=sha256:6a6dd7fbee581909eeec6a756cff1d7f7c376063b14e4a298dc4980309e55970
- # via halo
-thrift==0.16.0 \
- --hash=sha256:2b5b6488fcded21f9d312aa23c9ff6a0195d0f6ae26ddbd5ad9e3e25dfc14408
- # via parquet-tools
-tzdata==2025.2 \
- --hash=sha256:1a403fada01ff9221ca8044d701868fa132215d84beb92242d9acd2147f667a8 \
- --hash=sha256:b60a638fcc0daffadf82fe0f57e53d06bdec2f36c4df66280ae79bce6bd6f2b9
- # via pandas
-urllib3==2.6.2 \
- --hash=sha256:016f9c98bb7e98085cb2b4b17b87d2c702975664e4f060c6532e64d1c1a5e797 \
- --hash=sha256:ec21cddfe7724fc7cb4ba4bea7aa8e2ef36f607a4bab81aa6ce42a13dc3f03dd
- # via botocore
diff --git a/ui/fixtures/reset-dev-postgres.sh b/ui/fixtures/reset-dev-postgres.sh
index 7c5977ddb6c..90c1c73e0e8 100755
--- a/ui/fixtures/reset-dev-postgres.sh
+++ b/ui/fixtures/reset-dev-postgres.sh
@@ -4,11 +4,25 @@ set -e
# Development database reset script
# Drops and recreates the dev database, then runs migrations
-DB_NAME="${POSTGRES_DEV_DB:-postgres_migration_dev}"
+DB_NAME="${POSTGRES_DEV_DB:-tensorzero-e2e-tests}"
DB_HOST="${POSTGRES_HOST:-localhost}"
DB_PORT="${POSTGRES_PORT:-5432}"
DB_USER="${POSTGRES_USER:-postgres}"
DB_PASSWORD="${POSTGRES_PASSWORD:-postgres}"
+DEV_DATABASE_URL="postgres://$DB_USER:$DB_PASSWORD@$DB_HOST:$DB_PORT/$DB_NAME"
+
+if [[ -z "${TENSORZERO_POSTGRES_URL:-}" ]]; then
+ echo "Error: TENSORZERO_POSTGRES_URL is not set (expected $DEV_DATABASE_URL)." >&2
+ exit 1
+fi
+
+if [[ "$TENSORZERO_POSTGRES_URL" != "$DEV_DATABASE_URL" ]]; then
+ echo "Error: TENSORZERO_POSTGRES_URL does not match dev database." >&2
+ echo "Expected: $DEV_DATABASE_URL" >&2
+ echo "Found: $TENSORZERO_POSTGRES_URL" >&2
+ echo "To override the database name, set POSTGRES_DEV_DB=." >&2
+ exit 1
+fi
echo "==> Resetting dev database: $DB_NAME"
diff --git a/requirements.txt b/ui/fixtures/small-fixtures/.gitkeep
similarity index 100%
rename from requirements.txt
rename to ui/fixtures/small-fixtures/.gitkeep
diff --git a/ui/package.json b/ui/package.json
index dfb2523e17c..7a82eeb9800 100644
--- a/ui/package.json
+++ b/ui/package.json
@@ -54,8 +54,10 @@
"react": "^19.2.1",
"react-dom": "^19.2.1",
"react-hook-form": "^7.60.0",
+ "react-markdown": "^10.1.0",
"react-router": "^7.12.0",
"recharts": "^2.15.4",
+ "remark-gfm": "^4.0.1",
"smol-toml": "^1.4.1",
"tailwind-merge": "^3.3.1",
"tailwindcss-animate": "^1.0.7",