better handling ai messages

Author: darkoatanasovski

examples/nextjs-ai-chatbot/components/messages.tsx

@@ -7,7 +7,6 @@ import { useApp } from "@/contexts/app";
 import { useEffect, useRef } from "react";
 import { useParams } from "next/navigation";
 import Image from "next/image";
-import { GetToolOutput } from "@/utils/helpers";
 import ToolsOutput from "./tools";
 
 export default function Messages() {

examples/nextjs-ai-chatbot/contexts/app.tsx

@@ -112,7 +112,11 @@ function AppProviderInner({ children }: { children: ReactNode }) {
         .then((data) => {
           const messages = data.map((message: any) => ({
             id: message.id,
-            role: message.user.id === "ai-bot" ? "assistant" : "user",
+            role:
+              message.user.id === "ai-bot" ||
+              message.message_type === "agent_response"
+                ? "assistant"
+                : "user",
             parts: [
               {
                 type: "text",

packages/node-sdk/src/ai-sdk-helpers.ts

@@ -86,6 +86,9 @@ export class AISDKStreamStorage {
         await this.streamStorage.sendMessage(channelId, {
           text,
           userId: this.streamStorage.getBotUserId(),
+          metadata: {
+            message_type: "agent_response",
+          },
         });
       }
     }
@@ -106,6 +109,9 @@ export class AISDKStreamStorage {
             toolOutput
           )}\n\`\`\``,
           userId: this.streamStorage.getBotUserId(),
+          metadata: {
+            message_type: "agent_response",
+          },
         });
       }
     }