Add files via upload

web/static/js/chat.js

@@ -711,9 +711,18 @@ function addAttackChainButton(conversationId) {
     }
 
     if (conversationId) {
-        attackChainBtn.disabled = false;
-        attackChainBtn.title = '查看当前对话的攻击链';
-        attackChainBtn.onclick = () => showAttackChain(conversationId);
+        const isRunning = typeof isConversationTaskRunning === 'function'
+            ? isConversationTaskRunning(conversationId)
+            : false;
+        if (isRunning) {
+            attackChainBtn.disabled = true;
+            attackChainBtn.title = '当前对话正在执行，请稍后再生成攻击链';
+            attackChainBtn.onclick = null;
+        } else {
+            attackChainBtn.disabled = false;
+            attackChainBtn.title = '查看当前对话的攻击链';
+            attackChainBtn.onclick = () => showAttackChain(conversationId);
+        }
     } else {
         attackChainBtn.disabled = true;
         attackChainBtn.title = '请选择一个对话以查看攻击链';
@@ -721,6 +730,10 @@ function addAttackChainButton(conversationId) {
     }
 }
 
+function updateAttackChainAvailability() {
+    addAttackChainButton(currentConversationId);
+}
+
 // 显示攻击链模态框
 async function showAttackChain(conversationId) {
     // 防止重复点击