diff --git a/src/api/dataSet.ts b/src/api/dataSet.ts
new file mode 100644
index 0000000000000000000000000000000000000000..a780114f887a61f72924bdb2506216492025333c
--- /dev/null
+++ b/src/api/dataSet.ts
@@ -0,0 +1,109 @@
+import request from '@/utils/request';
+
+class dataSetAPI {
+  // 获取数据集管理数据
+  static queryDataSetList(data: any) {
+    return request({
+      url: `/dataset/list`,
+      method: 'post',
+      data: data,
+    });
+  }
+
+  /** 创建数据集管理 */
+  static createDataSet(data: any) {
+    return request({
+      url: `/dataset`,
+      method: 'post',
+      data: data,
+    });
+  }
+
+  /** 删除数据集管理*/
+  static delDataSet(dataSetId: string) {
+    return request({
+      url: `/dataSet?databaseId=${dataSetId}`,
+      method: 'delete',
+    });
+  }
+
+  /**更新数据集管理 */
+  static updateDataSet(data: any) {
+    return request({
+      url: `/dataset`,
+      method: 'put',
+      data: data,
+    });
+  }
+
+  /**获取单个数据集的详情数据 */
+  static querySingleDataSetInfo(data: any) {
+    return request({
+      url: `/dataset/data`,
+      method: 'post',
+      data: data,
+    });
+  }
+
+  /**获取数据集任务状态 */
+  static queryDataSetTask() {
+    return request({
+      url: `/dataset/task`,
+      method: 'get',
+    });
+  }
+
+  /** 下载数据集 */
+  static downloadDataSet(taskId: string) {
+    return request({
+      url: `/dataset/download`,
+      method: 'get',
+      data: { taskId },
+    });
+  }
+
+  /**生成数据集 */
+  static generateDataSet(data: any) {
+    return request({
+      url: `/dataset/generate`,
+      method: 'post',
+      data,
+    });
+  }
+
+  /**导入数据集 */
+  static importDataSet(payload: { data: any; params: any }, options: any) {
+    return request({
+      url: `/dataset/import?kbId=${payload.params.kbId}`,
+      method: 'post',
+      data: payload.data,
+      headers: {
+        'Content-Type': 'multipart/form-data',
+      },
+      onUploadProgress(e) {
+        const rate = Math.floor((e.loaded / (e.total as number)) * 100);
+        if (rate < 100) {
+          options.onProgress(rate);
+        }
+      },
+    });
+  }
+
+  /** 导出数据集*/
+  static exportDataSet(datasetIds: string[]) {
+    return request({
+      url: `/dataset/export`,
+      method: 'post',
+      data: datasetIds,
+    });
+  }
+
+  static queryLlmData() {
+    return request({
+      url: `/other/llm`,
+      method: 'get',
+    });
+  }
+}
+
+export default dataSetAPI;
diff --git a/src/enums/KnowledgeEnum.ts b/src/enums/KnowledgeEnum.ts
index ec74a57cb1e1a9e6f245e225287fedf70d14e72b..b310b72942eb5ab53d8e6467b0c83bf6fd5cf4d4 100644
--- a/src/enums/KnowledgeEnum.ts
+++ b/src/enums/KnowledgeEnum.ts
@@ -10,11 +10,12 @@ export const enum StatusEnum {
 }
 
 export const enum DataSetStatusEnum {
-  FAIL = 'failed',
-  CANCEL = 'canceled',
-  GENERATE_ING = 'pending',
-  SUCCESS = 'success',
-  RUNNING = 'running',
+ PENDING = 'pending',
+RUNNING = 'running',
+SUCCESS ='success',
+FAILED = 'failed',
+CANCELED = 'canceled',
+ DELETED ='deleted'
 }
 
 export const enum MenuType {
diff --git a/src/views/dataSet/index.vue b/src/views/dataSet/index.vue
index ed5570b6d18379bad44cc06120b7a04e71604e59..145bb50ab34dd8eb1e27ed54723ac93766f98e4a 100644
--- a/src/views/dataSet/index.vue
+++ b/src/views/dataSet/index.vue
@@ -12,13 +12,20 @@
     class="dataSet-container">
     <div>
       <div>
+        <el-button
+            type="primary"
+            style="margin-right: 8px"
+            @click="handleImportDataSet"
+            class="importFileBtn">
+            {{ $t('导入数据集') }}
+          </el-button>
         <el-dropdown
           placement="bottom"
           popper-class="dropdown-container dataSet-ops-dowlon"
           @visible-change="handleBatchDownBth"
-          :disabled="!(selectionDataSetList.length > 0)">
+          :disabled="!selectionDataSetList.length">
           <el-button
-            :class="batchDownBth ? 'upBtn' : 'downBtn'">
+            :class="batchDownBth ? 'upBtn' : 'downBtn'" :disabled="!selectionDataSetList.length">
             {{ $t('btnText.batchOps') }}
             <el-icon
               class="el-icon--right"
@@ -34,12 +41,12 @@
           <template #dropdown>
             <el-dropdown-menu>
               <el-dropdown-item
-                :disabled="!(selectionDataSetList.length > 0)"
+                :disabled="!selectionDataSetList.length"
                 @click="handleSelectExportDataSet">
                 {{ $t('btnText.batchExport') }}
               </el-dropdown-item>
               <el-dropdown-item
-                :disabled="!(selectionDataSetList.length > 0)"
+                :disabled="!selectionDataSetList.length"
                 @click="handleSelectDeleteDataSet">
                 {{ $t('btnText.batchDelete') }}
               </el-dropdown-item>
@@ -81,19 +88,19 @@
             </template>
           </el-table-column>
           <el-table-column
-            prop="dataSetDesc"
+            prop="description"
             :label="$t('数据集简介')"
             show-overflow-tooltip>
             <template #default="scope">
-              <span>{{ scope.row.dataSetDesc }}</span>
+              <span>{{ scope.row.description }}</span>
             </template>
           </el-table-column>
           <el-table-column
-            prop="dataSetNumber"
+            prop="dataCnt"
             :label="$t('数据集条目')"
             sortable />
           <el-table-column
-            prop="isDataClean"
+            prop="isDataCleared"
             :label="$t('是否进行数据清洗')">
             <template #header>
               <div class="custom-header">
@@ -102,7 +109,7 @@
                   ref="enableRef"
                   @click.stop
                   :class="
-                    searchPayload?.enabled?.length > 0 || enableFilterVisible
+                    searchPayload?.enabled?.length || enableFilterVisible
                       ? 'searchIconIsActive'
                       : ''
                   ">
@@ -132,9 +139,8 @@
                 style="--el-switch-on-color: #24ab36; --el-switch-off-color: #c3cedf" />
             </template>
           </el-table-column>
-
           <el-table-column
-            prop="dataSetStatus"
+            prop="generateStatus"
             :label="$t('状态')"
             width="220">
             <template #header>
@@ -169,29 +175,29 @@
             </template>
             <template #default="scope">
               <div
-                v-if="scope.row.dataSetStatus === DataSetStatusEnum.FAIL"
+                v-if="scope.row.generateTask.generateStatus === DataSetStatusEnum.FAILED"
                 class="statusFail">
                 {{ $t('生成失败') }}
               </div>
               <div
-                v-if="scope.row.dataSetStatus === DataSetStatusEnum.SUCCESS"
+                v-if="scope.row.generateTask.generateStatus === DataSetStatusEnum.SUCCESS"
                 class="statusSuccess">
                 {{ $t('生成完毕') }}
               </div>
               <div
-                v-if="scope.row.dataSetStatus === DataSetStatusEnum.CANCEL"
+                v-if="scope.row.generateTask.generateStatus === DataSetStatusEnum.CANCELED"
                 class="statusCancel">
                 {{ $t('取消生成') }}
               </div>
               <div
-                v-if="scope.row.dataSetStatus === DataSetStatusEnum.GENERATE_ING"
+                v-if="scope.row.generateTask.generateStatus === DataSetStatusEnum.PENDING"
                 class="statusWaitIng">
                 <div class="icon-box icon-loading"></div>
                 {{ $t('等待生成') }}
               </div>
               <div
                 class="statusGenerate"
-                v-if="scope.row.dataSetStatus === DataSetStatusEnum.RUNNING">
+                v-if="scope.row.generateTask.generateStatus === DataSetStatusEnum.RUNNING">
                 <div class="percent-box">
                   <el-progress
                     :percentage="
@@ -215,7 +221,7 @@
             </template>
           </el-table-column>
           <el-table-column
-            prop="dataSetScore"
+            prop="score"
             :label="`${$t('数据集质量分数')}(1~100)`"
             width="240"
             show-overflow-tooltip
@@ -232,7 +238,7 @@
                 <el-icon
                   ref="creatorRef"
                   :class="
-                    searchPayload?.creator?.length > 0 || creatorVisible ? 'searchIconIsActive' : ''
+                    searchPayload?.creator?.length || creatorVisible ? 'searchIconIsActive' : ''
                   ">
                   <IconFilter />
                 </el-icon>
@@ -259,7 +265,11 @@
             class-name="upload-time-cell"
             :label="$t('完成时间')"
             sortable
-            @click.stop></el-table-column>
+            @click.stop>
+            <template #default="scope">
+               {{ scope.row.generateTask.createdTime }}
+            </template>
+          </el-table-column>
 
           <el-table-column
             prop="action"
@@ -268,7 +278,7 @@
             fixed="right">
             <template #default="scope">
               <el-button
-                v-if="scope.row.dataSetStatus === DataSetStatusEnum.RUNNING"
+                v-if="scope.row.generateTask.generateStatus === DataSetStatusEnum.RUNNING"
                 text
                 @click="handlePauseDataSet(scope.row, 'cancel')">
                 {{ $t('暂停') }}
@@ -276,10 +286,9 @@
               <el-button
                 v-if="
                   [
-                    DataSetStatusEnum.FAIL,
-                    DataSetStatusEnum.CANCEL,
-                    DataSetStatusEnum.ANALYSIS_ING,
-                  ].includes(scope.row.dataSetStatus)
+                    DataSetStatusEnum.FAILED,
+                    DataSetStatusEnum.CANCELED,
+                  ].includes(scope.row.generateTask.generateStatus)
                 "
                 text
                 @click="handleRunDataSet(scope.row, 'run')">
