Skip to content

Commit f35d32b

Browse files
committed
Update example.
1 parent dec5bce commit f35d32b

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

README.md

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -37,11 +37,10 @@ For each example, you also need to download the GGUF model and start the Llama.c
3737
## Example Route
3838

3939
```ts
40-
import { ModelFusionTextStream } from "@modelfusion/vercel-ai";
40+
import { ModelFusionTextStream, asChatMessages } from "@modelfusion/vercel-ai";
4141
import { Message, StreamingTextResponse } from "ai";
4242
import {
4343
Llama2Prompt,
44-
ChatMessage,
4544
llamacpp,
4645
streamText,
4746
trimChatPrompt,
@@ -73,10 +72,7 @@ export async function POST(req: Request) {
7372
"Follow the user's instructions carefully.",
7473

7574
// map Vercel AI SDK Message to ModelFusion ChatMessage:
76-
messages: messages.filter(
77-
// only user and assistant roles are supported:
78-
(message) => message.role === "user" || message.role === "assistant"
79-
) as ChatMessage[],
75+
messages: asChatMessages(messages),
8076
},
8177
})
8278
);

0 commit comments

Comments
 (0)