fix delete

src/components/submissions/ChainModal.vue

@@ -131,17 +131,47 @@
                 "
               >
                 <span>{{ round.question }}</span>
-                <span
+                <div style="display: flex; flex-direction: row; align-items: center; gap: 4px; flex-shrink: 0">
-                  v-if="round.prompt_level"
+                  <span
-                  :style="{
+                    v-if="round.source"
-                    flexShrink: 0,
+                    :style="{
-                    fontSize: '11px',
+                      fontSize: '10px',
-                    fontWeight: 'bold',
+                      padding: '1px 5px',
-                    color: levelColors[round.prompt_level],
+                      borderRadius: '4px',
-                    marginTop: '2px',
+                      background: round.source === 'conversation' ? '#e8f0fe' : '#f0f0f0',
-                  }"
+                      color: round.source === 'conversation' ? '#2060c0' : '#888',
-                  >L{{ round.prompt_level }}</span
+                      fontWeight: 500,
-                >
+                      whiteSpace: 'nowrap',
+                    }"
+                  >{{ round.source === "conversation" ? "对话" : "手动" }}</span>
+                  <span
+                    v-if="round.prompt_level"
+                    :style="{
+                      fontSize: '11px',
+                      fontWeight: 'bold',
+                      color: levelColors[round.prompt_level],
+                    }"
+                    >L{{ round.prompt_level }}</span
+                  >
+                  <n-popconfirm
+                    v-if="round.assistantMsgId && canDelete"
+                    :show-icon="false"
+                    @positive-click="deleteRound(index)"
+                  >
+                    <template #trigger>
+                      <n-button
+                        text
+                        size="tiny"
+                        type="error"
+                        style="margin-left: 2px"
+                        @click.stop
+                      >
+                        <Icon icon="lucide:trash-2" :width="12" />
+                      </n-button>
+                    </template>
+                    确定删除这一轮？
+                  </n-popconfirm>
+                </div>
               </div>
             </div>
           </div>
@@ -171,16 +201,21 @@
 
 <script setup lang="ts">
 import { computed, ref, watch } from "vue"
+import { NPopconfirm, NButton } from "naive-ui"
 import { Icon } from "@iconify/vue"
 import { Prompt } from "../../api"
 import type { PromptMessage } from "../../utils/type"
+import { user, roleSuper } from "../../store/user"
 
 const props = defineProps<{
   show: boolean
   userId: number
   taskId: number
+  username?: string
 }>()
 
+const canDelete = computed(() => roleSuper.value || (!!props.username && props.username === user.username))
+
 defineEmits<{ "update:show": [value: boolean] }>()
 
 const loading = ref(false)
