feat: annotation queue testset export

web/oss/src/components/DrillInView/TraceSpanDrillInView.tsx

@@ -47,6 +47,7 @@ import {
 } from "./decodedJsonHelpers"
 import type {DrillInContentProps} from "./DrillInContent"
 import {EntityDrillInView} from "./EntityDrillInView"
+import {getDefaultJsonViewMode} from "./viewModes"
 const ImagePreview = dynamic(() => import("@agenta/ui").then((mod) => mod.ImagePreview), {
     ssr: false,
 })
@@ -113,9 +114,9 @@ type RawSpanViewMode = "json" | "yaml"
  * Summary:
  * - `json` / `yaml`: faithful — data as stored, no cleanup.
  * - `decoded-json`: JSON editor, cleaned (unwrap nested stringified JSON,
- *   decode escaped newlines). Default for non-message data.
+ *   decode escaped newlines).
  * - `beautified-json`: custom component tree (chat bubbles, per-key fields,
- *   envelope unwrap, noise stripping). Default for `viewModePreset="message"`.
+ *   envelope unwrap, noise stripping). Default for structured JSON data.
  * - `text` / `markdown`: prose editor.
  */
 type RawSpanDisplayMode = RawSpanViewMode | "decoded-json" | "beautified-json" | "text" | "markdown"
@@ -129,15 +130,6 @@ const RAW_SPAN_VIEW_MODE_LABELS: Record<RawSpanDisplayMode, string> = {
     markdown: "Markdown",
 }
 
