62 lines
2.4 KiB
Python
62 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
from odoo import models, fields, api
|
|
|
|
|
|
class AiTokenUsage(models.Model):
|
|
"""Token 使用量记录"""
|
|
_name = 'ai.token.usage'
|
|
_description = 'AI Token 使用记录'
|
|
_order = 'create_date desc'
|
|
|
|
user_id = fields.Many2one(
|
|
'res.users', string='用户', required=True,
|
|
default=lambda self: self.env.uid, index=True,
|
|
)
|
|
conversation_id = fields.Many2one(
|
|
'ai.conversation', string='对话', ondelete='set null', index=True,
|
|
)
|
|
provider_id = fields.Many2one(
|
|
'ai.provider', string='服务商', ondelete='set null',
|
|
)
|
|
model_name = fields.Char(string='模型')
|
|
prompt_tokens = fields.Integer(string='输入 Token', default=0)
|
|
completion_tokens = fields.Integer(string='输出 Token', default=0)
|
|
total_tokens = fields.Integer(string='总 Token', default=0)
|
|
create_date = fields.Datetime(string='时间', readonly=True)
|
|
|
|
# 汇总统计（SQL视图替代方案：直接在列表中分组展示即可）
|
|
|
|
|
|
class AiTokenUsageSummary(models.Model):
|
|
"""用户 Token 汇总（数据库视图）"""
|
|
_name = 'ai.token.usage.summary'
|
|
_description = 'AI Token 使用汇总'
|
|
_auto = False
|
|
_order = 'total_tokens desc'
|
|
|
|
user_id = fields.Many2one('res.users', string='用户', readonly=True)
|
|
provider_id = fields.Many2one('ai.provider', string='服务商', readonly=True)
|
|
model_name = fields.Char(string='模型', readonly=True)
|
|
call_count = fields.Integer(string='调用次数', readonly=True)
|
|
prompt_tokens = fields.Integer(string='输入 Token', readonly=True)
|
|
completion_tokens = fields.Integer(string='输出 Token', readonly=True)
|
|
total_tokens = fields.Integer(string='总 Token', readonly=True)
|
|
|
|
def init(self):
|
|
self.env.cr.execute("""
|
|
DROP VIEW IF EXISTS ai_token_usage_summary;
|
|
CREATE OR REPLACE VIEW ai_token_usage_summary AS (
|
|
SELECT
|
|
row_number() OVER () AS id,
|
|
user_id,
|
|
provider_id,
|
|
model_name,
|
|
COUNT(*) AS call_count,
|
|
SUM(prompt_tokens) AS prompt_tokens,
|
|
SUM(completion_tokens) AS completion_tokens,
|
|
SUM(total_tokens) AS total_tokens
|
|
FROM ai_token_usage
|
|
GROUP BY user_id, provider_id, model_name
|
|
);
|
|
""")
|