veryfront 0.1.1132 → 0.1.1135

package/esm/src/agent/schemas/agent.schema.d.ts

@@ -91,6 +91,17 @@ export declare const getMessagePartSchema: () => import("../../internal-agents/s
     toolCallId: import("../../internal-agents/schema.js").Schema<string>;
     toolName: import("../../internal-agents/schema.js").Schema<string>;
     args: import("../../internal-agents/schema.js").Schema<Record<string, unknown>>;
+}> | import("../../extensions/schema/schema-validator.js").InferShape<{
+    type: import("../../internal-agents/schema.js").Schema<"source-url">;
+    sourceId: import("../../internal-agents/schema.js").Schema<string>;
+    url: import("../../internal-agents/schema.js").Schema<string>;
+    title: import("../../internal-agents/schema.js").Schema<string | undefined>;
+}> | import("../../extensions/schema/schema-validator.js").InferShape<{
+    type: import("../../internal-agents/schema.js").Schema<"source-document">;
+    sourceId: import("../../internal-agents/schema.js").Schema<string>;
+    title: import("../../internal-agents/schema.js").Schema<string>;
+    mediaType: import("../../internal-agents/schema.js").Schema<string | undefined>;
+    filename: import("../../internal-agents/schema.js").Schema<string | undefined>;
 }> | import("../../extensions/schema/schema-validator.js").InferShape<{
     type: import("../../internal-agents/schema.js").Schema<"image">;
     url: import("../../internal-agents/schema.js").Schema<string>;
@@ -136,6 +147,17 @@ export declare const getMessageSchema: () => import("../../internal-agents/schem
         toolCallId: import("../../internal-agents/schema.js").Schema<string>;
         toolName: import("../../internal-agents/schema.js").Schema<string>;
         args: import("../../internal-agents/schema.js").Schema<Record<string, unknown>>;
+    }> | import("../../extensions/schema/schema-validator.js").InferShape<{
+        type: import("../../internal-agents/schema.js").Schema<"source-url">;
+        sourceId: import("../../internal-agents/schema.js").Schema<string>;
+        url: import("../../internal-agents/schema.js").Schema<string>;
+        title: import("../../internal-agents/schema.js").Schema<string | undefined>;
+    }> | import("../../extensions/schema/schema-validator.js").InferShape<{
+        type: import("../../internal-agents/schema.js").Schema<"source-document">;
+        sourceId: import("../../internal-agents/schema.js").Schema<string>;
+        title: import("../../internal-agents/schema.js").Schema<string>;
+        mediaType: import("../../internal-agents/schema.js").Schema<string | undefined>;
+        filename: import("../../internal-agents/schema.js").Schema<string | undefined>;
     }> | import("../../extensions/schema/schema-validator.js").InferShape<{
         type: import("../../internal-agents/schema.js").Schema<"image">;
         url: import("../../internal-agents/schema.js").Schema<string>;
@@ -201,6 +223,17 @@ export declare const getAgentResponseSchema: () => import("../../internal-agents
             toolCallId: import("../../internal-agents/schema.js").Schema<string>;
             toolName: import("../../internal-agents/schema.js").Schema<string>;
             args: import("../../internal-agents/schema.js").Schema<Record<string, unknown>>;
+        }> | import("../../extensions/schema/schema-validator.js").InferShape<{
+            type: import("../../internal-agents/schema.js").Schema<"source-url">;
+            sourceId: import("../../internal-agents/schema.js").Schema<string>;
+            url: import("../../internal-agents/schema.js").Schema<string>;
+            title: import("../../internal-agents/schema.js").Schema<string | undefined>;
+        }> | import("../../extensions/schema/schema-validator.js").InferShape<{
+            type: import("../../internal-agents/schema.js").Schema<"source-document">;
+            sourceId: import("../../internal-agents/schema.js").Schema<string>;
+            title: import("../../internal-agents/schema.js").Schema<string>;
+            mediaType: import("../../internal-agents/schema.js").Schema<string | undefined>;
+            filename: import("../../internal-agents/schema.js").Schema<string | undefined>;
         }> | import("../../extensions/schema/schema-validator.js").InferShape<{
             type: import("../../internal-agents/schema.js").Schema<"image">;
             url: import("../../internal-agents/schema.js").Schema<string>;
@@ -289,6 +322,17 @@ export declare const getAgentContextSchema: () => import("../../internal-agents/
             toolCallId: import("../../internal-agents/schema.js").Schema<string>;
             toolName: import("../../internal-agents/schema.js").Schema<string>;
             args: import("../../internal-agents/schema.js").Schema<Record<string, unknown>>;
+        }> | import("../../extensions/schema/schema-validator.js").InferShape<{
+            type: import("../../internal-agents/schema.js").Schema<"source-url">;
+            sourceId: import("../../internal-agents/schema.js").Schema<string>;
+            url: import("../../internal-agents/schema.js").Schema<string>;
+            title: import("../../internal-agents/schema.js").Schema<string | undefined>;
+        }> | import("../../extensions/schema/schema-validator.js").InferShape<{
+            type: import("../../internal-agents/schema.js").Schema<"source-document">;
+            sourceId: import("../../internal-agents/schema.js").Schema<string>;
+            title: import("../../internal-agents/schema.js").Schema<string>;
+            mediaType: import("../../internal-agents/schema.js").Schema<string | undefined>;
+            filename: import("../../internal-agents/schema.js").Schema<string | undefined>;
         }> | import("../../extensions/schema/schema-validator.js").InferShape<{
             type: import("../../internal-agents/schema.js").Schema<"image">;
             url: import("../../internal-agents/schema.js").Schema<string>;

package/esm/src/agent/schemas/agent.schema.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent.schema.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/schemas/agent.schema.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAmB,MAAM,kCAAkC,CAAC;AAErF,eAAO,MAAM,sBAAsB,qGAElC,CAAC;AAEF,eAAO,MAAM,oBAAoB,sIAWhC,CAAC;AAEF,eAAO,MAAM,qBAAqB;;;;;GASjC,CAAC;AAEF,eAAO,MAAM,mBAAmB;;;;;GAO/B,CAAC;AAEF,eAAO,MAAM,6BAA6B;;;;;;;GASzC,CAAC;AAEF,eAAO,MAAM,8BAA8B;;;;;;;GAS1C,CAAC;AAEF,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;GAKjC,CAAC;AAEF,eAAO,MAAM,uBAAuB;;;;;;GAQnC,CAAC;AAEF,yEAAyE;AACzE,eAAO,MAAM,yCAAyC;;;;;CAM5C,CAAC;AAYX,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BhC,CAAC;AAEF,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAQ5B,CAAC;AAEF,eAAO,MAAM,uBAAuB;;;;GAMnC,CAAC;AAEF,eAAO,MAAM,iBAAiB;;;;;;;;;GAW7B,CAAC;AAEF,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkClC,CAAC;AAEF,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GASjC,CAAC;AAGF,8CAA8C;AAC9C,MAAM,MAAM,aAAa,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC,CAAC;AACnF,4CAA4C;AAC5C,MAAM,MAAM,WAAW,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,oBAAoB,CAAC,CAAC,CAAC;AAC/E,oCAAoC;AACpC,MAAM,MAAM,YAAY,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,qBAAqB,CAAC,CAAC,CAAC;AACjF,kCAAkC;AAClC,MAAM,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,mBAAmB,CAAC,CAAC,CAAC;AAC7E,oDAAoD;AACpD,MAAM,MAAM,oBAAoB,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,6BAA6B,CAAC,CAAC,CAAC;AACjG,gDAAgD;AAChD,MAAM,MAAM,qBAAqB,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,8BAA8B,CAAC,CAAC,CAAC;AACnG,0CAA0C;AAC1C,MAAM,MAAM,YAAY,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,qBAAqB,CAAC,CAAC,CAAC;AACjF,4CAA4C;AAC5C,MAAM,MAAM,cAAc,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC,CAAC;AACrF,4CAA4C;AAC5C,MAAM,MAAM,WAAW,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,oBAAoB,CAAC,CAAC,CAAC;AAC/E,uCAAuC;AACvC,MAAM,MAAM,OAAO,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,gBAAgB,CAAC,CAAC,CAAC;AACvE,gDAAgD;AAChD,MAAM,MAAM,cAAc,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC,CAAC;AACrF,yCAAyC;AACzC,MAAM,MAAM,QAAQ,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,iBAAiB,CAAC,CAAC,CAAC;AACzE,kCAAkC;AAClC,MAAM,MAAM,aAAa,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC,CAAC;AACnF,yBAAyB;AACzB,MAAM,MAAM,YAAY,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,qBAAqB,CAAC,CAAC,CAAC"}
+{"version":3,"file":"agent.schema.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/schemas/agent.schema.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAmB,MAAM,kCAAkC,CAAC;AAErF,eAAO,MAAM,sBAAsB,qGAElC,CAAC;AAEF,eAAO,MAAM,oBAAoB,sIAWhC,CAAC;AAEF,eAAO,MAAM,qBAAqB;;;;;GASjC,CAAC;AAEF,eAAO,MAAM,mBAAmB;;;;;GAO/B,CAAC;AAEF,eAAO,MAAM,6BAA6B;;;;;;;GASzC,CAAC;AAEF,eAAO,MAAM,8BAA8B;;;;;;;GAS1C,CAAC;AAEF,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;GAKjC,CAAC;AAEF,eAAO,MAAM,uBAAuB;;;;;;GAQnC,CAAC;AAEF,yEAAyE;AACzE,eAAO,MAAM,yCAAyC;;;;;CAM5C,CAAC;AA6BX,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BhC,CAAC;AAEF,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAQ5B,CAAC;AAEF,eAAO,MAAM,uBAAuB;;;;GAMnC,CAAC;AAEF,eAAO,MAAM,iBAAiB;;;;;;;;;GAW7B,CAAC;AAEF,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkClC,CAAC;AAEF,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GASjC,CAAC;AAGF,8CAA8C;AAC9C,MAAM,MAAM,aAAa,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC,CAAC;AACnF,4CAA4C;AAC5C,MAAM,MAAM,WAAW,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,oBAAoB,CAAC,CAAC,CAAC;AAC/E,oCAAoC;AACpC,MAAM,MAAM,YAAY,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,qBAAqB,CAAC,CAAC,CAAC;AACjF,kCAAkC;AAClC,MAAM,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,mBAAmB,CAAC,CAAC,CAAC;AAC7E,oDAAoD;AACpD,MAAM,MAAM,oBAAoB,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,6BAA6B,CAAC,CAAC,CAAC;AACjG,gDAAgD;AAChD,MAAM,MAAM,qBAAqB,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,8BAA8B,CAAC,CAAC,CAAC;AACnG,0CAA0C;AAC1C,MAAM,MAAM,YAAY,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,qBAAqB,CAAC,CAAC,CAAC;AACjF,4CAA4C;AAC5C,MAAM,MAAM,cAAc,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC,CAAC;AACrF,4CAA4C;AAC5C,MAAM,MAAM,WAAW,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,oBAAoB,CAAC,CAAC,CAAC;AAC/E,uCAAuC;AACvC,MAAM,MAAM,OAAO,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,gBAAgB,CAAC,CAAC,CAAC;AACvE,gDAAgD;AAChD,MAAM,MAAM,cAAc,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC,CAAC;AACrF,yCAAyC;AACzC,MAAM,MAAM,QAAQ,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,iBAAiB,CAAC,CAAC,CAAC;AACzE,kCAAkC;AAClC,MAAM,MAAM,aAAa,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC,CAAC;AACnF,yBAAyB;AACzB,MAAM,MAAM,YAAY,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,qBAAqB,CAAC,CAAC,CAAC"}

package/esm/src/agent/schemas/agent.schema.js

@@ -64,6 +64,19 @@ const inlineToolCallPartShape = (v) => v.object({
     toolName: v.string(),
     args: v.record(v.string(), v.unknown()),
 });
+const sourceUrlPartShape = (v) => v.object({
+    type: v.literal("source-url"),
+    sourceId: v.string(),
+    url: v.string(),
+    title: v.string().optional(),
+});
+const sourceDocumentPartShape = (v) => v.object({
+    type: v.literal("source-document"),
+    sourceId: v.string(),
+    title: v.string(),
+    mediaType: v.string().optional(),
+    filename: v.string().optional(),
+});
 export const getMessagePartSchema = defineSchema((v) => v.union([
     v.object({
         type: v.literal("text"),
@@ -78,6 +91,8 @@ export const getMessagePartSchema = defineSchema((v) => v.union([
     getToolCallPartSchema(),
     inlineToolCallPartShape(v),
     getToolResultPartSchema(),
+    sourceUrlPartShape(v),
+    sourceDocumentPartShape(v),
     v.object({
         type: v.literal("image"),
         url: v.string(),

package/esm/src/agent/streaming/chat-ui-message-stream.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chat-ui-message-stream.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/streaming/chat-ui-message-stream.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,gBAAgB,EAAmB,MAAM,wBAAwB,CAAC;AAChF,OAAO,KAAK,EAEV,aAAa,EACb,kBAAkB,EAClB,eAAe,EAChB,MAAM,qBAAqB,CAAC;AAgB7B,kEAAkE;AAClE,MAAM,MAAM,6BAA6B,GAAG;IAC1C,IAAI,EAAE,QAAQ,CAAC;IACf,YAAY,EAAE,gBAAgB,CAAC;IAC/B,eAAe,EAAE,gBAAgB,CAAC;IAClC,UAAU,EAAE;QACV,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;QACpB,iBAAiB,EAAE;YACjB,aAAa,CAAC,EAAE,MAAM,CAAC;YACvB,eAAe,CAAC,EAAE,MAAM,CAAC;YACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;SAC3B,CAAC;QACF,kBAAkB,EAAE;YAClB,UAAU,CAAC,EAAE,MAAM,CAAC;YACpB,eAAe,CAAC,EAAE,MAAM,CAAC;SAC1B,CAAC;QACF,mBAAmB,CAAC,EAAE,MAAM,CAAC;QAC7B,oBAAoB,CAAC,EAAE,MAAM,CAAC;QAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,oBAAoB,CAAC,EAAE,MAAM,CAAC;QAC9B,qBAAqB,CAAC,EAAE,MAAM,CAAC;QAC/B,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,uBAAuB,CAAC,EAAE,MAAM,CAAC;QACjC,wBAAwB,CAAC,EAAE,MAAM,CAAC;QAClC,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,UAAU,CAAC,EAAE,SAAS,GAAG,SAAS,GAAG,SAAS,CAAC;QAC/C,WAAW,CAAC,EAAE,QAAQ,GAAG,UAAU,CAAC;QACpC,kBAAkB,CAAC,EAAE,UAAU,GAAG,SAAS,GAAG,SAAS,CAAC;KACzD,CAAC;CACH,CAAC;AAEF,6DAA6D;AAC7D,MAAM,MAAM,yBAAyB,CAAC,gBAAgB,GAAG,eAAe,IAAI;IAC1E,QAAQ,EAAE,KAAK,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC,CAAC;IACjD,cAAc,EAAE,KAAK,CAAC;IACtB,eAAe,EAAE,aAAa,CAAC,gBAAgB,CAAC,CAAC;IACjD,SAAS,EAAE,KAAK,CAAC;IACjB,YAAY,EAAE,gBAAgB,CAAC;CAChC,CAAC;AAEF,kDAAkD;AAClD,MAAM,MAAM,0BAA0B,CAAC,gBAAgB,GAAG,eAAe,IAAI;IAC3E,iBAAiB,CAAC,EAAE,MAAM,MAAM,CAAC;IACjC,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,MAAM,CAAC;IACrC,eAAe,CAAC,EAAE,CAChB,KAAK,EAAE;QAAE,IAAI,EAAE,6BAA6B,CAAA;KAAE,KAC3C,gBAAgB,GAAG,SAAS,CAAC;IAClC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,yBAAyB,CAAC,gBAAgB,CAAC,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACzF,mBAAmB,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,KAAK,IAAI,CAAC;CAClF,CAAC;AAodF,sDAAsD;AACtD,wBAAgB,uCAAuC,CAAC,gBAAgB,GAAG,eAAe,EACxF,KAAK,EAAE;IAAE,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAA;CAAE,EAC7C,OAAO,GAAE,0BAA0B,CAAC,gBAAgB,CAAM,GACzD,aAAa,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,CAoGpD"}
+{"version":3,"file":"chat-ui-message-stream.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/streaming/chat-ui-message-stream.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,gBAAgB,EAAmB,MAAM,wBAAwB,CAAC;AAChF,OAAO,KAAK,EAEV,aAAa,EACb,kBAAkB,EAClB,eAAe,EAChB,MAAM,qBAAqB,CAAC;AAiB7B,kEAAkE;AAClE,MAAM,MAAM,6BAA6B,GAAG;IAC1C,IAAI,EAAE,QAAQ,CAAC;IACf,YAAY,EAAE,gBAAgB,CAAC;IAC/B,eAAe,EAAE,gBAAgB,CAAC;IAClC,UAAU,EAAE;QACV,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;QACpB,iBAAiB,EAAE;YACjB,aAAa,CAAC,EAAE,MAAM,CAAC;YACvB,eAAe,CAAC,EAAE,MAAM,CAAC;YACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;SAC3B,CAAC;QACF,kBAAkB,EAAE;YAClB,UAAU,CAAC,EAAE,MAAM,CAAC;YACpB,eAAe,CAAC,EAAE,MAAM,CAAC;SAC1B,CAAC;QACF,mBAAmB,CAAC,EAAE,MAAM,CAAC;QAC7B,oBAAoB,CAAC,EAAE,MAAM,CAAC;QAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,oBAAoB,CAAC,EAAE,MAAM,CAAC;QAC9B,qBAAqB,CAAC,EAAE,MAAM,CAAC;QAC/B,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,uBAAuB,CAAC,EAAE,MAAM,CAAC;QACjC,wBAAwB,CAAC,EAAE,MAAM,CAAC;QAClC,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,UAAU,CAAC,EAAE,SAAS,GAAG,SAAS,GAAG,SAAS,CAAC;QAC/C,WAAW,CAAC,EAAE,QAAQ,GAAG,UAAU,CAAC;QACpC,kBAAkB,CAAC,EAAE,UAAU,GAAG,SAAS,GAAG,SAAS,CAAC;KACzD,CAAC;CACH,CAAC;AAEF,6DAA6D;AAC7D,MAAM,MAAM,yBAAyB,CAAC,gBAAgB,GAAG,eAAe,IAAI;IAC1E,QAAQ,EAAE,KAAK,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC,CAAC;IACjD,cAAc,EAAE,KAAK,CAAC;IACtB,eAAe,EAAE,aAAa,CAAC,gBAAgB,CAAC,CAAC;IACjD,SAAS,EAAE,KAAK,CAAC;IACjB,YAAY,EAAE,gBAAgB,CAAC;CAChC,CAAC;AAEF,kDAAkD;AAClD,MAAM,MAAM,0BAA0B,CAAC,gBAAgB,GAAG,eAAe,IAAI;IAC3E,iBAAiB,CAAC,EAAE,MAAM,MAAM,CAAC;IACjC,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,MAAM,CAAC;IACrC,eAAe,CAAC,EAAE,CAChB,KAAK,EAAE;QAAE,IAAI,EAAE,6BAA6B,CAAA;KAAE,KAC3C,gBAAgB,GAAG,SAAS,CAAC;IAClC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,yBAAyB,CAAC,gBAAgB,CAAC,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACzF,mBAAmB,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,KAAK,IAAI,CAAC;CAClF,CAAC;AAihBF,sDAAsD;AACtD,wBAAgB,uCAAuC,CAAC,gBAAgB,GAAG,eAAe,EACxF,KAAK,EAAE;IAAE,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAA;CAAE,EAC7C,OAAO,GAAE,0BAA0B,CAAC,gBAAgB,CAAM,GACzD,aAAa,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,CAgIpD"}

package/esm/src/agent/streaming/chat-ui-message-stream.js

@@ -2,11 +2,14 @@ import * as dntShim from "../../../_dnt.shims.js";
 import { createAgUiRuntimeChatStreamEncoder, } from "../ag-ui/runtime-chat-stream-encoder.js";
 import { mergeToolInputDelta, parseToolInputObject, streamDataStreamEvents, stripLeadingEmptyObjectPlaceholder, } from "./data-stream.js";
 import { normalizeChatMessageMetadata, normalizeChatUiMessageStream, } from "../../chat/chat-ui-message-helpers.js";
+import { deriveKnowledgeSourceDocumentChunk } from "../../chat/knowledge-source-document.js";
 function createFrameworkUiMessageState() {
     return {
         textBlocks: new Map(),
         reasoningBlocks: new Map(),
         toolParts: new Map(),
+        sourceDocumentParts: new Map(),
+        sourceUrlParts: new Map(),
         dataParts: [],
         pendingToolDeltas: new Map(),
         nextOrder: 0,
@@ -66,10 +69,10 @@ function getParsedStreamedToolInput(inputText) {
     }
 }
 function observeChatStreamEvent(input) {
-    const { event, responseMessageId, state } = input;
+    const { event, responseMessageId, state, replaceExistingSourceDocument = false } = input;
     switch (event.type) {
         case "text-start": {
-            const id = event.id || responseMessageId;
+            const id = event.contentId || event.id || responseMessageId;
             if (!state.textBlocks.has(id)) {
                 state.textBlocks.set(id, { id, order: state.nextOrder, text: "" });
                 state.nextOrder += 1;
@@ -77,7 +80,7 @@ function observeChatStreamEvent(input) {
             return;
         }
         case "text-delta": {
-            const id = event.id || responseMessageId;
+            const id = event.contentId || event.id || responseMessageId;
             const existingBlock = state.textBlocks.get(id);
             if (existingBlock) {
                 existingBlock.text += event.delta;
@@ -226,6 +229,38 @@ function observeChatStreamEvent(input) {
             state.nextOrder += 1;
             return;
         }
+        case "source-document": {
+            const existingSource = state.sourceDocumentParts.get(event.sourceId);
+            if (existingSource && !replaceExistingSourceDocument) {
+                return;
+            }
+            state.sourceDocumentParts.set(event.sourceId, {
+                type: "source-document",
+                sourceId: event.sourceId,
+                mediaType: event.mediaType,
+                title: event.title,
+                ...(event.filename ? { filename: event.filename } : {}),
+                order: existingSource?.order ?? state.nextOrder,
+            });
+            if (!existingSource) {
+                state.nextOrder += 1;
+            }
+            return;
+        }
+        case "source-url": {
+            if (state.sourceUrlParts.has(event.sourceId)) {
+                return;
+            }
+            state.sourceUrlParts.set(event.sourceId, {
+                type: "source-url",
+                sourceId: event.sourceId,
+                url: event.url,
+                ...(event.title ? { title: event.title } : {}),
+                order: state.nextOrder,
+            });
+            state.nextOrder += 1;
+            return;
+        }
         default: {
             if (!event.type.startsWith("data-")) {
                 return;
@@ -322,6 +357,14 @@ function buildResponseMessageParts(state) {
             part,
         });
     }
+    for (const sourceDocumentPart of state.sourceDocumentParts.values()) {
+        const { order, ...part } = sourceDocumentPart;
+        orderedParts.push({ order, part });
+    }
+    for (const sourceUrlPart of state.sourceUrlParts.values()) {
+        const { order, ...part } = sourceUrlPart;
+        orderedParts.push({ order, part });
+    }
     for (const dataPart of state.dataParts) {
         orderedParts.push({
             order: dataPart.order,
@@ -391,6 +434,7 @@ export function createChatUiMessageStreamFromDataStream(input, options = {}) {
         onError: options.onError,
     });
     const materializedToolCallIds = new Set();
+    const derivedSourceDocumentIds = new Set();
     let finishReason = "stop";
     return normalizeChatUiMessageStream((async function* () {
         const ensureStepStarted = function* (shouldStartStep) {
@@ -411,15 +455,41 @@ export function createChatUiMessageStreamFromDataStream(input, options = {}) {
             const chatEvents = chatEventEncoder.encode(event);
             finishReason = chatEventEncoder.state.finishReason;
             for (const chatEvent of chatEvents) {
+                const replacesDerivedSourceDocument = chatEvent.type === "source-document" &&
+                    state.sourceDocumentParts.has(chatEvent.sourceId) &&
+                    derivedSourceDocumentIds.delete(chatEvent.sourceId);
+                const isDuplicateSourceDocument = chatEvent.type === "source-document" &&
+                    state.sourceDocumentParts.has(chatEvent.sourceId) &&
+                    !replacesDerivedSourceDocument;
+                const isDuplicateSourceUrl = chatEvent.type === "source-url" &&
+                    state.sourceUrlParts.has(chatEvent.sourceId);
                 observeChatStreamEvent({
                     event: chatEvent,
                     responseMessageId,
                     state,
+                    replaceExistingSourceDocument: replacesDerivedSourceDocument,
                 });
                 const chunk = toUiChunk(chatEvent);
-                if (chunk) {
+                if (chunk && !isDuplicateSourceDocument && !isDuplicateSourceUrl) {
                     yield chunk;
                 }
+                if (chatEvent.type !== "tool-output-available") {
+                    continue;
+                }
+                const sourceChunk = deriveKnowledgeSourceDocumentChunk({
+                    toolName: state.toolParts.get(chatEvent.toolCallId)?.toolName,
+                    output: chatEvent.output,
+                });
+                if (!sourceChunk || state.sourceDocumentParts.has(sourceChunk.sourceId)) {
+                    continue;
+                }
+                derivedSourceDocumentIds.add(sourceChunk.sourceId);
+                observeChatStreamEvent({
+                    event: sourceChunk,
+                    responseMessageId,
+                    state,
+                });
+                yield sourceChunk;
             }
         }
         for (const [toolCallId, pendingToolDelta] of state.pendingToolDeltas.entries()) {

package/esm/src/agent/streaming/fork-runtime-part-mapper.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"fork-runtime-part-mapper.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/streaming/fork-runtime-part-mapper.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AAM1E,OAAO,KAAK,EAAE,QAAQ,EAAE,eAAe,EAAE,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AAElG,KAAK,gBAAgB,GAAG,OAAO,CAAC,QAAQ,EAAE;IAAE,IAAI,EAAE,WAAW,CAAA;CAAE,CAAC,CAAC;AACjE,KAAK,kBAAkB,GAAG,OAAO,CAAC,QAAQ,EAAE;IAAE,IAAI,EAAE,aAAa,CAAA;CAAE,CAAC,CAAC;AAGrE,MAAM,WAAW,wBAAwB;IACvC,aAAa,EAAE,OAAO,CAAC;IACvB,aAAa,EAAE,OAAO,CAAC;IACvB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,kBAAkB,EAAE,OAAO,CAAC;IAC5B,cAAc,EAAE,OAAO,CAAC;CACzB;AAED,sCAAsC;AACtC,MAAM,WAAW,uBAAuB;IACtC,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,wBAAwB,CAAC,CAAC;IACjD,kBAAkB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAChC,oBAAoB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAClC,MAAM,CAAC,EAAE,uBAAuB,CAAC;CAClC;AAED,KAAK,wBAAwB,GAAG,wBAAwB,GAAG;IACzD,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAChC,CAAC;AAEF,6CAA6C;AAC7C,MAAM,MAAM,6BAA6B,GAAG;IAC1C,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,wBAAwB,CAAC,CAAC;IACjD,kBAAkB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAChC,oBAAoB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAClC,MAAM,CAAC,EAAE,uBAAuB,CAAC;CAClC,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG,6BAA6B,CAAC;AAUjE,mCAAmC;AACnC,wBAAgB,uBAAuB,CACrC,IAAI,EAAE,eAAe,EACrB,KAAK,EAAE,uBAAuB,GAC7B,KAAK,CAAC,gBAAgB,GAAG,kBAAkB,CAAC,CAyD9C;AAMD,wBAAgB,0BAA0B,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAe5F;AAsBD,oDAAoD;AACpD,wBAAgB,mCAAmC,CACjD,KAAK,GAAE;IAAE,MAAM,CAAC,EAAE,uBAAuB,CAAA;CAAO,GAC/C,6BAA6B,CAO/B;AAED,6CAA6C;AAC7C,wBAAgB,8BAA8B,CAC5C,KAAK,EAAE,sBAAsB,EAC7B,KAAK,EAAE,6BAA6B,GACnC,QAAQ,EAAE,CAqJZ;AAED;;GAEG;AACH,wBAAgB,0BAA0B,CACxC,KAAK,GAAE;IAAE,MAAM,CAAC,EAAE,uBAAuB,CAAA;CAAO,GAC/C,6BAA6B,CAE/B;AAED;;GAEG;AACH,wBAAgB,4BAA4B,CAC1C,KAAK,EAAE,sBAAsB,EAC7B,KAAK,EAAE,6BAA6B,GACnC,QAAQ,EAAE,CAEZ"}
+{"version":3,"file":"fork-runtime-part-mapper.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/streaming/fork-runtime-part-mapper.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AAM1E,OAAO,KAAK,EAAE,QAAQ,EAAE,eAAe,EAAE,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AAElG,KAAK,gBAAgB,GAAG,OAAO,CAAC,QAAQ,EAAE;IAAE,IAAI,EAAE,WAAW,CAAA;CAAE,CAAC,CAAC;AACjE,KAAK,kBAAkB,GAAG,OAAO,CAAC,QAAQ,EAAE;IAAE,IAAI,EAAE,aAAa,CAAA;CAAE,CAAC,CAAC;AAIrE,MAAM,WAAW,wBAAwB;IACvC,aAAa,EAAE,OAAO,CAAC;IACvB,aAAa,EAAE,OAAO,CAAC;IACvB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,kBAAkB,EAAE,OAAO,CAAC;IAC5B,cAAc,EAAE,OAAO,CAAC;CACzB;AAED,sCAAsC;AACtC,MAAM,WAAW,uBAAuB;IACtC,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,wBAAwB,CAAC,CAAC;IACjD,kBAAkB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAChC,oBAAoB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAClC,MAAM,CAAC,EAAE,uBAAuB,CAAC;CAClC;AAED,KAAK,wBAAwB,GAAG,wBAAwB,GAAG;IACzD,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAChC,CAAC;AAEF,6CAA6C;AAC7C,MAAM,MAAM,6BAA6B,GAAG;IAC1C,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,wBAAwB,CAAC,CAAC;IACjD,kBAAkB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAChC,oBAAoB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAClC,MAAM,CAAC,EAAE,uBAAuB,CAAC;CAClC,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG,6BAA6B,CAAC;AAUjE,mCAAmC;AACnC,wBAAgB,uBAAuB,CACrC,IAAI,EAAE,eAAe,EACrB,KAAK,EAAE,uBAAuB,GAC7B,KAAK,CAAC,gBAAgB,GAAG,kBAAkB,CAAC,CAyD9C;AAMD,wBAAgB,0BAA0B,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAe5F;AA0ED,oDAAoD;AACpD,wBAAgB,mCAAmC,CACjD,KAAK,GAAE;IAAE,MAAM,CAAC,EAAE,uBAAuB,CAAA;CAAO,GAC/C,6BAA6B,CAO/B;AAED,6CAA6C;AAC7C,wBAAgB,8BAA8B,CAC5C,KAAK,EAAE,sBAAsB,EAC7B,KAAK,EAAE,6BAA6B,GACnC,QAAQ,EAAE,CAgKZ;AAED;;GAEG;AACH,wBAAgB,0BAA0B,CACxC,KAAK,GAAE;IAAE,MAAM,CAAC,EAAE,uBAAuB,CAAA;CAAO,GAC/C,6BAA6B,CAE/B;AAED;;GAEG;AACH,wBAAgB,4BAA4B,CAC1C,KAAK,EAAE,sBAAsB,EAC7B,KAAK,EAAE,6BAA6B,GACnC,QAAQ,EAAE,CAEZ"}

package/esm/src/agent/streaming/fork-runtime-part-mapper.js

@@ -1,4 +1,5 @@
 import { isRecord } from "../../chat/conversation.js";
+import { toRenderableCustomChunk } from "../../chat/ag-ui-helpers.js";
 import { safeJsonParse } from "../../chat/provider-errors.js";
 import { mergeToolInputDelta, parseToolInputObject, stripLeadingEmptyObjectPlaceholder, } from "./data-stream.js";
 function warnForkRuntimeStream(logger, message, metadata) {
@@ -86,6 +87,48 @@ function buildToolCallPartIfNeeded(toolCallId, state) {
         },
     ];
 }
+function mapRenderableSourceToForkPart(value) {
+    const chunk = toRenderableCustomChunk(value);
+    if (!chunk) {
+        return null;
+    }
+    if (chunk.type === "source-url") {
+        return {
+            type: "source",
+            id: chunk.sourceId,
+            sourceType: "url",
+            url: chunk.url,
+            ...(chunk.title ? { title: chunk.title } : {}),
+        };
+    }
+    if (chunk.type === "source-document") {
+        return {
+            type: "source",
+            id: chunk.sourceId,
+            sourceType: "document",
+            mediaType: chunk.mediaType,
+            title: chunk.title,
+            ...(chunk.filename ? { filename: chunk.filename } : {}),
+        };
+    }
+    return null;
+}
+function mapWrappedSourceDataToForkPart(event) {
+    if (!isRecord(event.data)) {
+        return null;
+    }
+    const name = event.data.name;
+    if (name !== "source-document" && name !== "source-url") {
+        return null;
+    }
+    const sourcePart = mapRenderableSourceToForkPart(event.data.value);
+    if (!sourcePart ||
+        (name === "source-document" && sourcePart.sourceType !== "document") ||
+        (name === "source-url" && sourcePart.sourceType !== "url")) {
+        return null;
+    }
+    return sourcePart;
+}
 /** State for create fork runtime stream mapping. */
 export function createForkRuntimeStreamMappingState(input = {}) {
     return {
@@ -104,6 +147,15 @@ export function mapAgUiRuntimeEventToForkParts(event, state) {
                 : [];
         case "text-delta":
             return typeof event.delta === "string" ? [{ type: "text-delta", text: event.delta }] : [];
+        case "source-document":
+        case "source-url": {
+            const sourcePart = mapRenderableSourceToForkPart(event);
+            return sourcePart ? [sourcePart] : [];
+        }
+        case "data": {
+            const sourcePart = mapWrappedSourceDataToForkPart(event);
+            return sourcePart ? [sourcePart] : [];
+        }
         case "tool-input-start": {
             const toolCallId = typeof event.toolCallId === "string" ? event.toolCallId : null;
             const toolName = typeof event.toolName === "string" ? event.toolName : null;

package/esm/src/agent/streaming/fork-runtime-types.d.ts

@@ -1,7 +1,9 @@
+import type { HostedStreamSourcePart } from "../../chat/hosted-ui-chunk-mapping.js";
 interface ForkStreamPart {
     type: "reasoning-delta" | "text-delta";
     text: string;
 }
+export type ForkSourcePart = HostedStreamSourcePart;
 interface ForkToolInputStartPart {
     type: "tool-input-start";
     toolCallId: string;
@@ -54,7 +56,7 @@ export interface ForkRuntimeStep {
     finishReason: string | null;
 }
 /** Public API contract for fork part. */
-export type ForkPart = ForkStreamPart | ForkToolInputStartPart | ForkToolInputDeltaPart | ForkToolCallPart | ForkToolResultPart | ForkToolErrorPart | ForkErrorPart;
+export type ForkPart = ForkStreamPart | ForkSourcePart | ForkToolInputStartPart | ForkToolInputDeltaPart | ForkToolCallPart | ForkToolResultPart | ForkToolErrorPart | ForkErrorPart;
 /** Public API contract for fork runtime stream logger. */
 export type ForkRuntimeStreamLogger = {
     warn: (message: string, metadata?: Record<string, unknown>) => void;

package/esm/src/agent/streaming/fork-runtime-types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"fork-runtime-types.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/streaming/fork-runtime-types.ts"],"names":[],"mappings":"AAAA,UAAU,cAAc;IACtB,IAAI,EAAE,iBAAiB,GAAG,YAAY,CAAC;IACvC,IAAI,EAAE,MAAM,CAAC;CACd;AAED,UAAU,sBAAsB;IAC9B,IAAI,EAAE,kBAAkB,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,UAAU,sBAAsB;IAC9B,IAAI,EAAE,kBAAkB,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,UAAU,gBAAgB;IACxB,IAAI,EAAE,WAAW,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,OAAO,CAAC;CAChB;AAED,UAAU,kBAAkB;IAC1B,IAAI,EAAE,aAAa,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,UAAU,iBAAiB;IACzB,IAAI,EAAE,YAAY,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,OAAO,CAAC;IACf,KAAK,EAAE,KAAK,CAAC;CACd;AAED,UAAU,aAAa;IACrB,IAAI,EAAE,OAAO,CAAC;IACd,KAAK,EAAE,KAAK,CAAC;CACd;AAED,iDAAiD;AACjD,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,SAAS,EAAE,KAAK,CAAC;QACf,UAAU,EAAE,MAAM,CAAC;QACnB,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,OAAO,CAAC;KAChB,CAAC,CAAC;IACH,WAAW,EAAE,KAAK,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,OAAO,CAAC;QACf,MAAM,EAAE,OAAO,CAAC;KACjB,CAAC,CAAC;IACH,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;CAC7B;AAED,yCAAyC;AACzC,MAAM,MAAM,QAAQ,GAChB,cAAc,GACd,sBAAsB,GACtB,sBAAsB,GACtB,gBAAgB,GAChB,kBAAkB,GAClB,iBAAiB,GACjB,aAAa,CAAC;AAElB,0DAA0D;AAC1D,MAAM,MAAM,uBAAuB,GAAG;IACpC,IAAI,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;CACrE,CAAC"}
+{"version":3,"file":"fork-runtime-types.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/streaming/fork-runtime-types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,uCAAuC,CAAC;AAEpF,UAAU,cAAc;IACtB,IAAI,EAAE,iBAAiB,GAAG,YAAY,CAAC;IACvC,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,MAAM,cAAc,GAAG,sBAAsB,CAAC;AAEpD,UAAU,sBAAsB;IAC9B,IAAI,EAAE,kBAAkB,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,UAAU,sBAAsB;IAC9B,IAAI,EAAE,kBAAkB,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,UAAU,gBAAgB;IACxB,IAAI,EAAE,WAAW,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,OAAO,CAAC;CAChB;AAED,UAAU,kBAAkB;IAC1B,IAAI,EAAE,aAAa,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,UAAU,iBAAiB;IACzB,IAAI,EAAE,YAAY,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,OAAO,CAAC;IACf,KAAK,EAAE,KAAK,CAAC;CACd;AAED,UAAU,aAAa;IACrB,IAAI,EAAE,OAAO,CAAC;IACd,KAAK,EAAE,KAAK,CAAC;CACd;AAED,iDAAiD;AACjD,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,SAAS,EAAE,KAAK,CAAC;QACf,UAAU,EAAE,MAAM,CAAC;QACnB,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,OAAO,CAAC;KAChB,CAAC,CAAC;IACH,WAAW,EAAE,KAAK,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,OAAO,CAAC;QACf,MAAM,EAAE,OAAO,CAAC;KACjB,CAAC,CAAC;IACH,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;CAC7B;AAED,yCAAyC;AACzC,MAAM,MAAM,QAAQ,GAChB,cAAc,GACd,cAAc,GACd,sBAAsB,GACtB,sBAAsB,GACtB,gBAAgB,GAChB,kBAAkB,GAClB,iBAAiB,GACjB,aAAa,CAAC;AAElB,0DAA0D;AAC1D,MAAM,MAAM,uBAAuB,GAAG;IACpC,IAAI,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;CACrE,CAAC"}

package/esm/src/agent/streaming/mirrored-tool-chunk-state.d.ts

@@ -47,6 +47,7 @@ export interface CreateHostedMirroredUiStreamInput {
     mirroredToolChunkState: MirroredToolChunkState;
     appendChunk?: (chunk: ChatUiMessageChunk<ChatMessageMetadata>) => Promise<void> | void;
     setMirroredOutput?: (value: boolean) => void;
+    registerPendingDerivedSourceFlush?: (flush: () => Promise<void>) => void;
     logger?: HostedMirroredUiStreamLogger;
 }
 /** Input payload for close hosted mirrored open tool calls. */

package/esm/src/agent/streaming/mirrored-tool-chunk-state.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mirrored-tool-chunk-state.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/streaming/mirrored-tool-chunk-state.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAGtF,yDAAyD;AACzD,wBAAgB,4BAA4B,CAC1C,KAAK,EAAE,kBAAkB,CAAC,mBAAmB,CAAC,GAC7C,OAAO,CAoBT;AAED,qCAAqC;AACrC,MAAM,WAAW,sBAAsB;IACrC,kBAAkB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAChC,yBAAyB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACvC,0BAA0B,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACxC,sBAAsB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACpC,uBAAuB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACrC,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACpC;AAED,4CAA4C;AAC5C,wBAAgB,4BAA4B,IAAI,sBAAsB,CASrE;AAED,2CAA2C;AAC3C,wBAAgB,2BAA2B,CACzC,KAAK,EAAE,sBAAsB,GAC5B,sBAAsB,CASxB;AAED,4CAA4C;AAC5C,wBAAgB,4BAA4B,CAC1C,KAAK,EAAE,sBAAsB,EAC7B,KAAK,EAAE,kBAAkB,CAAC,mBAAmB,CAAC,GAC7C,IAAI,CAsCN;AAED,+CAA+C;AAC/C,MAAM,WAAW,aAAa;IAC5B,eAAe,EAAE,KAAK,CAAC;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACjE,gBAAgB,EAAE,KAAK,CAAC;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACnE;AAED,qEAAqE;AACrE,MAAM,WAAW,gCAAgC;IAC/C,IAAI,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;CACrE;AAED,gEAAgE;AAChE,MAAM,WAAW,4BAA6B,SAAQ,gCAAgC;IACpF,KAAK,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;CACtE;AAED,kEAAkE;AAClE,MAAM,WAAW,8BAA8B;IAC7C,OAAO,EAAE,CAAC,KAAK,EAAE,kBAAkB,CAAC,mBAAmB,CAAC,KAAK,IAAI,CAAC;IAClE,OAAO,EAAE,MAAM,IAAI,CAAC;CACrB;AAED,0DAA0D;AAC1D,MAAM,WAAW,iCAAiC;IAChD,YAAY,EAAE,aAAa,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC;IACrE,kBAAkB,EAAE,8BAA8B,CAAC;IACnD,sBAAsB,EAAE,sBAAsB,CAAC;IAC/C,WAAW,CAAC,EAAE,CACZ,KAAK,EAAE,kBAAkB,CAAC,mBAAmB,CAAC,KAC3C,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC1B,iBAAiB,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;IAC7C,MAAM,CAAC,EAAE,4BAA4B,CAAC;CACvC;AAED,+DAA+D;AAC/D,MAAM,WAAW,qCAAqC;IACpD,sBAAsB,EAAE,sBAAsB,CAAC;IAC/C,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,CACX,KAAK,EAAE,kBAAkB,CAAC,mBAAmB,CAAC,KAC3C,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC1B,MAAM,CAAC,EAAE,gCAAgC,CAAC;CAC3C;AAUD,+CAA+C;AAC/C,wBAAgB,+BAA+B,CAAC,WAAW,EAAE,OAAO,GAAG,MAAM,CAQ5E;AAED,+BAA+B;AAC/B,wBAAgB,oBAAoB,CAClC,KAAK,EAAE,sBAAsB,GAC5B,aAAa,CAyBf;AAED,oDAAoD;AACpD,wBAAsB,gCAAgC,CACpD,KAAK,EAAE,qCAAqC,GAC3C,OAAO,CAAC,IAAI,CAAC,CA0Cf;AAED,wCAAwC;AACxC,wBAAuB,4BAA4B,CACjD,KAAK,EAAE,iCAAiC,GACvC,aAAa,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,CAuDxD"}
+{"version":3,"file":"mirrored-tool-chunk-state.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/streaming/mirrored-tool-chunk-state.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAGtF,yDAAyD;AACzD,wBAAgB,4BAA4B,CAC1C,KAAK,EAAE,kBAAkB,CAAC,mBAAmB,CAAC,GAC7C,OAAO,CAoBT;AAED,qCAAqC;AACrC,MAAM,WAAW,sBAAsB;IACrC,kBAAkB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAChC,yBAAyB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACvC,0BAA0B,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACxC,sBAAsB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACpC,uBAAuB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACrC,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACpC;AAED,4CAA4C;AAC5C,wBAAgB,4BAA4B,IAAI,sBAAsB,CASrE;AAED,2CAA2C;AAC3C,wBAAgB,2BAA2B,CACzC,KAAK,EAAE,sBAAsB,GAC5B,sBAAsB,CASxB;AAED,4CAA4C;AAC5C,wBAAgB,4BAA4B,CAC1C,KAAK,EAAE,sBAAsB,EAC7B,KAAK,EAAE,kBAAkB,CAAC,mBAAmB,CAAC,GAC7C,IAAI,CAsCN;AAED,+CAA+C;AAC/C,MAAM,WAAW,aAAa;IAC5B,eAAe,EAAE,KAAK,CAAC;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACjE,gBAAgB,EAAE,KAAK,CAAC;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACnE;AAED,qEAAqE;AACrE,MAAM,WAAW,gCAAgC;IAC/C,IAAI,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;CACrE;AAED,gEAAgE;AAChE,MAAM,WAAW,4BAA6B,SAAQ,gCAAgC;IACpF,KAAK,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;CACtE;AAED,kEAAkE;AAClE,MAAM,WAAW,8BAA8B;IAC7C,OAAO,EAAE,CAAC,KAAK,EAAE,kBAAkB,CAAC,mBAAmB,CAAC,KAAK,IAAI,CAAC;IAClE,OAAO,EAAE,MAAM,IAAI,CAAC;CACrB;AAED,0DAA0D;AAC1D,MAAM,WAAW,iCAAiC;IAChD,YAAY,EAAE,aAAa,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC;IACrE,kBAAkB,EAAE,8BAA8B,CAAC;IACnD,sBAAsB,EAAE,sBAAsB,CAAC;IAC/C,WAAW,CAAC,EAAE,CACZ,KAAK,EAAE,kBAAkB,CAAC,mBAAmB,CAAC,KAC3C,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC1B,iBAAiB,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;IAC7C,iCAAiC,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC;IACzE,MAAM,CAAC,EAAE,4BAA4B,CAAC;CACvC;AAED,+DAA+D;AAC/D,MAAM,WAAW,qCAAqC;IACpD,sBAAsB,EAAE,sBAAsB,CAAC;IAC/C,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,CACX,KAAK,EAAE,kBAAkB,CAAC,mBAAmB,CAAC,KAC3C,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC1B,MAAM,CAAC,EAAE,gCAAgC,CAAC;CAC3C;AAUD,+CAA+C;AAC/C,wBAAgB,+BAA+B,CAAC,WAAW,EAAE,OAAO,GAAG,MAAM,CAQ5E;AAED,+BAA+B;AAC/B,wBAAgB,oBAAoB,CAClC,KAAK,EAAE,sBAAsB,GAC5B,aAAa,CAyBf;AAED,oDAAoD;AACpD,wBAAsB,gCAAgC,CACpD,KAAK,EAAE,qCAAqC,GAC3C,OAAO,CAAC,IAAI,CAAC,CA0Cf;AAED,wCAAwC;AACxC,wBAAuB,4BAA4B,CACjD,KAAK,EAAE,iCAAiC,GACvC,aAAa,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,CAwIxD"}

package/esm/src/agent/streaming/mirrored-tool-chunk-state.js

@@ -1,4 +1,4 @@
-import { deriveKnowledgeSourceDocumentChunk } from "./knowledge-source-document.js";
+import { deriveKnowledgeSourceDocumentChunk } from "../../chat/knowledge-source-document.js";
 /** Check whether a durable chunk mirrors tool output. */
 export function isDurableMirroredOutputChunk(chunk) {
     switch (chunk.type) {
@@ -151,39 +151,106 @@ export async function closeHostedMirroredOpenToolCalls(input) {
 export async function* createHostedMirroredUiStream(input) {
     let streamError = null;
     const emittedKnowledgeSourceIds = new Set();
+    const mirroredKnowledgeSourceIds = new Set();
+    const emittedDerivedKnowledgeSourceIds = new Set();
+    let pendingDerivedSource = null;
+    const mirrorChunk = async (chunk) => {
+        input.rootStreamWatchdog.observe(chunk);
+        if (isDurableMirroredOutputChunk(chunk)) {
+            input.setMirroredOutput?.(true);
+        }
+        recordMirroredToolChunkState(input.mirroredToolChunkState, chunk);
+        if (input.appendChunk) {
+            await Promise.resolve(input.appendChunk(chunk)).catch((error) => {
+                input.logger?.error("Durable run mirror failed to handle chunk", {
+                    chunkType: chunk.type,
+                    error: error instanceof Error ? error.message : String(error),
+                });
+            });
+        }
+    };
+    const takePendingDerivedSource = () => {
+        const source = pendingDerivedSource;
+        pendingDerivedSource = null;
+        if (!source || emittedKnowledgeSourceIds.has(source.sourceId)) {
+            return null;
+        }
+        emittedKnowledgeSourceIds.add(source.sourceId);
+        return source;
+    };
+    const mirrorSourceOnce = async (source) => {
+        if (mirroredKnowledgeSourceIds.has(source.sourceId)) {
+            return;
+        }
+        mirroredKnowledgeSourceIds.add(source.sourceId);
+        await mirrorChunk(source);
+    };
+    input.registerPendingDerivedSourceFlush?.(async () => {
+        if (!pendingDerivedSource || emittedKnowledgeSourceIds.has(pendingDerivedSource.sourceId)) {
+            return;
+        }
+        await mirrorSourceOnce(pendingDerivedSource);
+    });
     try {
         for await (const sourceChunk of input.sourceStream) {
+            if (pendingDerivedSource &&
+                sourceChunk.type === "source-document" &&
+                sourceChunk.sourceId === pendingDerivedSource.sourceId) {
+                pendingDerivedSource = null;
+                if (!emittedKnowledgeSourceIds.has(sourceChunk.sourceId)) {
+                    emittedKnowledgeSourceIds.add(sourceChunk.sourceId);
+                    mirroredKnowledgeSourceIds.add(sourceChunk.sourceId);
+                    await mirrorChunk(sourceChunk);
+                    yield sourceChunk;
+                }
+                continue;
+            }
+            const pendingSource = takePendingDerivedSource();
+            if (pendingSource) {
+                emittedDerivedKnowledgeSourceIds.add(pendingSource.sourceId);
+                await mirrorSourceOnce(pendingSource);
+                yield pendingSource;
+            }
+            if (sourceChunk.type === "source-document") {
+                if (emittedKnowledgeSourceIds.has(sourceChunk.sourceId)) {
+                    if (emittedDerivedKnowledgeSourceIds.delete(sourceChunk.sourceId)) {
+                        await mirrorChunk(sourceChunk);
+                        yield sourceChunk;
+                    }
+                    continue;
+                }
+                emittedKnowledgeSourceIds.add(sourceChunk.sourceId);
+                await mirrorSourceOnce(sourceChunk);
+                yield sourceChunk;
+                continue;
+            }
             const derivedSource = sourceChunk.type === "tool-output-available"
                 ? deriveKnowledgeSourceDocumentChunk({
                     toolName: input.mirroredToolChunkState.toolCallNames.get(sourceChunk.toolCallId),
                     output: sourceChunk.output,
                 })
                 : null;
-            const chunks = [sourceChunk];
             if (derivedSource && !emittedKnowledgeSourceIds.has(derivedSource.sourceId)) {
-                emittedKnowledgeSourceIds.add(derivedSource.sourceId);
-                chunks.push(derivedSource);
-            }
-            for (const chunk of chunks) {
-                input.rootStreamWatchdog.observe(chunk);
-                if (isDurableMirroredOutputChunk(chunk)) {
-                    input.setMirroredOutput?.(true);
-                }
-                recordMirroredToolChunkState(input.mirroredToolChunkState, chunk);
-                if (input.appendChunk) {
-                    await Promise.resolve(input.appendChunk(chunk)).catch((error) => {
-                        input.logger?.error("Durable run mirror failed to handle chunk", {
-                            chunkType: chunk.type,
-                            error: error instanceof Error ? error.message : String(error),
-                        });
-                    });
-                }
-                yield chunk;
+                pendingDerivedSource = derivedSource;
             }
+            await mirrorChunk(sourceChunk);
+            yield sourceChunk;
+        }
+        const pendingSource = takePendingDerivedSource();
+        if (pendingSource) {
+            emittedDerivedKnowledgeSourceIds.add(pendingSource.sourceId);
+            await mirrorSourceOnce(pendingSource);
+            yield pendingSource;
         }
     }
     catch (error) {
         streamError = error;
+        const pendingSource = takePendingDerivedSource();
+        if (pendingSource) {
+            emittedDerivedKnowledgeSourceIds.add(pendingSource.sourceId);
+            await mirrorSourceOnce(pendingSource);
+            yield pendingSource;
+        }
         throw error;
     }
     finally {

package/esm/src/chat/conversation.d.ts

@@ -48,6 +48,12 @@ export declare const getMessagePartSchema: () => import("../internal-agents/sche
     url: import("../internal-agents/schema.js").Schema<string | undefined>;
     title: import("../internal-agents/schema.js").Schema<string | undefined>;
     quote: import("../internal-agents/schema.js").Schema<string | undefined>;
+}> | import("../extensions/schema/schema-validator.js").InferShape<{
+    type: import("../internal-agents/schema.js").Schema<"source_document">;
+    source_id: import("../internal-agents/schema.js").Schema<string>;
+    media_type: import("../internal-agents/schema.js").Schema<string>;
+    title: import("../internal-agents/schema.js").Schema<string | undefined>;
+    filename: import("../internal-agents/schema.js").Schema<string | undefined>;
 }> | import("../extensions/schema/schema-validator.js").InferShape<{
     type: import("../internal-agents/schema.js").Schema<"step_start">;
 }> | import("../extensions/schema/schema-validator.js").InferShape<{
@@ -131,6 +137,12 @@ export declare const getApiMessageSchema: () => import("../internal-agents/schem
         url: import("../internal-agents/schema.js").Schema<string | undefined>;
         title: import("../internal-agents/schema.js").Schema<string | undefined>;
         quote: import("../internal-agents/schema.js").Schema<string | undefined>;
+    }> | import("../extensions/schema/schema-validator.js").InferShape<{
+        type: import("../internal-agents/schema.js").Schema<"source_document">;
+        source_id: import("../internal-agents/schema.js").Schema<string>;
+        media_type: import("../internal-agents/schema.js").Schema<string>;
+        title: import("../internal-agents/schema.js").Schema<string | undefined>;
+        filename: import("../internal-agents/schema.js").Schema<string | undefined>;
     }> | import("../extensions/schema/schema-validator.js").InferShape<{
         type: import("../internal-agents/schema.js").Schema<"step_start">;
     }> | import("../extensions/schema/schema-validator.js").InferShape<{

package/esm/src/chat/conversation.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"conversation.d.ts","sourceRoot":"","sources":["../../../src/src/chat/conversation.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,+BAA+B,CAAC;AACjE,OAAO,KAAK,EAEV,aAAa,EACb,iBAAiB,EACjB,oBAAoB,EACrB,MAAM,YAAY,CAAC;AAEpB,QAAA,MAAM,gCAAgC,eAAuD,CAAC;AAE9F,8DAA8D;AAC9D,MAAM,MAAM,gCAAgC,GAAG,oBAAoB,GAAG;IACpE,CAAC,gCAAgC,CAAC,CAAC,EAAE,MAAM,CAAC;CAC7C,CAAC;AAEF,oFAAoF;AACpF,wBAAgB,+BAA+B,CAAC,OAAO,EAAE,oBAAoB,GAAG,MAAM,GAAG,SAAS,CAEjG;AAED,6EAA6E;AAC7E,wBAAgB,gCAAgC,CAC9C,OAAO,EAAE,oBAAoB,EAC7B,QAAQ,EAAE,MAAM,GACf,oBAAoB,CAQtB;AAED,uFAAuF;AACvF,wBAAgB,gCAAgC,CAAC,CAAC,SAAS,oBAAoB,EAC7E,MAAM,EAAE,oBAAoB,EAC5B,MAAM,EAAE,CAAC,GACR,CAAC,CAGH;AAED,uCAAuC;AACvC,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkDhC,CAAC;AAEF,4CAA4C;AAC5C,MAAM,MAAM,WAAW,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,oBAAoB,CAAC,CAAC,CAAC;AAE/E,4CAA4C;AAC5C,eAAO,MAAM,yBAAyB,6DAErC,CAAC;AACF,iDAAiD;AACjD,MAAM,MAAM,gBAAgB,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,yBAAyB,CAAC,CAAC,CAAC;AAEzF,yCAAyC;AACzC,eAAO,MAAM,sBAAsB,6DAElC,CAAC;AACF,8CAA8C;AAC9C,MAAM,MAAM,aAAa,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC,CAAC;AAEnF,2CAA2C;AAC3C,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;GAiBpC,CAAC;AAEF,gDAAgD;AAChD,MAAM,MAAM,eAAe,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,wBAAwB,CAAC,CAAC,CAAC;AAEvF,sCAAsC;AACtC,eAAO,MAAM,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoB/B,CAAC;AAEF,6BAA6B;AAC7B,MAAM,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,mBAAmB,CAAC,CAAC,CAAC;AAE7E,8CAA8C;AAC9C,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,WAAW,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,OAAO,CAAC;IACf,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B;AAED,gDAAgD;AAChD,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,aAAa,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,OAAO,CAAC;IAChB,eAAe,CAAC,EAAE,OAAO,CAAC;CAC3B;AAED,uCAAuC;AACvC,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACd;AAED,4CAA4C;AAC5C,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,WAAW,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,kDAAkD;AAClD,KAAK,UAAU,GAAG,OAAO,CAAC,iBAAiB,EAAE;IAAE,UAAU,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC,CAAC;AAGpF,iCAAiC;AACjC,eAAO,MAAM,YAAY,QACuD,CAAC;AAEjF,uCAAuC;AACvC,wBAAgB,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,GAAG,KAAK,IAAI,MAAM,CAExE;AAED,yBAAyB;AACzB,wBAAgB,eAAe,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAG1D;AAED,0BAA0B;AAC1B,wBAAgB,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS,GAAG,WAAW,GAAG,OAAO,CAiB9F;AAED,2BAA2B;AAC3B,wBAAgB,QAAQ,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAEzE;AAED,2BAA2B;AAC3B,wBAAgB,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,CAMtF;AAoBD,gCAAgC;AAChC,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,CAQvD;AAED,uDAAuD;AACvD,wBAAgB,YAAY,CAC1B,IAAI,EAAE,iBAAiB,GACtB,IAAI,IAAI,iBAAiB,GAAG;IAAE,IAAI,EAAE,QAAQ,MAAM,EAAE,CAAC;IAAC,IAAI,EAAE,OAAO,CAAA;CAAE,CAEvE;AAED,mDAAmD;AACnD,wBAAgB,YAAY,CAAC,IAAI,EAAE,iBAAiB,GAAG,IAAI,IAAI,UAAU,CAOxE;AAED,2BAA2B;AAC3B,wBAAgB,aAAa,CAAC,IAAI,EAAE,UAAU,GAAG,MAAM,GAAG,SAAS,CAOlE;AAaD,uBAAuB;AACvB,wBAAgB,aAAa,CAC3B,KAAK,EAAE,WAAW,EAAE,EACpB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,MAAM,EACb,IAAI,EAAE;IACJ,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B,GACA,IAAI,CAwCN;AA2BD,uDAAuD;AACvD,wBAAgB,gCAAgC,CAAC,OAAO,EAAE,aAAa,GAAG,WAAW,EAAE,CAsEtF;AAiBD,sDAAsD;AACtD,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,aAAa,GAAG,OAAO,CAEtE;AAED,6CAA6C;AAC7C,wBAAgB,gCAAgC,CAAC,OAAO,EAAE,aAAa,GAAG,aAAa,CAEtF;AAED,6CAA6C;AAC7C,wBAAgB,gCAAgC,CAC9C,OAAO,EAAE,aAAa,EACtB,SAAS,EAAE,MAAM,GAChB,aAAa,CAaf;AA8BD,iDAAiD;AACjD,wBAAgB,cAAc,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,YAAY,CAOpE;AAED,mDAAmD;AACnD,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,cAAc,CAOxE;AAED,4CAA4C;AAC5C,wBAAgB,UAAU,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,YAAY,CAEhE;AAED,iDAAiD;AACjD,wBAAgB,eAAe,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,iBAAiB,CAK1E;AAED,2CAA2C;AAC3C,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,oBAAoB,GAAG,MAAM,CAoB5E;AAmdD,sDAAsD;AACtD,wBAAgB,wCAAwC,CACtD,QAAQ,EAAE,aAAa,EAAE,GACxB,oBAAoB,EAAE,CAmCxB"}
+{"version":3,"file":"conversation.d.ts","sourceRoot":"","sources":["../../../src/src/chat/conversation.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,+BAA+B,CAAC;AACjE,OAAO,KAAK,EAEV,aAAa,EACb,iBAAiB,EACjB,oBAAoB,EACrB,MAAM,YAAY,CAAC;AAEpB,QAAA,MAAM,gCAAgC,eAAuD,CAAC;AAE9F,8DAA8D;AAC9D,MAAM,MAAM,gCAAgC,GAAG,oBAAoB,GAAG;IACpE,CAAC,gCAAgC,CAAC,CAAC,EAAE,MAAM,CAAC;CAC7C,CAAC;AAEF,oFAAoF;AACpF,wBAAgB,+BAA+B,CAAC,OAAO,EAAE,oBAAoB,GAAG,MAAM,GAAG,SAAS,CAEjG;AAED,6EAA6E;AAC7E,wBAAgB,gCAAgC,CAC9C,OAAO,EAAE,oBAAoB,EAC7B,QAAQ,EAAE,MAAM,GACf,oBAAoB,CAQtB;AAED,uFAAuF;AACvF,wBAAgB,gCAAgC,CAAC,CAAC,SAAS,oBAAoB,EAC7E,MAAM,EAAE,oBAAoB,EAC5B,MAAM,EAAE,CAAC,GACR,CAAC,CAGH;AAED,uCAAuC;AACvC,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAyDhC,CAAC;AAEF,4CAA4C;AAC5C,MAAM,MAAM,WAAW,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,oBAAoB,CAAC,CAAC,CAAC;AAE/E,4CAA4C;AAC5C,eAAO,MAAM,yBAAyB,6DAErC,CAAC;AACF,iDAAiD;AACjD,MAAM,MAAM,gBAAgB,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,yBAAyB,CAAC,CAAC,CAAC;AAEzF,yCAAyC;AACzC,eAAO,MAAM,sBAAsB,6DAElC,CAAC;AACF,8CAA8C;AAC9C,MAAM,MAAM,aAAa,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC,CAAC;AAEnF,2CAA2C;AAC3C,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;GAiBpC,CAAC;AAEF,gDAAgD;AAChD,MAAM,MAAM,eAAe,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,wBAAwB,CAAC,CAAC,CAAC;AAEvF,sCAAsC;AACtC,eAAO,MAAM,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoB/B,CAAC;AAEF,6BAA6B;AAC7B,MAAM,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC,OAAO,mBAAmB,CAAC,CAAC,CAAC;AAE7E,8CAA8C;AAC9C,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,WAAW,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,OAAO,CAAC;IACf,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B;AAED,gDAAgD;AAChD,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,aAAa,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,OAAO,CAAC;IAChB,eAAe,CAAC,EAAE,OAAO,CAAC;CAC3B;AAED,uCAAuC;AACvC,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACd;AAED,4CAA4C;AAC5C,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,WAAW,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,kDAAkD;AAClD,KAAK,UAAU,GAAG,OAAO,CAAC,iBAAiB,EAAE;IAAE,UAAU,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC,CAAC;AAGpF,iCAAiC;AACjC,eAAO,MAAM,YAAY,QACuD,CAAC;AAEjF,uCAAuC;AACvC,wBAAgB,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,GAAG,KAAK,IAAI,MAAM,CAExE;AAED,yBAAyB;AACzB,wBAAgB,eAAe,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAG1D;AAED,0BAA0B;AAC1B,wBAAgB,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS,GAAG,WAAW,GAAG,OAAO,CAiB9F;AAED,2BAA2B;AAC3B,wBAAgB,QAAQ,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAEzE;AAED,2BAA2B;AAC3B,wBAAgB,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,CAMtF;AAoBD,gCAAgC;AAChC,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,CAQvD;AAED,uDAAuD;AACvD,wBAAgB,YAAY,CAC1B,IAAI,EAAE,iBAAiB,GACtB,IAAI,IAAI,iBAAiB,GAAG;IAAE,IAAI,EAAE,QAAQ,MAAM,EAAE,CAAC;IAAC,IAAI,EAAE,OAAO,CAAA;CAAE,CAEvE;AAED,mDAAmD;AACnD,wBAAgB,YAAY,CAAC,IAAI,EAAE,iBAAiB,GAAG,IAAI,IAAI,UAAU,CAOxE;AAED,2BAA2B;AAC3B,wBAAgB,aAAa,CAAC,IAAI,EAAE,UAAU,GAAG,MAAM,GAAG,SAAS,CAOlE;AAaD,uBAAuB;AACvB,wBAAgB,aAAa,CAC3B,KAAK,EAAE,WAAW,EAAE,EACpB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,MAAM,EACb,IAAI,EAAE;IACJ,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B,GACA,IAAI,CAwCN;AA2BD,uDAAuD;AACvD,wBAAgB,gCAAgC,CAAC,OAAO,EAAE,aAAa,GAAG,WAAW,EAAE,CAiFtF;AAiBD,sDAAsD;AACtD,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,aAAa,GAAG,OAAO,CAEtE;AAED,6CAA6C;AAC7C,wBAAgB,gCAAgC,CAAC,OAAO,EAAE,aAAa,GAAG,aAAa,CAEtF;AAED,6CAA6C;AAC7C,wBAAgB,gCAAgC,CAC9C,OAAO,EAAE,aAAa,EACtB,SAAS,EAAE,MAAM,GAChB,aAAa,CAaf;AA8BD,iDAAiD;AACjD,wBAAgB,cAAc,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,YAAY,CAOpE;AAED,mDAAmD;AACnD,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,cAAc,CAOxE;AAED,4CAA4C;AAC5C,wBAAgB,UAAU,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,YAAY,CAEhE;AAED,iDAAiD;AACjD,wBAAgB,eAAe,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,iBAAiB,CAK1E;AAED,2CAA2C;AAC3C,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,oBAAoB,GAAG,MAAM,CAoB5E;AAmdD,sDAAsD;AACtD,wBAAgB,wCAAwC,CACtD,QAAQ,EAAE,aAAa,EAAE,GACxB,oBAAoB,EAAE,CAmCxB"}

package/esm/src/chat/conversation.js

@@ -61,6 +61,13 @@ export const getMessagePartSchema = defineSchema((v) => v.discriminatedUnion("ty
         title: v.string().optional(),
         quote: v.string().optional(),
     }),
+    v.object({
+        type: v.literal("source_document"),
+        source_id: v.string(),
+        media_type: v.string(),
+        title: v.string().optional(),
+        filename: v.string().optional(),
+    }),
     v.object({ type: v.literal("step_start") }),
     v.object({
         type: v.literal("error"),
@@ -291,10 +298,20 @@ export function toConversationPartsFromUiMessage(message) {
             continue;
         }
         if (part.type === "source-document") {
+            if (!part.mediaType) {
+                parts.push({
+                    type: "citation",
+                    source_id: part.sourceId,
+                    title: part.title,
+                });
+                continue;
+            }
             parts.push({
-                type: "citation",
+                type: "source_document",
                 source_id: part.sourceId,
+                media_type: part.mediaType,
                 title: part.title,
+                ...(part.filename ? { filename: part.filename } : {}),
             });
             continue;
         }