wangjunxiao/tianshu-engine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: improve AI response JSON parsing robustness

Browse Source
This commit is contained in:
cocos02
2026-06-15 14:06:32 +08:00
parent 7b3ad95549
commit b1ea2e6474
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
vite.config.ts
View File

@@ -118,7 +118,9 @@ function apiSavePlugin() {
let resolvedSessionId = sessionId
let aiText = ''
for (const line of stdout.trim().split('\n')) {
for (const rawLine of stdout.split(/\r?\n/)) {
const line = rawLine.trim()
if (!line) continue
try {
const event = JSON.parse(line)
if (!resolvedSessionId && event.sessionID) resolvedSessionId = event.sessionID
@@ -129,16 +131,16 @@ function apiSavePlugin() {
}
if (mode === 'json') {
const jsonMatch = aiText.match(/```json\n?([\s\S]*?)\n?```/)
const jsonStr = jsonMatch ? jsonMatch[1] : aiText
const codeBlock = aiText.match(/```(?:json)?\s*\n?([\s\S]*?)\n?\s*```/)
let jsonStr = codeBlock ? codeBlock[1] : aiText
try {
JSON.parse(jsonStr)
res.writeHead(200, { 'Content-Type': 'application/json' })
res.end(JSON.stringify({ result: jsonStr, sessionId: resolvedSessionId || '' }))
} catch {
res.writeHead(500)
res.end(JSON.stringify({ error: 'invalid JSON returned', raw: stdout }))
const bareMatch = aiText.match(/(\{[\s\S]*\}|\[[\s\S]*\])/)
jsonStr = bareMatch ? bareMatch[0] : aiText
}
res.writeHead(200, { 'Content-Type': 'application/json' })
res.end(JSON.stringify({ result: jsonStr, sessionId: resolvedSessionId || '' }))
} else {
res.writeHead(200, { 'Content-Type': 'application/json' })
res.end(JSON.stringify({ result: aiText || 'done', sessionId: resolvedSessionId || '' }))