i/next-ai-draw-io
1
0
Fork 0
mirror of https://github.com/DayuanJiang/next-ai-draw-io.git synced 2026-01-02 22:32:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: update system message to reflect diagram terminology and integrate OpenAI model

Browse Source
This commit is contained in:
dayuan.jiang
2025-03-19 12:02:07 +00:00
parent 7caaa358e4
commit bb753c4140
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
app/api/chat/route.ts
View File

@@ -1,4 +1,5 @@
import { google } from "@ai-sdk/google"; import { google } from "@ai-sdk/google";
import { openai } from "@ai-sdk/openai";
import { streamText } from "ai"; import { streamText } from "ai";
import { z } from "zod"; import { z } from "zod";
import { readFileSync } from 'fs'; import { readFileSync } from 'fs';
@@ -17,7 +18,7 @@ export async function POST(req: Request) {
// Read and escape the guide content // Read and escape the guide content
const systemMessage = ` const systemMessage = `
You are a helpful assistant that can create, edit, and display flowcharts using draw.io through xml strings. You are a helpful assistant that can create, edit, and display diagram using draw.io through xml strings.
You can use the following tools: You can use the following tools:
---Tool1--- ---Tool1---
tool name: display_flow_chart tool name: display_flow_chart
@@ -27,15 +28,15 @@ parameters: {
} }
---Tool2--- ---Tool2---
tool name: fetch_flow_chart tool name: fetch_flow_chart
description: Get the current flowchart XML from draw.io description: Get the current diagram XML from draw.io
parameters: {} parameters: {}
---End of tools--- ---End of tools---
When you need to modify the flowchart, you need fetch the current flowchart XML from draw.io and then modify it and display it again. When you need to modify the diagram, you need fetch the current diagram XML from draw.io and then modify it and display it again.
here is a guide for the XML format: ${guide} here is a guide for the XML format: ${guide}
You can use the tools to create and manipulate flowcharts. You can use the tools to create and manipulate diagram.
You can also answer questions and provide explanations. You can also answer questions and provide explanations.
If user want you to draw something rather than flowchart, you can use the combination of the shape to draw it. If user want you to draw something rather than diagram, you can use the combination of the shape to draw it.
`; `;
// Add system message if only user message is provided // Add system message if only user message is provided
@@ -44,7 +45,8 @@ If user want you to draw something rather than flowchart, you can use the combin
: messages; : messages;
const result = streamText({ const result = streamText({
model: google("gemini-2.0-flash"), // model: google("gemini-2.0-flash"),
model: openai("gpt-4o"),
messages: enhancedMessages, messages: enhancedMessages,
tools: { tools: {
// Client-side tool that will be executed on the client // Client-side tool that will be executed on the client