@@ -287,13 +296,13 @@
               </el-button>
               <el-button
                 text
-                :disabled="scope.row.dataSetStatus === DataSetStatusEnum.RUNNING"
+                :disabled="scope.row.generateTask.generateStatus === DataSetStatusEnum.RUNNING"
                 @click="handleEditDataSet(scope.row)">
                 {{ $t('btnText.edit') }}
               </el-button>
               <el-button
                 text
-                :disabled="scope.row.dataSetStatus === DataSetStatusEnum.SUCCESS"
+                :disabled="scope.row.generateTask.generateStatus === DataSetStatusEnum.SUCCESS"
                 @click="handleDataSetEval(scope.row)">
                 {{ $t('评测') }}
               </el-button>
@@ -303,7 +312,7 @@
                 {{ $t('导出') }}
               </el-button>
               <el-button
-                :disabled="scope.row.dataSetStatus === DataSetStatusEnum.RUNNING"
+                :disabled="scope.row.generateTask.generateStatus === DataSetStatusEnum.RUNNING"
                 text
                 @click="handleDeleteDataSet(scope.row)">
                 {{ $t('btnText.delete') }}
@@ -312,7 +321,7 @@
           </el-table-column>
         </el-table>
         <el-pagination
-          v-if="fileTableList.data?.length > 0"
+          v-if="fileTableList.data?.length"
           v-model:current-page="currentPage"
           v-model:page-size="currentPageSize"
           :page-sizes="pagination.pageSizes"