-const getDefaultRawSpanViewMode = (
-    availableModes: RawSpanDisplayMode[],
-    {preferBeautified = false}: {preferBeautified?: boolean} = {},
-): RawSpanDisplayMode => {
-    if (preferBeautified && availableModes.includes("beautified-json")) return "beautified-json"
-    if (availableModes.includes("decoded-json")) return "decoded-json"
-    return availableModes[0] ?? "json"
-}
-
 // Value-simplification and beautified rendering live in ./BeautifiedJsonView.
 
 const LanguageAwareViewer = ({
@@ -394,9 +386,7 @@ export const TraceSpanDrillInView = memo(
             return ["json", "yaml", "decoded-json", "beautified-json"] as RawSpanDisplayMode[]
         }, [viewModePreset, isStringValue, hasStructuredValue, parsedStructuredString])
         const [viewMode, setViewMode] = useState<RawSpanDisplayMode>(() =>
-            getDefaultRawSpanViewMode(availableViewModes, {
-                preferBeautified: viewModePreset === "message",
-            }),
+            getDefaultJsonViewMode(availableViewModes),
         )
 
         const isCodeMode = viewMode === "json" || viewMode === "yaml" || viewMode === "decoded-json"
@@ -441,13 +431,9 @@ export const TraceSpanDrillInView = memo(
 
         useEffect(() => {
             if (!availableViewModes.includes(viewMode)) {
-                setViewMode(
-                    getDefaultRawSpanViewMode(availableViewModes, {
-                        preferBeautified: viewModePreset === "message",
-                    }),
-                )
+                setViewMode(getDefaultJsonViewMode(availableViewModes))
             }
-        }, [availableViewModes, viewMode, viewModePreset])
+        }, [availableViewModes, viewMode])
 
         useEffect(() => {
             closeSearch()

web/oss/src/components/DrillInView/VIEW_MODES.md

@@ -40,8 +40,7 @@ out as a YAML string containing stringified JSON. Intended for readability only.
 
 ### `decoded-json` — Decoded
 
-**This is the default for non-message data.** Same JSON code editor as `json`,
-but the source is passed through the pipeline in
+Same JSON code editor as `json`, but the source is passed through the pipeline in
 [decodedJsonHelpers.ts](./decodedJsonHelpers.ts) before pretty-printing:
 
 1. If the raw value is a string, use its structure-parsed form —
@@ -70,9 +69,9 @@ is now `decoded-json`. Do not reintroduce the old name.
 
 ### `beautified-json` — Reshaped
 
-**This is the default for `viewModePreset="message"`.** Not JSON at all —
-renders via [BeautifiedJsonView](./BeautifiedJsonView.tsx), which uses a
-custom React layout:
+**This is the default for structured JSON data.** Not JSON at all — renders via
+[BeautifiedJsonView](./BeautifiedJsonView.tsx), which uses a custom React
+layout:
 
 - Chat-like arrays and single messages → chat bubbles (role label + content
   editor with markdown support).
@@ -103,8 +102,7 @@ string is markdown-formatted (headings, lists, code fences).
 
 Both panels use the same default-selection logic:
 
-- If `viewModePreset === "message"` and `beautified-json` is available,
-  default to `beautified-json`.
+- If `beautified-json` is available, default to `beautified-json`.
 - Otherwise, if `decoded-json` is available, default to `decoded-json`.
 - Otherwise the first available mode wins (in practice `text` for plain
   strings that do not parse as JSON).

web/oss/src/components/DrillInView/viewModes.ts

@@ -0,0 +1,10 @@
+export const getDefaultJsonViewMode = <TMode extends string>(
+    availableModes: readonly TMode[],
+): TMode => {
+    const hasMode = (mode: string): mode is TMode => availableModes.includes(mode as TMode)
+
+    if (hasMode("beautified-json")) return "beautified-json"
+    if (hasMode("decoded-json")) return "decoded-json"
+
+    return (availableModes[0] ?? "json") as TMode
+}

web/oss/src/components/SharedDrawers/TraceDrawer/components/AccordionTreePanel.tsx

@@ -31,6 +31,7 @@ import {
     normalizeEscapedLineBreaks,
     parseStructuredJson,
 } from "@/oss/components/DrillInView/decodedJsonHelpers"
+import {getDefaultJsonViewMode} from "@/oss/components/DrillInView/viewModes"
 import EnhancedButton from "@/oss/components/EnhancedUIs/Button"
 import {copyToClipboard} from "@/oss/lib/helpers/copyToClipboard"
 import {getStringOrJson, sanitizeDataWithBlobUrls} from "@/oss/lib/helpers/utils"
@@ -64,9 +65,9 @@ type AccordionTreePanelProps = {
  * Summary:
  * - `json` / `yaml`: faithful — data as stored, no cleanup.
  * - `decoded-json`: JSON editor, cleaned (unwrap nested stringified JSON,
- *   decode escaped newlines). Default for non-message data.
+ *   decode escaped newlines).
  * - `beautified-json`: custom component tree (chat bubbles, per-key fields,
- *   envelope unwrap, noise stripping). Default for `viewModePreset="message"`.
+ *   envelope unwrap, noise stripping). Default for structured JSON data.
  * - `text` / `markdown`: prose editor.
  */
 type PanelViewMode = "json" | "yaml" | "decoded-json" | "beautified-json" | "text" | "markdown"
@@ -80,15 +81,6 @@ const PANEL_VIEW_MODE_LABELS: Record<PanelViewMode, string> = {
     markdown: "Markdown",
 }
 
-const getDefaultPanelViewMode = (
-    availableModes: PanelViewMode[],
-    {preferBeautified = false}: {preferBeautified?: boolean} = {},
-): PanelViewMode => {
-    if (preferBeautified && availableModes.includes("beautified-json")) return "beautified-json"
-    if (availableModes.includes("decoded-json")) return "decoded-json"
-    return availableModes[0] ?? "json"
-}
-
 const useStyles = createUseStyles((theme: JSSTheme) => ({
     collapseContainer: ({bgColor}: {bgColor?: string}) => ({
         backgroundColor: "unset",
@@ -336,20 +328,14 @@ const AccordionTreePanel = ({
         return ["json", "yaml", "decoded-json", "beautified-json"]
     }, [viewModePreset, isStringValue, hasStructuredValue, parsedStructuredString])
     const [panelViewMode, setPanelViewMode] = useState<PanelViewMode>(() =>
-        getDefaultPanelViewMode(availableViewModes, {
-            preferBeautified: viewModePreset === "message",
-        }),
+        getDefaultJsonViewMode(availableViewModes),
     )
 
     useEffect(() => {
         if (!availableViewModes.includes(panelViewMode)) {
-            setPanelViewMode(
-                getDefaultPanelViewMode(availableViewModes, {
-                    preferBeautified: viewModePreset === "message",
-                }),
-            )
+            setPanelViewMode(getDefaultJsonViewMode(availableViewModes))
         }
-    }, [availableViewModes, panelViewMode, viewModePreset])
+    }, [availableViewModes, panelViewMode])
 
     const isCodeMode =
         panelViewMode === "json" || panelViewMode === "yaml" || panelViewMode === "decoded-json"