61db646805
后端： 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。
69 lines
2.3 KiB
Go
69 lines
2.3 KiB
Go
package llm
|
|
|
|
import (
|
|
"crypto/sha1"
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
|
|
llmcontracts "github.com/LoveLosita/smartflow/backend/shared/contracts/llm"
|
|
"github.com/cloudwego/eino/schema"
|
|
)
|
|
|
|
// EnsureTextBillingIdentity 负责在老调用点未显式提供 event_id 时兜底补稳定事件号。
|
|
//
|
|
// 兼容策略：
|
|
// 1. 只在 user_id、request_id 已具备且 event_id 为空时触发，避免覆盖显式幂等键；
|
|
// 2. stage 优先从 GenerateOptions.Metadata["stage"] 读取，兼容 agent 现有大量调用点；
|
|
// 3. 输入摘要使用 messages 的稳定哈希，确保同一 request_id 下不同阶段/不同输入不会串账。
|
|
func EnsureTextBillingIdentity(billing BillingContext, options llmcontracts.GenerateOptions, messages []*schema.Message) BillingContext {
|
|
return ensureBillingIdentity(billing, readStageFromMetadata(options.Metadata), messages)
|
|
}
|
|
|
|
// EnsureResponsesBillingIdentity 负责给 Responses 调用补稳定事件号。
|
|
func EnsureResponsesBillingIdentity(billing BillingContext, messages []llmcontracts.ResponsesMessage) BillingContext {
|
|
return ensureBillingIdentity(billing, "", messages)
|
|
}
|
|
|
|
func ensureBillingIdentity(billing BillingContext, stage string, payload any) BillingContext {
|
|
billing = billing.Normalize()
|
|
if billing.UserID == 0 || strings.TrimSpace(billing.RequestID) == "" || strings.TrimSpace(billing.EventID) != "" {
|
|
return billing
|
|
}
|
|
|
|
stage = strings.TrimSpace(stage)
|
|
billing.EventID = buildStableBillingEventID(billing.RequestID, stage, hashPayload(payload))
|
|
return billing
|
|
}
|
|
|
|
func readStageFromMetadata(metadata map[string]any) string {
|
|
if len(metadata) == 0 {
|
|
return ""
|
|
}
|
|
raw, ok := metadata["stage"]
|
|
if !ok || raw == nil {
|
|
return ""
|
|
}
|
|
return strings.TrimSpace(fmt.Sprint(raw))
|
|
}
|
|
|
|
func hashPayload(payload any) string {
|
|
raw, err := json.Marshal(payload)
|
|
if err != nil || len(raw) == 0 {
|
|
return ""
|
|
}
|
|
sum := sha1.Sum(raw)
|
|
return hex.EncodeToString(sum[:8])
|
|
}
|
|
|
|
func buildStableBillingEventID(requestID, stage, payloadDigest string) string {
|
|
requestID = strings.TrimSpace(requestID)
|
|
stage = strings.TrimSpace(stage)
|
|
payloadDigest = strings.TrimSpace(payloadDigest)
|
|
|
|
base := requestID + "|" + stage + "|" + payloadDigest
|
|
sum := sha1.Sum([]byte(base))
|
|
return requestID + ":" + hex.EncodeToString(sum[:8])
|
|
}
|