alex/LocalAGI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

allow to set configs for actions too

Browse Source
This commit is contained in:
Ettore Di Giacinto
2024-04-08 22:58:53 +02:00
parent 66b1847644
commit cf3c4549da
3 changed files with 61 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
example/webui/agentpool.go
View File

@@ -18,7 +18,10 @@ type ConnectorConfig struct {
Config string `json:"config"`
}
type ActionsConfig string
type ActionsConfig struct {
Name string `json:"name"` // e.g. search
Config string `json:"config"`
}
type AgentConfig struct {
Connector []ConnectorConfig `json:"connectors" form:"connectors" `
@@ -123,15 +126,19 @@ var AvailableActions = []string{ActionSearch}
func (a *AgentConfig) availableActions() []Action {
actions := []Action{}
if len(a.Actions) == 0 {
// Return search as default
return []Action{external.NewSearch(3)}
}
for _, action := range a.Actions {
fmt.Println("Set Action", action)
switch action {
var config map[string]string
if err := json.Unmarshal([]byte(action.Config), &config); err != nil {
fmt.Println("Error unmarshalling connector config", err)
continue
}
fmt.Println("Config", config)
switch action.Name {
case ActionSearch:
actions = append(actions, external.NewSearch(3))
actions = append(actions, external.NewSearch(config))
}
}
@@ -151,14 +158,16 @@ func (a *AgentConfig) availableConnectors() []Connector {
for _, c := range a.Connector {
fmt.Println("Set Connector", c)
var config map[string]string
if err := json.Unmarshal([]byte(c.Config), &config); err != nil {
fmt.Println("Error unmarshalling connector config", err)
continue
}
fmt.Println("Config", config)
switch c.Type {
case ConnectorTelegram:
var config map[string]string
if err := json.Unmarshal([]byte(c.Config), &config); err != nil {
fmt.Println("Error unmarshalling connector config", err)
continue
}
fmt.Println("Config", config)
cc, err := connector.NewTelegramConnector(config)
if err != nil {
fmt.Println("Error creating telegram connector", err)