Use internal API for services/actions when using the pool
Signed-off-by: mudler <mudler@localai.io>
This commit is contained in:
33
core/state/internal.go
Normal file
33
core/state/internal.go
Normal 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
|
||||
}
|
||||
@@ -591,6 +591,7 @@ func (a *AgentPool) save() error {
|
||||
}
|
||||
return os.WriteFile(a.file, data, 0644)
|
||||
}
|
||||
|
||||
func (a *AgentPool) GetAgent(name string) *Agent {
|
||||
a.Lock()
|
||||
defer a.Unlock()
|
||||
|
||||
Reference in New Issue
Block a user