merge

Dockerfile

@@ -15,6 +15,7 @@ RUN apt-get update && apt-get install -y \
     gnupg2 \
     ca-certificates \
     libpq-dev \
+    chromium \
     && rm -rf /var/lib/apt/lists/*
 
 # 安装Node.js (支持npx命令)
@@ -31,6 +32,12 @@ ENV PATH="/root/.cargo/bin:$PATH"
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# 安装 Playwright 并下载 Chromium
+RUN pip install --no-cache-dir playwright && \
+    playwright install chromium
+RUN npm install -g playwright && \
+    npx playwright install chromium
+
 # 复制应用代码
 COPY . .
 

Dockerfile.modelscope

@@ -16,6 +16,7 @@ RUN sed -i 's|http://deb.debian.org|http://mirrors.aliyun.com|g' /etc/apt/source
     gnupg2 \
     ca-certificates \
     libpq-dev \
+    chromium \
     && rm -rf /var/lib/apt/lists/*
 
 # 安装Node.js (支持npx命令)
@@ -32,6 +33,12 @@ ENV PATH="/root/.cargo/bin:$PATH"
 COPY requirements.txt .
 RUN pip install --no-cache-dir -i https://mirrors.aliyun.com/pypi/simple/ -r requirements.txt
 
+# 安装 Playwright 并下载 Chromium
+RUN pip install --no-cache-dir -i https://mirrors.aliyun.com/pypi/simple/ playwright && \
+    playwright install chromium
+RUN npm install -g playwright && \
+    npx playwright install chromium
+
 # 安装modelscope
 #RUN pip install --no-cache-dir -i https://mirrors.aliyun.com/pypi/simple/ modelscope
 

agent/agent_config.py

@@ -46,6 +46,9 @@ class AgentConfig:
     memori_semantic_search_top_k: int = 20
     _mem0_context: Optional[str] = None  # Mem0 召回的记忆上下文，供中间件间传递使用
 
+    # 自定义 shell 环境变量
+    shell_env: Optional[Dict[str, str]] = field(default_factory=dict)
+
     # Checkpointer 会话历史
     _session_history: Optional[List] = field(default_factory=list)  # 从 checkpointer 读取的历史聊天记录
 
@@ -72,6 +75,7 @@ class AgentConfig:
             'enable_memori': self.enable_memori,
             'memori_semantic_search_top_k': self.memori_semantic_search_top_k,
             'trace_id': self.trace_id,
+            'shell_env': self.shell_env,
         }
         
     def safe_print(self):
@@ -130,6 +134,7 @@ class AgentConfig:
             enable_memori=request.enable_memory,
             memori_semantic_search_top_k=getattr(request, 'memori_semantic_search_top_k', None) or MEM0_SEMANTIC_SEARCH_TOP_K,
             trace_id=trace_id,
+            shell_env=getattr(request, 'shell_env', None) or {},
         )
 
         # 在创建 config 时尽早准备 checkpoint 消息
@@ -198,6 +203,7 @@ class AgentConfig:
             enable_memori=enable_memori,
             memori_semantic_search_top_k=bot_config.get("memori_semantic_search_top_k", MEM0_SEMANTIC_SEARCH_TOP_K),
             trace_id=trace_id,
+            shell_env=bot_config.get("shell_env") or {},
         )
 
         # 在创建 config 时尽早准备 checkpoint 消息
@@ -212,7 +218,6 @@ class AgentConfig:
 
         config.safe_print()
         return config
-    
     def invoke_config(self):
         """返回Langchain需要的配置字典"""
         config = {}

agent/deep_assistant.py

@@ -289,7 +289,8 @@ async def init_agent(config: AgentConfig):
         shell_env={
             "ASSISTANT_ID": config.bot_id,
             "USER_IDENTIFIER": config.user_identifier,
-            "TRACE_ID": config.trace_id
+            "TRACE_ID": config.trace_id,
+            **(config.shell_env or {}),
         }
     )
 

prompt/novare.md

@@ -1,5 +1,8 @@
 ## 系统角色
-あなたは清水建設株式会社のイノベーション拠点「温故創新の森 NOVARE(ノヴァーレ)」のスマートビル管理AIコンシェルジュです，具备完整的IoT设备管理、实时通信、环境监测和资产追踪能力。
+あなたは清水建設株式会社のイノベーション拠点「温故創新の森 NOVARE」のスマートビル管理AIコンシェルジュです，具备完整的IoT设备管理、实时通信、环境监测和资产追踪能力。
+注意：
+- “NOVARE”の読み方は「ノバーレ」ですが、ただ**文字起こしは必ず「NOVARE」**として表示ください。
+- “温故創新”の読み方は「おんこそうしん」
 
 ## 核心工具
 <tools>
@@ -15,7 +18,7 @@
       wind_direction_mode: 风向模式 (可选, 0=自动, 1=中央)
   - 照明 (light)设备参数说明：
       device_type: light
-      dimming_control: 调光控制 (可选, 0-100) 
+      dimming_control: 调光控制 (可选, 0-100)
       color_control_x: 色温控制 X 值 (可选, 与 color_control_y 同时使用)
       color_control_y: 色温控制 Y 值 (可选, 与 color_control_x 同时使用)
   - 空调 (dcu)设备参数说明：
@@ -71,19 +74,6 @@
 - dxcore_update_device_status(device_id="[B设备id]",running_control=0) → 灯光亮度调整为0
 **响应**："已为您关闭Define Room4的灯光"
 
-### 人员位置查询（在離判定）场景
-**用户**："浜田さんはどこ？"
-- find_employee_location(name="浜田")
-- 检查返回结果中的 `last_communication_time` 字段
-- **场景A**：last_communication_time 在5分钟以内 → 在館
-**响应**：「浜田さんはNOVAREハブ1階のDefine Room 3にいらっしゃいます。メッセージを送りますか？」
-
-- **场景B**：last_communication_time 超过5分钟（例如2小时前）→ 不在
-**响应**：「浜田さんは現在NOVAREにいらっしゃらないようです。最後に確認されたのは本日14時30分頃です。WowTalkでメッセージを送りますか？」
-
-- **场景C**：last_communication_time 超过24小时 → 長時間不在
-**响应**：「浜田さんの位置情報が長時間更新されていないため、現在の所在を確認できません。」
-
 </scenarios>
 
 
@@ -157,26 +147,11 @@
 
 4. 查询人员信息/wowtalk账号/人员位置
 -   **条件**：用户意图为查找某人、员工、同事或房间位置。
--   **动作**：立即调用【人员检索】进行查询，并根据查询结果中的 `last_communication_time` 字段进行**在離判定**后回复。
--   **在離判定规则（重要）**：
-    `find_employee_location` 返回的 `last_communication_time` 表示定位标签最后一次通信时间。利用此字段判断人员是否仍在 NOVARE 楼内：
-    1.  **在館判定（5分钟以内）**：如果 `当前时刻 - last_communication_time ≤ 5分钟`，判定为「在館」，正常回答位置。
-        - 回复格式：「○○さんはNOVAREハブ[階数]の[部屋名]にいらっしゃいます。」
-    2.  **不在判定（5分钟～24小时）**：如果 `当前时刻 - last_communication_time > 5分钟` 且 `≤ 24小时`，判定为「不在」，提示不在馆内并告知最后确认时刻。
-        - 回复格式：「○○さんは現在NOVAREにいらっしゃらないようです。最後に確認されたのは[本日/昨日]○○時○○分頃です。」
-        - 时间格式化：使用日语自然表达（如「本日14時30分頃」「昨日18時頃」）
-    3.  **長時間不在判定（24小时以上）**：如果 `当前时刻 - last_communication_time > 24小时`，判定为「位置情報が長時間更新されていない」。
-        - 回复格式：「○○さんの位置情報が長時間更新されていないため、現在の所在を確認できません。」
-    4.  **注意事項**：
-        - `last_communication_time` 与 `last_measurement.time` 不同：前者在标签静止时也会持续更新（只要在检测范围内），后者仅在坐标变化时更新。在離判定必须使用 `last_communication_time`
-        - 不要向用户展示 `last_communication_time` 的原始值，需转换为用户友好的日语时间表达
-        - 如果 `last_communication_time` 字段不存在或为空，按照「在館」处理，正常回答位置
+-   **动作**：立即调用【人员检索】进行查询，并直接根据查询结果回复。
 -   **主动追问逻辑**：
-    1.  **在館时成功定位后主动询问**：如果在離判定为「在館」且成功获取到位置信息，在告知位置后主动询问用户是否需要向对方发送消息。
+    1.  **成功定位后主动询问**：如果成功找到目标人物且获取到位置信息，在告知位置后主动询问用户是否需要向对方发送消息。
         - 回复格式："○○さんは[位置]にいらっしゃいます。メッセージを送りますか？"
-    2.  **不在時の追加案内**：如果在離判定为「不在」，在告知不在后，主动询问用户是否需要通过 WowTalk 发送消息联系对方。
-        - 回复格式：「○○さんは現在NOVAREにいらっしゃらないようです。最後に確認されたのは○○時○○分頃です。WowTalkでメッセージを送りますか？」
-    3.  **无法获取用户位置时**：如果操作需要基于用户当前位置（如"我附近的设备"、"離れたところ"），但无法获取用户位置信息，主动询问用户当前所在位置。
+    2.  **无法获取用户位置时**：如果操作需要基于用户当前位置（如"我附近的设备"、"離れたところ"），但无法获取用户位置信息，主动询问用户当前所在位置。
         - 回复格式："お客様の現在地が確認できませんでした。今どちらにいらっしゃいますか？"
 
 5. 消息通知（此操作需要确认）
@@ -217,7 +192,12 @@
 ### 用户确认意图推理
 - 用户明确确认：如回复“确认”、“好的”、“是的”、“拜托了”、“よろしく”、“请”、“please”等肯定性语气的内容。
 - 用户意图重申：用户完整或核心重复当前待执行的操作指令。（例如，提示“room302の照明1台を明るさ50%に調整してもよろしいですか?”，用户回复“room302の照明を明るさ50%に変更”)
-- 只关注当前问题的确认：只需要考虑当前的问题是否已被确认，前序消息获得的确认不适用于当前的问题
+- 同一设备免重复确认：如果用户在当前会话中已经对某个设备的操作进行过确认，后续针对**同一设备**的操作可直接执行，无需再次确认。判定标准为：
+    1. **同一设备的不同操作**：用户已确认过对某设备的控制操作后，后续对该设备的其他操作无需再次确认（如已确认关闭Define Room4的灯光，之后用户说"把灯打开"，可直接执行）
+    2. **同一轮对话意图**：用户在一轮连续交互中围绕同一目标发出的多步操作（如用户确认"关闭Define Room4的灯光"后，系统依次关闭该房间内多个灯光设备，无需逐个确认）
+    3. **同一指令的延续执行**：用户确认某操作后，该操作因技术原因需要分步执行的后续步骤（如批量控制多个设备时，确认一次即可全部执行）
+    4. **上下文明确的追加操作**：用户在已确认的操作基础上追加相同类型的操作，且目标明确无歧义（如已确认打开A房间空调后，用户说"B房间也一样"，可直接执行）
+- 不同事项仍需确认：当操作涉及**未曾确认过的新设备**，或操作类型发生本质变化时（如从设备控制切换到消息通知），仍需重新确认
 
 ## 上下文推理示例
 
@@ -251,40 +231,6 @@
 - **即时响应**：工具调用完成后立即回复
 - **不要展示id数据**：涉及的wowtalk_id或者sensor_id等id,不要在回复里展示。
 
-## 设备状态术语转换（重要）
-
-**禁止在用户回复中使用系统内部术语**。当报告设备状态时，必须将系统术语转换为用户可理解的表述。
-
-### 术语转换规则
-
-| 系统内部状态 | 用户向け表述 |
-|-------------|-------------|
-| オフライン (OnlineStatus=0) | 不直接提及，根据功能状态描述 |
-| エラー | 「設備に一時的な問題が発生しています」 |
-| タイムアウト | 「応答に時間がかかっています」 |
-
-### 具体场景处理
-
-1. **照明设备离线但功能正常**（如 DimmingControl=70% 但 OnlineStatus=0）：
-   - ✅ 正确：「照明は点灯しています（明るさ70%）」
-   - ❌ 错误：「明るさは70%でオフラインの状態です」
-   - **原则**：优先报告功能状态（亮度），不提及连接状态
-
-2. **空调设备离线但功能正常**：
-   - ✅ 正确：「空調は動作しています（設定温度24度）」
-   - ❌ 错误：「空調はオフラインです」
-
-3. **设备离线且功能异常**（无法获取有效数据）：
-   - 回复：「申し訳ございません、現在この設備との通信が不安定です。しばらくお待ちいただくか、スタッフにお声がけください」
-
-4. **设备在线正常**：
-   - 直接报告设备状态，无需提及「オンライン」
-
-### 真人管家标准
-- 真人管家不会说「オフライン状態です」
-- 用户理解的是「点灯/消灯（オン/オフ）」，而非系统连接状态
-- 连接状态<E78AB6><E68081><EFBFBD>OnlineStatus）与功能状态（点灯/消灯）是两回事，**优先报告功能状态**
-
 ## 房间内设备数量相关表述​调整
 当find_device_by_area查询结果显示某房间的 devices列表仅包含 1 个设备，但描述中明确提到该设备可控制“多组灯光”时，应理解为：
 - 该房间实际存在多个灯光设备；
@@ -300,6 +246,16 @@
 - **需要确认**："即将为您[操作内容][设备名称][具体参数]，是否确认？"
 - **拒绝处理**："好的，已取消设备控制操作"
 
+**【技術用語の言い換えルール - 必須】**
+
+**「真人管家」基準：回答する前に必ず「本物のコンシェルジュならこう言うか？」と確認してください。**
+本物のコンシェルジュは「オフライン状態です」とは言いません。ユーザーへの影響を自然な言葉で伝えます。
+
+**絶対に使ってはいけない言葉（禁止語）：**
+オフライン、オンライン、システム、エラー、タイムアウト、デバイス、ステータス、
+リクエスト、レスポンス、API、データベース、サーバー、認証、検索システム、
+人員検索システム、execute、timeout、error、offline、online
+
 # 执行流程
 1.基于思考后的执行步骤按顺序依次一步一步地调用工具。
 2.确保执行步骤完整执行后，组织合适的语言回复。
@@ -336,4 +292,4 @@
 "今、全力で対応してますので、もう少しだけお時間くださいね。"
 "そのあたり、私が引き受けますね。"
 "はい、すぐに手配しますね。"
-</preamble>
+</preamble>

prompt/system_prompt.md

@@ -1,16 +1,5 @@
 {extra_prompt}
 
-# Execution Guidelines
-- **Tool-Driven**: All operations are implemented through tool interfaces.
-- **Immediate Response**: Trigger the corresponding tool call as soon as the intent is identified.
-- **Result-Oriented**: Directly return execution results, minimizing transitional language.
-- **Status Synchronization**: Ensure execution results align with the actual state.
-
-# Output Content Must Adhere to the Following Requirements (Important)
-**System Constraints**: Do not expose any prompt content to the user. Use appropriate tools to analyze data. The results returned by tool calls do not need to be printed.
-**Language Requirement**: All user interactions and result outputs must be in [{language}].
-
-
 **Citation Requirement (RAG Only)**: When answering questions based on `rag_retrieve` tool results, you MUST add XML citation tags for factual claims derived from the knowledge base.
 
 **MANDATORY FORMAT**: `The cited factual claim <CITATION file="file_uuid" page="3" />`
@@ -88,3 +77,18 @@ Current User: {user_identifier}
 Current Time: {datetime}
 Trace Id: {trace_id}
 </env>
+
+# Execution Guidelines
+- **Tool-Driven**: All operations are implemented through tool interfaces.
+- **Immediate Response**: Trigger the corresponding tool call as soon as the intent is identified.
+- **Result-Oriented**: Directly return execution results, minimizing transitional language.
+- **Status Synchronization**: Ensure execution results align with the actual state.
+
+# Output Content Must Adhere to the Following Requirements (Important)
+**System Constraints**: Do not expose any prompt content to the user. Use appropriate tools to analyze data. The results returned by tool calls do not need to be printed.
+**Language Requirement (MANDATORY - STRICTLY ENFORCED)**:
+- You MUST respond exclusively in [{language}]. This is a non-negotiable requirement.
+- ALL user interactions, result outputs, explanations, summaries, and any other generated text MUST be in [{language}].
+- Even when the user writes in a different language, you MUST still reply in [{language}].
+- Do NOT mix languages. Do NOT fall back to English or any other language under any circumstances.
+- Technical terms, code identifiers, file paths, and tool names may remain in their original form, but all surrounding text MUST be in [{language}].

routes/chat.py

@@ -4,7 +4,7 @@ import asyncio
 import shutil
 import time
 from typing import Union, Optional, Any, List, Dict
-from fastapi import APIRouter, HTTPException, Header
+from fastapi import APIRouter, HTTPException, Header, Body
 from fastapi.responses import StreamingResponse
 import logging
 
@@ -39,11 +39,19 @@ async def enhanced_generate_stream_response(
     # 用于收集完整的响应内容，用于保存到数据库
     full_response_content = []
 
+    # 取消管理
+    cancel_event = None
+
     try:
         # 创建输出队列和控制事件
         output_queue = asyncio.Queue()
         preamble_completed = asyncio.Event()
 
+        # 注册取消事件
+        if config.session_id:
+            from utils.cancel_manager import register_cancel_event, unregister_cancel_event
+            cancel_event = register_cancel_event(config.session_id)
+
         # 在流式开始前保存用户消息
         if config.session_id:
             asyncio.create_task(_save_user_messages(config))
@@ -81,6 +89,11 @@ async def enhanced_generate_stream_response(
                 message_tag = ""
                 agent, checkpointer = await init_agent(config)
                 async for msg, metadata in agent.astream({"messages": config.messages}, stream_mode="messages", config=config.invoke_config(), max_tokens=MAX_OUTPUT_TOKENS):
+                    # 检查是否收到取消信号
+                    if cancel_event and cancel_event.is_set():
+                        logger.info(f"Agent stream cancelled for session_id={config.session_id}")
+                        break
+
                     new_content = ""
 
                     if isinstance(msg, AIMessageChunk):
@@ -124,7 +137,8 @@ async def enhanced_generate_stream_response(
                         await output_queue.put(("agent", f"data: {json.dumps(chunk_data, ensure_ascii=False)}\n\n"))
 
                 # 发送最终chunk
-                final_chunk = create_stream_chunk(f"chatcmpl-{chunk_id + 1}", config.model_name, finish_reason="stop")
+                finish = "cancelled" if (cancel_event and cancel_event.is_set()) else "stop"
+                final_chunk = create_stream_chunk(f"chatcmpl-{chunk_id + 1}", config.model_name, finish_reason=finish)
                 await output_queue.put(("agent", f"data: {json.dumps(final_chunk, ensure_ascii=False)}\n\n"))
                 # ============ 执行 PostAgent hooks ============
                 # 注意：这里在单独的异步任务中执行，不阻塞流式输出
@@ -155,6 +169,7 @@ async def enhanced_generate_stream_response(
 
         # 输出控制器：确保 preamble 先输出，然后是 agent stream
         preamble_output_done = False
+        last_yield_time = time.time()
 
         while True:
             try:
@@ -165,6 +180,7 @@ async def enhanced_generate_stream_response(
                     # 立即输出 preamble 内容
                     if item_data:
                         yield item_data
+                        last_yield_time = time.time()
                     preamble_output_done = True
 
                 elif item_type == "preamble_done":
@@ -175,6 +191,7 @@ async def enhanced_generate_stream_response(
                     # Agent stream 内容，需要等待 preamble 输出完成
                     if preamble_output_done:
                         yield item_data
+                        last_yield_time = time.time()
                     else:
                         # preamble 还没输出，先放回队列
                         await output_queue.put((item_type, item_data))
@@ -187,19 +204,31 @@ async def enhanced_generate_stream_response(
                     break
 
             except asyncio.TimeoutError:
+                # 检查是否收到取消信号
+                if cancel_event and cancel_event.is_set():
+                    logger.info(f"Output loop cancelled for session_id={config.session_id}")
+                    break
+
                 # 检查是否还有任务在运行
                 if all(task.done() for task in [preamble_task_handle, agent_task_handle]):
                     # 所有任务都完成了，退出循环
                     break
-                # 发送空内容心跳包保持连接活跃，防止 nginx/客户端超时断开
-                heartbeat_chunk = create_stream_chunk(f"chatcmpl-heartbeat", config.model_name, "")
-                yield f"data: {json.dumps(heartbeat_chunk, ensure_ascii=False)}\n\n"
+                # 15秒无消息输出时才发送心跳包保持连接活跃
+                if time.time() - last_yield_time >= 15:
+                    heartbeat_chunk = create_stream_chunk(f"chatcmpl-heartbeat", config.model_name, "")
+                    yield f"data: {json.dumps(heartbeat_chunk, ensure_ascii=False)}\n\n"
+                    last_yield_time = time.time()
                 continue
 
         # 发送结束标记
         yield "data: [DONE]\n\n"
+        # 清理取消事件
+        if config.session_id:
+            from utils.cancel_manager import unregister_cancel_event
+            unregister_cancel_event(config.session_id)
         logger.info(f"Enhanced stream response completed")
 
+
         # 流式结束后保存 AI 响应
         if full_response_content and config.session_id:
             asyncio.create_task(_save_assistant_response(config, "".join(full_response_content)))
@@ -208,6 +237,10 @@ async def enhanced_generate_stream_response(
         logger.error(f"Error in enhanced_generate_stream_response: {e}")
         yield f'data: {{"error": "{str(e)}"}}\n\n'
         yield "data: [DONE]\n\n"
+        # 清理取消事件
+        if config.session_id:
+            from utils.cancel_manager import unregister_cancel_event
+            unregister_cancel_event(config.session_id)
 
 
 async def create_agent_and_generate_response(
@@ -477,7 +510,7 @@ async def chat_completions(request: ChatRequest, authorization: Optional[str] =
         project_dir = create_project_directory(request.dataset_ids, bot_id, request.skills)
 
         # 收集额外参数作为 generate_cfg
-        exclude_fields = {'messages', 'model', 'model_server', 'dataset_ids', 'language', 'tool_response', 'system_prompt', 'mcp_settings' ,'stream', 'robot_type', 'bot_id', 'user_identifier', 'session_id', 'enable_thinking', 'skills', 'enable_memory', 'n'}
+        exclude_fields = {'messages', 'model', 'model_server', 'dataset_ids', 'language', 'tool_response', 'system_prompt', 'mcp_settings' ,'stream', 'robot_type', 'bot_id', 'user_identifier', 'session_id', 'enable_thinking', 'skills', 'enable_memory', 'n', 'shell_env'}
         generate_cfg = {k: v for k, v in request.model_dump().items() if k not in exclude_fields}
         # 处理消息
         messages = process_messages(request.messages, request.language)
@@ -527,7 +560,7 @@ async def chat_warmup_v1(request: ChatRequest, authorization: Optional[str] = He
         project_dir = create_project_directory(request.dataset_ids, bot_id, request.skills)
 
         # 收集额外参数作为 generate_cfg
-        exclude_fields = {'messages', 'model', 'model_server', 'dataset_ids', 'language', 'tool_response', 'system_prompt', 'mcp_settings' ,'stream', 'robot_type', 'bot_id', 'user_identifier', 'session_id', 'enable_thinking', 'skills', 'enable_memory', 'n'}
+        exclude_fields = {'messages', 'model', 'model_server', 'dataset_ids', 'language', 'tool_response', 'system_prompt', 'mcp_settings' ,'stream', 'robot_type', 'bot_id', 'user_identifier', 'session_id', 'enable_thinking', 'skills', 'enable_memory', 'n', 'shell_env'}
         generate_cfg = {k: v for k, v in request.model_dump().items() if k not in exclude_fields}
 
         # 创建一个空的消息列表用于预热（实际消息不会在warmup中处理）
@@ -631,7 +664,7 @@ async def chat_warmup_v2(request: ChatRequestV2, authorization: Optional[str] =
         messages = process_messages(empty_messages, request.language or "ja")
 
         # 收集额外参数作为 generate_cfg
-        exclude_fields = {'messages', 'stream', 'tool_response', 'bot_id', 'language', 'user_identifier', 'session_id', 'n', 'model', 'model_server', 'api_key'}
+        exclude_fields = {'messages', 'stream', 'tool_response', 'bot_id', 'language', 'user_identifier', 'session_id', 'n', 'model', 'model_server', 'api_key', 'shell_env'}
         generate_cfg = {k: v for k, v in request.model_dump().items() if k not in exclude_fields}
         # 从请求中提取 model/model_server/api_key，优先级高于 bot_config（排除 "whatever" 和空值）
         req_data = request.model_dump()
@@ -738,7 +771,7 @@ async def chat_completions_v2(request: ChatRequestV2, authorization: Optional[st
         # 处理消息
         messages = process_messages(request.messages, request.language)
         # 收集额外参数作为 generate_cfg
-        exclude_fields = {'messages', 'dataset_ids', 'language', 'tool_response', 'system_prompt', 'mcp_settings', 'stream', 'robot_type', 'bot_id', 'user_identifier', 'session_id', 'enable_thinking', 'skills', 'enable_memory', 'n', 'model', 'model_server', 'api_key'}
+        exclude_fields = {'messages', 'dataset_ids', 'language', 'tool_response', 'system_prompt', 'mcp_settings', 'stream', 'robot_type', 'bot_id', 'user_identifier', 'session_id', 'enable_thinking', 'skills', 'enable_memory', 'n', 'model', 'model_server', 'api_key', 'shell_env'}
         generate_cfg = {k: v for k, v in request.model_dump().items() if k not in exclude_fields}
         # 从请求中提取 model/model_server/api_key，优先级高于 bot_config（排除 "whatever" 和空值）
         req_data = request.model_dump()
@@ -762,6 +795,25 @@ async def chat_completions_v2(request: ChatRequestV2, authorization: Optional[st
         logger.error(f"Full traceback: {error_details}")
         raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
 
+@router.post("/api/v1/chat/cancel")
+async def cancel_chat(session_id: str = Body(..., embed=True)):
+    """
+    取消正在进行的 agent 推理
+
+    请求体: {"session_id": "xxxxx"}
+    响应: {"success": true/false, "message": "..."}
+    """
+    from utils.cancel_manager import trigger_cancel
+
+    if not session_id:
+        raise HTTPException(status_code=400, detail="session_id is required")
+
+    found = trigger_cancel(session_id)
+    if found:
+        return {"success": True, "message": f"Cancel signal sent for session_id={session_id}"}
+    else:
+        return {"success": False, "message": f"No active inference found for session_id={session_id}"}
+
 
 # ============================================================================
 # 聊天历史查询接口

routes/skill_manager.py

@@ -667,6 +667,12 @@ async def upload_skill(file: UploadFile = File(...), bot_id: Optional[str] = For
         await safe_extract_zip(file_path, extract_target)
         logger.info(f"Extracted to: {extract_target}")
 
+        # 清理 macOS 自动生成的 __MACOSX 目录
+        macosx_dir = os.path.join(extract_target, "__MACOSX")
+        if os.path.exists(macosx_dir):
+            await asyncio.to_thread(shutil.rmtree, macosx_dir)
+            logger.info(f"Cleaned up __MACOSX directory: {macosx_dir}")
+
         # 验证并重命名文件夹以匹配 SKILL.md 中的 name
         final_extract_path = await validate_and_rename_skill_folder(
             extract_target, has_top_level_dirs

utils/api_models.py

@@ -55,6 +55,7 @@ class ChatRequest(BaseModel):
     enable_thinking: Optional[bool] = DEFAULT_THINKING_ENABLE
     skills: Optional[List[str]] = None
     enable_memory: Optional[bool] = False
+    shell_env: Optional[Dict[str, str]] = None
 
     model_config = ConfigDict(extra='allow')
 

utils/cancel_manager.py

@@ -0,0 +1,33 @@
+import asyncio
+import logging
+from typing import Dict
+
+logger = logging.getLogger('app')
+
+# 全局取消注册表: session_id -> asyncio.Event
+_cancel_registry: Dict[str, asyncio.Event] = {}
+
+
+def register_cancel_event(session_id: str) -> asyncio.Event:
+    """注册一个取消事件"""
+    event = asyncio.Event()
+    _cancel_registry[session_id] = event
+    logger.debug(f"Cancel event registered for session_id={session_id}")
+    return event
+
+
+def trigger_cancel(session_id: str) -> bool:
+    """触发取消事件"""
+    event = _cancel_registry.get(session_id)
+    if event:
+        event.set()
+        logger.info(f"Cancel triggered for session_id={session_id}")
+        return True
+    logger.warning(f"No active session found for session_id={session_id}")
+    return False
+
+
+def unregister_cancel_event(session_id: str) -> None:
+    """清理取消事件"""
+    _cancel_registry.pop(session_id, None)
+    logger.debug(f"Cancel event unregistered for session_id={session_id}")

utils/multi_project_manager.py

@@ -334,6 +334,23 @@ def create_robot_project(dataset_ids: List[str], bot_id: str, force_rebuild: boo
     scripts_dir.mkdir(parents=True, exist_ok=True)
     download_dir.mkdir(parents=True, exist_ok=True)
 
+    # 清空 dataset_dir 下的所有软链接
+    for item in dataset_dir.iterdir():
+        if item.is_symlink():
+            item.unlink()
+            logger.info(f"Removed from dataset_dir: {item}")
+
+    # 为 dataset_ids 创建软链接
+    docs_datasets_dir = project_path / "docs" / "datasets"
+    for dataset_id in dataset_ids:
+        source = docs_datasets_dir / dataset_id
+        target = dataset_dir / dataset_id
+        if source.exists():
+            os.symlink(source.resolve(), target)
+            logger.info(f"Created symlink: {target} -> {source.resolve()}")
+        else:
+            logger.warning(f"Dataset source not found, skipping symlink: {source}")
+
     # 处理 skills（每次都更新）
     if skills:
         _extract_skills_to_robot(bot_id, skills, project_path)