diff --git a/args/tiktok.go b/args/tiktok.go index dd68e6f..cfe60fc 100644 --- a/args/tiktok.go +++ b/args/tiktok.go @@ -75,6 +75,11 @@ func (t *TikTokTranscriptionArguments) HasLanguagePreference() bool { return t.Language != "" } +// GetVideoURL returns the source video URL +func (t *TikTokTranscriptionArguments) GetVideoURL() string { + return t.VideoURL +} + // GetLanguageCode returns the language code, defaulting to "en-us" if not specified func (t *TikTokTranscriptionArguments) GetLanguageCode() string { if t.Language == "" { diff --git a/args/unmarshaller.go b/args/unmarshaller.go index e7adf95..353bf8f 100644 --- a/args/unmarshaller.go +++ b/args/unmarshaller.go @@ -39,6 +39,7 @@ type TikTokJobArguments interface { JobArguments ValidateForJobType(jobType types.JobType) error HasLanguagePreference() bool + GetVideoURL() string GetLanguageCode() string } @@ -159,12 +160,9 @@ func (t *TelemetryJobArguments) GetCapability() types.Capability { } // Type assertion helpers -func AsWebArguments(args JobArguments) (WebJobArguments, bool) { +func AsWebArguments(args JobArguments) (*WebSearchArguments, bool) { webArgs, ok := args.(*WebSearchArguments) - if !ok { - return nil, false - } - return webArgs, true + return webArgs, ok } func AsTwitterArguments(args JobArguments) (TwitterJobArguments, bool) {