143 lines
3.4 KiB
JavaScript
143 lines
3.4 KiB
JavaScript
const Sequelize = require('sequelize');
|
||
|
||
/**
|
||
* AI调用记录表模型
|
||
* 记录所有AI API调用的详细信息和Token使用情况
|
||
*/
|
||
module.exports = (db) => {
|
||
const ai_call_records = db.define("ai_call_records", {
|
||
user_id: {
|
||
comment: '用户ID（如果是用户触发的调用）',
|
||
type: Sequelize.INTEGER,
|
||
allowNull: true,
|
||
defaultValue: null
|
||
},
|
||
sn_code: {
|
||
comment: '设备序列号',
|
||
type: Sequelize.STRING(50),
|
||
allowNull: true,
|
||
defaultValue: null
|
||
},
|
||
service_type: {
|
||
comment: '服务类型（如：chat, completion, embedding等）',
|
||
type: Sequelize.STRING(50),
|
||
allowNull: false,
|
||
defaultValue: ''
|
||
},
|
||
model_name: {
|
||
comment: 'AI模型名称（如：gpt-4, gpt-3.5-turbo等）',
|
||
type: Sequelize.STRING(100),
|
||
allowNull: false,
|
||
defaultValue: ''
|
||
},
|
||
prompt_tokens: {
|
||
comment: '输入Token数量',
|
||
type: Sequelize.INTEGER,
|
||
allowNull: false,
|
||
defaultValue: 0
|
||
},
|
||
completion_tokens: {
|
||
comment: '输出Token数量',
|
||
type: Sequelize.INTEGER,
|
||
allowNull: false,
|
||
defaultValue: 0
|
||
},
|
||
total_tokens: {
|
||
comment: '总Token数量',
|
||
type: Sequelize.INTEGER,
|
||
allowNull: false,
|
||
defaultValue: 0
|
||
},
|
||
request_content: {
|
||
comment: '请求内容（用户输入的prompt）',
|
||
type: Sequelize.TEXT,
|
||
allowNull: true,
|
||
defaultValue: null
|
||
},
|
||
response_content: {
|
||
comment: '响应内容（AI返回的结果）',
|
||
type: Sequelize.TEXT,
|
||
allowNull: true,
|
||
defaultValue: null
|
||
},
|
||
cost_amount: {
|
||
comment: '本次调用费用（元）',
|
||
type: Sequelize.DECIMAL(10, 4),
|
||
allowNull: true,
|
||
defaultValue: null
|
||
},
|
||
status: {
|
||
comment: '调用状态（success=成功, failed=失败, timeout=超时）',
|
||
type: Sequelize.STRING(20),
|
||
allowNull: false,
|
||
defaultValue: 'success'
|
||
},
|
||
error_message: {
|
||
comment: '错误信息（如果调用失败）',
|
||
type: Sequelize.TEXT,
|
||
allowNull: true,
|
||
defaultValue: null
|
||
},
|
||
response_time: {
|
||
comment: '响应时间（毫秒）',
|
||
type: Sequelize.INTEGER,
|
||
allowNull: true,
|
||
defaultValue: null
|
||
},
|
||
api_provider: {
|
||
comment: 'API提供商（openai, azure, anthropic等）',
|
||
type: Sequelize.STRING(50),
|
||
allowNull: false,
|
||
defaultValue: 'openai'
|
||
},
|
||
business_type: {
|
||
comment: '业务类型（job_filter, chat, resume_optimization等）',
|
||
type: Sequelize.STRING(50),
|
||
allowNull: true,
|
||
defaultValue: null
|
||
},
|
||
reference_id: {
|
||
comment: '关联业务ID（如job_posting_id, chat_record_id等）',
|
||
type: Sequelize.INTEGER,
|
||
allowNull: true,
|
||
defaultValue: null
|
||
},
|
||
}, {
|
||
timestamps: false,
|
||
indexes: [
|
||
{
|
||
unique: false,
|
||
fields: ['user_id']
|
||
},
|
||
{
|
||
unique: false,
|
||
fields: ['sn_code']
|
||
},
|
||
{
|
||
unique: false,
|
||
fields: ['service_type']
|
||
},
|
||
{
|
||
unique: false,
|
||
fields: ['status']
|
||
},
|
||
{
|
||
unique: false,
|
||
fields: ['create_time']
|
||
},
|
||
{
|
||
unique: false,
|
||
fields: ['business_type']
|
||
},
|
||
{
|
||
unique: false,
|
||
fields: ['reference_id']
|
||
}
|
||
]
|
||
});
|
||
|
||
// ai_call_records.sync({ force: true });
|
||
|
||
return ai_call_records;
|
||
}
|