Use internal API for services/actions when using the pool

Signed-off-by: mudler <mudler@localai.io>
This commit is contained in:
mudler
2025-03-28 16:12:42 +01:00
parent 2c273392cd
commit 05af5d9695
4 changed files with 37 additions and 3 deletions

33
core/state/internal.go Normal file
View File

@@ -0,0 +1,33 @@
package state
import (
. "github.com/mudler/LocalAgent/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
}