diff --git a/cmd/stream.go b/cmd/stream.go index b0585dc..3083a44 100644 --- a/cmd/stream.go +++ b/cmd/stream.go @@ -15,20 +15,25 @@ const ( // streamCmd represents the new command for streaming functionality var streamCmd = &cobra.Command{ - Use: "stream [file]", - Short: "Stream data to your agent", - Long: `The stream command allows you to initiate a data stream to your agent.`, + Use: "stream --file [filename]", + Short: "Parse stream data from agent", + Long: `Allows you to parse a data stream to your agent response.`, Run: agentStream, } func init() { streamCmd.PersistentFlags().String(streamCmdFileFlag, "", "Parse agent responses from a file") + rootCmd.AddCommand(streamCmd) } func agentStream(cmd *cobra.Command, args []string) { fmt.Println("stream command executed successfully") - file := args[0] + file, _ := cmd.Flags().GetString(streamCmdFileFlag) + if file == "" { + fmt.Fprintln(os.Stderr, "Error: --file [file] is required") + os.Exit(1) + } result, err := stream.ParseFile(file) if err != nil {