refactor(documentor): embed Agent

agents/documentor/documentor.go

@@ -10,7 +10,7 @@ import (
 )
 
 type Documentor struct {
-	inner agent.Agent
+	agent.Agent
 }
 
 // NewDocumentor returns a Documentor agent.
@@ -42,10 +42,5 @@ func NewDocumentor(ctx context.Context, cfg *Config, model model.LLM) (*Document
 		return nil, err
 	}
 
-	return &Documentor{inner: da}, nil
-}
-
-// Agent returns the inner agent interface (higher abstraction may not be necessary but we will see).
-func (d *Documentor) Agent() agent.Agent {
-	return d.inner
+	return &Documentor{Agent: da}, nil
 }

cmd/documentor.go

@@ -87,14 +87,14 @@ func NewDocumentorCmd() *cobra.Command {
 
 			slog.Info(
 				"created agent",
-				"agent_name", docAgent.Agent().Name(),
-				"agent_description", docAgent.Agent().Description(),
+				"agent_name", docAgent.Name(),
+				"agent_description", docAgent.Description(),
 			)
 
 			sessService := session.InMemoryService()
 			r, err := runner.New(runner.Config{
 				AppName:        "documentor",
-				Agent:          docAgent.Agent(),
+				Agent:          docAgent,
 				SessionService: sessService,
 			})
 			if err != nil {

tools/tools.go

@@ -28,16 +28,12 @@ func GetToolByEnum(kind Kind, deps *Deps) (tool.Tool, error) {
 			return nil, fmt.Errorf("fetch_repo_tree requires WorkDir")
 		}
 		return NewFetchRepoTreeTool(cfg.WorkDir)
-
 	case ReadFile:
 		return NewReadFileTool()
-
 	case SearchRepo:
 		return NewSearchRepoTool()
-
 	case WriteFile:
 		return NewWriteFileTool()
-
 	default:
 		return nil, fmt.Errorf("invalid tool kind: %q", kind)
 	}