ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-10-15 23:55:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: allow adding tags when creating collections via api (#2268)

Browse Source 
* feat: allow adding tags when creating collections via api

* fix
This commit is contained in:
heheer
2024-08-05 18:33:58 +08:00
committed by GitHub
parent fe71efbbd2
commit 025d405fe2
5 changed files with 45 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
packages/service/core/dataset/collection/controller.ts
View File

@@ -1,7 +1,4 @@
import {
TrainingModeEnum,
DatasetCollectionTypeEnum
} from '@fastgpt/global/core/dataset/constants';
import { TrainingModeEnum } from '@fastgpt/global/core/dataset/constants';
import type { CreateDatasetCollectionParams } from '@fastgpt/global/core/dataset/api.d';
import { MongoDatasetCollection } from './schema';
import {
@@ -15,6 +12,7 @@ import { deleteDatasetDataVector } from '../../../common/vectorStore/controller'
import { delFileByFileIdList } from '../../../common/file/gridfs/controller';
import { BucketNameEnum } from '@fastgpt/global/common/file/constants';
import { ClientSession } from '../../../common/mongo';
import { createOrGetCollectionTags } from './utils';
export async function createOneCollection({
teamId,
@@ -39,6 +37,7 @@ export async function createOneCollection({
rawTextLength,
metadata = {},
session,
tags,
...props
}: CreateDatasetCollectionParams & {
teamId: string;
@@ -46,6 +45,7 @@ export async function createOneCollection({
[key: string]: any;
session?: ClientSession;
}) {
const collectionTags = await createOrGetCollectionTags({ tags, teamId, datasetId, session });
const [collection] = await MongoDatasetCollection.create(
[
{
@@ -69,7 +69,8 @@ export async function createOneCollection({
rawTextLength,
hashRawText,
metadata
metadata,
tags: collectionTags
}
],
{ session }