import { h as ElButton, aj as refresh_default, ak as plus_default, aa as delete_default, al as ElTable, am as ElTableColumn, q as ElTag, r as ElText, ay as view_default, ao as vLoading, ap as ElPagination, T as ElCard, w as ElForm, x as ElFormItem, y as ElInput, aq as ElSelect, ar as ElOption, a1 as ElUpload, az as upload_default, v as ElDialog, a3 as ElDescriptions, a4 as ElDescriptionsItem, E as ElMessage, D as ElMessageBox } from "./element-plus.CkEW9frc.js";
import { a as KB_STATUS_MAP, K as KnowledgeBaseAPI } from "./knowledge_base.CHSLq1jf.js";
import { E as EmbeddingConfigAPI } from "./ai_config.BnEV7HHL.js";
import { J as defineComponent, t as onMounted, S as openBlock, _ as createElementBlock, $ as createVNode, a0 as withCtx, a1 as createBaseVNode, o as unref, a9 as createTextVNode, a6 as withDirectives, T as createBlock, aa as toDisplayString, a8 as createCommentVNode, H as Fragment, ay as renderList, r as ref, ak as reactive } from "./.pnpm.BW3P1y8f.js";
import { _ as _export_sfc } from "./_plugin-vue_export-helper.1tPrXgE0.js";
import "./index.CMd5bD1r.js";
import "./codemirror.CvJAcn2d.js";
const _hoisted_1 = { class: "app-container" };
const _hoisted_2 = { class: "table-toolbar" };
const _hoisted_3 = {
  key: 0,
  class: "embedding-info"
};
const _hoisted_4 = { class: "config-name" };
const _sfc_main = /* @__PURE__ */ defineComponent({
  ...{
    name: "KnowledgeBase",
    inheritAttrs: false
  },
  __name: "index",
  setup(__props) {
    const loading = ref(false);
    const tableData = ref([]);
    const total = ref(0);
    const selection = ref([]);
    const queryParams = reactive({
      page_no: 1,
      page_size: 10,
      name: void 0,
      kb_status: void 0
    });
    const embeddingOptions = ref([]);
    const uploadRef = ref();
    const fileList = ref([]);
    async function loadData() {
      loading.value = true;
      try {
        const res = await KnowledgeBaseAPI.list(queryParams);
        tableData.value = res.data.data.items;
        total.value = res.data.data.total;
      } finally {
        loading.value = false;
      }
    }
    async function loadEmbeddingOptions() {
      try {
        const res = await EmbeddingConfigAPI.list({ page_no: 1, page_size: 100 });
        embeddingOptions.value = res.data.data.items;
      } catch (error) {
        console.error("加载向量化配置失败", error);
      }
    }
    function handleSelectionChange(rows) {
      selection.value = rows;
    }
    function getStatusType(status) {
      var _a;
      return ((_a = KB_STATUS_MAP[status]) == null ? void 0 : _a.type) || "info";
    }
    const dialogVisible = ref(false);
    const dialogTitle = ref("新增知识库");
    const formRef = ref();
    const formData = reactive({
      id: void 0,
      name: void 0,
      embedding_config_id: void 0,
      description: void 0
    });
    const formRules = {
      name: [{ required: true, message: "请输入知识库名称", trigger: "blur" }]
    };
    function handleAdd() {
      dialogTitle.value = "新增知识库";
      const defaultEmbedding = embeddingOptions.value.find((item) => item.is_default === 1);
      Object.assign(formData, {
        id: void 0,
        name: void 0,
        embedding_config_id: defaultEmbedding == null ? void 0 : defaultEmbedding.id,
        description: void 0
      });
      fileList.value = [];
      dialogVisible.value = true;
    }
    async function handleSubmit() {
      var _a;
      await ((_a = formRef.value) == null ? void 0 : _a.validate());
      const files = fileList.value.map((f) => f.raw).filter(Boolean);
      await KnowledgeBaseAPI.create(formData, files);
      ElMessage.success("创建成功");
      dialogVisible.value = false;
      loadData();
    }
    function handleExceed() {
      ElMessage.warning("最多只能上传10个文件");
    }
    async function handleDelete(row) {
      await ElMessageBox.confirm("确定删除该知识库吗？删除后相关向量数据也将被删除。", "警告", { type: "warning" });
      await KnowledgeBaseAPI.delete([row.id]);
      ElMessage.success("删除成功");
      loadData();
    }
    async function handleBatchDelete() {
      await ElMessageBox.confirm(`确定删除选中的${selection.value.length}个知识库吗？`, "警告", { type: "warning" });
      const ids = selection.value.map((item) => item.id);
      await KnowledgeBaseAPI.delete(ids);
      ElMessage.success("删除成功");
      loadData();
    }
    const detailVisible = ref(false);
    const detailData = ref({});
    async function handleDetail(row) {
      try {
        const res = await KnowledgeBaseAPI.detail(row.id);
        detailData.value = res.data.data;
        detailVisible.value = true;
      } catch (error) {
        console.error("获取详情失败", error);
      }
    }
    async function handleRetry(row) {
      await ElMessageBox.confirm("确定重新向量化该知识库吗？", "提示", { type: "info" });
      await KnowledgeBaseAPI.retry(row.id);
      ElMessage.success("已启动重新向量化");
      loadData();
    }
    onMounted(() => {
      loadData();
      loadEmbeddingOptions();
    });
    return (_ctx, _cache) => {
      const _component_el_button = ElButton;
      const _component_el_table_column = ElTableColumn;
      const _component_el_tag = ElTag;
      const _component_el_text = ElText;
      const _component_el_table = ElTable;
      const _component_el_pagination = ElPagination;
      const _component_el_card = ElCard;
      const _component_el_input = ElInput;
      const _component_el_form_item = ElFormItem;
      const _component_el_option = ElOption;
      const _component_el_select = ElSelect;
      const _component_el_upload = ElUpload;
      const _component_el_form = ElForm;
      const _component_el_dialog = ElDialog;
      const _component_el_descriptions_item = ElDescriptionsItem;
      const _component_el_descriptions = ElDescriptions;
      const _directive_loading = vLoading;
      return openBlock(), createElementBlock("div", _hoisted_1, [
        createVNode(_component_el_card, null, {
          header: withCtx(() => [..._cache[9] || (_cache[9] = [
            createBaseVNode("div", { class: "card-header" }, [
              createBaseVNode("span", null, "知识库管理")
            ], -1)
          ])]),
          default: withCtx(() => [
            createBaseVNode("div", _hoisted_2, [
              createVNode(_component_el_button, {
                icon: unref(refresh_default),
                onClick: loadData
              }, {
                default: withCtx(() => [..._cache[10] || (_cache[10] = [
                  createTextVNode("刷新", -1)
                ])]),
                _: 1
              }, 8, ["icon"]),
              createVNode(_component_el_button, {
                type: "primary",
                icon: unref(plus_default),
                onClick: handleAdd
              }, {
                default: withCtx(() => [..._cache[11] || (_cache[11] = [
                  createTextVNode("新增知识库", -1)
                ])]),
                _: 1
              }, 8, ["icon"]),
              createVNode(_component_el_button, {
                type: "danger",
                icon: unref(delete_default),
                disabled: selection.value.length === 0,
                onClick: handleBatchDelete
              }, {
                default: withCtx(() => [..._cache[12] || (_cache[12] = [
                  createTextVNode("批量删除", -1)
                ])]),
                _: 1
              }, 8, ["icon", "disabled"])
            ]),
            withDirectives((openBlock(), createBlock(_component_el_table, {
              data: tableData.value,
              onSelectionChange: handleSelectionChange
            }, {
              default: withCtx(() => [
                createVNode(_component_el_table_column, {
                  type: "selection",
                  width: "50"
                }),
                createVNode(_component_el_table_column, {
                  prop: "name",
                  label: "知识库名称",
                  "min-width": "150"
                }),
                createVNode(_component_el_table_column, {
                  label: "向量化配置",
                  "min-width": "200"
                }, {
                  default: withCtx(({ row }) => [
                    row.embedding_config ? (openBlock(), createElementBlock("div", _hoisted_3, [
                      createVNode(_component_el_tag, {
                        size: "small",
                        type: row.embedding_config.embedding_type === 0 ? "primary" : "success"
                      }, {
                        default: withCtx(() => [
                          createTextVNode(toDisplayString(row.embedding_config.embedding_type === 0 ? "本地" : "远程"), 1)
                        ]),
                        _: 2
                      }, 1032, ["type"]),
                      createBaseVNode("span", _hoisted_4, toDisplayString(row.embedding_config.name), 1)
                    ])) : (openBlock(), createBlock(_component_el_text, {
                      key: 1,
                      type: "info"
                    }, {
                      default: withCtx(() => [..._cache[13] || (_cache[13] = [
                        createTextVNode("未配置", -1)
                      ])]),
                      _: 1
                    }))
                  ]),
                  _: 1
                }),
                createVNode(_component_el_table_column, {
                  prop: "collection_name",
                  label: "集合名称",
                  "min-width": "180",
                  "show-overflow-tooltip": ""
                }),
                createVNode(_component_el_table_column, {
                  prop: "document_count",
                  label: "文档数",
                  width: "100",
                  align: "center"
                }),
                createVNode(_component_el_table_column, {
                  prop: "vector_count",
                  label: "向量数",
                  width: "100",
                  align: "center"
                }),
                createVNode(_component_el_table_column, {
                  prop: "kb_status",
                  label: "状态",
                  width: "100",
                  align: "center"
                }, {
                  default: withCtx(({ row }) => [
                    createVNode(_component_el_tag, {
                      type: getStatusType(row.kb_status)
                    }, {
                      default: withCtx(() => {
                        var _a;
                        return [
                          createTextVNode(toDisplayString(((_a = unref(KB_STATUS_MAP)[row.kb_status]) == null ? void 0 : _a.label) || "未知"), 1)
                        ];
                      }),
                      _: 2
                    }, 1032, ["type"])
                  ]),
                  _: 1
                }),
                createVNode(_component_el_table_column, {
                  prop: "description",
                  label: "备注",
                  "min-width": "150",
                  "show-overflow-tooltip": ""
                }),
                createVNode(_component_el_table_column, {
                  prop: "created_time",
                  label: "创建时间",
                  width: "170"
                }),
                createVNode(_component_el_table_column, {
                  label: "操作",
                  width: "180",
                  fixed: "right"
                }, {
                  default: withCtx(({ row }) => [
                    row.kb_status === 3 ? (openBlock(), createBlock(_component_el_button, {
                      key: 0,
                      type: "warning",
                      link: "",
                      icon: unref(refresh_default),
                      onClick: ($event) => handleRetry(row)
                    }, {
                      default: withCtx(() => [..._cache[14] || (_cache[14] = [
                        createTextVNode("重试", -1)
                      ])]),
                      _: 1
                    }, 8, ["icon", "onClick"])) : createCommentVNode("", true),
                    createVNode(_component_el_button, {
                      type: "primary",
                      link: "",
                      icon: unref(view_default),
                      onClick: ($event) => handleDetail(row)
                    }, {
                      default: withCtx(() => [..._cache[15] || (_cache[15] = [
                        createTextVNode("详情", -1)
                      ])]),
                      _: 1
                    }, 8, ["icon", "onClick"]),
                    createVNode(_component_el_button, {
                      type: "danger",
                      link: "",
                      icon: unref(delete_default),
                      onClick: ($event) => handleDelete(row)
                    }, {
                      default: withCtx(() => [..._cache[16] || (_cache[16] = [
                        createTextVNode("删除", -1)
                      ])]),
                      _: 1
                    }, 8, ["icon", "onClick"])
                  ]),
                  _: 1
                })
              ]),
              _: 1
            }, 8, ["data"])), [
              [_directive_loading, loading.value]
            ]),
            createVNode(_component_el_pagination, {
              "current-page": queryParams.page_no,
              "onUpdate:currentPage": _cache[0] || (_cache[0] = ($event) => queryParams.page_no = $event),
              "page-size": queryParams.page_size,
              "onUpdate:pageSize": _cache[1] || (_cache[1] = ($event) => queryParams.page_size = $event),
              total: total.value,
              "page-sizes": [10, 20, 50, 100],
              layout: "total, sizes, prev, pager, next, jumper",
              onSizeChange: loadData,
              onCurrentChange: loadData
            }, null, 8, ["current-page", "page-size", "total"])
          ]),
          _: 1
        }),
        createVNode(_component_el_dialog, {
          modelValue: dialogVisible.value,
          "onUpdate:modelValue": _cache[7] || (_cache[7] = ($event) => dialogVisible.value = $event),
          title: dialogTitle.value,
          width: "600px",
          "destroy-on-close": ""
        }, {
          footer: withCtx(() => [
            createVNode(_component_el_button, {
              onClick: _cache[6] || (_cache[6] = ($event) => dialogVisible.value = false)
            }, {
              default: withCtx(() => [..._cache[19] || (_cache[19] = [
                createTextVNode("取消", -1)
              ])]),
              _: 1
            }),
            createVNode(_component_el_button, {
              type: "primary",
              onClick: handleSubmit
            }, {
              default: withCtx(() => [..._cache[20] || (_cache[20] = [
                createTextVNode("确定", -1)
              ])]),
              _: 1
            })
          ]),
          default: withCtx(() => [
            createVNode(_component_el_form, {
              ref_key: "formRef",
              ref: formRef,
              model: formData,
              rules: formRules,
              "label-width": "100px"
            }, {
              default: withCtx(() => [
                createVNode(_component_el_form_item, {
                  label: "知识库名称",
                  prop: "name"
                }, {
                  default: withCtx(() => [
                    createVNode(_component_el_input, {
                      modelValue: formData.name,
                      "onUpdate:modelValue": _cache[2] || (_cache[2] = ($event) => formData.name = $event),
                      placeholder: "请输入知识库名称"
                    }, null, 8, ["modelValue"])
                  ]),
                  _: 1
                }),
                createVNode(_component_el_form_item, {
                  label: "向量化配置",
                  prop: "embedding_config_id"
                }, {
                  default: withCtx(() => [
                    createVNode(_component_el_select, {
                      modelValue: formData.embedding_config_id,
                      "onUpdate:modelValue": _cache[3] || (_cache[3] = ($event) => formData.embedding_config_id = $event),
                      placeholder: "请选择向量化配置",
                      style: { "width": "100%" },
                      clearable: ""
                    }, {
                      default: withCtx(() => [
                        (openBlock(true), createElementBlock(Fragment, null, renderList(embeddingOptions.value, (item) => {
                          return openBlock(), createBlock(_component_el_option, {
                            key: item.id,
                            label: `${item.name} (${item.embedding_type === 0 ? "本地" : "远程"} - ${item.model_name})`,
                            value: item.id
                          }, null, 8, ["label", "value"]);
                        }), 128))
                      ]),
                      _: 1
                    }, 8, ["modelValue"])
                  ]),
                  _: 1
                }),
                !formData.id ? (openBlock(), createBlock(_component_el_form_item, {
                  key: 0,
                  label: "上传文件",
                  prop: "files"
                }, {
                  default: withCtx(() => [
                    createVNode(_component_el_upload, {
                      ref_key: "uploadRef",
                      ref: uploadRef,
                      "file-list": fileList.value,
                      "onUpdate:fileList": _cache[4] || (_cache[4] = ($event) => fileList.value = $event),
                      "auto-upload": false,
                      limit: 10,
                      multiple: "",
                      accept: ".txt,.pdf,.doc,.docx,.md",
                      "on-exceed": handleExceed
                    }, {
                      tip: withCtx(() => [..._cache[18] || (_cache[18] = [
                        createBaseVNode("div", { class: "el-upload__tip" }, "支持 txt、pdf、doc、docx、md 格式，最多上传10个文件", -1)
                      ])]),
                      default: withCtx(() => [
                        createVNode(_component_el_button, {
                          type: "primary",
                          icon: unref(upload_default)
                        }, {
                          default: withCtx(() => [..._cache[17] || (_cache[17] = [
                            createTextVNode("选择文件", -1)
                          ])]),
                          _: 1
                        }, 8, ["icon"])
                      ]),
                      _: 1
                    }, 8, ["file-list"])
                  ]),
                  _: 1
                })) : createCommentVNode("", true),
                createVNode(_component_el_form_item, {
                  label: "备注",
                  prop: "description"
                }, {
                  default: withCtx(() => [
                    createVNode(_component_el_input, {
                      modelValue: formData.description,
                      "onUpdate:modelValue": _cache[5] || (_cache[5] = ($event) => formData.description = $event),
                      type: "textarea",
                      rows: 3,
                      placeholder: "请输入备注"
                    }, null, 8, ["modelValue"])
                  ]),
                  _: 1
                })
              ]),
              _: 1
            }, 8, ["model"])
          ]),
          _: 1
        }, 8, ["modelValue", "title"]),
        createVNode(_component_el_dialog, {
          modelValue: detailVisible.value,
          "onUpdate:modelValue": _cache[8] || (_cache[8] = ($event) => detailVisible.value = $event),
          title: "知识库详情",
          width: "700px"
        }, {
          default: withCtx(() => [
            createVNode(_component_el_descriptions, {
              column: 2,
              border: ""
            }, {
              default: withCtx(() => [
                createVNode(_component_el_descriptions_item, { label: "知识库名称" }, {
                  default: withCtx(() => [
                    createTextVNode(toDisplayString(detailData.value.name), 1)
                  ]),
                  _: 1
                }),
                createVNode(_component_el_descriptions_item, { label: "状态" }, {
                  default: withCtx(() => [
                    createVNode(_component_el_tag, {
                      type: getStatusType(detailData.value.kb_status)
                    }, {
                      default: withCtx(() => {
                        var _a;
                        return [
                          createTextVNode(toDisplayString(((_a = unref(KB_STATUS_MAP)[detailData.value.kb_status]) == null ? void 0 : _a.label) || "未知"), 1)
                        ];
                      }),
                      _: 1
                    }, 8, ["type"])
                  ]),
                  _: 1
                }),
                createVNode(_component_el_descriptions_item, {
                  label: "向量化配置",
                  span: 2
                }, {
                  default: withCtx(() => [
                    detailData.value.embedding_config ? (openBlock(), createElementBlock(Fragment, { key: 0 }, [
                      createVNode(_component_el_tag, {
                        size: "small",
                        type: detailData.value.embedding_config.embedding_type === 0 ? "primary" : "success"
                      }, {
                        default: withCtx(() => [
                          createTextVNode(toDisplayString(detailData.value.embedding_config.embedding_type === 0 ? "本地" : "远程"), 1)
                        ]),
                        _: 1
                      }, 8, ["type"]),
                      createTextVNode(" " + toDisplayString(detailData.value.embedding_config.name) + " (" + toDisplayString(detailData.value.embedding_config.model_name) + ") ", 1)
                    ], 64)) : (openBlock(), createElementBlock(Fragment, { key: 1 }, [
                      createTextVNode("未配置")
                    ], 64))
                  ]),
                  _: 1
                }),
                createVNode(_component_el_descriptions_item, {
                  label: "ChromaDB集合名称",
                  span: 2
                }, {
                  default: withCtx(() => [
                    createTextVNode(toDisplayString(detailData.value.collection_name), 1)
                  ]),
                  _: 1
                }),
                createVNode(_component_el_descriptions_item, { label: "文档数量" }, {
                  default: withCtx(() => [
                    createTextVNode(toDisplayString(detailData.value.document_count), 1)
                  ]),
                  _: 1
                }),
                createVNode(_component_el_descriptions_item, { label: "向量数量" }, {
                  default: withCtx(() => [
                    createTextVNode(toDisplayString(detailData.value.vector_count), 1)
                  ]),
                  _: 1
                }),
                createVNode(_component_el_descriptions_item, {
                  label: "备注",
                  span: 2
                }, {
                  default: withCtx(() => [
                    createTextVNode(toDisplayString(detailData.value.description || "-"), 1)
                  ]),
                  _: 1
                }),
                createVNode(_component_el_descriptions_item, { label: "创建时间" }, {
                  default: withCtx(() => [
                    createTextVNode(toDisplayString(detailData.value.created_time), 1)
                  ]),
                  _: 1
                }),
                createVNode(_component_el_descriptions_item, { label: "更新时间" }, {
                  default: withCtx(() => [
                    createTextVNode(toDisplayString(detailData.value.updated_time), 1)
                  ]),
                  _: 1
                }),
                detailData.value.error_message ? (openBlock(), createBlock(_component_el_descriptions_item, {
                  key: 0,
                  label: "错误信息",
                  span: 2
                }, {
                  default: withCtx(() => [
                    createVNode(_component_el_text, { type: "danger" }, {
                      default: withCtx(() => [
                        createTextVNode(toDisplayString(detailData.value.error_message), 1)
                      ]),
                      _: 1
                    })
                  ]),
                  _: 1
                })) : createCommentVNode("", true)
              ]),
              _: 1
            })
          ]),
          _: 1
        }, 8, ["modelValue"])
      ]);
    };
  }
});
const index = /* @__PURE__ */ _export_sfc(_sfc_main, [["__scopeId", "data-v-c71b4110"]]);
export {
  index as default
};