74 lines
1.6 KiB
JavaScript
74 lines
1.6 KiB
JavaScript
|
module.exports = (sequelize, DataTypes) => {
|
||
|
|
const ToolExecution = sequelize.define('ToolExecution', {
|
||
|
|
id: {
|
||
|
|
type: DataTypes.UUID,
|
||
|
|
defaultValue: DataTypes.UUIDV4,
|
||
|
|
primaryKey: true
|
||
|
|
},
|
||
|
|
plan_id: {
|
||
|
|
type: DataTypes.UUID,
|
||
|
|
allowNull: false,
|
||
|
|
references: {
|
||
|
|
model: 'plans',
|
||
|
|
key: 'id'
|
||
|
|
}
|
||
|
|
},
|
||
|
|
document_id: {
|
||
|
|
type: DataTypes.UUID,
|
||
|
|
allowNull: true,
|
||
|
|
references: {
|
||
|
|
model: 'documents',
|
||
|
|
key: 'id'
|
||
|
|
}
|
||
|
|
},
|
||
|
|
tool_name: {
|
||
|
|
type: DataTypes.STRING,
|
||
|
|
allowNull: false
|
||
|
|
},
|
||
|
|
tool_type: {
|
||
|
|
type: DataTypes.ENUM('query_expander', 'extraction', 'report1', 'report2', 'web_search', 'encyclopedia_pdf'),
|
||
|
|
allowNull: false
|
||
|
|
},
|
||
|
|
input_parameters: {
|
||
|
|
type: DataTypes.JSONB,
|
||
|
|
allowNull: false
|
||
|
|
},
|
||
|
|
output_result: {
|
||
|
|
type: DataTypes.JSONB,
|
||
|
|
allowNull: true
|
||
|
|
},
|
||
|
|
status: {
|
||
|
|
type: DataTypes.ENUM('pending', 'running', 'completed', 'failed'),
|
||
|
|
defaultValue: 'pending'
|
||
|
|
},
|
||
|
|
execution_time: {
|
||
|
|
type: DataTypes.FLOAT,
|
||
|
|
allowNull: true
|
||
|
|
},
|
||
|
|
error_message: {
|
||
|
|
type: DataTypes.TEXT,
|
||
|
|
allowNull: true
|
||
|
|
},
|
||
|
|
tokens_used: {
|
||
|
|
type: DataTypes.INTEGER,
|
||
|
|
defaultValue: 0
|
||
|
|
},
|
||
|
|
metadata: {
|
||
|
|
type: DataTypes.JSONB,
|
||
|
|
defaultValue: {}
|
||
|
|
}
|
||
|
|
}, {
|
||
|
|
tableName: 'tool_executions',
|
||
|
|
indexes: [
|
||
|
|
{ fields: ['plan_id'] },
|
||
|
|
{ fields: ['document_id'] },
|
||
|
|
{ fields: ['tool_name'] },
|
||
|
|
{ fields: ['tool_type'] },
|
||
|
|
{ fields: ['status'] },
|
||
|
|
{ fields: ['created_at'] }
|
||
|
|
]
|
||
|
|
});
|
||
|
|
|
||
|
|
return ToolExecution;
|
||
|
|
};
|