bug fix

Author: jihe520

README.md

@@ -44,11 +44,14 @@
 - [x] 添加正确文献引用
 - [x] 更多测试案例
 - [ ] docker 部署
-- [ ] 引入用户的交互（选择模型，@agent重写，等等）
+- [ ] human in loop: 引入用户的交互（选择模型，@agent重写，等等）
+- [ ] feedback: evaluate the result and modify
 - [x] codeinterpreter 接入云端 如 e2b 等供应商..
 - [ ] 多语言: R 语言, matlab
-- [ ] 绘图 napki,draw.io
+- [ ] 绘图 napki,draw.io,plantuml,svg, mermaid.js
 - [ ] 添加 benchmark
+- [ ] web search tool
+- [ ] RAG 知识库
 
 ## 视频demo
 
@@ -70,15 +73,14 @@
 
 
 
-1. 配置模型
+1. 配置环境变量
 
 复制`/backend/.env.dev.example`到`/backend/.env.dev`(删除`.example` 后缀)
 
 **配置环境变量**
 
 推荐模型能力较强的、参数量大的模型。
 
-
 复制`/fronted/.env.example`到`/fronted/.env`(删除`.example` 后缀)
 
 

backend/app/core/llm/llm.py

@@ -1,5 +1,5 @@
 import json
-from app.utils.common_utils import transform_link
+from app.utils.common_utils import transform_link, split_footnotes
 from app.utils.log_util import logger
 import time
 from app.schemas.response import (
@@ -89,6 +89,7 @@ async def send_message(self, response, agent_name, sub_title=None):
                 agent_msg: CoderMessage = CoderMessage(content=content)
             case AgentType.WRITER:
                 # 处理 Markdown 格式的图片语法
+                content, _ = split_footnotes(content)
                 content = transform_link(self.task_id, content)
                 agent_msg: WriterMessage = WriterMessage(
                     content=content,

backend/app/core/prompts.py

@@ -118,14 +118,14 @@ def get_writer_prompt(
         skill：熟练掌握{format_output}排版,如图片、**公式**、表格、列表等
         output：你需要按照要求的格式排版,只输出正确的{format_output}排版的内容
         
-        1. 当你输入图像引用时候，使用![image_name](image_name.png)
+        1. 当你输入图像引用时候，使用![image_name](image_name.png),图片引用需换行，在段后。
         2. 你不需要输出markdown的这个```markdown格式，只需要输出markdown的内容，
-        3. LaTex: 行内公式（Inline Formula） 和 块级公式（Block Formula）
+        3. LaTex: 行内公式（Inline Formula）$ 和 块级公式（Block Formula）$$ 包裹
         4. 严格按照参考用户输入的格式模板以及**正确的编号顺序**
         5. 不需要询问用户 
         6. 当提到图片时，请使用提供的图片列表中的文件名
         7. when you write,check if you need to use tools search_papers to cite. if you need, markdown Footnote e.g.[^1]
-        8. List all references at the end in markdown footnote format.
+        8. List all references at the end in markdown footnote format. Dont't use title # ,just list in the end.
         9. Include an empty line between each citation for better readability.
         10. 对于问题背景和模型介绍，需查询文献调用tools search_papers
         """

backend/app/models/user_output.py

@@ -2,6 +2,7 @@
 import re
 from app.utils.data_recorder import DataRecorder
 from app.schemas.A2A import WriterResponse
+from app.utils.log_util import logger
 
 
 class UserOutput:
@@ -41,6 +42,7 @@ def get_model_build_solve(self) -> str:
         return model_build_solve
 
     def get_result_to_save(self, ques_count):
+        logger.info(f"开始处理结果保存，问题数量: {ques_count}")
         # 动态顺序获取拼接res value，正确拼接顺序
         ques_str = [f"ques{i}" for i in range(1, ques_count + 1)]
         seq = [
@@ -54,6 +56,7 @@ def get_result_to_save(self, ques_count):
             "sensitivity_analysis",
             "judge",
         ]
+        logger.debug(f"处理序列: {seq}")
 
         # 收集所有内容和脚注
         all_content = []
@@ -62,20 +65,26 @@ def get_result_to_save(self, ques_count):
         footnote_mapping = {}  # 用于存储原始编号到新编号的映射
 
         # 第一遍：收集所有引用并建立映射
+        logger.info("开始第一遍处理：收集引用并建立映射")
         for key in seq:
             if key not in self.res:
+                logger.debug(f"跳过不存在的键: {key}")
                 continue
 
             content = self.res[key]["response_content"]
             footnotes = self.res[key]["footnotes"]
 
             if footnotes:
+                logger.debug(f"处理 {key} 的脚注，数量: {len(footnotes)}")
                 for num, content in footnotes:  # 直接解构元组
                     if num not in footnote_mapping:
                         footnote_mapping[num] = str(footnote_counter)
                         footnote_counter += 1
 
+        logger.info(f"脚注映射完成，共有 {len(footnote_mapping)} 个脚注")
+
         # 第二遍：更新内容和脚注
+        logger.info("开始第二遍处理：更新内容和脚注")
         for key in seq:
             if key not in self.res:
                 continue
@@ -85,6 +94,7 @@ def get_result_to_save(self, ques_count):
 
             # 更新内容中的引用编号
             if footnotes:
+                logger.debug(f"更新 {key} 的内容和脚注")
                 # 更新正文中的引用
                 for old_num, new_num in footnote_mapping.items():
                     content = content.replace(f"[^{old_num}]", f"[^{new_num}]")
@@ -109,6 +119,7 @@ def get_result_to_save(self, ques_count):
             )
             final_content += "\n\n" + "\n".join(sorted_footnotes)
 
+        logger.info(f"结果处理完成，最终内容长度: {len(final_content)}")
         return final_content
 
     def save_result(self, ques_count):

backend/app/utils/common_utils.py

@@ -116,5 +116,5 @@ def split_footnotes(text: str) -> tuple[str, list[tuple[str, str]]]:
 
     # 匹配脚注定义
     footnotes = re.findall(r"\[\^(\d+)\]:\s*(.+?)(?=\n\[\^|\n\n|\Z)", text, re.DOTALL)
-
+    logger.info(f"main_text:{main_text} \n footnotes:{footnotes}")
     return main_text, footnotes

frontend/.gitignore

@@ -28,3 +28,6 @@ dist-ssr
 .env.development
 .env.production
 .env.test
+
+
+test

frontend/src/pages/task/index.vue

@@ -60,11 +60,11 @@ onBeforeUnmount(() => {
 <template>
   <div class="fixed inset-0">
     <ResizablePanelGroup direction="horizontal" class="h-full rounded-lg border">
-      <ResizablePanel :default-size="30" class="h-full">
+      <ResizablePanel :default-size="40" class="h-full">
         <ChatArea :messages="taskStore.chatMessages" />
       </ResizablePanel>
       <ResizableHandle />
-      <ResizablePanel :default-size="70" class="h-full min-w-0">
+      <ResizablePanel :default-size="60" 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 flex justify-between">
@@ -124,12 +124,4 @@ onBeforeUnmount(() => {
   </div>
 </template>
 
-<style scoped>
-:deep(body),
-:deep(html) {
-  overflow: hidden;
-  height: 100%;
-  margin: 0;
-  padding: 0;
-}
-</style>
+<style scoped></style>

frontend/src/stores/task.ts

@@ -2,12 +2,13 @@ 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/20250524-115938-d4c84576.json'
+import messageData from '@/test/20250524-133152-0f9d6be8.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