@deepagents/context 0.22.0 → 0.23.0

package/dist/index.js

@@ -412,6 +412,18 @@ import {
   createUIMessageStream as createUIMessageStream2,
   generateId as generateId3
 } from "ai";
+function toMessageFragment(item) {
+  if (isFragment(item) && isMessageFragment(item)) {
+    return item;
+  }
+  return message(item);
+}
+function chatMessageToUIMessage(item) {
+  if (isFragment(item) && isMessageFragment(item)) {
+    return item.codec.decode();
+  }
+  return item;
+}
 async function chat(agent2, messages, options) {
   const context = agent2.context;
   if (!context) {
@@ -422,17 +434,20 @@ async function chat(agent2, messages, options) {
   if (messages.length === 0) {
     throw new Error("messages must not be empty");
   }
-  const lastMessage = messages[messages.length - 1];
+  const lastItem = messages[messages.length - 1];
+  const lastFragment = toMessageFragment(lastItem);
+  const lastUIMessage = chatMessageToUIMessage(lastItem);
   let assistantMsgId;
-  if (lastMessage.role === "assistant") {
-    context.set(message(lastMessage));
+  if (lastUIMessage.role === "assistant") {
+    context.set(lastFragment);
     await context.save({ branch: false });
-    assistantMsgId = lastMessage.id;
+    assistantMsgId = lastUIMessage.id;
   } else {
-    context.set(message(lastMessage));
+    context.set(lastFragment);
     await context.save();
     assistantMsgId = options?.generateMessageId?.() ?? generateId3();
   }
+  const uiMessages = messages.map(chatMessageToUIMessage);
   const streamContextVariables = options?.contextVariables === void 0 ? {} : options.contextVariables;
   const result = await agent2.stream(streamContextVariables, {
     transform: options?.transform
@@ -443,12 +458,12 @@ async function chat(agent2, messages, options) {
     sendFinish: true,
     sendReasoning: true,
     sendSources: true,
-    originalMessages: messages,
+    originalMessages: uiMessages,
     generateMessageId: () => assistantMsgId,
     messageMetadata: options?.messageMetadata
   });
   return createUIMessageStream2({
-    originalMessages: messages,
+    originalMessages: uiMessages,
     generateId: () => assistantMsgId,
     onStepFinish: async ({ responseMessage }) => {
       const normalizedMessage = {
@@ -4106,7 +4121,7 @@ var SqliteContextStore = class extends ContextStore {
     ).all(chatId);
     const nodes = messageRows.map((row) => {
       const data = JSON.parse(row.data);
-      const content = typeof data === "string" ? data : JSON.stringify(data);
+      const content = typeof data === "string" ? data : Array.isArray(data.parts) ? data.parts.filter((p) => p.type === "text").map((p) => p.text).join(" ") : JSON.stringify(data);
       return {
         id: row.id,
         parentId: row.parentId,
@@ -6485,6 +6500,7 @@ export {
   assistant,
   assistantText,
   chat,
+  chatMessageToUIMessage,
   clarification,
   correction,
   createAdaptivePollingState,
@@ -6538,6 +6554,7 @@ export {
   structuredOutput,
   styleGuide,
   term,
+  toMessageFragment,
   useSandbox,
   user,
   userContext,