add

Author: jihe520

backend/app/core/prompts.py

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

backend/app/tools/code_interpreter.py

@@ -75,32 +75,23 @@ async def _upload_all_files(self):
                 logger.error(f"工作目录不存在: {self.work_dir}")
                 raise FileNotFoundError(f"工作目录不存在: {self.work_dir}")
 
-            # 只上传数据集文件
             files = [
                 f for f in os.listdir(self.work_dir) if f.endswith((".csv", ".xlsx"))
             ]
             logger.info(f"工作目录中的文件列表: {files}")
 
             for file in files:
                 file_path = os.path.join(self.work_dir, file)
-                logger.info(f"处理文件: {file_path}")
-
                 if os.path.isfile(file_path):
                     try:
                         with open(file_path, "rb") as f:
                             content = f.read()
-                            # 使用 files API 上传文件
+                            # 使用官方推荐的 files.write 方法
                             await self.sbx.files.write(f"/home/user/{file}", content)
                             logger.info(f"成功上传文件到沙箱: {file}")
                     except Exception as e:
                         logger.error(f"上传文件 {file} 失败: {str(e)}")
                         raise
-                else:
-                    logger.info(f"跳过目录: {file_path}")
-
-            # 验证上传的文件
-            uploaded_files = await self.sbx.files.list("/home/user")
-            logger.info(f"沙箱中的文件列表: {[f.name for f in uploaded_files]}")
 
         except Exception as e:
             logger.error(f"文件上传过程失败: {str(e)}")

frontend/src/components/Bubble.vue

@@ -184,7 +184,6 @@ const renderedContent = computed(() => {
   /* 浅灰 */
   color: #0f172a;
   box-shadow: 0 2px 8px rgba(16, 185, 129, 0.08);
-  border: 1px solid #38bdf8;
 }
 
 /* WriterAgent 气泡颜色 */
@@ -193,6 +192,5 @@ const renderedContent = computed(() => {
   /* 浅黄 */
   color: #92400e;
   box-shadow: 0 2px 8px rgba(251, 191, 36, 0.08);
-  border: 1px solid #facc15;
 }
 </style>

frontend/src/components/ChatArea.vue

@@ -23,7 +23,7 @@ const sendMessage = () => {
 </script>
 
 <template>
-  <div class="flex h-full flex-col p-3 bg-gray-50">
+  <div class="flex h-full flex-col p-3">
     <div ref="scrollRef" class="flex-1 overflow-y-auto">
       <template v-for="message in props.messages" :key="message.id">
         <div class="mb-3">

frontend/src/components/WriterEditor.vue

@@ -137,16 +137,27 @@ watch(() => props.messages, async (messages) => {
 }
 
 .prose table {
-  @apply w-full border-collapse border border-gray-800;
+  @apply w-full border-collapse my-6 !border-2 !border-gray-400;
+}
+
+.prose th {
+  @apply !bg-gray-200 p-3 text-left !font-bold !text-gray-900 !border !border-gray-400;
 }
 
-.prose th,
 .prose td {
-  @apply border border-gray-800 p-1 text-center;
+  @apply p-3 !text-gray-900 !border !border-gray-400;
 }
 
-.prose th {
-  @apply font-bold text-gray-900 text-center;
+.prose tr {
+  @apply !border !border-gray-400;
+}
+
+.prose tr:nth-child(even) {
+  @apply !bg-gray-50;
+}
+
+.prose tr:hover {
+  @apply !bg-gray-100;
 }
 
 .prose code {

frontend/src/pages/task/index.vue

@@ -21,6 +21,7 @@ import { onMounted, onBeforeUnmount, ref } from 'vue'
 import { useTaskStore } from '@/stores/task'
 import { ScrollArea } from '@/components/ui/scroll-area'
 import { getWriterSeque } from '@/apis/commonApi';
+import { Button } from '@/components/ui/button';
 
 const props = defineProps<{ task_id: string }>()
 const taskStore = useTaskStore()
@@ -52,8 +53,8 @@ onBeforeUnmount(() => {
       <ResizablePanel :default-size="70" class="h-full min-w-0">
         <div class="flex h-full flex-col min-w-0">
           <Tabs default-value="coder" class="w-full h-full flex flex-col">
-            <div class="border-b px-4 py-1">
-              <TabsList class="justify-center">
+            <div class="border-b px-4 py-1 flex justify-between">
+              <TabsList class="">
                 <TabsTrigger value="coder" class="text-sm">
                   CoderAgent
                 </TabsTrigger>
@@ -62,6 +63,9 @@ onBeforeUnmount(() => {
                 </TabsTrigger>
               </TabsList>
               <!--  TODO: 其他选项 -->
+              <Button @click="taskStore.downloadMessages" class="flex justify-end">
+                下载消息
+              </Button>
             </div>
 
             <TabsContent value="coder" class="h-full p-1 flex-1 overflow-auto">
@@ -83,10 +87,7 @@ onBeforeUnmount(() => {
         </div>
       </ResizablePanel>
     </ResizablePanelGroup>
-    <button @click="taskStore.downloadMessages"
-      class="absolute top-2 right-2 z-10 bg-blue-500 text-white px-3 py-1 rounded">
-      下载消息
-    </button>
+
   </div>
 </template>
 

frontend/src/stores/task.ts

@@ -2,12 +2,12 @@ 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-232525-876ee531.json'
+// import messageData from '@/test/20250430-232525-876ee531.json'
 
 export const useTaskStore = defineStore('task', () => {
   // 初始化时直接加载测试数据，确保页面首次渲染时有数据
-  const messages = ref<Message[]>(messageData as Message[])
-  // const messages = ref<Message[]>([])
+  // const messages = ref<Message[]>(messageData as Message[])
+  const messages = ref<Message[]>([])
   let ws: TaskWebSocket | null = null
 
   // 连接 WebSocket

frontend/src/utils/markdown.ts

@@ -40,25 +40,14 @@ const renderer: Partial<RendererObject> = {
     })
 
     // 处理块级公式（使用 $$ 包裹）
-    if (text.startsWith('$$') && text.endsWith('$$')) {
-      const tex = text.slice(2, -2).trim()
-      return `<div class="math-block">${renderMath(tex, true)}</div>`
-    }
+    const blockMathPattern = /\$\$([\s\S]*?)\$\$/g
+    text = text.replace(blockMathPattern, (_, tex) => {
+      return `<div class="math-block">${renderMath(tex.trim(), 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"')) {
-      text = text.replace(/\((.*?)\)/g, (match, tex) => {
-        // 检查是否是数学公式（包含数学符号）
-        if (/[+\-*/=^_{}\\]/.test(tex)) {
-          return renderMath(tex.trim(), false)
-        }
-        return match // 如果不是数学公式，保持原样
-      })
-    }
-
     // 还原图片占位符
     text = text.replace(/__IMAGE_PLACEHOLDER_(\d+)__/g, (_, index) => {
       const [, alt, src] = images[parseInt(index)]