Update packages.

Author: lgrammel

app/api/llama2/route.ts

@@ -1,8 +1,7 @@
-import { ModelFusionTextStream } from "@modelfusion/vercel-ai";
+import { ModelFusionTextStream, asChatMessages } from "@modelfusion/vercel-ai";
 import { Message, StreamingTextResponse } from "ai";
 import {
   Llama2Prompt,
-  ChatMessage,
   llamacpp,
   streamText,
   trimChatPrompt,
@@ -34,10 +33,7 @@ export async function POST(req: Request) {
           "Follow the user's instructions carefully.",
 
         // map Vercel AI SDK Message to ModelFusion ChatMessage:
-        messages: messages.filter(
-          // only user and assistant roles are supported:
-          (message) => message.role === "user" || message.role === "assistant"
-        ) as ChatMessage[],
+        messages: asChatMessages(messages),
       },
     })
   );

app/api/mistral/route.ts

@@ -1,12 +1,6 @@
-import { ModelFusionTextStream } from "@modelfusion/vercel-ai";
+import { ModelFusionTextStream, asChatMessages } from "@modelfusion/vercel-ai";
 import { Message, StreamingTextResponse } from "ai";
-import {
-  ChatMessage,
-  TextPrompt,
-  llamacpp,
-  streamText,
-  trimChatPrompt,
-} from "modelfusion";
+import { TextPrompt, llamacpp, streamText, trimChatPrompt } from "modelfusion";
 
 export const runtime = "edge";
 
@@ -34,10 +28,7 @@ export async function POST(req: Request) {
           "Follow the user's instructions carefully.",
 
         // map Vercel AI SDK Message to ModelFusion ChatMessage:
-        messages: messages.filter(
-          // only user and assistant roles are supported:
-          (message) => message.role === "user" || message.role === "assistant"
-        ) as ChatMessage[],
+        messages: asChatMessages(messages),
       },
     })
   );

app/api/openhermes/route.ts

@@ -1,8 +1,7 @@
-import { ModelFusionTextStream } from "@modelfusion/vercel-ai";
+import { ModelFusionTextStream, asChatMessages } from "@modelfusion/vercel-ai";
 import { Message, StreamingTextResponse } from "ai";
 import {
   ChatMLPrompt,
-  ChatMessage,
   llamacpp,
   streamText,
   trimChatPrompt,
@@ -34,10 +33,7 @@ export async function POST(req: Request) {
           "Follow the user's instructions carefully.",
 
         // map Vercel AI SDK Message to ModelFusion ChatMessage:
-        messages: messages.filter(
-          // only user and assistant roles are supported:
-          (message) => message.role === "user" || message.role === "assistant"
-        ) as ChatMessage[],
+        messages: asChatMessages(messages),
       },
     })
   );

package-lock.json

@@ -9,10 +9,10 @@
     "lint": "next lint"
   },
   "dependencies": {
-    "@modelfusion/vercel-ai": "1.0.0",
+    "@modelfusion/vercel-ai": "1.1.0",
     "ai": "2.2.28",
     "next": "14.0.3",
-    "modelfusion": ">= 0.104.0",
+    "modelfusion": ">= 0.105.0",
     "react": "18.2.0",
     "react-dom": "^18.2.0"
   },