test: cover tool call hydration preview

2025-11-15 17:33:56 +01:00
parent af7d7cedb0
commit e26219c967
4 changed files with 136 additions and 18 deletions
@@ -0,0 +1,40 @@
+import {
+  extractCommandDetails,
+  extractEditEntries,
+  extractReadEntries,
+  type CommandDetails,
+  type EditEntry,
+  type ReadEntry,
+} from "@/utils/tool-call-parsers";
+
+export type ToolCallPreviewSource = {
+  args?: unknown;
+  result?: unknown;
+  parsedEditEntries?: EditEntry[] | undefined;
+  parsedReadEntries?: ReadEntry[] | undefined;
+  parsedCommandDetails?: CommandDetails | null | undefined;
+};
+
+export type ToolCallPreview = {
+  editEntries: EditEntry[];
+  readEntries: ReadEntry[];
+  commandDetails: CommandDetails | null | undefined;
+};
+
+export function resolveToolCallPreview({
+  args,
+  result,
+  parsedEditEntries,
+  parsedReadEntries,
+  parsedCommandDetails,
+}: ToolCallPreviewSource): ToolCallPreview {
+  const fallbackEditEntries = extractEditEntries(args, result);
+  const fallbackReadEntries = extractReadEntries(result, args);
+  const fallbackCommandDetails = extractCommandDetails(args, result);
+
+  return {
+    editEntries: parsedEditEntries ?? fallbackEditEntries,
+    readEntries: parsedReadEntries ?? fallbackReadEntries,
+    commandDetails: parsedCommandDetails ?? fallbackCommandDetails,
+  };
+}