34 lines
609 B
Go
34 lines
609 B
Go
package state
|
|
|
|
import (
|
|
. "github.com/mudler/LocalAGI/core/agent"
|
|
)
|
|
|
|
type AgentPoolInternalAPI struct {
|
|
*AgentPool
|
|
}
|
|
|
|
func (a *AgentPool) InternalAPI() *AgentPoolInternalAPI {
|
|
return &AgentPoolInternalAPI{a}
|
|
}
|
|
|
|
func (a *AgentPoolInternalAPI) GetAgent(name string) *Agent {
|
|
return a.agents[name]
|
|
}
|
|
|
|
func (a *AgentPoolInternalAPI) AllAgents() []string {
|
|
var agents []string
|
|
for agent := range a.agents {
|
|
agents = append(agents, agent)
|
|
}
|
|
return agents
|
|
}
|
|
|
|
func (a *AgentPoolInternalAPI) GetConfig(name string) *AgentConfig {
|
|
agent, exists := a.pool[name]
|
|
if !exists {
|
|
return nil
|
|
}
|
|
return &agent
|
|
}
|