Files
smartmate/backend/services/tokenstore/sv/credit_charge.go
Losita 61db646805 Version: 0.9.80.dev.260506
后端:
1. LLM 独立服务与统一计费出口落地:新增 `cmd/llm`、`client/llm` 与 `services/llm/rpc`,补齐 BillingContext、CreditBalanceGuard、价格规则解析、stream usage 归集与 `credit.charge.requested` outbox 发布,active-scheduler / agent / course / memory / gateway fallback 全部改走 llm zrpc,不再各自本地初始化模型。
2. TokenStore 收口为 Credit 权威账本:新增 credit account / ledger / product / order / price-rule / reward-rule 能力与 Redis 快照缓存,扩展 tokenstore rpc/client 支撑余额快照、消耗看板、商品、订单、流水、价格规则和奖励规则,并接入 LLM charge 事件消费完成 Credit 扣费落账。
3. 计费旧链路下线与网关切口切换:`/token-store` 语义整体切到 `/credit-store`,agent chat 移除旧 TokenQuotaGuard,userauth 的 CheckTokenQuota / AdjustTokenUsage 改为废弃,聊天历史落库不再同步旧 token 额度账本,course 图片解析请求补 user_id 进入新计费口径。

前端:
4. 计划广场从 mock 数据切到真实接口:新增 forum api/types,首页支持真实列表、标签、搜索、防抖、点赞、导入和发布计划,详情页补齐帖子详情、评论树、回复和删除评论链路,同时补上“至少一个标签”的前后端约束与默认标签兜底。
5. 商店页切到 Credit 体系并重做展示:顶部改为余额 + Credit/Token 消耗看板,支持 24h/7d/30d/all 周期切换;套餐区展示原价与当前价;历史区改为当前用户 Credit 流水并支持查看更多,整体视觉和交互同步收口。

仓库:
6. 配置与本地启动体系补齐 llm / outbox 编排:`config.example.yaml` 增加 llm rpc 和统一 outbox service 配置,`dev-common.ps1` 把 llm 纳入多服务依赖并自动建 Kafka topic,`docker-compose.yml` 同步初始化 agent/task/memory/active-scheduler/notification/taskclass-forum/llm/token-store 全量 outbox topic。
2026-05-06 20:16:53 +08:00

113 lines
3.7 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
package sv
import (
"context"
"fmt"
"strings"
"time"
storemodel "github.com/LoveLosita/smartflow/backend/services/tokenstore/model"
creditcontracts "github.com/LoveLosita/smartflow/backend/shared/contracts/creditstore"
sharedevents "github.com/LoveLosita/smartflow/backend/shared/events"
)
type creditChargeMetadata struct {
Scene string `json:"scene"`
RequestID string `json:"request_id"`
ConversationID string `json:"conversation_id"`
ModelAlias string `json:"model_alias"`
ProviderName string `json:"provider_name"`
ModelName string `json:"model_name"`
InputTokens int64 `json:"input_tokens"`
OutputTokens int64 `json:"output_tokens"`
CachedTokens int64 `json:"cached_tokens"`
ReasoningTokens int64 `json:"reasoning_tokens"`
TotalTokens int64 `json:"total_tokens"`
RMBCostMicros int64 `json:"rmb_cost_micros"`
CreditCost int64 `json:"credit_cost"`
SkipCharge bool `json:"skip_charge"`
TriggeredAt time.Time `json:"triggered_at"`
}
// RecordCreditCharge 负责把 LLM 扣费事件写入 Credit 权威账本。
func (s *Service) RecordCreditCharge(ctx context.Context, payload sharedevents.CreditChargeRequestedPayload) (*creditcontracts.CreditTransactionView, error) {
if err := s.Ready(); err != nil {
return nil, err
}
if err := payload.Validate(); err != nil {
return nil, err
}
sourceRefID := strings.TrimSpace(payload.RequestID)
if sourceRefID == "" {
sourceRefID = strings.TrimSpace(payload.ConversationID)
}
var sourceRefIDPtr *string
if sourceRefID != "" {
sourceRefIDPtr = &sourceRefID
}
amount := -payload.CreditCost
status := storemodel.CreditLedgerStatusApplied
if payload.SkipCharge {
amount = 0
status = storemodel.CreditLedgerStatusSkipped
}
ledger, _, err := s.applyCreditLedger(ctx, applyCreditLedgerRequest{
EventID: strings.TrimSpace(payload.EventID),
UserID: payload.UserID,
Source: storemodel.CreditLedgerSourceCharge,
SourceLabel: creditSourceLabel(storemodel.CreditLedgerSourceCharge, ""),
Direction: storemodel.CreditLedgerDirectionExpense,
SourceRefID: sourceRefIDPtr,
Amount: amount,
Status: status,
Description: creditChargeDescription(payload),
MetadataJSON: creditMetadataJSON(creditChargeMetadataFromPayload(payload)),
CreatedAt: payload.TriggeredAt,
})
if err != nil {
return nil, err
}
view := creditTransactionViewFromModel(*ledger)
return &view, nil
}
func creditChargeMetadataFromPayload(payload sharedevents.CreditChargeRequestedPayload) creditChargeMetadata {
return creditChargeMetadata{
Scene: payload.Scene,
RequestID: payload.RequestID,
ConversationID: payload.ConversationID,
ModelAlias: payload.ModelAlias,
ProviderName: payload.ProviderName,
ModelName: payload.ModelName,
InputTokens: payload.InputTokens,
OutputTokens: payload.OutputTokens,
CachedTokens: payload.CachedTokens,
ReasoningTokens: payload.ReasoningTokens,
TotalTokens: payload.TotalTokens,
RMBCostMicros: payload.RMBCostMicros,
CreditCost: payload.CreditCost,
SkipCharge: payload.SkipCharge,
TriggeredAt: payload.TriggeredAt,
}
}
func creditChargeDescription(payload sharedevents.CreditChargeRequestedPayload) string {
modelText := strings.TrimSpace(payload.ModelAlias)
if modelText == "" {
modelText = strings.TrimSpace(payload.ModelName)
}
sceneText := strings.TrimSpace(payload.Scene)
switch {
case sceneText != "" && modelText != "":
return fmt.Sprintf("AI 调用扣费(%s / %s", sceneText, modelText)
case modelText != "":
return fmt.Sprintf("AI 调用扣费(%s", modelText)
default:
return "AI 调用扣费"
}
}