refactor: update variable names and improve code readability; add Makefile for linting and installation

This commit is contained in:
Laisky.Cai
2025-01-23 03:32:41 +00:00
parent 81153f764b
commit 531f832d1e
12 changed files with 659 additions and 60 deletions

View File

@@ -33,7 +33,7 @@ func SaveTmpFile(filename string, data io.Reader) (string, error) {
}
// GetAudioTokens returns the number of tokens in an audio file.
func GetAudioTokens(ctx context.Context, audio io.Reader, tokensPerSecond int) (int, error) {
func GetAudioTokens(ctx context.Context, audio io.Reader, tokensPerSecond float64) (int, error) {
filename, err := SaveTmpFile("audio", audio)
if err != nil {
return 0, errors.Wrap(err, "failed to save audio to temporary file")
@@ -45,7 +45,7 @@ func GetAudioTokens(ctx context.Context, audio io.Reader, tokensPerSecond int) (
return 0, errors.Wrap(err, "failed to get audio tokens")
}
return int(math.Ceil(duration)) * tokensPerSecond, nil
return int(math.Ceil(duration * tokensPerSecond)), nil
}
// GetAudioDuration returns the duration of an audio file in seconds.