diff --git a/cmd/rootCmd.go b/cmd/rootCmd.go index 64b6c1a..f7cb407 100644 --- a/cmd/rootCmd.go +++ b/cmd/rootCmd.go @@ -9,11 +9,9 @@ import ( ) var rootCmd = &cobra.Command{ - Use: "gh-debug-cli", Short: "A CLI tool for debugging", Long: `This CLI tool allows you to debug your agent by chatting with it locally.`, Run: func(cmd *cobra.Command, args []string) { - fmt.Println("Use 'gh-debug-cli --help' to see available commands") }, } diff --git a/cmd/stream.go b/cmd/stream.go index b0585dc..1d90fda 100644 --- a/cmd/stream.go +++ b/cmd/stream.go @@ -15,9 +15,9 @@ 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, } @@ -28,7 +28,11 @@ func init() { 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 {