194 lines
2.4 KiB
JavaScript
194 lines
2.4 KiB
JavaScript
const config = require('../../config');
|
|
const providers = config.providers;
|
|
const crypto = require('crypto');
|
|
const bcrypt = require('bcrypt');
|
|
const moment = require('moment');
|
|
|
|
module.exports = function(sequelize, DataTypes) {
|
|
const proxy_routes = sequelize.define(
|
|
'proxy_routes',
|
|
{
|
|
id: {
|
|
type: DataTypes.UUID,
|
|
defaultValue: DataTypes.UUIDV4,
|
|
primaryKey: true,
|
|
},
|
|
|
|
route_type: {
|
|
type: DataTypes.ENUM,
|
|
|
|
|
|
|
|
values: [
|
|
|
|
"models",
|
|
|
|
|
|
"chat_completions",
|
|
|
|
|
|
"messages"
|
|
|
|
],
|
|
|
|
},
|
|
|
|
path: {
|
|
type: DataTypes.TEXT,
|
|
|
|
|
|
|
|
},
|
|
|
|
method: {
|
|
type: DataTypes.ENUM,
|
|
|
|
|
|
|
|
values: [
|
|
|
|
"GET",
|
|
|
|
|
|
"POST"
|
|
|
|
],
|
|
|
|
},
|
|
|
|
model_prefix_rule: {
|
|
type: DataTypes.ENUM,
|
|
|
|
|
|
|
|
values: [
|
|
|
|
"openai_gpt_o",
|
|
|
|
|
|
"anthropic_claude",
|
|
|
|
|
|
"mixed"
|
|
|
|
],
|
|
|
|
},
|
|
|
|
stream_supported: {
|
|
type: DataTypes.BOOLEAN,
|
|
|
|
allowNull: false,
|
|
defaultValue: false,
|
|
|
|
|
|
|
|
},
|
|
|
|
tool_calls_supported: {
|
|
type: DataTypes.BOOLEAN,
|
|
|
|
allowNull: false,
|
|
defaultValue: false,
|
|
|
|
|
|
|
|
},
|
|
|
|
transform_notes: {
|
|
type: DataTypes.TEXT,
|
|
|
|
|
|
|
|
},
|
|
|
|
enabled: {
|
|
type: DataTypes.BOOLEAN,
|
|
|
|
allowNull: false,
|
|
defaultValue: false,
|
|
|
|
|
|
|
|
},
|
|
|
|
importHash: {
|
|
type: DataTypes.STRING(255),
|
|
allowNull: true,
|
|
unique: true,
|
|
},
|
|
},
|
|
{
|
|
timestamps: true,
|
|
paranoid: true,
|
|
freezeTableName: true,
|
|
},
|
|
);
|
|
|
|
proxy_routes.associate = (db) => {
|
|
|
|
|
|
/// loop through entities and it's fields, and if ref === current e[name] and create relation has many on parent entity
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
db.proxy_routes.hasMany(db.api_requests, {
|
|
as: 'api_requests_proxy_route',
|
|
foreignKey: {
|
|
name: 'proxy_routeId',
|
|
},
|
|
constraints: false,
|
|
});
|
|
|
|
|
|
|
|
|
|
|
|
//end loop
|
|
|
|
|
|
|
|
db.proxy_routes.belongsTo(db.proxy_services, {
|
|
as: 'proxy_service',
|
|
foreignKey: {
|
|
name: 'proxy_serviceId',
|
|
},
|
|
constraints: false,
|
|
});
|
|
|
|
db.proxy_routes.belongsTo(db.ai_endpoints, {
|
|
as: 'upstream_endpoint',
|
|
foreignKey: {
|
|
name: 'upstream_endpointId',
|
|
},
|
|
constraints: false,
|
|
});
|
|
|
|
|
|
|
|
|
|
db.proxy_routes.belongsTo(db.users, {
|
|
as: 'createdBy',
|
|
});
|
|
|
|
db.proxy_routes.belongsTo(db.users, {
|
|
as: 'updatedBy',
|
|
});
|
|
};
|
|
|
|
|
|
|
|
return proxy_routes;
|
|
};
|
|
|
|
|