m3tam3re/fabric
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use default language param default:"" to avoid changes in the output.

Browse Source
This commit is contained in:
csquarechen 2024-09-14 10:49:55 +08:00
parent b96415d445
commit d574670084
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
cli/flags.go
View File

@ -40,7 +40,7 @@ type Flags struct {
YouTubeTranscript bool `long:"transcript" description:"Grab transcript from YouTube video and send to chat"` YouTubeTranscript bool `long:"transcript" description:"Grab transcript from YouTube video and send to chat"`
YouTubeComments bool `long:"comments" description:"Grab comments from YouTube video and send to chat"` YouTubeComments bool `long:"comments" description:"Grab comments from YouTube video and send to chat"`
DryRun bool `long:"dry-run" description:"Show what would be sent to the model without actually sending it"` DryRun bool `long:"dry-run" description:"Show what would be sent to the model without actually sending it"`
Language string `short:"g" long:"language" description:"Language of the chat" default:"en"` Language string `short:"g" long:"language" description:"Language of the chat" default:""`
} }
// Init Initialize flags. returns a Flags struct and an error // Init Initialize flags. returns a Flags struct and an error
@ -107,10 +107,11 @@ func (o *Flags) BuildChatRequest() (ret *common.ChatRequest) {
PatternVariables: o.PatternVariables, PatternVariables: o.PatternVariables,
Message: o.Message, Message: o.Message,
} }
langTag, err := language.Parse(o.Language) if o.Language != "" {
if err != nil { langTag, err := language.Parse(o.Language)
langTag = language.English if err == nil {
ret.Language = langTag.String()
}
} }
ret.Language = langTag.String()
return return
} }