Explorar o código

fix: LLM may not return `<think>` tag, cause thinking time keep increase (#13962)

NFish hai 2 meses
pai
achega
ece25bce1a
Modificáronse 2 ficheiros con 7 adicións e 6 borrados
  1. 3 3
      web/.env.example
  2. 4 3
      web/app/components/base/markdown.tsx

+ 3 - 3
web/.env.example

@@ -10,10 +10,10 @@ NEXT_PUBLIC_API_PREFIX=http://localhost:5001/console/api
 # console or api domain.
 # example: http://udify.app/api
 NEXT_PUBLIC_PUBLIC_API_PREFIX=http://localhost:5001/api
-# The APIFREX for MARKETPLACE
-NEXT_PUBLIC_MARKETPLACE_API_PREFIX=http://localhost:5002/api
+# The API PREFIX for MARKETPLACE
+NEXT_PUBLIC_MARKETPLACE_API_PREFIX=https://marketplace.dify.ai/api/v1
 # The URL for MARKETPLACE
-NEXT_PUBLIC_MARKETPLACE_URL_PREFIX=
+NEXT_PUBLIC_MARKETPLACE_URL_PREFIX=https://marketplace.dify.ai
 
 # SENTRY
 NEXT_PUBLIC_SENTRY_DSN=

+ 4 - 3
web/app/components/base/markdown.tsx

@@ -68,12 +68,13 @@ const preprocessLaTeX = (content: string) => {
 }
 
 const preprocessThinkTag = (content: string) => {
-  if (!content.trim().startsWith('<think>\n'))
+  if (!(content.trim().startsWith('<think>\n') || content.trim().startsWith('<details style=')))
     return content
 
   return flow([
-    (str: string) => str.replace('<think>\n', '<details>\n'),
-    (str: string) => str.replace('\n</think>', '\n[ENDTHINKFLAG]</details>'),
+    (str: string) => str.replaceAll('<think>\n', '<details>\n'),
+    (str: string) => str.replaceAll('\n</think>', '\n[ENDTHINKFLAG]</details>'),
+    (str: string) => str.replaceAll('\n</details>', '\n[ENDTHINKFLAG]</details>'),
   ])(content)
 }