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

fix: Corrected a bug where the current XML file was not attached to the user message.

Browse Source
This commit is contained in:
dayuan.jiang
2025-03-24 02:38:27 +00:00
parent 05d1adc179
commit 19dacfad30
2 changed files with 21 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -13,8 +13,9 @@ export const maxDuration = 30;
const guide = readFileSync(resolve('./app/api/chat/xml_guide.md'), 'utf8');
export async function POST(req: Request) {
const { messages } = await req.json();
const body = await req.json();
const { messages, data = {} } = body;
// Read and escape the guide content
const systemMessage = `
@@ -50,12 +51,22 @@ ${guide}
"""
`;
// Add system message if only user message is provided
const enhancedMessages = messages.length === 1
? [{ role: "system", content: systemMessage }, ...messages]
const lastMessage = messages[messages.length - 1];
const formattedContent = `
Current diagram XML:
"""xml
${data.xml || ''}
"""
User input:
"""md
${lastMessage.content}
"""`;
: messages;
let enhancedMessages = messages.length === 1
? [{ role: "system", content: systemMessage }, { ...lastMessage, content: formattedContent }]
: [...messages.slice(0, -1), { ...lastMessage, content: formattedContent }];
console.log(enhancedMessages);
const result = streamText({
// model: google("gemini-2.0-flash"),
model: openai("gpt-4o"),