refactor: 重构流模式逻辑

This commit is contained in:
CalciumIon
2024-07-15 18:04:05 +08:00
parent 0f687aab9a
commit 7029065892
4 changed files with 114 additions and 123 deletions

View File

@@ -36,3 +36,7 @@ func GenerateFinalUsageResponse(id string, createAt int64, model string, usage d
Usage: &usage,
}
}
func ValidUsage(usage *dto.Usage) bool {
return usage != nil && (usage.PromptTokens != 0 || usage.CompletionTokens != 0)
}