95 lines
1.8 KiB
TypeScript
95 lines
1.8 KiB
TypeScript
import { sequelize } from '@/modules/sequelize.ts';
|
|
import { DataTypes, Model } from 'sequelize';
|
|
|
|
export class AiAgent extends Model {
|
|
declare id: string;
|
|
declare type: string;
|
|
declare model: string;
|
|
declare baseUrl: string;
|
|
declare apiKey: string;
|
|
declare temperature: number;
|
|
declare cache: string;
|
|
declare cacheName: string;
|
|
declare status: string;
|
|
declare data: any;
|
|
declare description: string;
|
|
declare key: string;
|
|
}
|
|
|
|
// 获取AIAgent的属性
|
|
export type AiProperties = {
|
|
id: string;
|
|
type: string;
|
|
model: string;
|
|
baseUrl: string;
|
|
apiKey?: string;
|
|
temperature?: number;
|
|
cache?: string;
|
|
cacheName?: string;
|
|
data?: any;
|
|
description?: string;
|
|
};
|
|
AiAgent.init(
|
|
{
|
|
id: {
|
|
type: DataTypes.UUID,
|
|
primaryKey: true,
|
|
defaultValue: DataTypes.UUIDV4,
|
|
},
|
|
type: {
|
|
type: DataTypes.STRING,
|
|
allowNull: false,
|
|
},
|
|
description: {
|
|
type: DataTypes.TEXT,
|
|
allowNull: true,
|
|
},
|
|
status: {
|
|
type: DataTypes.STRING,
|
|
defaultValue: 'open',
|
|
},
|
|
model: {
|
|
type: DataTypes.STRING,
|
|
allowNull: false,
|
|
},
|
|
baseUrl: {
|
|
type: DataTypes.STRING,
|
|
allowNull: false,
|
|
},
|
|
apiKey: {
|
|
type: DataTypes.STRING,
|
|
allowNull: false,
|
|
},
|
|
key: {
|
|
type: DataTypes.STRING,
|
|
allowNull: false,
|
|
unique: true,
|
|
},
|
|
temperature: {
|
|
type: DataTypes.FLOAT,
|
|
allowNull: true,
|
|
},
|
|
cache: {
|
|
type: DataTypes.STRING,
|
|
allowNull: true,
|
|
},
|
|
cacheName: {
|
|
type: DataTypes.STRING,
|
|
allowNull: true,
|
|
},
|
|
data: {
|
|
type: DataTypes.JSON,
|
|
allowNull: true,
|
|
defaultValue: {},
|
|
},
|
|
},
|
|
{
|
|
sequelize,
|
|
tableName: 'ai_agent',
|
|
paranoid: true,
|
|
},
|
|
);
|
|
AiAgent.sync({ alter: true, logging: false }).catch((e) => {
|
|
console.error('AiAgent sync error', e);
|
|
});
|