@@ -324,6 +333,11 @@
     </div>
   </div>
   <CreateEvaluate :dialogEvaluateVisible="dialogEvaluateVisible" :rowData="rowData" :close="handleCloseDialogue" />
+  <DataSetDrawer
+    v-if="dataSetDrawerVisible"
+    :dataSetDrawerVisible="dataSetDrawerVisible"
+    :dataSetRow="dataSetRow"
+    :handleDataSetProps="handleDataSetProps" />
 </template>
 
 <script setup lang="ts">
@@ -332,17 +346,22 @@ import { useGroupStore } from '@/store/modules/group';
 import { IconCaretDown, IconFilter, IconCaretUp } from '@computing/opendesign-icons';
 import { DataSetStatusEnum } from '@/enums/KnowledgeEnum';
 import FilterContainr from '@/components/TableFilter/index.vue';
+import DataSetDrawer from './dataSetDrawer.vue';
 import CreateEvaluate from '@/views/dataSet/craeteEvaluate.vue';
+import dataSetAPI from '@/api/dataSet';
+import router from '@/router';
+const route = useRoute();
+const { t } = useI18n();
 
 import { ref } from 'vue';
 import '@/styles/dataSet.scss';
 const store = useGroupStore();
-const selectionDataSetList = ref([]);
+const selectionDataSetList = ref<any>([]);
 const batchDownBth = ref(false);
 const dataSetTableList = ref([
-     {dataSetName:'数据集06',dataSetDesc:'数据集06描述',dataSetNumber:10,
-     dataSetClean:true,dataSetStatus:DataSetStatusEnum.FAIL,dataSetCreator:'admin',
-     dataSetCreatime:'2022-03-22 12:22:22',dataSetScore:99},
+     {dataSetName:'数据集06',description:'数据集06描述',dataCnt:10,
+     isDataCleared:true,generateTask:{generateStatus:DataSetStatusEnum.FAILED,createdTime:'2022-03-22 12:22:22'},dataSetCreator:'admin',
+     score:99},
 ]);
 const loading = ref(false);
 const checkedFilterList = ref([]);
