refactor: 重构流模式逻辑

2024-07-15 18:04:05 +08:00
parent 7b20b4ed3d
commit 917c16903c
4 changed files with 114 additions and 123 deletions
@@ -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)
+}