mirror of
https://github.com/labring/FastGPT.git
synced 2026-05-07 01:02:55 +08:00
chore: vitest support (#4026)
* chore: vitest * chore: move test files * chore: support vitest * fix: exclude test files * chore(ci): add test workflow * feat: remove read env
This commit is contained in:
Finley Ge
@@ -1,130 +0,0 @@
|
||||
import { MongoMemoryReplSet } from 'mongodb-memory-server';
|
||||
import mongoose from 'mongoose';
|
||||
import { parseHeaderCertMock } from '@fastgpt/service/test/utils';
|
||||
import { initMockData, root } from './db/init';
|
||||
import { faker } from '@faker-js/faker/locale/zh_CN';
|
||||
|
||||
jest.mock('nanoid', () => {
|
||||
return {
|
||||
nanoid: () => {}
|
||||
};
|
||||
});
|
||||
|
||||
jest.mock('@fastgpt/global/common/string/tools', () => {
|
||||
return {
|
||||
hashStr(str: string) {
|
||||
return str;
|
||||
},
|
||||
getNanoid() {
|
||||
return faker.string.alphanumeric(12);
|
||||
}
|
||||
};
|
||||
});
|
||||
|
||||
jest.mock('@fastgpt/service/common/system/log', () => ({
|
||||
addLog: {
|
||||
log: jest.fn(),
|
||||
warn: jest.fn((...prop) => {
|
||||
console.warn(prop);
|
||||
}),
|
||||
error: jest.fn((...prop) => {
|
||||
console.error(prop);
|
||||
}),
|
||||
info: jest.fn(),
|
||||
debug: jest.fn()
|
||||
}
|
||||
}));
|
||||
|
||||
jest.setMock(
|
||||
'@fastgpt/service/support/permission/controller',
|
||||
(() => {
|
||||
const origin = jest.requireActual<
|
||||
typeof import('@fastgpt/service/support/permission/controller')
|
||||
>('@fastgpt/service/support/permission/controller');
|
||||
|
||||
return {
|
||||
...origin,
|
||||
parseHeaderCert: parseHeaderCertMock
|
||||
};
|
||||
})()
|
||||
);
|
||||
|
||||
jest.mock('@/service/middleware/entry', () => {
|
||||
return {
|
||||
NextAPI: (...args: any) => {
|
||||
return async function api(req: any, res: any) {
|
||||
try {
|
||||
let response = null;
|
||||
for (const handler of args) {
|
||||
response = await handler(req, res);
|
||||
}
|
||||
return {
|
||||
code: 200,
|
||||
data: response
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
code: 500,
|
||||
error
|
||||
};
|
||||
}
|
||||
};
|
||||
}
|
||||
};
|
||||
});
|
||||
|
||||
beforeAll(async () => {
|
||||
// 新建一个内存数据库，然后让 mongoose 连接这个数据库
|
||||
if (!global.mongod || !global.mongodb) {
|
||||
const replSet = new MongoMemoryReplSet({
|
||||
instanceOpts: [
|
||||
{
|
||||
storageEngine: 'wiredTiger'
|
||||
},
|
||||
{
|
||||
storageEngine: 'wiredTiger'
|
||||
}
|
||||
]
|
||||
});
|
||||
replSet.start();
|
||||
await replSet.waitUntilRunning();
|
||||
const uri = replSet.getUri();
|
||||
// const mongod = await MongoMemoryServer.create({
|
||||
// instance: {
|
||||
// replSet: 'testset'
|
||||
// }
|
||||
// });
|
||||
// global.mongod = mongod;
|
||||
global.replSet = replSet;
|
||||
global.mongodb = mongoose;
|
||||
|
||||
await global.mongodb.connect(uri, {
|
||||
dbName: 'fastgpt_test',
|
||||
bufferCommands: true,
|
||||
maxConnecting: 50,
|
||||
maxPoolSize: 50,
|
||||
minPoolSize: 20,
|
||||
connectTimeoutMS: 60000,
|
||||
waitQueueTimeoutMS: 60000,
|
||||
socketTimeoutMS: 60000,
|
||||
maxIdleTimeMS: 300000,
|
||||
retryWrites: true,
|
||||
retryReads: true
|
||||
});
|
||||
|
||||
await initMockData();
|
||||
console.log(root);
|
||||
}
|
||||
});
|
||||
|
||||
afterAll(async () => {
|
||||
if (global.mongodb) {
|
||||
await global.mongodb.disconnect();
|
||||
}
|
||||
if (global.replSet) {
|
||||
await global.replSet.stop();
|
||||
}
|
||||
if (global.mongod) {
|
||||
await global.mongod.stop();
|
||||
}
|
||||
});
|
||||
@@ -1,56 +0,0 @@
|
||||
import { DefaultGroupName } from '@fastgpt/global/support/user/team/group/constant';
|
||||
import { MongoApp } from '@fastgpt/service/core/app/schema';
|
||||
import { MongoMemberGroupModel } from '@fastgpt/service/support/permission/memberGroup/memberGroupSchema';
|
||||
import { MongoUser } from '@fastgpt/service/support/user/schema';
|
||||
import { MongoTeamMember } from '@fastgpt/service/support/user/team/teamMemberSchema';
|
||||
import { MongoTeam } from '@fastgpt/service/support/user/team/teamSchema';
|
||||
|
||||
export const root = {
|
||||
uid: '',
|
||||
tmbId: '',
|
||||
teamId: '',
|
||||
isRoot: true,
|
||||
appId: ''
|
||||
};
|
||||
|
||||
export const initMockData = async () => {
|
||||
const [rootUser] = await MongoUser.create([
|
||||
{
|
||||
username: 'root',
|
||||
password: '123456'
|
||||
}
|
||||
]);
|
||||
root.uid = String(rootUser._id);
|
||||
const [rootTeam] = await MongoTeam.create([
|
||||
{
|
||||
name: 'root Team'
|
||||
}
|
||||
]);
|
||||
root.teamId = String(rootTeam._id);
|
||||
const [rootTmb] = await MongoTeamMember.create([
|
||||
{
|
||||
teamId: rootTeam._id,
|
||||
name: 'owner',
|
||||
role: 'owner',
|
||||
userId: rootUser._id,
|
||||
status: 'active'
|
||||
}
|
||||
]);
|
||||
root.tmbId = String(rootTmb._id);
|
||||
await MongoMemberGroupModel.create([
|
||||
{
|
||||
name: DefaultGroupName,
|
||||
teamId: rootTeam._id
|
||||
}
|
||||
]);
|
||||
|
||||
const [rootApp] = await MongoApp.create([
|
||||
{
|
||||
name: 'root Test App',
|
||||
teamId: rootTeam._id,
|
||||
tmbId: rootTmb._id
|
||||
}
|
||||
]);
|
||||
|
||||
root.appId = String(rootApp._id);
|
||||
};
|
||||
@@ -1,61 +0,0 @@
|
||||
import '@/pages/api/__mocks__/base';
|
||||
import { root } from '@/pages/api/__mocks__/db/init';
|
||||
import { getTestRequest } from '@fastgpt/service/test/utils';
|
||||
import { AppErrEnum } from '@fastgpt/global/common/error/code/app';
|
||||
import handler from './demo';
|
||||
|
||||
// Import the schema
|
||||
import { MongoOutLink } from '@fastgpt/service/support/outLink/schema';
|
||||
|
||||
beforeAll(async () => {
|
||||
// await MongoOutLink.create({
|
||||
// shareId: 'aaa',
|
||||
// appId: root.appId,
|
||||
// tmbId: root.tmbId,
|
||||
// teamId: root.teamId,
|
||||
// type: 'share',
|
||||
// name: 'aaa'
|
||||
// })
|
||||
});
|
||||
|
||||
test('Should return a list of outLink', async () => {
|
||||
// Mock request
|
||||
const res = (await handler(
|
||||
...getTestRequest({
|
||||
query: {
|
||||
appId: root.appId,
|
||||
type: 'share'
|
||||
},
|
||||
user: root
|
||||
})
|
||||
)) as any;
|
||||
|
||||
expect(res.code).toBe(200);
|
||||
expect(res.data.length).toBe(2);
|
||||
});
|
||||
|
||||
test('appId is required', async () => {
|
||||
const res = (await handler(
|
||||
...getTestRequest({
|
||||
query: {
|
||||
type: 'share'
|
||||
},
|
||||
user: root
|
||||
})
|
||||
)) as any;
|
||||
expect(res.code).toBe(500);
|
||||
expect(res.error).toBe(AppErrEnum.unExist);
|
||||
});
|
||||
|
||||
test('if type is not provided, return nothing', async () => {
|
||||
const res = (await handler(
|
||||
...getTestRequest({
|
||||
query: {
|
||||
appId: root.appId
|
||||
},
|
||||
user: root
|
||||
})
|
||||
)) as any;
|
||||
expect(res.code).toBe(200);
|
||||
expect(res.data.length).toBe(0);
|
||||
});
|
||||
@@ -1,17 +0,0 @@
|
||||
import type { ApiRequestProps, ApiResponseType } from '@fastgpt/service/type/next';
|
||||
import { NextAPI } from '@/service/middleware/entry';
|
||||
|
||||
export type demoQuery = {};
|
||||
|
||||
export type demoBody = {};
|
||||
|
||||
export type demoResponse = {};
|
||||
|
||||
async function handler(
|
||||
req: ApiRequestProps<demoBody, demoQuery>,
|
||||
res: ApiResponseType<any>
|
||||
): Promise<demoResponse> {
|
||||
return {};
|
||||
}
|
||||
|
||||
export default NextAPI(handler);
|
||||
-11
@@ -1,11 +0,0 @@
|
||||
import type { MongoMemoryReplSet, MongoMemoryServer } from 'mongodb-memory-server';
|
||||
declare global {
|
||||
var mongod: MongoMemoryServer | undefined;
|
||||
var replSet: MongoMemoryReplSet | undefined;
|
||||
}
|
||||
|
||||
export type RequestResponse<T = any> = {
|
||||
code: number;
|
||||
error?: string;
|
||||
data?: T;
|
||||
};
|
||||
@@ -1,55 +0,0 @@
|
||||
import '@/pages/api/__mocks__/base';
|
||||
import { root } from '@/pages/api/__mocks__/db/init';
|
||||
import { getTestRequest } from '@fastgpt/service/test/utils';
|
||||
import handler, { getLatestVersionQuery, getLatestVersionResponse } from './latest';
|
||||
import { MongoAppVersion } from '@fastgpt/service/core/app/version/schema';
|
||||
|
||||
beforeAll(async () => {
|
||||
// 创建3个测试数据，其中2个是已发布的
|
||||
await MongoAppVersion.create([
|
||||
{
|
||||
appId: root.appId,
|
||||
nodes: [1],
|
||||
edges: [],
|
||||
chatConfig: {},
|
||||
isPublish: false,
|
||||
versionName: 'v1',
|
||||
tmbId: root.tmbId,
|
||||
time: new Date('2023-01-01')
|
||||
},
|
||||
{
|
||||
appId: root.appId,
|
||||
nodes: [2],
|
||||
edges: [],
|
||||
chatConfig: {},
|
||||
isPublish: true,
|
||||
versionName: 'v2',
|
||||
tmbId: root.tmbId,
|
||||
time: new Date('2023-01-02')
|
||||
},
|
||||
{
|
||||
appId: root.appId,
|
||||
nodes: [3],
|
||||
edges: [],
|
||||
chatConfig: {},
|
||||
isPublish: false,
|
||||
versionName: 'v3',
|
||||
tmbId: root.tmbId,
|
||||
time: new Date('2023-01-03')
|
||||
}
|
||||
]);
|
||||
});
|
||||
|
||||
test('获取最新版本并检查', async () => {
|
||||
const _res = (await handler(
|
||||
...getTestRequest<{}, getLatestVersionQuery>({
|
||||
query: {
|
||||
appId: root.appId
|
||||
},
|
||||
user: root
|
||||
})
|
||||
)) as any;
|
||||
const res = _res.data as getLatestVersionResponse;
|
||||
|
||||
expect(res.nodes[0]).toEqual(2);
|
||||
});
|
||||
@@ -1,69 +0,0 @@
|
||||
import '@/pages/api/__mocks__/base';
|
||||
import { root } from '@/pages/api/__mocks__/db/init';
|
||||
import { getTestRequest } from '@fastgpt/service/test/utils';
|
||||
import handler, { versionListBody, versionListResponse } from './list';
|
||||
|
||||
// Import the schema
|
||||
import { MongoAppVersion } from '@fastgpt/service/core/app/version/schema';
|
||||
|
||||
const total = 22;
|
||||
|
||||
beforeAll(async () => {
|
||||
const arr = new Array(total).fill(0);
|
||||
await MongoAppVersion.insertMany(
|
||||
arr.map((_, index) => ({
|
||||
appId: root.appId,
|
||||
nodes: [],
|
||||
edges: [],
|
||||
chatConfig: {},
|
||||
isPublish: index % 2 === 0,
|
||||
versionName: `v` + index,
|
||||
tmbId: root.tmbId,
|
||||
time: new Date(index * 1000)
|
||||
}))
|
||||
);
|
||||
});
|
||||
|
||||
test('Get version list and check', async () => {
|
||||
const offset = 0;
|
||||
const pageSize = 10;
|
||||
|
||||
const _res = (await handler(
|
||||
...getTestRequest<{}, versionListBody>({
|
||||
body: {
|
||||
offset,
|
||||
pageSize,
|
||||
appId: root.appId
|
||||
},
|
||||
user: root
|
||||
})
|
||||
)) as any;
|
||||
const res = _res.data as versionListResponse;
|
||||
|
||||
expect(res.total).toBe(total);
|
||||
expect(res.list.length).toBe(pageSize);
|
||||
expect(res.list[0].versionName).toBe('v21');
|
||||
expect(res.list[9].versionName).toBe('v12');
|
||||
});
|
||||
|
||||
test('Get version list with offset 20', async () => {
|
||||
const offset = 20;
|
||||
const pageSize = 10;
|
||||
|
||||
const _res = (await handler(
|
||||
...getTestRequest<{}, versionListBody>({
|
||||
body: {
|
||||
offset,
|
||||
pageSize,
|
||||
appId: root.appId
|
||||
},
|
||||
user: root
|
||||
})
|
||||
)) as any;
|
||||
const res = _res.data as versionListResponse;
|
||||
|
||||
expect(res.total).toBe(total);
|
||||
expect(res.list.length).toBe(2);
|
||||
expect(res.list[0].versionName).toBe('v1');
|
||||
expect(res.list[1].versionName).toBe('v0');
|
||||
});
|
||||
@@ -0,0 +1,35 @@
|
||||
import { MongoApp } from '@fastgpt/service/core/app/schema';
|
||||
import { MongoAppVersion } from '@fastgpt/service/core/app/version/schema';
|
||||
import { getRootUser } from '@test/datas/users';
|
||||
import { Call } from '@test/utils/request';
|
||||
import { describe, expect, it } from 'vitest';
|
||||
import handler, { type versionListBody, type versionListResponse } from './list';
|
||||
|
||||
describe('app version list test', () => {
|
||||
it('should return app version list', async () => {
|
||||
const root = await getRootUser();
|
||||
const app = await MongoApp.create({
|
||||
name: 'test',
|
||||
tmbId: root.tmbId,
|
||||
teamId: root.teamId
|
||||
});
|
||||
await MongoAppVersion.create(
|
||||
[...Array(10).keys()].map((i) => ({
|
||||
tmbId: root.tmbId,
|
||||
appId: app._id,
|
||||
versionName: `v${i}`
|
||||
}))
|
||||
);
|
||||
const res = await Call<versionListBody, {}, versionListResponse>(handler, {
|
||||
auth: root,
|
||||
body: {
|
||||
pageSize: 10,
|
||||
offset: 0,
|
||||
appId: app._id
|
||||
}
|
||||
});
|
||||
expect(res.code).toBe(200);
|
||||
expect(res.data.total).toBe(10);
|
||||
expect(res.data.list.length).toBe(10);
|
||||
});
|
||||
});
|
||||
@@ -1,36 +0,0 @@
|
||||
import '@/pages/api/__mocks__/base';
|
||||
import { root } from '@/pages/api/__mocks__/db/init';
|
||||
import { getTestRequest } from '@fastgpt/service/test/utils';
|
||||
import handler from './publish';
|
||||
import { MongoAppVersion } from '@fastgpt/service/core/app/version/schema';
|
||||
import { PostPublishAppProps } from '@/global/core/app/api';
|
||||
import { AppTypeEnum } from '@fastgpt/global/core/app/constants';
|
||||
|
||||
describe('发布应用版本测试', () => {
|
||||
test('发布一个未发布的版本', async () => {
|
||||
const publishData: PostPublishAppProps = {
|
||||
nodes: [],
|
||||
edges: [],
|
||||
chatConfig: {},
|
||||
isPublish: false,
|
||||
versionName: '1'
|
||||
};
|
||||
|
||||
await handler(
|
||||
...getTestRequest<{ appId: string }, PostPublishAppProps>({
|
||||
body: publishData,
|
||||
query: { appId: root.appId },
|
||||
user: root
|
||||
})
|
||||
);
|
||||
|
||||
// 检查数据库是否插入成功
|
||||
const insertedVersion = await MongoAppVersion.countDocuments();
|
||||
|
||||
console.log(insertedVersion, '==-');
|
||||
|
||||
// expect(insertedVersion).toBeTruthy();
|
||||
// expect(insertedVersion?.isPublish).toBe(false);
|
||||
// expect(insertedVersion?.versionName).toBe('1');
|
||||
});
|
||||
});
|
||||
@@ -1,67 +0,0 @@
|
||||
import '../../__mocks__/base';
|
||||
import { root } from '../../__mocks__/db/init';
|
||||
import { getTestRequest } from '@fastgpt/service/test/utils';
|
||||
import type { OutLinkListQuery } from './list';
|
||||
import { AppErrEnum } from '@fastgpt/global/common/error/code/app';
|
||||
import handler from './list';
|
||||
import { MongoOutLink } from '@fastgpt/service/support/outLink/schema';
|
||||
|
||||
beforeAll(async () => {
|
||||
await MongoOutLink.create({
|
||||
shareId: 'aaa',
|
||||
appId: root.appId,
|
||||
tmbId: root.tmbId,
|
||||
teamId: root.teamId,
|
||||
type: 'share',
|
||||
name: 'aaa'
|
||||
});
|
||||
await MongoOutLink.create({
|
||||
shareId: 'bbb',
|
||||
appId: root.appId,
|
||||
tmbId: root.tmbId,
|
||||
teamId: root.teamId,
|
||||
type: 'share',
|
||||
name: 'bbb'
|
||||
});
|
||||
});
|
||||
|
||||
test('Should return a list of outLink', async () => {
|
||||
const res = (await handler(
|
||||
...getTestRequest<OutLinkListQuery>({
|
||||
query: {
|
||||
appId: root.appId,
|
||||
type: 'share'
|
||||
},
|
||||
user: root
|
||||
})
|
||||
)) as any;
|
||||
|
||||
expect(res.code).toBe(200);
|
||||
expect(res.data.length).toBe(2);
|
||||
});
|
||||
|
||||
test('appId is required', async () => {
|
||||
const res = (await handler(
|
||||
...getTestRequest<OutLinkListQuery>({
|
||||
query: {
|
||||
type: 'share'
|
||||
},
|
||||
user: root
|
||||
})
|
||||
)) as any;
|
||||
expect(res.code).toBe(500);
|
||||
expect(res.error).toBe(AppErrEnum.unExist);
|
||||
});
|
||||
|
||||
test('if type is not provided, return nothing', async () => {
|
||||
const res = (await handler(
|
||||
...getTestRequest<OutLinkListQuery>({
|
||||
query: {
|
||||
appId: root.appId
|
||||
},
|
||||
user: root
|
||||
})
|
||||
)) as any;
|
||||
expect(res.code).toBe(200);
|
||||
expect(res.data.length).toBe(0);
|
||||
});
|
||||
@@ -1,54 +0,0 @@
|
||||
import '../../__mocks__/base';
|
||||
import { getTestRequest } from '@fastgpt/service/test/utils';
|
||||
import handler, { OutLinkUpdateBody, OutLinkUpdateQuery } from './update';
|
||||
import { MongoOutLink } from '@fastgpt/service/support/outLink/schema';
|
||||
import { CommonErrEnum } from '@fastgpt/global/common/error/code/common';
|
||||
import { root } from '../../__mocks__/db/init';
|
||||
|
||||
beforeAll(async () => {
|
||||
await MongoOutLink.create({
|
||||
shareId: 'aaa',
|
||||
appId: root.appId,
|
||||
tmbId: root.tmbId,
|
||||
teamId: root.teamId,
|
||||
type: 'share',
|
||||
name: 'aaa'
|
||||
});
|
||||
});
|
||||
|
||||
test('Update Outlink', async () => {
|
||||
const outlink = await MongoOutLink.findOne({ name: 'aaa' }).lean();
|
||||
if (!outlink) {
|
||||
throw new Error('Outlink not found');
|
||||
}
|
||||
|
||||
const res = (await handler(
|
||||
...getTestRequest<OutLinkUpdateQuery, OutLinkUpdateBody>({
|
||||
body: {
|
||||
_id: outlink._id,
|
||||
name: 'changed'
|
||||
},
|
||||
user: root
|
||||
})
|
||||
)) as any;
|
||||
|
||||
console.log(res);
|
||||
expect(res.code).toBe(200);
|
||||
|
||||
const link = await MongoOutLink.findById(outlink._id).lean();
|
||||
expect(link?.name).toBe('changed');
|
||||
});
|
||||
|
||||
test('Did not post _id', async () => {
|
||||
const res = (await handler(
|
||||
...getTestRequest<OutLinkUpdateQuery, OutLinkUpdateBody>({
|
||||
body: {
|
||||
name: 'changed'
|
||||
},
|
||||
user: root
|
||||
})
|
||||
)) as any;
|
||||
|
||||
expect(res.code).toBe(500);
|
||||
expect(res.error).toBe(CommonErrEnum.missingParams);
|
||||
});
|
||||
@@ -1,145 +0,0 @@
|
||||
import '@/pages/api/__mocks__/base';
|
||||
import { parseReasoningStreamContent } from '@fastgpt/service/core/ai/utils';
|
||||
|
||||
test('Parse reasoning stream content test', async () => {
|
||||
const partList = [
|
||||
{
|
||||
data: [{ content: '你好1' }, { content: '你好2' }, { content: '你好3' }],
|
||||
correct: { answer: '你好1你好2你好3', reasoning: '' }
|
||||
},
|
||||
{
|
||||
data: [
|
||||
{ reasoning_content: '这是' },
|
||||
{ reasoning_content: '思考' },
|
||||
{ reasoning_content: '过程' },
|
||||
{ content: '你好1' },
|
||||
{ content: '你好2' },
|
||||
{ content: '你好3' }
|
||||
],
|
||||
correct: { answer: '你好1你好2你好3', reasoning: '这是思考过程' }
|
||||
},
|
||||
{
|
||||
data: [
|
||||
{ content: '<t' },
|
||||
{ content: 'hink>' },
|
||||
{ content: '这是' },
|
||||
{ content: '思考' },
|
||||
{ content: '过程' },
|
||||
{ content: '</think>' },
|
||||
{ content: '你好1' },
|
||||
{ content: '你好2' },
|
||||
{ content: '你好3' }
|
||||
],
|
||||
correct: { answer: '你好1你好2你好3', reasoning: '这是思考过程' }
|
||||
},
|
||||
{
|
||||
data: [
|
||||
{ content: '<think>' },
|
||||
{ content: '这是' },
|
||||
{ content: '思考' },
|
||||
{ content: '过程' },
|
||||
{ content: '</think>' },
|
||||
{ content: '你好1' },
|
||||
{ content: '你好2' },
|
||||
{ content: '你好3' }
|
||||
],
|
||||
correct: { answer: '你好1你好2你好3', reasoning: '这是思考过程' }
|
||||
},
|
||||
{
|
||||
data: [
|
||||
{ content: '<think>这是' },
|
||||
{ content: '思考' },
|
||||
{ content: '过程' },
|
||||
{ content: '</think>' },
|
||||
{ content: '你好1' },
|
||||
{ content: '你好2' },
|
||||
{ content: '你好3' }
|
||||
],
|
||||
correct: { answer: '你好1你好2你好3', reasoning: '这是思考过程' }
|
||||
},
|
||||
{
|
||||
data: [
|
||||
{ content: '<think>这是' },
|
||||
{ content: '思考' },
|
||||
{ content: '过程</' },
|
||||
{ content: 'think>' },
|
||||
{ content: '你好1' },
|
||||
{ content: '你好2' },
|
||||
{ content: '你好3' }
|
||||
],
|
||||
correct: { answer: '你好1你好2你好3', reasoning: '这是思考过程' }
|
||||
},
|
||||
{
|
||||
data: [
|
||||
{ content: '<think>这是' },
|
||||
{ content: '思考' },
|
||||
{ content: '过程</think>' },
|
||||
{ content: '你好1' },
|
||||
{ content: '你好2' },
|
||||
{ content: '你好3' }
|
||||
],
|
||||
correct: { answer: '你好1你好2你好3', reasoning: '这是思考过程' }
|
||||
},
|
||||
{
|
||||
data: [
|
||||
{ content: '<think>这是' },
|
||||
{ content: '思考' },
|
||||
{ content: '过程</think>你好1' },
|
||||
{ content: '你好2' },
|
||||
{ content: '你好3' }
|
||||
],
|
||||
correct: { answer: '你好1你好2你好3', reasoning: '这是思考过程' }
|
||||
},
|
||||
{
|
||||
data: [
|
||||
{ content: '<think>这是' },
|
||||
{ content: '思考' },
|
||||
{ content: '过程</th' },
|
||||
{ content: '假的' },
|
||||
{ content: '你好2' },
|
||||
{ content: '你好3' },
|
||||
{ content: '过程</think>你好1' },
|
||||
{ content: '你好2' },
|
||||
{ content: '你好3' }
|
||||
],
|
||||
correct: { answer: '你好1你好2你好3', reasoning: '这是思考过程</th假的你好2你好3过程' }
|
||||
},
|
||||
{
|
||||
data: [
|
||||
{ content: '<think>这是' },
|
||||
{ content: '思考' },
|
||||
{ content: '过程</th' },
|
||||
{ content: '假的' },
|
||||
{ content: '你好2' },
|
||||
{ content: '你好3' }
|
||||
],
|
||||
correct: { answer: '', reasoning: '这是思考过程</th假的你好2你好3' }
|
||||
}
|
||||
];
|
||||
|
||||
partList.forEach((part) => {
|
||||
const { parsePart } = parseReasoningStreamContent();
|
||||
|
||||
let answer = '';
|
||||
let reasoning = '';
|
||||
part.data.forEach((item) => {
|
||||
const formatPart = {
|
||||
choices: [
|
||||
{
|
||||
delta: {
|
||||
role: 'assistant',
|
||||
content: item.content,
|
||||
reasoning_content: item.reasoning_content
|
||||
}
|
||||
}
|
||||
]
|
||||
};
|
||||
const [reasoningContent, content] = parsePart(formatPart, true);
|
||||
answer += content;
|
||||
reasoning += reasoningContent;
|
||||
});
|
||||
|
||||
expect(answer).toBe(part.correct.answer);
|
||||
expect(reasoning).toBe(part.correct.reasoning);
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,358 @@
|
||||
{
|
||||
"nodes": [
|
||||
{
|
||||
"nodeId": "userGuide",
|
||||
"name": "common:core.module.template.system_config",
|
||||
"intro": "common:core.module.template.system_config_info",
|
||||
"avatar": "core/workflow/template/systemConfig",
|
||||
"flowNodeType": "userGuide",
|
||||
"position": {
|
||||
"x": 220.4077028616387,
|
||||
"y": -429.3158723159836
|
||||
},
|
||||
"version": "481",
|
||||
"inputs": [
|
||||
{
|
||||
"key": "welcomeText",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"valueType": "string",
|
||||
"label": "core.app.Welcome Text",
|
||||
"value": ""
|
||||
},
|
||||
{
|
||||
"key": "variables",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"valueType": "any",
|
||||
"label": "core.app.Chat Variable",
|
||||
"value": []
|
||||
},
|
||||
{
|
||||
"key": "questionGuide",
|
||||
"valueType": "any",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"label": "core.app.Question Guide",
|
||||
"value": {
|
||||
"open": false
|
||||
}
|
||||
},
|
||||
{
|
||||
"key": "tts",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"valueType": "any",
|
||||
"label": "",
|
||||
"value": {
|
||||
"type": "web"
|
||||
}
|
||||
},
|
||||
{
|
||||
"key": "whisper",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"valueType": "any",
|
||||
"label": "",
|
||||
"value": {
|
||||
"open": false,
|
||||
"autoSend": false,
|
||||
"autoTTSResponse": false
|
||||
}
|
||||
},
|
||||
{
|
||||
"key": "scheduleTrigger",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"valueType": "any",
|
||||
"label": "",
|
||||
"value": null
|
||||
}
|
||||
],
|
||||
"outputs": []
|
||||
},
|
||||
{
|
||||
"nodeId": "448745",
|
||||
"name": "common:core.module.template.work_start",
|
||||
"intro": "",
|
||||
"avatar": "core/workflow/template/workflowStart",
|
||||
"flowNodeType": "workflowStart",
|
||||
"position": {
|
||||
"x": 773.4174945178407,
|
||||
"y": -331.3158723159836
|
||||
},
|
||||
"version": "481",
|
||||
"inputs": [
|
||||
{
|
||||
"key": "userChatInput",
|
||||
"renderTypeList": [
|
||||
"reference",
|
||||
"textarea"
|
||||
],
|
||||
"valueType": "string",
|
||||
"label": "common:core.module.input.label.user question",
|
||||
"required": true,
|
||||
"toolDescription": "用户问题",
|
||||
"debugLabel": ""
|
||||
}
|
||||
],
|
||||
"outputs": [
|
||||
{
|
||||
"id": "userChatInput",
|
||||
"key": "userChatInput",
|
||||
"label": "common:core.module.input.label.user question",
|
||||
"type": "static",
|
||||
"valueType": "string",
|
||||
"description": ""
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"nodeId": "nlv8iMRsvgkA",
|
||||
"name": "批量执行",
|
||||
"intro": "输入一个数组，遍历数组并将每一个数组元素作为输入元素，执行工作流。",
|
||||
"avatar": "core/workflow/template/loop",
|
||||
"flowNodeType": "loop",
|
||||
"showStatus": true,
|
||||
"position": {
|
||||
"x": 1236,
|
||||
"y": -593
|
||||
},
|
||||
"version": "4811",
|
||||
"inputs": [
|
||||
{
|
||||
"key": "loopInputArray",
|
||||
"renderTypeList": [
|
||||
"reference"
|
||||
],
|
||||
"valueType": "arrayNumber",
|
||||
"required": true,
|
||||
"label": "数组",
|
||||
"value": [
|
||||
[
|
||||
"VARIABLE_NODE_ID",
|
||||
"list"
|
||||
]
|
||||
],
|
||||
"valueDesc": "",
|
||||
"description": "",
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "childrenNodeIdList",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"valueType": "arrayString",
|
||||
"label": "",
|
||||
"value": [
|
||||
"tRxC7faEoGuE",
|
||||
"cGnptXbKAyMN"
|
||||
]
|
||||
},
|
||||
{
|
||||
"key": "nodeWidth",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"valueType": "number",
|
||||
"label": "",
|
||||
"value": 1246.6404923618281
|
||||
},
|
||||
{
|
||||
"key": "nodeHeight",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"valueType": "number",
|
||||
"label": "",
|
||||
"value": 642.1566957382456
|
||||
},
|
||||
{
|
||||
"key": "loopNodeInputHeight",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"valueType": "number",
|
||||
"label": "",
|
||||
"value": 83,
|
||||
"valueDesc": "",
|
||||
"description": "",
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
}
|
||||
],
|
||||
"outputs": [
|
||||
{
|
||||
"id": "loopArray",
|
||||
"key": "loopArray",
|
||||
"label": "数组执行结果",
|
||||
"type": "static",
|
||||
"valueType": "arrayAny",
|
||||
"valueDesc": "",
|
||||
"description": ""
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"nodeId": "tRxC7faEoGuE",
|
||||
"parentNodeId": "nlv8iMRsvgkA",
|
||||
"name": "开始",
|
||||
"avatar": "core/workflow/template/loopStart",
|
||||
"flowNodeType": "loopStart",
|
||||
"showStatus": false,
|
||||
"position": {
|
||||
"x": 1305.782937883576,
|
||||
"y": -270.30845154767246
|
||||
},
|
||||
"version": "4811",
|
||||
"inputs": [
|
||||
{
|
||||
"key": "loopStartInput",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"valueType": "any",
|
||||
"label": "",
|
||||
"required": true,
|
||||
"value": ""
|
||||
},
|
||||
{
|
||||
"key": "loopStartIndex",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"valueType": "number",
|
||||
"label": "workflow:Array_element_index"
|
||||
}
|
||||
],
|
||||
"outputs": [
|
||||
{
|
||||
"id": "loopStartIndex",
|
||||
"key": "loopStartIndex",
|
||||
"label": "workflow:Array_element_index",
|
||||
"type": "static",
|
||||
"valueType": "number"
|
||||
},
|
||||
{
|
||||
"id": "loopStartInput",
|
||||
"key": "loopStartInput",
|
||||
"label": "数组元素",
|
||||
"type": "static",
|
||||
"valueType": "number"
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"nodeId": "cGnptXbKAyMN",
|
||||
"parentNodeId": "nlv8iMRsvgkA",
|
||||
"name": "结束",
|
||||
"avatar": "core/workflow/template/loopEnd",
|
||||
"flowNodeType": "loopEnd",
|
||||
"showStatus": false,
|
||||
"position": {
|
||||
"x": 1929.4234302454042,
|
||||
"y": 135.8482441905731
|
||||
},
|
||||
"version": "4811",
|
||||
"inputs": [
|
||||
{
|
||||
"key": "loopEndInput",
|
||||
"renderTypeList": [
|
||||
"reference"
|
||||
],
|
||||
"valueType": "any",
|
||||
"label": "",
|
||||
"required": true,
|
||||
"value": []
|
||||
}
|
||||
],
|
||||
"outputs": []
|
||||
},
|
||||
{
|
||||
"nodeId": "zpOBWBxfyUap",
|
||||
"parentNodeId": "nlv8iMRsvgkA",
|
||||
"name": "指定回复",
|
||||
"intro": "该模块可以直接回复一段指定的内容。常用于引导、提示。非字符串内容传入时，会转成字符串进行输出。",
|
||||
"avatar": "core/workflow/template/reply",
|
||||
"flowNodeType": "answerNode",
|
||||
"position": {
|
||||
"x": 1806.423430245404,
|
||||
"y": -217.4185397094268
|
||||
},
|
||||
"version": "481",
|
||||
"inputs": [
|
||||
{
|
||||
"key": "text",
|
||||
"renderTypeList": [
|
||||
"textarea",
|
||||
"reference"
|
||||
],
|
||||
"valueType": "any",
|
||||
"required": true,
|
||||
"label": "回复的内容",
|
||||
"description": "可以使用 \\n 来实现连续换行。\n可以通过外部模块输入实现回复，外部模块输入时会覆盖当前填写的内容。\n如传入非字符串类型数据将会自动转成字符串",
|
||||
"placeholder": "common:core.module.input.description.Response content",
|
||||
"value": "{{$tRxC7faEoGuE.loopStartInput$}}",
|
||||
"valueDesc": "",
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
}
|
||||
],
|
||||
"outputs": []
|
||||
}
|
||||
],
|
||||
"edges": [
|
||||
{
|
||||
"source": "448745",
|
||||
"target": "nlv8iMRsvgkA",
|
||||
"sourceHandle": "448745-source-right",
|
||||
"targetHandle": "nlv8iMRsvgkA-target-left"
|
||||
},
|
||||
{
|
||||
"source": "tRxC7faEoGuE",
|
||||
"target": "zpOBWBxfyUap",
|
||||
"sourceHandle": "tRxC7faEoGuE-source-right",
|
||||
"targetHandle": "zpOBWBxfyUap-target-left"
|
||||
}
|
||||
],
|
||||
"chatConfig": {
|
||||
"variables": [
|
||||
{
|
||||
"id": "04sm7m",
|
||||
"key": "list",
|
||||
"label": "list",
|
||||
"type": "custom",
|
||||
"description": "",
|
||||
"required": false,
|
||||
"valueType": "arrayNumber",
|
||||
"list": [
|
||||
{
|
||||
"value": "",
|
||||
"label": ""
|
||||
}
|
||||
],
|
||||
"defaultValue": "[1,2,3]",
|
||||
"enums": [
|
||||
{
|
||||
"value": "",
|
||||
"label": ""
|
||||
}
|
||||
]
|
||||
}
|
||||
],
|
||||
"_id": "67a8d281b54c01f7bd95c995",
|
||||
"scheduledTriggerConfig": {
|
||||
"cronString": "",
|
||||
"timezone": "Asia/Shanghai",
|
||||
"defaultPrompt": ""
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,319 @@
|
||||
{
|
||||
"nodes": [
|
||||
{
|
||||
"nodeId": "userGuide",
|
||||
"name": "系统配置",
|
||||
"intro": "",
|
||||
"avatar": "core/workflow/template/systemConfig",
|
||||
"flowNodeType": "userGuide",
|
||||
"position": {
|
||||
"x": 531.2422736065552,
|
||||
"y": -486.7611729549753
|
||||
},
|
||||
"version": "481",
|
||||
"inputs": [],
|
||||
"outputs": []
|
||||
},
|
||||
{
|
||||
"nodeId": "workflowStartNodeId",
|
||||
"name": "流程开始",
|
||||
"intro": "",
|
||||
"avatar": "core/workflow/template/workflowStart",
|
||||
"flowNodeType": "workflowStart",
|
||||
"position": {
|
||||
"x": 531.2422736065552,
|
||||
"y": 244.69591764653183
|
||||
},
|
||||
"version": "481",
|
||||
"inputs": [
|
||||
{
|
||||
"key": "userChatInput",
|
||||
"renderTypeList": [
|
||||
"reference",
|
||||
"textarea"
|
||||
],
|
||||
"valueType": "string",
|
||||
"label": "workflow:user_question",
|
||||
"toolDescription": "用户问题",
|
||||
"required": true,
|
||||
"debugLabel": ""
|
||||
}
|
||||
],
|
||||
"outputs": [
|
||||
{
|
||||
"id": "userChatInput",
|
||||
"key": "userChatInput",
|
||||
"label": "common:core.module.input.label.user question",
|
||||
"type": "static",
|
||||
"valueType": "string",
|
||||
"description": ""
|
||||
},
|
||||
{
|
||||
"id": "userFiles",
|
||||
"key": "userFiles",
|
||||
"label": "app:workflow.user_file_input",
|
||||
"description": "app:workflow.user_file_input_desc",
|
||||
"type": "static",
|
||||
"valueType": "arrayString"
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"nodeId": "7BdojPlukIQw",
|
||||
"name": "AI 对话",
|
||||
"intro": "AI 大模型对话",
|
||||
"avatar": "core/workflow/template/aiChat",
|
||||
"flowNodeType": "chatNode",
|
||||
"showStatus": true,
|
||||
"position": {
|
||||
"x": 1106.3238387960757,
|
||||
"y": -350.6030674683474
|
||||
},
|
||||
"version": "4813",
|
||||
"inputs": [
|
||||
{
|
||||
"key": "model",
|
||||
"renderTypeList": [
|
||||
"settingLLMModel",
|
||||
"reference"
|
||||
],
|
||||
"label": "",
|
||||
"valueType": "string",
|
||||
"value": "deepseek-reasoner",
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "temperature",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"label": "",
|
||||
"value": 0,
|
||||
"valueType": "number",
|
||||
"min": 0,
|
||||
"max": 10,
|
||||
"step": 1,
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "maxToken",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"label": "",
|
||||
"value": 8000,
|
||||
"valueType": "number",
|
||||
"min": 100,
|
||||
"max": 4000,
|
||||
"step": 50,
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "isResponseAnswerText",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"label": "",
|
||||
"value": true,
|
||||
"valueType": "boolean",
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "aiChatQuoteRole",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"label": "",
|
||||
"valueType": "string",
|
||||
"value": "system",
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "quoteTemplate",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"label": "",
|
||||
"valueType": "string",
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "quotePrompt",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"label": "",
|
||||
"valueType": "string",
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "aiChatVision",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"label": "",
|
||||
"valueType": "boolean",
|
||||
"value": true,
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "aiChatReasoning",
|
||||
"renderTypeList": [
|
||||
"hidden"
|
||||
],
|
||||
"label": "",
|
||||
"valueType": "boolean",
|
||||
"value": true,
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "systemPrompt",
|
||||
"renderTypeList": [
|
||||
"textarea",
|
||||
"reference"
|
||||
],
|
||||
"max": 3000,
|
||||
"valueType": "string",
|
||||
"label": "core.ai.Prompt",
|
||||
"description": "core.app.tip.systemPromptTip",
|
||||
"placeholder": "core.app.tip.chatNodeSystemPromptTip",
|
||||
"value": "",
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "history",
|
||||
"renderTypeList": [
|
||||
"numberInput",
|
||||
"reference"
|
||||
],
|
||||
"valueType": "chatHistory",
|
||||
"label": "core.module.input.label.chat history",
|
||||
"required": true,
|
||||
"min": 0,
|
||||
"max": 50,
|
||||
"value": 6,
|
||||
"description": "workflow:max_dialog_rounds",
|
||||
"debugLabel": "",
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "quoteQA",
|
||||
"renderTypeList": [
|
||||
"settingDatasetQuotePrompt"
|
||||
],
|
||||
"label": "",
|
||||
"debugLabel": "知识库引用",
|
||||
"description": "",
|
||||
"valueType": "datasetQuote",
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "fileUrlList",
|
||||
"renderTypeList": [
|
||||
"reference",
|
||||
"input"
|
||||
],
|
||||
"label": "app:file_quote_link",
|
||||
"debugLabel": "文件链接",
|
||||
"valueType": "arrayString",
|
||||
"value": [
|
||||
[
|
||||
"workflowStartNodeId",
|
||||
"userFiles"
|
||||
]
|
||||
],
|
||||
"toolDescription": ""
|
||||
},
|
||||
{
|
||||
"key": "userChatInput",
|
||||
"renderTypeList": [
|
||||
"reference",
|
||||
"textarea"
|
||||
],
|
||||
"valueType": "string",
|
||||
"label": "common:core.module.input.label.user question",
|
||||
"required": true,
|
||||
"toolDescription": "用户问题",
|
||||
"value": [
|
||||
"workflowStartNodeId",
|
||||
"userChatInput"
|
||||
],
|
||||
"debugLabel": ""
|
||||
}
|
||||
],
|
||||
"outputs": [
|
||||
{
|
||||
"id": "history",
|
||||
"key": "history",
|
||||
"required": true,
|
||||
"label": "common:core.module.output.label.New context",
|
||||
"description": "将本次回复内容拼接上历史记录，作为新的上下文返回",
|
||||
"valueType": "chatHistory",
|
||||
"valueDesc": "{\n obj: System | Human | AI;\n value: string;\n}[]",
|
||||
"type": "static"
|
||||
},
|
||||
{
|
||||
"id": "answerText",
|
||||
"key": "answerText",
|
||||
"required": true,
|
||||
"label": "common:core.module.output.label.Ai response content",
|
||||
"description": "将在 stream 回复完毕后触发",
|
||||
"valueType": "string",
|
||||
"type": "static"
|
||||
}
|
||||
]
|
||||
}
|
||||
],
|
||||
"edges": [
|
||||
{
|
||||
"source": "workflowStartNodeId",
|
||||
"target": "7BdojPlukIQw",
|
||||
"sourceHandle": "workflowStartNodeId-source-right",
|
||||
"targetHandle": "7BdojPlukIQw-target-left"
|
||||
}
|
||||
],
|
||||
"chatConfig": {
|
||||
"questionGuide": false,
|
||||
"ttsConfig": {
|
||||
"type": "web"
|
||||
},
|
||||
"whisperConfig": {
|
||||
"open": false,
|
||||
"autoSend": false,
|
||||
"autoTTSResponse": false
|
||||
},
|
||||
"scheduledTriggerConfig": {
|
||||
"cronString": "",
|
||||
"timezone": "Asia/Shanghai",
|
||||
"defaultPrompt": ""
|
||||
},
|
||||
"chatInputGuide": {
|
||||
"open": false,
|
||||
"textList": [],
|
||||
"customUrl": ""
|
||||
},
|
||||
"instruction": "",
|
||||
"autoExecute": {
|
||||
"open": false,
|
||||
"defaultPrompt": ""
|
||||
},
|
||||
"welcomeText": "",
|
||||
"variables": [],
|
||||
"fileSelectConfig": {
|
||||
"canSelectFile": false,
|
||||
"canSelectImg": false,
|
||||
"maxFiles": 10
|
||||
},
|
||||
"_id": "66f4c2f5e9e4e93a95141004"
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,82 @@
|
||||
import { readFileSync } from 'fs';
|
||||
import { resolve } from 'path';
|
||||
import { describe, it, expect, beforeEach, vi } from 'vitest';
|
||||
import { dispatchWorkFlow } from '@fastgpt/service/core/workflow/dispatch';
|
||||
import {
|
||||
getWorkflowEntryNodeIds,
|
||||
storeNodes2RuntimeNodes
|
||||
} from '@fastgpt/global/core/workflow/runtime/utils';
|
||||
import { ChatItemValueTypeEnum } from '@fastgpt/global/core/chat/constants';
|
||||
vi.mock(import('@fastgpt/service/common/string/tiktoken'), async (importOriginal) => {
|
||||
const mod = await importOriginal();
|
||||
return {
|
||||
...mod,
|
||||
countGptMessagesTokens: async () => {
|
||||
return 1;
|
||||
}
|
||||
};
|
||||
});
|
||||
|
||||
vi.mock(import('@fastgpt/service/support/wallet/usage/utils'), async (importOriginal) => {
|
||||
const mod = await importOriginal();
|
||||
return {
|
||||
...mod,
|
||||
formatModelChars2Points: () => ({
|
||||
modelName: 'test',
|
||||
totalPoints: 1
|
||||
})
|
||||
};
|
||||
});
|
||||
|
||||
const testWorkflow = async (path: string) => {
|
||||
const workflowStr = readFileSync(resolve(path), 'utf-8');
|
||||
const workflow = JSON.parse(workflowStr);
|
||||
const { nodes, edges, chatConfig } = workflow;
|
||||
let runtimeNodes = storeNodes2RuntimeNodes(nodes, getWorkflowEntryNodeIds(nodes));
|
||||
const variables = {};
|
||||
const { assistantResponses, flowResponses } = await dispatchWorkFlow({
|
||||
mode: 'test',
|
||||
runningAppInfo: {
|
||||
id: 'test',
|
||||
teamId: 'test',
|
||||
tmbId: 'test'
|
||||
},
|
||||
runningUserInfo: {
|
||||
tmbId: 'test',
|
||||
teamId: 'test'
|
||||
},
|
||||
timezone: 'Asia/Shanghai',
|
||||
externalProvider: {},
|
||||
uid: 'test',
|
||||
runtimeNodes,
|
||||
runtimeEdges: edges,
|
||||
variables,
|
||||
query: [
|
||||
{
|
||||
type: ChatItemValueTypeEnum.text,
|
||||
text: {
|
||||
content: '你是谁'
|
||||
}
|
||||
}
|
||||
],
|
||||
chatConfig,
|
||||
histories: [],
|
||||
stream: false,
|
||||
maxRunTimes: 5
|
||||
});
|
||||
expect(assistantResponses).toBeDefined();
|
||||
expect(assistantResponses[0].text?.content).toBeDefined();
|
||||
return {
|
||||
assistantResponses,
|
||||
flowResponses
|
||||
};
|
||||
};
|
||||
|
||||
it('Workflow test: simple workflow', async () => {
|
||||
// create a simple app
|
||||
await testWorkflow('projects/app/src/test/workflow/simple.json');
|
||||
});
|
||||
|
||||
it('Workflow test: output test', async () => {
|
||||
console.log(await testWorkflow('projects/app/src/test/workflow/loopTest.json'));
|
||||
});
|
||||
Reference in New Issue
Block a user