d6df056687
- 数据层：messages 表增加 part_type 字段，新建 file_versions 表支持版本追踪 - 后端：saveWorkspace 版本追踪、saveAgentOutput 源头分离、generateBriefMessage 成员简报 - 后端：applyDocumentEdit 增量编辑、buildWorkflowStep phase-aware 工作流引擎 - API：文件版本查询/回退接口 - 前端：part_type 驱动渲染，产物面板版本历史 - 新增写手团队（主编/搜索员/策划编辑/合规审查员）配置 - store 模块、scheduler 模块、web-search skill Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
30 lines
830 B
Go
30 lines
830 B
Go
package store
|
|
|
|
type TokenStats struct {
|
|
PromptTokens int
|
|
CompletionTokens int
|
|
TotalTokens int
|
|
}
|
|
|
|
func (s *Store) InsertTokenUsage(roomID, agent string, prompt, completion, total int) error {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
_, err := s.db.Exec(
|
|
`INSERT INTO token_usage (room_id, agent, prompt_tokens, completion_tokens, total_tokens) VALUES (?, ?, ?, ?, ?)`,
|
|
roomID, agent, prompt, completion, total,
|
|
)
|
|
return err
|
|
}
|
|
|
|
func (s *Store) GetRoomTokenStats(roomID string) (*TokenStats, error) {
|
|
var ts TokenStats
|
|
err := s.db.QueryRow(
|
|
`SELECT COALESCE(SUM(prompt_tokens),0), COALESCE(SUM(completion_tokens),0), COALESCE(SUM(total_tokens),0)
|
|
FROM token_usage WHERE room_id = ?`, roomID,
|
|
).Scan(&ts.PromptTokens, &ts.CompletionTokens, &ts.TotalTokens)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &ts, nil
|
|
}
|