Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
refactor stream cmd
  • Loading branch information
calvinmvrk committed Nov 19, 2024
commit cfbacdbf988ee6494c4f3bd0f98b54cfaefec666
13 changes: 9 additions & 4 deletions cmd/stream.go
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Comment thread
calvinmvrk marked this conversation as resolved.
Outdated
}

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 {
Expand Down