Consolidated a bunch of reused code to agents

cmd/agent/cmd.go

@@ -7,14 +7,8 @@ import (
 	"os"
 	"strings"
 
-	knowledge2 "gitea.stevedudenhoeffer.com/steve/answer/pkg/agents"
-
+	"gitea.stevedudenhoeffer.com/steve/answer/pkg/agents"
 	gollm "gitea.stevedudenhoeffer.com/steve/go-llm"
-
-	"gitea.stevedudenhoeffer.com/steve/answer/pkg/agents/shared"
-
-	"gitea.stevedudenhoeffer.com/steve/answer/pkg/agents/searcher"
-
 	"github.com/joho/godotenv"
 	"github.com/urfave/cli"
 )
@@ -102,18 +96,9 @@ func main() {
 			}
 			question := strings.Join(c.Args(), " ")
 
-			search := searcher.Agent{
-				Model: m,
+			agent := agents.NewAgent(m, nil).WithMaxCalls(200)
 
-				OnDone: func(ctx context.Context, knowledge shared.Knowledge) error {
-					slog.Info("done", "knowledge", knowledge)
-					return nil
-				},
-				MaxReads: 20,
-			}
-
-			processor := knowledge2.KnowledgeProcessor{Model: m}
-			knowledge, err := search.Search(ctx, question, question)
+			knowledge, err := agent.SearchAndRead(ctx, question, []string{question}, true, 10)
 
 			if err != nil {
 				panic(err)
@@ -121,9 +106,12 @@ func main() {
 
 			slog.Info("knowledge", "knowledge", knowledge)
 
-			sum, err := processor.Process(ctx, knowledge)
+			res, err := agent.AnswerQuestionWithKnowledge(ctx, knowledge)
+			if err != nil {
+				panic(err)
+			}
 
-			fmt.Println(sum)
+			fmt.Println(res)
 			return nil
 		},
 	}