Commit 661ee799 by Archer Committed by GitHub

fix: CQ module output (#445)

parent 60ee1601
...@@ -7,3 +7,24 @@ export const InformTypeMap = { ...@@ -7,3 +7,24 @@ export const InformTypeMap = {
label: '系统通知' label: '系统通知'
} }
}; };
export enum TeamMemberRoleEnum {
owner = 'owner',
admin = 'admin',
member = 'member',
visitor = 'visitor'
}
export const TeamMemberRoleMap = {
[TeamMemberRoleEnum.owner]: {
label: 'user.team.role.owner'
},
[TeamMemberRoleEnum.admin]: {
label: 'user.team.role.admin'
},
[TeamMemberRoleEnum.member]: {
label: 'user.team.role.member'
},
[TeamMemberRoleEnum.visitor]: {
label: 'user.team.role.visitor'
}
};
export type CreateTeamProps = {
ownerId: string;
name: string;
avatar?: string;
};
export type UpdateTeamProps = {
id: string;
name?: string;
avatar?: string;
};
export type updateTeamBalanceProps = {
id: string;
balance: number;
};
export type CreateTeamMemberProps = {
ownerId: string;
teamId: string;
userId: string;
name?: string;
};
import { InformTypeEnum } from './constant'; import { InformTypeEnum, TeamMemberRoleEnum } from './constant';
export type UserModelSchema = { export type UserModelSchema = {
_id: string; _id: string;
...@@ -30,3 +30,19 @@ export type UserInformSchema = { ...@@ -30,3 +30,19 @@ export type UserInformSchema = {
content: string; content: string;
read: boolean; read: boolean;
}; };
export type TeamSchema = {
_id: string;
name: string;
ownerId: string;
avatar: string;
createTime: Date;
};
export type TeamMemberSchema = {
_id: string;
name: string;
teamId: string;
userId: string;
role: `${TeamMemberRoleEnum}`;
};
...@@ -4,6 +4,8 @@ import { hashStr } from '@fastgpt/global/common/string/tools'; ...@@ -4,6 +4,8 @@ import { hashStr } from '@fastgpt/global/common/string/tools';
import { PRICE_SCALE } from '@fastgpt/global/common/bill/constants'; import { PRICE_SCALE } from '@fastgpt/global/common/bill/constants';
import type { UserModelSchema } from '@fastgpt/global/support/user/type'; import type { UserModelSchema } from '@fastgpt/global/support/user/type';
export const userCollectionName = 'users';
const UserSchema = new Schema({ const UserSchema = new Schema({
username: { username: {
// 可以是手机/邮箱,新的验证都只用手机 // 可以是手机/邮箱,新的验证都只用手机
...@@ -60,4 +62,5 @@ const UserSchema = new Schema({ ...@@ -60,4 +62,5 @@ const UserSchema = new Schema({
} }
}); });
export const MongoUser: Model<UserModelSchema> = models['user'] || model('user', UserSchema); export const MongoUser: Model<UserModelSchema> =
models[userCollectionName] || model(userCollectionName, UserSchema);
...@@ -121,14 +121,13 @@ const NodeCQNode = ({ data }: NodeProps<FlowModuleItemType>) => { ...@@ -121,14 +121,13 @@ const NodeCQNode = ({ data }: NodeProps<FlowModuleItemType>) => {
onChangeNode({ onChangeNode({
moduleId, moduleId,
type: 'updateOutput', type: 'addOutput',
key: agentKey, value: {
value: outputs.concat({
key, key,
label: '', label: '',
type: FlowNodeOutputTypeEnum.hidden, type: FlowNodeOutputTypeEnum.hidden,
targets: [] targets: []
}) }
}); });
}} }}
> >
......
...@@ -127,6 +127,9 @@ ${systemPrompt} ...@@ -127,6 +127,9 @@ ${systemPrompt}
tokens: response.usage?.total_tokens || 0 tokens: response.usage?.total_tokens || 0
}; };
} catch (error) { } catch (error) {
console.log(agentFunction.parameters);
console.log(response.choices?.[0]?.message);
console.log('Your model may not support function_call', error); console.log('Your model may not support function_call', error);
return { return {
......
...@@ -141,6 +141,9 @@ async function functionCall({ ...@@ -141,6 +141,9 @@ async function functionCall({
try { try {
return JSON.parse(response.choices?.[0]?.message?.function_call?.arguments || '{}'); return JSON.parse(response.choices?.[0]?.message?.function_call?.arguments || '{}');
} catch (error) { } catch (error) {
console.log(agentFunction.parameters);
console.log(response.choices?.[0]?.message);
console.log('Your model may not support function_call', error);
return {}; return {};
} }
})(); })();
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment