mirror of
https://github.com/labring/FastGPT.git
synced 2025-10-14 07:00:47 +00:00
perf: mcp save raw schema (#5030)
* perf: mcp save raw schema * fix: test * code * perf: json schema test * perf: mcp
This commit is contained in:
Archer
122
test/cases/global/core/app/jsonschema.test.ts
Normal file
122
test/cases/global/core/app/jsonschema.test.ts
Normal file
@@ -0,0 +1,122 @@
|
||||
import { describe, expect, it } from 'vitest';
|
||||
import type { JSONSchemaInputType } from '@fastgpt/global/core/app/jsonschema';
|
||||
import { jsonSchema2NodeInput } from '@fastgpt/global/core/app/jsonschema';
|
||||
|
||||
describe('jsonSchema2NodeInput', () => {
|
||||
it('should return correct node input', () => {
|
||||
const jsonSchema: JSONSchemaInputType = {
|
||||
type: 'object',
|
||||
properties: {
|
||||
name: { type: 'string' },
|
||||
select: { type: 'string', enum: ['11', '22'] },
|
||||
age: { type: 'number', minimum: 0, maximum: 100 },
|
||||
boolean: { type: 'boolean' },
|
||||
object: { type: 'object' },
|
||||
strArr: { type: 'array', items: { type: 'string' } },
|
||||
numArr: { type: 'array', items: { type: 'number' } },
|
||||
boolArr: { type: 'array', items: { type: 'boolean' } },
|
||||
objArr: { type: 'array', items: { type: 'object' } },
|
||||
anyArr: { type: 'array', items: { type: 'array' } }
|
||||
},
|
||||
required: ['name', 'age']
|
||||
};
|
||||
const expectResponse = [
|
||||
{
|
||||
key: 'name',
|
||||
label: 'name',
|
||||
valueType: 'string',
|
||||
toolDescription: 'name',
|
||||
required: true,
|
||||
renderTypeList: ['input']
|
||||
},
|
||||
{
|
||||
key: 'select',
|
||||
label: 'select',
|
||||
valueType: 'string',
|
||||
toolDescription: 'select',
|
||||
required: false,
|
||||
value: '11',
|
||||
renderTypeList: ['select'],
|
||||
list: [
|
||||
{
|
||||
label: '11',
|
||||
value: '11'
|
||||
},
|
||||
{
|
||||
label: '22',
|
||||
value: '22'
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
key: 'age',
|
||||
label: 'age',
|
||||
valueType: 'number',
|
||||
toolDescription: 'age',
|
||||
required: true,
|
||||
renderTypeList: ['numberInput'],
|
||||
max: 100,
|
||||
min: 0
|
||||
},
|
||||
{
|
||||
key: 'boolean',
|
||||
label: 'boolean',
|
||||
valueType: 'boolean',
|
||||
toolDescription: 'boolean',
|
||||
required: false,
|
||||
renderTypeList: ['switch']
|
||||
},
|
||||
{
|
||||
key: 'object',
|
||||
label: 'object',
|
||||
valueType: 'object',
|
||||
toolDescription: 'object',
|
||||
required: false,
|
||||
renderTypeList: ['JSONEditor']
|
||||
},
|
||||
{
|
||||
key: 'strArr',
|
||||
label: 'strArr',
|
||||
valueType: 'arrayString',
|
||||
toolDescription: 'strArr',
|
||||
required: false,
|
||||
renderTypeList: ['JSONEditor']
|
||||
},
|
||||
{
|
||||
key: 'numArr',
|
||||
label: 'numArr',
|
||||
valueType: 'arrayNumber',
|
||||
toolDescription: 'numArr',
|
||||
required: false,
|
||||
renderTypeList: ['JSONEditor']
|
||||
},
|
||||
{
|
||||
key: 'boolArr',
|
||||
label: 'boolArr',
|
||||
valueType: 'arrayBoolean',
|
||||
toolDescription: 'boolArr',
|
||||
required: false,
|
||||
renderTypeList: ['JSONEditor']
|
||||
},
|
||||
{
|
||||
key: 'objArr',
|
||||
label: 'objArr',
|
||||
valueType: 'arrayObject',
|
||||
toolDescription: 'objArr',
|
||||
required: false,
|
||||
renderTypeList: ['JSONEditor']
|
||||
},
|
||||
{
|
||||
key: 'anyArr',
|
||||
label: 'anyArr',
|
||||
valueType: 'arrayAny',
|
||||
toolDescription: 'anyArr',
|
||||
required: false,
|
||||
renderTypeList: ['JSONEditor']
|
||||
}
|
||||
];
|
||||
const result = jsonSchema2NodeInput(jsonSchema);
|
||||
|
||||
expect(result).toEqual(expectResponse);
|
||||
});
|
||||
});
|
||||
@@ -1,293 +0,0 @@
|
||||
import { describe, expect, it, vi, beforeEach } from 'vitest';
|
||||
import {
|
||||
GET,
|
||||
POST,
|
||||
PUT,
|
||||
DELETE,
|
||||
responseSuccess,
|
||||
checkRes,
|
||||
responseError,
|
||||
getChannelList,
|
||||
getChannelProviders,
|
||||
postCreateChannel,
|
||||
putChannelStatus,
|
||||
putChannel,
|
||||
deleteChannel,
|
||||
getChannelLog,
|
||||
getLogDetail,
|
||||
getDashboardV2,
|
||||
instance
|
||||
} from '@/web/core/ai/channel';
|
||||
import axios from 'axios';
|
||||
|
||||
vi.mock('axios', () => ({
|
||||
default: {
|
||||
create: vi.fn(() => ({
|
||||
request: vi.fn(),
|
||||
interceptors: {
|
||||
response: {
|
||||
use: vi.fn()
|
||||
}
|
||||
}
|
||||
}))
|
||||
}
|
||||
}));
|
||||
|
||||
vi.mock('@fastgpt/web/common/system/utils', () => ({
|
||||
getWebReqUrl: () => 'http://localhost:3000'
|
||||
}));
|
||||
|
||||
describe('channel api', () => {
|
||||
const mockAxiosInstance = axios.create();
|
||||
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
it('should handle successful response', () => {
|
||||
const response = {
|
||||
data: {
|
||||
success: true,
|
||||
data: { test: 'data' }
|
||||
}
|
||||
};
|
||||
expect(responseSuccess(response)).toBe(response);
|
||||
});
|
||||
|
||||
it('should check response data', async () => {
|
||||
const validData = {
|
||||
success: true,
|
||||
message: 'ok',
|
||||
data: { test: 'data' }
|
||||
};
|
||||
await expect(Promise.resolve(checkRes(validData))).resolves.toEqual({ test: 'data' });
|
||||
|
||||
const invalidData = {
|
||||
success: false,
|
||||
message: 'error'
|
||||
};
|
||||
await expect(Promise.resolve(checkRes(invalidData))).rejects.toEqual(invalidData);
|
||||
|
||||
await expect(Promise.resolve(checkRes(undefined))).rejects.toBe('服务器异常');
|
||||
});
|
||||
|
||||
it('should handle response error', async () => {
|
||||
await expect(responseError('test error')).rejects.toEqual({
|
||||
message: 'test error'
|
||||
});
|
||||
|
||||
await expect(responseError({ response: { data: 'error data' } })).rejects.toBe('error data');
|
||||
|
||||
await expect(responseError(undefined)).rejects.toEqual({
|
||||
message: '未知错误'
|
||||
});
|
||||
});
|
||||
|
||||
it('should make GET request', async () => {
|
||||
vi.spyOn(instance, 'request').mockResolvedValueOnce({
|
||||
data: {
|
||||
success: true,
|
||||
data: { result: 'test' }
|
||||
}
|
||||
});
|
||||
|
||||
const result = await GET('/test');
|
||||
expect(result).toEqual({ result: 'test' });
|
||||
});
|
||||
|
||||
it('should make POST request', async () => {
|
||||
vi.spyOn(instance, 'request').mockResolvedValueOnce({
|
||||
data: {
|
||||
success: true,
|
||||
data: { id: 1 }
|
||||
}
|
||||
});
|
||||
|
||||
const result = await POST('/test', { name: 'test' });
|
||||
expect(result).toEqual({ id: 1 });
|
||||
});
|
||||
|
||||
it('should get channel list', async () => {
|
||||
vi.spyOn(instance, 'request').mockResolvedValueOnce({
|
||||
data: {
|
||||
success: true,
|
||||
data: { channels: [], total: 0 }
|
||||
}
|
||||
});
|
||||
|
||||
const result = await getChannelList();
|
||||
expect(result).toEqual({ channels: [], total: 0 });
|
||||
});
|
||||
|
||||
it('should get channel providers', async () => {
|
||||
const mockProviders = {
|
||||
1: {
|
||||
defaultBaseUrl: 'test.com',
|
||||
keyHelp: 'help',
|
||||
name: 'Test Provider'
|
||||
}
|
||||
};
|
||||
|
||||
vi.spyOn(instance, 'request').mockResolvedValueOnce({
|
||||
data: {
|
||||
success: true,
|
||||
data: mockProviders
|
||||
}
|
||||
});
|
||||
|
||||
const result = await getChannelProviders();
|
||||
expect(result).toEqual(mockProviders);
|
||||
});
|
||||
|
||||
it('should create channel', async () => {
|
||||
vi.spyOn(instance, 'request').mockResolvedValueOnce({
|
||||
data: {
|
||||
success: true,
|
||||
data: { id: 1 }
|
||||
}
|
||||
});
|
||||
|
||||
const result = await postCreateChannel({
|
||||
type: 1,
|
||||
name: 'test',
|
||||
base_url: 'test.com',
|
||||
models: [],
|
||||
model_mapping: {},
|
||||
key: 'key'
|
||||
});
|
||||
|
||||
expect(result).toEqual({ id: 1 });
|
||||
});
|
||||
|
||||
it('should update channel status', async () => {
|
||||
vi.spyOn(instance, 'request').mockResolvedValueOnce({
|
||||
data: {
|
||||
success: true,
|
||||
data: true
|
||||
}
|
||||
});
|
||||
|
||||
const result = await putChannelStatus(1, 'active');
|
||||
expect(result).toBe(true);
|
||||
});
|
||||
|
||||
it('should update channel', async () => {
|
||||
vi.spyOn(instance, 'request').mockResolvedValueOnce({
|
||||
data: {
|
||||
success: true,
|
||||
data: { id: 1 }
|
||||
}
|
||||
});
|
||||
|
||||
const result = await putChannel({
|
||||
id: 1,
|
||||
type: 1,
|
||||
name: 'test',
|
||||
base_url: 'test.com',
|
||||
models: [],
|
||||
model_mapping: {},
|
||||
key: 'key',
|
||||
status: 'active'
|
||||
});
|
||||
|
||||
expect(result).toEqual({ id: 1 });
|
||||
});
|
||||
|
||||
it('should delete channel', async () => {
|
||||
vi.spyOn(instance, 'request').mockResolvedValueOnce({
|
||||
data: {
|
||||
success: true,
|
||||
data: true
|
||||
}
|
||||
});
|
||||
|
||||
const result = await deleteChannel(1);
|
||||
expect(result).toBe(true);
|
||||
});
|
||||
|
||||
it('should get channel logs', async () => {
|
||||
vi.spyOn(instance, 'request').mockResolvedValueOnce({
|
||||
data: {
|
||||
success: true,
|
||||
data: {
|
||||
logs: [],
|
||||
total: 0
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
const result = await getChannelLog({
|
||||
start_timestamp: 0,
|
||||
end_timestamp: 1000,
|
||||
offset: 0,
|
||||
pageSize: 10
|
||||
});
|
||||
|
||||
expect(result).toEqual({
|
||||
list: [],
|
||||
total: 0
|
||||
});
|
||||
});
|
||||
|
||||
it('should get log detail', async () => {
|
||||
vi.spyOn(instance, 'request').mockResolvedValueOnce({
|
||||
data: {
|
||||
success: true,
|
||||
data: {
|
||||
request_body: '{}',
|
||||
response_body: '{}'
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
const result = await getLogDetail(1);
|
||||
expect(result).toEqual({
|
||||
request_body: '{}',
|
||||
response_body: '{}'
|
||||
});
|
||||
});
|
||||
|
||||
it('should get dashboard data', async () => {
|
||||
vi.spyOn(instance, 'request').mockResolvedValueOnce({
|
||||
data: {
|
||||
success: true,
|
||||
data: [
|
||||
{
|
||||
timestamp: 1000,
|
||||
summary: [
|
||||
{
|
||||
model: 'gpt-3.5',
|
||||
total_tokens: 100,
|
||||
request_count: 10,
|
||||
total_price: 0.002
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
});
|
||||
|
||||
const result = await getDashboardV2({
|
||||
channel: 1,
|
||||
model: 'gpt-3.5',
|
||||
start_timestamp: 0,
|
||||
end_timestamp: 1000,
|
||||
timezone: 'UTC',
|
||||
timespan: 'minute'
|
||||
});
|
||||
|
||||
expect(result).toEqual([
|
||||
{
|
||||
timestamp: 1000,
|
||||
summary: [
|
||||
{
|
||||
model: 'gpt-3.5',
|
||||
total_tokens: 100,
|
||||
request_count: 10,
|
||||
total_price: 0.002
|
||||
}
|
||||
]
|
||||
}
|
||||
]);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user