add

add

Author: jihe520

backend/app/core/prompts.py

@@ -75,7 +75,7 @@ def get_writer_prompt(
         skill：熟练掌握{format_output}排版,
         output：你需要按照要求的格式排版,只输出{format_output}排版的内容
         
-        1. 当你输入图像引用时候，image_name.png,就可正确引用显示
+        1. 当你输入图像引用时候，使用![image_name](image_name.png)
         2. 你不需要输出markdown的这个```markdown格式，只需要输出markdown的内容
         3. 严格按照参考用户输入的格式模板以及**正确的编号顺序**
         4. 不需要询问用户

backend/app/tools/code_interpreter.py

@@ -323,6 +323,7 @@ async def _push_to_websocket(self, content_to_display: list[OutputItem] | None):
     async def get_created_images(self, section: str) -> list[str]:
         """获取当前 section 创建的图片列表"""
         if not self.sbx:
+            logger.warning("沙箱环境未初始化")
             return []
 
         try:
@@ -335,6 +336,7 @@ async def get_created_images(self, section: str) -> list[str]:
             self.created_images = list(
                 set(self.section_output[section]["images"]) - set(self.created_images)
             )
+            logger.info(f"{section}-获取创建的图片列表: {self.created_images}")
             return self.created_images
         except Exception as e:
             logger.error(f"获取创建的图片列表失败: {str(e)}")
@@ -388,6 +390,10 @@ async def download_all_files_from_sandbox(self) -> None:
             # 下载新文件或更新已修改的文件
             for file in sandbox_files:
                 try:
+                    # 排除 .bash_logout、.bashrc 和 .profile 文件
+                    if file.name in [".bash_logout", ".bashrc", ".profile"]:
+                        continue
+
                     local_path = os.path.join(self.work_dir, file.name)
                     should_download = True
 
@@ -398,19 +404,14 @@ async def download_all_files_from_sandbox(self) -> None:
                         pass
 
                     if should_download:
-                        content = await self.sbx.files.read(file.path)
+                        # 使用 bytes 格式读取文件内容，确保正确处理二进制数据
+                        content = await self.sbx.files.read(file.path, format="bytes")
 
                         # 确保目标目录存在
                         os.makedirs(self.work_dir, exist_ok=True)
 
                         # 写入文件
                         with open(local_path, "wb") as f:
-                            if isinstance(content, str):
-                                if "," in content:  # 处理data URL格式
-                                    content = content.split(",")[1]
-                                    content = base64.b64decode(content)
-                                else:
-                                    content = content.encode("utf-8")
                             f.write(content)
                         logger.info(f"同步文件: {file.name}")
 

frontend/src/components/CoderEditor.vue

@@ -1,9 +1,5 @@
-<script lang="ts">
-export const description = 'A sidebar with a collapsible file tree.'
-export const iframeHeight = '800px'
-</script>
 <script setup lang="ts">
-import { ref, computed } from 'vue'
+import { ref } from 'vue'
 import {
   SidebarInset,
   SidebarProvider,
@@ -16,86 +12,6 @@ import {
   ResizablePanel,
   ResizableHandle,
 } from '@/components/ui/resizable'
-import type { CoderMessage } from '@/utils/response'
-
-interface Cell {
-  type: 'markdown' | 'code'
-  content: string
-  output: {
-    type: 'text' | 'table' | 'plot'
-    content?: string
-    data?: {
-      headers?: string[]
-      rows?: any[][]
-    }
-  } | null
-  isPreview?: boolean
-}
-
-const props = defineProps<{
-  messages: CoderMessage[]
-}>()
-
-// 将代码执行结果转换为Notebook单元格
-const cells = computed<Cell[]>(() => {
-  const notebookCells: Cell[] = []
-
-  for (const msg of props.messages) {
-    // 如果有普通内容，创建markdown单元格
-    if (msg.content) {
-      notebookCells.push({
-        type: 'markdown',
-        content: msg.content,
-        output: null,
-        isPreview: true
-      })
-    }
-
-    // 如果有代码，创建代码单元格
-    if (msg.code) {
-      const cell: Cell = {
-        type: 'code',
-        content: msg.code,
-        output: null
-      }
-
-      // 如果有执行结果，添加到输出
-      if (msg.code_results && msg.code_results.length > 0) {
-        const result = msg.code_results[0] // 暂时只取第一个结果
-
-        if (result.res_type === 'result') {
-          cell.output = {
-            type: 'table',
-            data: {
-              headers: [],
-              rows: []
-            }
-          }
-          try {
-            const data = JSON.parse(result.msg || '{}')
-            if (data.headers && data.rows) {
-              cell.output.data = data
-            }
-          } catch (e) {
-            cell.output = {
-              type: 'text',
-              content: result.msg || ''
-            }
-          }
-        } else {
-          cell.output = {
-            type: 'text',
-            content: result.msg || ''
-          }
-        }
-      }
-
-      notebookCells.push(cell)
-    }
-  }
-
-  return notebookCells
-})
 
 const isCollapsed = ref(false)
 
@@ -104,7 +20,6 @@ const handleCollapse = () => {
   isCollapsed.value = !isCollapsed.value
 }
 
-
 </script>
 <template>
   <SidebarProvider :collapsed="isCollapsed">
@@ -121,8 +36,8 @@ const handleCollapse = () => {
           <header class="flex h-10 shrink-0 items-center gap-2 border-b px-4">
             <SidebarTrigger class="-ml-1" @click="handleCollapse" />
           </header>
-          <div class="flex-1 min-h-0 min-w-0 overflow-auto">
-            <NotebookArea :cells="cells" class="h-full min-w-0" />
+          <div class="flex-1 min-h-0 min-w-0 overflow-auto h-full">
+            <NotebookArea class="h-full min-w-0 pb-4" />
           </div>
         </SidebarInset>
       </ResizablePanel>