@@ -190,7 +225,9 @@ const selectedRound = ref(0)
 const rounds = computed(() => {
   const result: {
     question: string
+    source: string | null
     prompt_level: number | null
+    assistantMsgId: number | null
     html: string | null
     css: string | null
     js: string | null
@@ -199,19 +236,25 @@ const rounds = computed(() => {
     if (msg.role !== "user") continue
     let html: string | null = null,
       css: string | null = null,
-      js: string | null = null
+      js: string | null = null,
+      assistantMsgId: number | null = null
     for (const reply of messages.value.slice(i + 1)) {
       if (reply.role === "user") break
-      if (reply.role === "assistant" && reply.code_html) {
+      if (reply.role === "assistant") {
-        html = reply.code_html
+        assistantMsgId = reply.id
-        css = reply.code_css
+        if (reply.code_html) {
-        js = reply.code_js
+          html = reply.code_html
+          css = reply.code_css
+          js = reply.code_js
+        }
         break
       }
     }
     result.push({
       question: msg.content,
+      source: msg.source ?? null,
       prompt_level: msg.prompt_level ?? null,
+      assistantMsgId,
       html,
       css,
       js,
@@ -220,6 +263,16 @@ const rounds = computed(() => {
   return result
 })
 
+async function deleteRound(index: number) {
+  const round = rounds.value[index]
+  if (!round.assistantMsgId) return
+  await Prompt.deleteMessagePair(round.assistantMsgId)
+  await loadMessages()
+  if (selectedRound.value >= rounds.value.length) {
+    selectedRound.value = Math.max(0, rounds.value.length - 1)
+  }
+}
+
 const levelColors: Record<number, string> = {
   1: "#aaa",
   2: "#6aab9c",

src/components/submissions/ExpandedSubTable.vue

@@ -24,7 +24,7 @@ import {
 import type { SubmissionOut } from "../../utils/type"
 import { TASK_TYPE } from "../../utils/const"
 import { parseTime } from "../../utils/helper"
-import { user } from "../../store/user"
+import { user, roleSuper } from "../../store/user"
 import { submission } from "../../store/submission"
 
 const props = defineProps<{
@@ -36,7 +36,7 @@ const props = defineProps<{
 const emit = defineEmits<{
   select: [id: string]
   delete: [row: SubmissionOut, parentId: string]
-  "show-chain": [userId: number, taskId: number]
+  "show-chain": [userId: number, taskId: number, username: string]
 }>()
 
 const isChallenge = computed(() => props.row.task_type === TASK_TYPE.Challenge)
@@ -76,57 +76,51 @@ const subColumns = computed((): DataTableColumn<SubmissionOut>[] => [
       )
     },
   },
-  ...(isChallenge.value
+  {
-    ? [
+    title: "操作",
-        {
+    key: "actions",
-          title: "提示词",
+    width: isChallenge.value ? 110 : 60,
-          key: "prompt",
+    render: (r: SubmissionOut) =>
-          width: 70,
+      h("div", { style: { display: "flex", gap: "8px" } }, [
-          render: (r: SubmissionOut) =>
+        ...(isChallenge.value
-            h(
+          ? [
-              NButton,
+              h(
-              {
+                NButton,
-                text: true,
+                {
-                type: "primary",
+                  text: true,
-                onClick: (e: Event) => {
+                  type: "primary",
-                  e.stopPropagation()
+                  onClick: (e: Event) => {
-                  emit("show-chain", r.userid, r.task_id)
+                    e.stopPropagation()
+                    emit("show-chain", r.userid, r.task_id, r.username)
+                  },
                 },
-              },
+                () => "查看",
-              () => "查看",
+              ),
-            ),
+            ]
-        } as DataTableColumn<SubmissionOut>,
+          : []),
-      ]
+        ...(r.username === user.username || roleSuper.value
-    : []),
+          ? [
-  ...(!isChallenge.value
+              h(
-    ? [
+                NPopconfirm,
-        {
+                { onPositiveClick: () => emit("delete", r, props.row.id) },
-          title: "操作",
+                {
-          key: "actions",
+                  trigger: () =>
-          width: 60,
+                    h(
-          render: (r: SubmissionOut) => {
+                      NButton,
-            if (r.username !== user.username) return null
+                      {
-            return h(
+                        text: true,
-              NPopconfirm,
+                        type: "error",
-              { onPositiveClick: () => emit("delete", r, props.row.id) },
+                        size: "small",
-              {
+                        onClick: (e: Event) => e.stopPropagation(),
-                trigger: () =>
+                      },
-                  h(
+                      () => "删除",
-                    NButton,
+                    ),
-                    {
+                  default: () => "确定删除这次提交？",
-                      text: true,
+                },
-                      type: "error",
+              ),
-                      size: "small",
+            ]
-                      onClick: (e: Event) => e.stopPropagation(),
+          : []),
-                    },
+      ]),
-                    () => "删除",
+  } as DataTableColumn<SubmissionOut>,
-                  ),
-                default: () => "确定删除这次提交？",
-              },
-            )
-          },
-        } as DataTableColumn<SubmissionOut>,
-      ]
-    : []),
 ])
 </script>

src/pages/Submissions.vue

@@ -107,6 +107,7 @@
     v-model:show="chainModal"
     :user-id="chainUserId"
     :task-id="chainTaskId"
+    :username="chainUsername"
   />
 </template>
 
@@ -160,6 +161,7 @@ const codeModal = ref(false)
 const chainModal = ref(false)
 const chainUserId = ref<number>(0)
 const chainTaskId = ref<number>(0)
+const chainUsername = ref<string>("")
 
 // 展开行
 const expandedKeys = ref<string[]>([])
@@ -201,9 +203,10 @@ async function clearAllFlags() {
   query.flag = null
 }
 
-function showChain(userId: number, taskId: number) {
+function showChain(userId: number, taskId: number, username: string) {
   chainUserId.value = userId
   chainTaskId.value = taskId
+  chainUsername.value = username
   chainModal.value = true
 }
 
@@ -219,7 +222,7 @@ const columns: DataTableColumn<SubmissionOut>[] = [
         loading: expandedLoading.has(row.id),
         onSelect: (id) => getSubmissionByID(id),
         onDelete: (r, parentId) => handleDelete(r, parentId),
-        "onShow-chain": (userId, taskId) => showChain(userId, taskId),
+        "onShow-chain": (userId, taskId, username) => showChain(userId, taskId, username),
       }),
   },
   {