persona

Overview ¶

Package persona provides functionality for defining and managing different LLM personas.

Overview ¶

The Persona system provides a way to define different roles or personalities for the LLM agent, each with its own set of tools and system prompts. A persona defines:

1. A set of tools available to the LLM
2. A system prompt that guides the LLM's behavior
3. A unique identifier and descriptive name

Usage ¶

Selecting a Persona:

You can select a persona when starting the chat application using the --persona flag:

aigent chat --persona software_engineer

Available Personas:

Currently, the following personas are available:

• Software Engineer (software_engineer): A persona focused on software engineering tasks, providing tools for code analysis, development, and debugging.

Creating a New Persona ¶

To create a new persona:

1. Implement the Persona interface defined in persona.go
2. Register the persona in the persona registry

Example implementation:

// MyNewPersona represents a custom persona for the LLM agent.
type MyNewPersona struct {
    tools     []llm.Tool
    executors map[string]llm.ToolExecutor
    logger    *slog.Logger
}

// NewMyNewPersona creates a new instance of the custom persona.
func NewMyNewPersona(logger *slog.Logger) *MyNewPersona {
    // Initialize with tools and executors
    // ...
    return persona
}

// Implement the Persona interface methods
// ...

// AddMCPTools adds tools from MCP servers to this persona.
func (p *MyNewPersona) AddMCPTools(mcpManager *mcp.Manager) {
    // Add MCP tools as needed
    // ...
}

Then register it in the persona registry:

// SetupPersonaRegistry creates and configures a persona registry with available personas.
func SetupPersonaRegistry(logger *slog.Logger) *persona.Registry {
    registry := persona.NewRegistry(
        persona.WithRegistryLogger(logger),
        persona.WithPersona(persona.NewSoftwareEngineer(logger)),
        persona.WithPersona(persona.NewMyNewPersona(logger)), // Add your new persona here
    )
    return registry
}

MCP Tools Support ¶

Personas can include tools from MCP (Machine Cognition Protocol) servers. The system works as follows:

1. All MCP tools defined in aigent.yaml are automatically added to every persona
2. Additional hard-coded MCP tools can be specified per persona using the GetMCPToolSpecs() method
3. These hard-coded tools would typically be from servers not defined in aigent.yaml

Example of defining additional MCP tools for a persona:

// GetMCPToolSpecs returns the MCP tool specifications for this persona.
func (p *MyPersona) GetMCPToolSpecs() []MCPToolSpec {
    return []MCPToolSpec{
        {
            // Additional server not in aigent.yaml
            ServerName: "specialized-tools",
        },
        {
            // Another additional server with specific tools
            ServerName: "advanced-tools",
            ToolNames:  []string{"special-tool-1", "special-tool-2"},
        },
    }
}

Architecture ¶

The persona system consists of:

• Persona interface: Defines the contract for all personas
• Registry: Manages available personas and provides access to them
• Integration with the chat system to use the selected persona's tools and system prompt

The Software Engineer persona is provided as the default implementation, which includes:

• A hardcoded "buildin" MCP server with tools like read_file and run_command
• Support for automatically adding all tools from MCP servers defined in aigent.yaml
• A system prompt tailored for software engineering tasks

Package persona provides functionality for defining and managing different LLM personas.