fix: optimize slice

YunYouJun · Feb 4, 2024 · 0587b88 · 0587b88
1 parent cd20221
commit 0587b88
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/server/api/generate.ts b/server/api/generate.ts
@@ -6,10 +6,21 @@ export default defineEventHandler(async (event) => {
   const data = await getCouplets(query.prompt as string)
   const { content } = data
 
-  const unwrapperContent = (content || '{}')?.replace('```json\n', '').replace('```', '')
+  let unWrapperContent = content || ''
+  const startPos = unWrapperContent.indexOf('{')
+  const endPos = unWrapperContent.lastIndexOf('}')
+
+  if (startPos === -1 || endPos === -1) {
+    // eslint-disable-next-line no-console
+    console.log(content)
+    return
+  }
+
+  unWrapperContent = unWrapperContent.slice(startPos, endPos + 1)
+  unWrapperContent = (unWrapperContent || '{}')?.replace('```json\n', '').replace('```', '')
   let coupletData: SprintFestivalCouplets | undefined
   try {
-    coupletData = JSON.parse(unwrapperContent) as SprintFestivalCouplets
+    coupletData = JSON.parse(unWrapperContent) as SprintFestivalCouplets
   }
   catch (e) {
     // eslint-disable-next-line no-console