1package agent
 2
 3import (
 4	"context"
 5	"errors"
 6
 7	"github.com/kujtimiihoxha/termai/internal/app"
 8	"github.com/kujtimiihoxha/termai/internal/config"
 9	"github.com/kujtimiihoxha/termai/internal/llm/models"
10	"github.com/kujtimiihoxha/termai/internal/llm/tools"
11)
12
13type taskAgent struct {
14	*agent
15}
16
17func (c *taskAgent) Generate(ctx context.Context, sessionID string, content string) error {
18	return c.generate(ctx, sessionID, content)
19}
20
21func NewTaskAgent(app *app.App) (Agent, error) {
22	model, ok := models.SupportedModels[config.Get().Model.Coder]
23	if !ok {
24		return nil, errors.New("model not supported")
25	}
26
27	agentProvider, titleGenerator, err := getAgentProviders(app.Context, model)
28	if err != nil {
29		return nil, err
30	}
31	return &taskAgent{
32		agent: &agent{
33			App: app,
34			tools: []tools.BaseTool{
35				tools.NewGlobTool(),
36				tools.NewGrepTool(),
37				tools.NewLsTool(),
38				tools.NewSourcegraphTool(),
39				tools.NewViewTool(app.LSPClients),
40			},
41			model:          model,
42			agent:          agentProvider,
43			titleGenerator: titleGenerator,
44		},
45	}, nil
46}