add

Author: jihe520

backend/app/core/prompts.py

@@ -76,10 +76,11 @@ def get_writer_prompt(
         output：你需要按照要求的格式排版,只输出{format_output}排版的内容
         
         1. 当你输入图像引用时候，使用![image_name](image_name.png)
-        2. 你不需要输出markdown的这个```markdown格式，只需要输出markdown的内容
-        3. 严格按照参考用户输入的格式模板以及**正确的编号顺序**
-        4. 不需要询问用户
-        5. 当提到图片时，请使用提供的图片列表中的文件名
+        2. 你不需要输出markdown的这个```markdown格式，只需要输出markdown的内容，
+        3. Latex公式使用$$ $$包裹
+        4. 严格按照参考用户输入的格式模板以及**正确的编号顺序**
+        5. 不需要询问用户
+        6. 当提到图片时，请使用提供的图片列表中的文件名
         """
 
 

backend/app/tools/code_interpreter.py

@@ -115,7 +115,7 @@ async def _pre_execute_code(self):
         )
         await self.execute_code(init_code)
 
-    def _truncate_text(self, text: str, max_length: int = 2000) -> str:
+    def _truncate_text(self, text: str, max_length: int = 1000) -> str:
         """截断文本，保留开头和结尾的重要信息"""
         if len(text) <= max_length:
             return text

frontend/src/components/WriterEditor.vue

@@ -3,7 +3,6 @@ import { onMounted, ref, watch, computed } from 'vue';
 import { renderMarkdown } from '@/utils/markdown';
 import type { WriterMessage } from '@/utils/response'
 import { ScrollArea } from '@/components/ui/scroll-area'
-import { getWriterSeque } from '@/apis/commonApi';
 
 interface ContentSection {
   id: number;
@@ -14,14 +13,10 @@ interface ContentSection {
 
 const props = defineProps<{
   messages: WriterMessage[]
+  writerSequence: string[]
 }>()
 
-const writerSequence = ref<string[]>([]);
 
-onMounted(async () => {
-  const res = await getWriterSeque();
-  writerSequence.value = Array.isArray(res.data) ? res.data : [];
-});
 
 const sections = ref<ContentSection[]>([]);
 let nextId = 0;
@@ -39,11 +34,11 @@ const appendContent = async (content: string, sub_title?: string) => {
 
 // 根据 writerSequence 排序内容
 const sortedSections = computed(() => {
-  if (!writerSequence.value.length) return sections.value;
+  if (!props.writerSequence.length) return sections.value;
 
   return [...sections.value].sort((a, b) => {
-    const aIndex = a.sub_title ? writerSequence.value.indexOf(a.sub_title) : Infinity;
-    const bIndex = b.sub_title ? writerSequence.value.indexOf(b.sub_title) : Infinity;
+    const aIndex = a.sub_title ? props.writerSequence.indexOf(a.sub_title) : Infinity;
+    const bIndex = b.sub_title ? props.writerSequence.indexOf(b.sub_title) : Infinity;
 
     if (aIndex === Infinity && bIndex === Infinity) return 0;
     if (aIndex === Infinity) return 1;
@@ -73,7 +68,7 @@ watch(() => props.messages, async (messages) => {
     <div class="max-w-4xl mx-auto overflow-y-auto space-y-6">
       <TransitionGroup name="section" tag="div" class="space-y-6">
         <div v-for="section in sortedSections" :key="section.id"
-          class="bg-white rounded-lg shadow-lg overflow-hidden transform transition-all duration-500">
+          class="bg-white rounded-lg shadow-lg transform transition-all duration-500">
           <div class="p-6">
             <div class="prose prose-slate max-w-none" v-html="section.renderedContent"></div>
           </div>
@@ -142,16 +137,16 @@ watch(() => props.messages, async (messages) => {
 }
 
 .prose table {
-  @apply w-full border-collapse border border-gray-300 my-4;
+  @apply w-full border-collapse border border-gray-800;
 }
 
 .prose th,
 .prose td {
-  @apply border border-gray-300 p-2;
+  @apply border border-gray-800 p-1 text-center;
 }
 
-.prose thead {
-  @apply bg-gray-50;
+.prose th {
+  @apply font-bold text-gray-900 text-center;
 }
 
 .prose code {

frontend/src/pages/task/index.vue

@@ -17,19 +17,25 @@ import {
 import CoderEditor from '@/components/CoderEditor.vue'
 import WriterEditor from '@/components/WriterEditor.vue'
 import ChatArea from '@/components/ChatArea.vue'
-import { onMounted, onBeforeUnmount } from 'vue'
+import { onMounted, onBeforeUnmount, ref } from 'vue'
 import { useTaskStore } from '@/stores/task'
 import { ScrollArea } from '@/components/ui/scroll-area'
+import { getWriterSeque } from '@/apis/commonApi';
 
 const props = defineProps<{ task_id: string }>()
 const taskStore = useTaskStore()
 
+const writerSequence = ref<string[]>([]);
+
 console.log('Task ID:', props.task_id)
 
-onMounted(() => {
+onMounted(async () => {
   taskStore.connectWebSocket(props.task_id)
+  const res = await getWriterSeque();
+  writerSequence.value = Array.isArray(res.data) ? res.data : [];
 })
 
+
 onBeforeUnmount(() => {
   taskStore.closeWebSocket()
 })
@@ -69,7 +75,7 @@ onBeforeUnmount(() => {
             <TabsContent value="writer" class="flex-1 p-1 min-w-0 h-full overflow-auto">
               <Card class="min-w-0 rounded-lg">
                 <CardContent class="p-2 h-full min-w-0 overflow-auto">
-                  <WriterEditor :messages="taskStore.writerMessages" />
+                  <WriterEditor :messages="taskStore.writerMessages" :writerSequence="writerSequence" />
                 </CardContent>
               </Card>
             </TabsContent>

frontend/src/stores/task.ts

@@ -2,7 +2,7 @@ import { defineStore } from 'pinia'
 import { ref, computed } from 'vue'
 import { TaskWebSocket } from '@/utils/websocket'
 import type { Message, CoderMessage, WriterMessage } from '@/utils/response'
-import messageData from '@/test/20250430-163507-03282e09.json'
+import messageData from '@/test/20250430-232525-876ee531.json'
 
 export const useTaskStore = defineStore('task', () => {
   // 初始化时直接加载测试数据，确保页面首次渲染时有数据

frontend/src/test/20250430-232525-876ee531.json

@@ -39,14 +39,14 @@ const renderer: Partial<RendererObject> = {
       return `__IMAGE_PLACEHOLDER_${imageIndex++}__`
     })
 
-    // 处理块级公式
-    if (text.startsWith('\\[') && text.endsWith('\\]')) {
+    // 处理块级公式（使用 $$ 包裹）
+    if (text.startsWith('$$') && text.endsWith('$$')) {
       const tex = text.slice(2, -2).trim()
       return `<div class="math-block">${renderMath(tex, true)}</div>`
     }
 
-    // 处理行内公式
-    text = text.replace(/\\\((.*?)\\\)/g, (_, tex) => renderMath(tex.trim(), false))
+    // 处理行内公式（使用 $ 包裹）
+    text = text.replace(/\$(.*?)\$/g, (_, tex) => renderMath(tex.trim(), false))
 
     // 处理带括号的公式，确保不是已经处理过的
     if (!text.includes('class="katex"')) {