@@ -357,8 +376,13 @@ const creatorRef = ref();
 const statusFilterVisible = ref(false);
 const creatorVisible = ref(false);
 const enableFilterVisible = ref(false);
+const dataSetDrawerVisible = ref(false);
+const dataSetRow = ref({});
 const dialogEvaluateVisible = ref(false);
 const rowData = ref({});
+const checkTableSelecData = ref([]);
+const pollingKfTimer = ref();
+const sortFilter = ref({});
 
 const handleCloseDialogue = () => {
   dialogEvaluateVisible.value = false;
@@ -376,22 +400,165 @@ const pagination = ref({
   layout: 'total,sizes,prev,pager,next,jumper',
 });
 const searchPayload = ref<any>({
-  name: '',
-  document_type_list: [],
-  chunk_size_order: '',
-  created_time_order: '',
-  status: [],
-  created_time_start: '',
-  created_time_end: '',
-  enabled: '',
-  parser_method: [],
+  isDataCleared: [],
 });
 const { handleKnowledgeTab } = store;
 
+const handleSearchPayload = () => {
+  const searchParams = Object.keys(searchPayload.value || {}).reduce((pre: any, item) => {
+    if (searchPayload.value?.[item]?.length && searchPayload.value?.[item] !== 'all') {
+      pre[item] = searchPayload.value[item];
+      if (item === 'enabled') {
+        pre[item] = JSON.parse(searchPayload.value[item]);
+      }
+    }
+
+    return pre;
+  }, {});
+  return searchParams || {};
+};
+
+const handleSortChange = (data: { column: any; prop: string; order: any }) => {
+  let sortKey = data.prop === 'score' ? 'scoreOrder' : null;
+  let sortValue = data.order ? (data.order === 'ascending' ? 'asc' : 'desc') : null;
+  sortFilter.value = sortValue
+    ? {
+        [sortKey || data.prop]: sortValue,
+      }
+    : {};
+  handleSearchOpsData(true, true);
+};
+
+const handlePollFileDataSet = () => {
+  dataSetAPI.queryDataSetList({
+    page_number: currentPage.value,
+    page_size: currentPageSize.value,
+    kbId: route.query.kb_id as string,
+    ...handleSearchPayload(),
+    ...sortFilter.value,
+  })
+    .then((res: any) => {
+      if (res.page_number === currentPage.value && fileTableList.data?.length) {
+        if (!res?.data_list?.length && currentPage.value !== 1) {
+          currentPage.value = 1;
+          handleSearchOpsData(true, true);
+          return;
+        }
+        fileTableList.data = fileTableList.data.map((item) => {
+          let fileData = res?.data_list?.filter((file: any) => file.id === item.id)?.[0];
+          return fileData || item;
+        });
+      }
+      if (res.data_list?.length) {
+        handCheckTableData(res.data_list);
+      }
+    })
+    .finally(() => {
+      loading.value = false;
+    });
+};
+
+const handleStartPollTimer = () => {
+  pollingKfTimer.value = setInterval(() => handlePollFileDataSet(), 15000);
+};
+
+const handCheckTableData = (tableList: { filter: (arg0: (checkItem: any) => any) => never[]; find: (arg0: (notCheckItem: any) => boolean) => any; }) => {
+  checkTableSelecData.value = tableList.filter((checkItem) => {
+    const selecData = tableList.find((notCheckItem) => notCheckItem?.id === checkItem?.id);
+    return selecData && ['pending', 'running'].includes(selecData.task.status);
+  });
+};
+
+const handleSearchOpsData = (loadingStatus: boolean, startPollTimer: boolean) => {
+  handleQueryDataSetList(
+    {
+      page_number: currentPage.value,
+      page_size: currentPageSize.value,
+      kbId: route.query.kb_id as string,
+      ...handleSearchPayload(),
+      ...sortFilter.value,
+    },
+    loadingStatus,
+    startPollTimer
+  );
+};
+
+const handleQueryDataSetList = (
+  payload: any,
+  loadingStatus: boolean,
+  pollTimer: boolean
+)=>{
+  if (pollTimer) {
+    handleCleartTimer();
+  }
+  loading.value = loadingStatus;
+  dataSetAPI.queryDataSetList(payload)
+    .then((res: any) => {
+      if (res.data_list?.length) {
+        handCheckTableData(res.data_list);
+      }
+      if (!res?.data_list?.length && currentPage.value !== 1) {
+        currentPage.value = 1;
+        handleSearchOpsData(true, true);
+        return;
+      }
+      fileTableList.data = res?.data_list || [];
+      currentPage.value = res.page_number;
+      currentPageSize.value = res.page_size;
+      totalCount.value = res.total;
+      if (pollTimer) {
+        handleStartPollTimer();
+      }
+    })
+    .finally(() => {
+      loading.value = false;
+    });
+}
+
+const handleCleartTimer = () => {
+  clearInterval(pollingKfTimer.value);
+  pollingKfTimer.value = null;
+};
+
+const handleSearchData = () => {
+  handleQueryDataSetList(
+    {
+      page_number: 1,
+      page_size: currentPageSize.value,
+      kbId: route.query.kb_id as string,
+      ...handleSearchPayload(),
+      ...sortFilter.value,
+    },
+    true,
+    true
+  );
+  enableFilterVisible.value = false;
+  statusFilterVisible.value = false;
+  creatorVisible.value = false
+};
+
+onMounted(() => {
+  const kbId = route.query.kb_id;
+  if (kbId?.length) {
+    handleQueryDataSetList(
+      {
+        kbId: kbId as string,
+        page_number: 1,
+        page_size: 20,
+      },
+      true,
+      true
+    );
+  }
+});
+
 const handleToCreate = () => {
   handleKnowledgeTab('document');
 };
-const handelEnableFilterProper = (filterList: any) => {};
+const handelEnableFilterProper = (filterList: any) => {
+  searchPayload.value.isDataCleared = filterList.length === 2 ? 'all' : filterList[0];
+  handleSearchData();
+};
 
 const handleBatchDownBth = () => {};
 
@@ -399,22 +566,28 @@ const handleSelectExportDataSet = () => {};
 
 const handleSelectDeleteDataSet = () => {};
 
-const handleSelectionChange = (newSelection: any[]) => {};
-
-const handleSortChange = (data: { column: any; prop: string; order: any }) => {};
+const handleSelectionChange = (newSelection: any[]) => {
+  selectionDataSetList.value = newSelection;
+};
 
 const checkSelecTable = (row: any) => {
   return true;
 };
 const handleJumpFileSection = (row: any) => {};
 
-const handelStatusFilterProper = (filterList: any) => {};
+const handelStatusFilterProper = (filterList: any) => {
+  searchPayload.value.generateStatus = filterList;
+  handleSearchData();
+};
 
 const handleDeleteDataSet = (row: any) => {};
 
 const handleExportDataSet = async (downloadData: any) => {};
 
-const handleEditDataSet = (row: any) => {};
+const handleEditDataSet = (row: any) => {
+  dataSetRow.value = row;
+  dataSetDrawerVisible.value = true;
+};
 
 const handleRunDataSet = (row: any, type: string) => {
   dialogEvaluateVisible.value = true;
@@ -433,4 +606,47 @@ const handleDataSetEval = (row: any) =>{
 
 const handlePauseDataSet = (row: any, p0: string) =>{}
 
+const handleImportDataSet = () =>{
+
+}
+
+watch(
+  () => t(''),
+  () => {
+    filterStatusList.value = [
+      {
+        label: t('生成失败'),
+        value: DataSetStatusEnum.FAILED,
+      },
+      {
+        label: t('生成完毕'),
+        value: DataSetStatusEnum.SUCCESS,
+      },
+      {
+        label: t('取消生成'),
+        value: DataSetStatusEnum.CANCELED,
+      },
+      {
+        label: t('等待生成'),
+        value: DataSetStatusEnum.PENDING,
+      },
+      {
+        label: t('生成中'),
+        value: DataSetStatusEnum.RUNNING,
+      },
+    ];
+    filterEnableList.value = [
+      { label: t('是'), value: 'true' },
+      { label: t('否'), value: 'false' },
+    ];
+  },
+  {
+    deep: true,
+    immediate: true,
+  }
+);
+
+const handleDataSetProps = () => {
+  dataSetDrawerVisible.value = false;
+};
 </script>
diff --git a/src/views/knowledgeFile/dataSetDialog.vue b/src/views/knowledgeFile/dataSetDialog.vue
index 2418a2ecef5675c2cd301b6fb48760144b1da54a..1e44c965ae2861417e88ef9772c7e6628b4aa03c 100644
--- a/src/views/knowledgeFile/dataSetDialog.vue
+++ b/src/views/knowledgeFile/dataSetDialog.vue
@@ -2,6 +2,7 @@
   <el-dialog
     v-if="props.generateDialogVisible"
     v-model="props.generateDialogVisible"
+    @close="handleGenerateDialogVisible"
     class="dataSet-edit-dialog"
     align-center
     width="550"
@@ -16,68 +17,72 @@
       label-position="left">
       <el-form-item
         :label="$t('数据集名称')"
-        prop="dataSetName"
+        prop="name"
         class="dataSetName">
-        <el-input :placeholder="$t('assetLibrary.message.pleasePlace')" />
+        <el-input
+          maxlength="150"
+          minlength="1"
+          v-model="ruleForm.name"
+          :placeholder="$t('assetLibrary.message.pleasePlace')" />
       </el-form-item>
       <el-form-item
         :label="$t('数据集简介')"
-        prop="dataSetDesc">
+        prop="description">
         <el-input
           :rows="4"
           show-word-limit
           type="textarea"
+          v-model="ruleForm.description"
           maxlength="200"
-          :placeholder="$t('assetLibrary.message.pleasePlace')"
-           />
+          :placeholder="$t('assetLibrary.message.pleasePlace')" />
       </el-form-item>
       <el-form-item
         :label="$t('文档数量')"
-        prop="fileNumber">
-        <span>3</span>
+        prop="documentIds">
+        <span>{{ ruleForm.documentIds.length }}</span>
       </el-form-item>
 
       <el-form-item
         :label="$t('数据集条目')"
-        prop="dataSetNumber"
+        prop="dataCnt"
         class="dataSetNumber">
         <el-input-number
           class="config-size"
-          v-model="ruleForm.dataSetNumber"
+          v-model="ruleForm.dataCnt"
           :min="1"
           :max="512" />
       </el-form-item>
       <el-form-item
         :label="$t('模型类型')"
-        prop="modelType">
+        prop="llmId">
         <el-select
-          v-model="ruleForm.modelType"
+          v-model="ruleForm.llmId"
           :placeholder="$t('assetLibrary.message.pleaseChoose')"
-          :class="ruleForm.modelType.length ? 'selectedType' : 'notSelectedType'"
+          :class="ruleForm.llmId.length ? 'selectedType' : 'notSelectedType'"
           :suffix-icon="IconCaretDown"
           :teleported="false">
           <el-option
-            v-for="item in []"
-            :key="item?.value"
-            :label="item?.label"
-            :value="item?.value" />
+            v-for="item in llmList"
+            :key="item?.llmId"
+            :label="item?.llmName"
+            :value="item?.llmId" />
         </el-select>
       </el-form-item>
       <el-form-item
         :label="$t('是否进行数据清洗')"
-        prop="isDataClean"
+        prop="isDataCleared"
         class="isDataClean">
         <el-switch
-          v-model="ruleForm.isDataClean"
+          v-model="ruleForm.isDataCleared"
           class="ml-2"
           style="--el-switch-on-color: #13ce66" />
       </el-form-item>
       <el-form-item
         :label="$t('是否补全上下文')"
-        prop="isComContext"
+        prop="isChunkRelated"
         class="isComContext">
         <el-switch
-        v-model="ruleForm.isComContext"
+          v-model="ruleForm.isChunkRelated"
           class="ml-2"
           style="--el-switch-on-color: #13ce66" />
       </el-form-item>
@@ -99,59 +104,124 @@
   </el-dialog>
 </template>
 <script setup>
-import {
-  IconCaretDown,
-} from '@computing/opendesign-icons';
+import { IconCaretDown } from '@computing/opendesign-icons';
 import '@/styles/dataSetDialog.scss';
+import dataSetAPI from '@/api/dataSet'
 import { ref } from 'vue';
-const { t, } = useI18n();
+const { t } = useI18n();
 const ruleForm = ref({
-  dataSetName:'',
-  dataSetDesc:'',
-  fileNumber:1,
-  dataSetNumber:1,
-  modelType:'',
-  isDataClean:true,
-  isComContext:false,
+  name: '',
+  description: '',
+  documentIds: [],
+  dataCnt: 1,
+  llmId: '',
+  isDataCleared: false,
+  isChunkRelated: false,
 });
 const ruleFormRef = ref();
 const isSubmitDisabled = ref(true);
-const rules = reactive({
-  dataSetName: [
+const llmList = ref([]);
+const rules = ref({
+  name: [
     {
       required: true,
       message: t('model.pleasePlace'),
       trigger: ['blur', 'change'],
     },
   ],
-  dataSetDesc:[
+  description: [
     {
       required: true,
       message: t('model.pleasePlace'),
       trigger: ['blur', 'change'],
     },
   ],
-  dataSetNumber:[
+  dataCnt: [
     {
       required: true,
       message: t('model.pleasePlace'),
       trigger: ['blur', 'change'],
     },
   ],
-  modelType:[
+  llmId: [
     {
       required: true,
       message: t('assetLibrary.message.pleaseChoose'),
       trigger: ['blur', 'change'],
     },
   ],
-})
+});
 const props = defineProps({
   generateDialogVisible: {
     type: Boolean,
     default: false,
   },
+  handleGenerateDataSet: {
+    type: Function,
+    default: () => {},
+  },
+  selectionFileData: {
+    type: Array,
+    default: [],
+  },
 });
 
-const submitForm = () => {};
+watch(
+  () => props.selectionFileData,
+  () => {
+    ruleForm.documentIds = props.selectionFileData.map((item) => item.id);
+  },
+  {
+    deep: true,
+    immediate: true,
+  }
+);
+
+watch(
+  () => ruleForm.value,
+  () => {
+    isSubmitDisabled.value = !Object.keys(rules.value).every((item) => {
+      return ruleForm.value[item].toString().length;
+    });
+  },
+  {
+    deep: true,
+    immediate: true,
+  }
+);
+
+const handleGenerateDialogVisible = () => {
+  props.handleGenerateDataSet(false);
+};
+
+const handleResetDataSet = () => {
+  ruleForm.value = {
+    name: '',
+    description: '',
+    documentIds: [],
+    dataCnt: 1,
+    llmId: '',
+    isDataCleared: false,
+    isChunkRelated: false,
+  };
+};
+
+const submitForm = ()=>{
+  dataSetAPI.createDataSet(ruleForm.value).then(res=>{
+    handleCancelVisible()
+  })
+}
+
+const handleCancelVisible = () => {
+  props.handleGenerateDataSet(false);
+  handleResetDataSet();
+};
+
+onMounted(()=>{
+  dataSetAPI.queryLlmData().then(res=>{
+    llmList.value = res.llms ||[];
+  })
+})
+
+
 </script>
diff --git a/src/views/knowledgeFile/index.vue b/src/views/knowledgeFile/index.vue
index 85bda8e3a3c2a529334905e794366c48b35e38d3..f6ecb394b87569da9a1fde7f60c09c0949650b96 100644
--- a/src/views/knowledgeFile/index.vue
+++ b/src/views/knowledgeFile/index.vue
@@ -654,7 +654,7 @@
       </el-form-item>
     </el-form>
   </el-dialog>
-  <DataSetDialog :generateDialogVisible="generateDialogVisible" />
+  <DataSetDialog :generateDialogVisible="generateDialogVisible" :handleGenerateDataSet="handleGenerateDataSet"/>
   <UploadProgress
     :isKnowledgeFileUpload="true"
     :showUploadNotify="uploadTaskListData.showUploadNotify"
@@ -1413,7 +1413,7 @@ const checkSelecTable = (row) => {
   return checkTableSelecData.value.every((item) => item?.id !== row?.id);
 };
 
-const handleGenerateDataSet = () => {
-  generateDialogVisible.value = true;
+const handleGenerateDataSet = (visible) => {
+  generateDialogVisible.value = visible;
 };
 </script>