Added generate code

Author: djthorpe

cmd/agent/generate.go

@@ -26,17 +26,17 @@ type GenerateCmd struct {
 func (cmd *GenerateCmd) Run(globals *Globals) error {
 	return runagent(globals, func(ctx context.Context, client llm.Agent) error {
 		// Get the model
-		agent, ok := client.(*agent.Agent)
+		a, ok := client.(*agent.Agent)
 		if !ok {
 			return fmt.Errorf("No agents found")
 		}
-		model, err := agent.GetModel(ctx, cmd.Model)
+		model, err := a.GetModel(ctx, cmd.Model)
 		if err != nil {
 			return err
 		}
 
 		// Create a session
-		session, err := model.Context(agent.WithStream(!cmd.NoStream))
+		session := model.Context(agent.WithStream(!cmd.NoStream))
 		if err != nil {
 			return err
 		}
@@ -48,21 +48,23 @@ func (cmd *GenerateCmd) Run(globals *Globals) error {
 				return nil
 			} else if err != nil {
 				return err
-			} else if err := session.AppendUserPrompt(strings.TrimSpace(input)); err != nil {
-				return err
 			}
 
-			// Ignore empty import
-			if session.Text() == "" {
+			// Ignore empty input
+			input = strings.TrimSpace(input)
+			if input == "" {
 				continue
 			}
 
 			// Feed input into the model
-			response, err := agent.Generate(ctx, model, session)
+			response, err := session.FromUser(ctx, input)
 			if err != nil {
 				return err
 			}
 			fmt.Println(response.Text())
+
+			// Update session
+			session = response
 		}
 	})
 }