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

feat(edit): allow to edit agents (#36)

Browse Source
This commit is contained in:
Ettore Di Giacinto
2025-03-11 22:32:13 +01:00
committed by GitHub
parent 9ff2fde44f
commit d451919414
6 changed files with 869 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

96
core/state/pool.go
View File

@@ -21,18 +21,18 @@ import (
type AgentPool struct { type AgentPool struct {
sync.Mutex sync.Mutex
file string file string
pooldir string pooldir string
pool AgentPoolData pool AgentPoolData
agents map[string]*Agent agents map[string]*Agent
managers map[string]sse.Manager managers map[string]sse.Manager
agentStatus map[string]*Status agentStatus map[string]*Status
apiURL, model, multimodalModel, localRAGAPI, apiKey string apiURL, defaultModel, defaultMultimodalModel, localRAGAPI, apiKey string
availableActions func(*AgentConfig) func(ctx context.Context) []Action availableActions func(*AgentConfig) func(ctx context.Context) []Action
connectors func(*AgentConfig) []Connector connectors func(*AgentConfig) []Connector
promptBlocks func(*AgentConfig) []PromptBlock promptBlocks func(*AgentConfig) []PromptBlock
timeout string timeout string
conversationLogs string conversationLogs string
} }
type Status struct { type Status struct {
@@ -66,7 +66,7 @@ func loadPoolFromFile(path string) (*AgentPoolData, error) {
} }
func NewAgentPool( func NewAgentPool(
model, multimodalModel, apiURL, apiKey, directory string, defaultModel, defaultMultimodalModel, apiURL, apiKey, directory string,
LocalRAGAPI string, LocalRAGAPI string,
availableActions func(*AgentConfig) func(ctx context.Context) []agent.Action, availableActions func(*AgentConfig) func(ctx context.Context) []agent.Action,
connectors func(*AgentConfig) []Connector, connectors func(*AgentConfig) []Connector,
@@ -87,22 +87,22 @@ func NewAgentPool(
if _, err := os.Stat(poolfile); err != nil { if _, err := os.Stat(poolfile); err != nil {
// file does not exist, create a new pool // file does not exist, create a new pool
return &AgentPool{ return &AgentPool{
file: poolfile, file: poolfile,
pooldir: directory, pooldir: directory,
apiURL: apiURL, apiURL: apiURL,
model: model, defaultModel: defaultModel,
multimodalModel: multimodalModel, defaultMultimodalModel: defaultMultimodalModel,
localRAGAPI: LocalRAGAPI, localRAGAPI: LocalRAGAPI,
apiKey: apiKey, apiKey: apiKey,
agents: make(map[string]*Agent), agents: make(map[string]*Agent),
pool: make(map[string]AgentConfig), pool: make(map[string]AgentConfig),
agentStatus: make(map[string]*Status), agentStatus: make(map[string]*Status),
managers: make(map[string]sse.Manager), managers: make(map[string]sse.Manager),
connectors: connectors, connectors: connectors,
availableActions: availableActions, availableActions: availableActions,
promptBlocks: promptBlocks, promptBlocks: promptBlocks,
timeout: timeout, timeout: timeout,
conversationLogs: conversationPath, conversationLogs: conversationPath,
}, nil }, nil
} }
@@ -111,22 +111,22 @@ func NewAgentPool(
return nil, err return nil, err
} }
return &AgentPool{ return &AgentPool{
file: poolfile, file: poolfile,
apiURL: apiURL, apiURL: apiURL,
pooldir: directory, pooldir: directory,
model: model, defaultModel: defaultModel,
multimodalModel: multimodalModel, defaultMultimodalModel: defaultMultimodalModel,
apiKey: apiKey, apiKey: apiKey,
agents: make(map[string]*Agent), agents: make(map[string]*Agent),
managers: make(map[string]sse.Manager), managers: make(map[string]sse.Manager),
agentStatus: map[string]*Status{}, agentStatus: map[string]*Status{},
pool: *poolData, pool: *poolData,
connectors: connectors, connectors: connectors,
localRAGAPI: LocalRAGAPI, localRAGAPI: LocalRAGAPI,
promptBlocks: promptBlocks, promptBlocks: promptBlocks,
availableActions: availableActions, availableActions: availableActions,
timeout: timeout, timeout: timeout,
conversationLogs: conversationPath, conversationLogs: conversationPath,
}, nil }, nil
} }
@@ -166,8 +166,8 @@ func (a *AgentPool) GetStatusHistory(name string) *Status {
func (a *AgentPool) startAgentWithConfig(name string, config *AgentConfig) error { func (a *AgentPool) startAgentWithConfig(name string, config *AgentConfig) error {
manager := sse.NewManager(5) manager := sse.NewManager(5)
ctx := context.Background() ctx := context.Background()
model := a.model model := a.defaultModel
multimodalModel := a.multimodalModel multimodalModel := a.defaultMultimodalModel
if config.MultimodalModel != "" { if config.MultimodalModel != "" {
multimodalModel = config.MultimodalModel multimodalModel = config.MultimodalModel
} }
@@ -340,7 +340,7 @@ func (a *AgentPool) startAgentWithConfig(name string, config *AgentConfig) error
go func() { go func() {
if err := agent.Run(); err != nil { if err := agent.Run(); err != nil {
xlog.Error("Agent stopped", "error", err.Error()) xlog.Error("Agent stopped", "error", err.Error(), "name", name)
panic(err) panic(err)
} }
}() }()

20
services/connectors/irc.go
View File

@@ -8,7 +8,7 @@ import (
"github.com/mudler/LocalAgent/core/agent" "github.com/mudler/LocalAgent/core/agent"
"github.com/mudler/LocalAgent/pkg/xlog" "github.com/mudler/LocalAgent/pkg/xlog"
"github.com/mudler/LocalAgent/services/actions" "github.com/mudler/LocalAgent/services/actions"
"github.com/thoj/go-ircevent" irc "github.com/thoj/go-ircevent"
) )
type IRC struct { type IRC struct {
@@ -53,21 +53,25 @@ func cleanUpMessage(message string, nickname string) string {
// isMentioned checks if the bot is mentioned in the message // isMentioned checks if the bot is mentioned in the message
func isMentioned(message string, nickname string) bool { func isMentioned(message string, nickname string) bool {
return strings.Contains(message, nickname+":") || return strings.Contains(message, nickname+":") ||
strings.Contains(message, nickname+",") || strings.Contains(message, nickname+",") ||
strings.HasPrefix(message, nickname) strings.HasPrefix(message, nickname)
} }
// Start connects to the IRC server and starts listening for messages // Start connects to the IRC server and starts listening for messages
func (i *IRC) Start(a *agent.Agent) { func (i *IRC) Start(a *agent.Agent) {
i.conn = irc.IRC(i.nickname, i.nickname) i.conn = irc.IRC(i.nickname, i.nickname)
i.conn.UseTLS = false if i.conn == nil {
xlog.Error("Failed to create IRC client")
return
}
i.conn.UseTLS = false
i.conn.AddCallback("001", func(e *irc.Event) { i.conn.AddCallback("001", func(e *irc.Event) {
xlog.Info("Connected to IRC server", "server", i.server) xlog.Info("Connected to IRC server", "server", i.server)
i.conn.Join(i.channel) i.conn.Join(i.channel)
xlog.Info("Joined channel", "channel", i.channel) xlog.Info("Joined channel", "channel", i.channel)
}) })
i.conn.AddCallback("JOIN", func(e *irc.Event) { i.conn.AddCallback("JOIN", func(e *irc.Event) {
if e.Nick == i.nickname { if e.Nick == i.nickname {
xlog.Info("Bot joined channel", "channel", e.Arguments[0]) xlog.Info("Bot joined channel", "channel", e.Arguments[0])
@@ -128,10 +132,10 @@ func (i *IRC) Start(a *agent.Agent) {
chunk = line chunk = line
line = "" line = ""
} }
// Send the message to the channel // Send the message to the channel
i.conn.Privmsg(channel, chunk) i.conn.Privmsg(channel, chunk)
// Small delay to prevent flooding // Small delay to prevent flooding
time.Sleep(500 * time.Millisecond) time.Sleep(500 * time.Millisecond)
} }

50
webui/app.go
View File

@@ -136,6 +136,56 @@ func (a *App) Create(pool *state.AgentPool) func(c *fiber.Ctx) error {
} }
} }
// NEW FUNCTION: Get agent configuration
func (a *App) GetAgentConfig(pool *state.AgentPool) func(c *fiber.Ctx) error {
return func(c *fiber.Ctx) error {
config := pool.GetConfig(c.Params("name"))
if config == nil {
return errorJSONMessage(c, "Agent not found")
}
return c.JSON(config)
}
}
// UpdateAgentConfig handles updating an agent's configuration
func (a *App) UpdateAgentConfig(pool *state.AgentPool) func(c *fiber.Ctx) error {
return func(c *fiber.Ctx) error {
agentName := c.Params("name")
// First check if agent exists
oldConfig := pool.GetConfig(agentName)
if oldConfig == nil {
return errorJSONMessage(c, "Agent not found")
}
// Parse the new configuration using the same approach as Create
newConfig := state.AgentConfig{}
if err := c.BodyParser(&newConfig); err != nil {
xlog.Error("Error parsing agent config", "error", err)
return errorJSONMessage(c, err.Error())
}
// Ensure the name doesn't change
newConfig.Name = agentName
// Remove the agent first
if err := pool.Remove(agentName); err != nil {
return errorJSONMessage(c, "Error removing agent: "+err.Error())
}
// Create agent with new config
if err := pool.CreateAgent(agentName, &newConfig); err != nil {
// Try to restore the old configuration if update fails
if restoreErr := pool.CreateAgent(agentName, oldConfig); restoreErr != nil {
return errorJSONMessage(c, fmt.Sprintf("Failed to update agent and restore failed: %v, %v", err, restoreErr))
}
return errorJSONMessage(c, "Error updating agent: "+err.Error())
}
return statusJSONMessage(c, "ok")
}
}
func (a *App) ExportAgent(pool *state.AgentPool) func(c *fiber.Ctx) error { func (a *App) ExportAgent(pool *state.AgentPool) func(c *fiber.Ctx) error {
return func(c *fiber.Ctx) error { return func(c *fiber.Ctx) error {
agent := pool.GetConfig(c.Params("name")) agent := pool.GetConfig(c.Params("name"))

13
webui/routes.go
View File

@@ -102,11 +102,18 @@ func (app *App) registerRoutes(pool *state.AgentPool, webapp *fiber.App) {
} }
return c.Render("views/settings", fiber.Map{ return c.Render("views/settings", fiber.Map{
// "Character": agent.Character, "Name": c.Params("name"),
"Name": c.Params("name"), "Status": status,
"Status": status, "Actions": services.AvailableActions,
"Connectors": services.AvailableConnectors,
"PromptBlocks": services.AvailableBlockPrompts,
}) })
}) })
// New API endpoints for getting and updating agent configuration
webapp.Get("/api/agent/:name/config", app.GetAgentConfig(pool))
webapp.Put("/api/agent/:name/config", app.UpdateAgentConfig(pool))
webapp.Post("/settings/import", app.ImportAgent(pool)) webapp.Post("/settings/import", app.ImportAgent(pool))
webapp.Get("/settings/export/:name", app.ExportAgent(pool)) webapp.Get("/settings/export/:name", app.ExportAgent(pool))
} }

143
webui/views/create.html
View File

@@ -78,7 +78,7 @@
<div class="mb-4"> <div class="mb-4">
<label for="kb_results">Knowledge Base Results</label> <label for="kb_results">Knowledge Base Results</label>
<input type="text" name="kb_results" id="kb_results" placeholder="3"> <input type="number" name="kb_results" id="kb_results" placeholder="3">
</div> </div>
<div class="mb-4"> <div class="mb-4">
@@ -278,19 +278,152 @@
form.addEventListener('submit', function(e) { form.addEventListener('submit', function(e) {
e.preventDefault(); e.preventDefault();
// Show a loading state // Show loading state
const createButton = document.getElementById('create-button'); const createButton = document.getElementById('create-button');
const originalButtonText = createButton.innerHTML; const originalButtonText = createButton.innerHTML;
createButton.innerHTML = '<i class="fas fa-spinner fa-spin"></i> Creating...'; createButton.innerHTML = '<i class="fas fa-spinner fa-spin"></i> Creating...';
createButton.disabled = true; createButton.disabled = true;
// Get form data // Build a structured data object
const formData = new FormData(form); const formData = new FormData(form);
const jsonData = {};
// Send the form data using fetch API // Process basic form fields
for (const [key, value] of formData.entries()) {
// Skip the array fields (connectors, actions, promptblocks) as they'll be processed separately
if (!key.includes('[') && !key.includes('].')) {
// Handle checkboxes
if (value === 'on') {
jsonData[key] = true;
}
// Handle numeric fields - specifically kb_results
else if (key === 'kb_results') {
// Convert to integer or default to 3 if empty
jsonData[key] = value ? parseInt(value, 10) : 3;
// Check if the parse was successful
if (isNaN(jsonData[key])) {
showToast('Knowledge Base Results must be a number', 'error');
createButton.innerHTML = originalButtonText;
createButton.disabled = false;
return; // Stop form submission
}
}
// Handle other numeric fields if needed
else if (key === 'periodic_runs' && value) {
// Try to parse as number if it looks like one
const numValue = parseInt(value, 10);
if (!isNaN(numValue) && String(numValue) === value) {
jsonData[key] = numValue;
} else {
jsonData[key] = value;
}
}
else {
jsonData[key] = value;
}
}
}
// Process connectors - KEEP CONFIG AS STRING
const connectors = [];
const connectorsElements = document.getElementsByClassName('connector');
for (let i = 0; i < connectorsElements.length; i++) {
const typeField = document.getElementById(`connectorType${i}`);
const configField = document.getElementById(`connectorConfig${i}`);
if (typeField && configField) {
try {
// Validate JSON but send as string
const configValue = configField.value.trim() || '{}';
// Parse to validate but don't use the parsed object
JSON.parse(configValue);
connectors.push({
type: typeField.value,
config: configValue // Send the raw string, not parsed JSON
});
} catch (err) {
console.error(`Error parsing connector ${i} config:`, err);
showToast(`Error in connector ${i+1} configuration: Invalid JSON`, 'error');
createButton.innerHTML = originalButtonText;
createButton.disabled = false;
return; // Stop form submission
}
}
}
jsonData.connectors = connectors;
// Process actions - KEEP CONFIG AS STRING
const actions = [];
const actionElements = document.getElementsByClassName('action');
for (let i = 0; i < actionElements.length; i++) {
const nameField = document.getElementById(`actionsName${i}`);
const configField = document.getElementById(`actionsConfig${i}`);
if (nameField && configField) {
try {
// Validate JSON but send as string
const configValue = configField.value.trim() || '{}';
// Parse to validate but don't use the parsed object
JSON.parse(configValue);
actions.push({
name: nameField.value,
config: configValue // Send the raw string, not parsed JSON
});
} catch (err) {
console.error(`Error parsing action ${i} config:`, err);
showToast(`Error in action ${i+1} configuration: Invalid JSON`, 'error');
createButton.innerHTML = originalButtonText;
createButton.disabled = false;
return; // Stop form submission
}
}
}
jsonData.actions = actions;
// Process prompt blocks - KEEP CONFIG AS STRING
const promptBlocks = [];
const promptBlockElements = document.getElementsByClassName('promptBlock');
for (let i = 0; i < promptBlockElements.length; i++) {
const nameField = document.getElementById(`promptName${i}`);
const configField = document.getElementById(`promptConfig${i}`);
if (nameField && configField) {
try {
// Validate JSON but send as string
const configValue = configField.value.trim() || '{}';
// Parse to validate but don't use the parsed object
JSON.parse(configValue);
promptBlocks.push({
name: nameField.value,
config: configValue // Send the raw string, not parsed JSON
});
} catch (err) {
console.error(`Error parsing prompt block ${i} config:`, err);
showToast(`Error in prompt block ${i+1} configuration: Invalid JSON`, 'error');
createButton.innerHTML = originalButtonText;
createButton.disabled = false;
return; // Stop form submission
}
}
}
jsonData.promptblocks = promptBlocks;
console.log('Sending data:', jsonData);
// Send the structured data as JSON
fetch('/create', { fetch('/create', {
method: 'POST', method: 'POST',
body: formData headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify(jsonData)
}) })
.then(response => response.json()) .then(response => response.json())
.then(data => { .then(data => {

614
webui/views/settings.html
View File

@@ -7,11 +7,11 @@
<body> <body>
{{template "views/partials/menu"}} {{template "views/partials/menu"}}
<!-- Toast for notifications --> <!-- Toast notification container -->
<div id="toast" class="toast"> <div id="toast" class="toast">
<span id="toast-message"></span> <span id="toast-message"></span>
</div> </div>
<div class="container"> <div class="container">
<header class="text-center mb-8"> <header class="text-center mb-8">
<h1 class="text-3xl md:text-5xl font-bold">Agent settings - {{.Name}}</h1> <h1 class="text-3xl md:text-5xl font-bold">Agent settings - {{.Name}}</h1>
@@ -34,6 +34,173 @@
</div> </div>
</div> </div>
<!-- New Agent Configuration Form Section -->
<div class="section-box">
<h2>Edit Agent Configuration</h2>
<form id="edit-agent-form">
<input type="hidden" name="name" id="name" value="{{.Name}}">
<div id="connectorsSection">
<!-- Connectors will be added here dynamically -->
</div>
<div class="button-container">
<button type="button" id="addConnectorButton" class="action-btn">
<i class="fas fa-plus-circle"></i> Add Connector
</button>
</div>
<div class="mb-4" id="action_box">
<!-- Actions will be added here dynamically -->
</div>
<div class="button-container">
<button id="action_button" type="button" class="action-btn">
<i class="fas fa-plus-circle"></i> Add Action
</button>
</div>
<div class="mb-4" id="dynamic_box">
<!-- Dynamic prompts will be added here dynamically -->
</div>
<div class="button-container">
<button id="dynamic_button" type="button" class="action-btn">
<i class="fas fa-plus-circle"></i> Add Dynamic Prompt
</button>
</div>
<div class="mb-4">
<label for="hud" class="checkbox-label">
<span class="checkbox-custom">
<input type="checkbox" name="hud" id="hud">
<span class="checkmark"></span>
</span>
HUD
</label>
</div>
<div class="mb-4">
<label for="enable_kb" class="checkbox-label">
<span class="checkbox-custom">
<input type="checkbox" name="enable_kb" id="enable_kb">
<span class="checkmark"></span>
</span>
Enable Knowledge Base
</label>
</div>
<div class="mb-4">
<label for="enable_reasoning" class="checkbox-label">
<span class="checkbox-custom">
<input type="checkbox" name="enable_reasoning" id="enable_reasoning">
<span class="checkmark"></span>
</span>
Enable Reasoning
</label>
</div>
<div class="mb-4">
<label for="kb_results">Knowledge Base Results</label>
<input type="number" name="kb_results" id="kb_results" placeholder="3">
</div>
<div class="mb-4">
<label for="standalone_job" class="checkbox-label">
<span class="checkbox-custom">
<input type="checkbox" name="standalone_job" id="standalone_job">
<span class="checkmark"></span>
</span>
Standalone Job
</label>
</div>
<div class="mb-4">
<label for="initiate_conversations" class="checkbox-label">
<span class="checkbox-custom">
<input type="checkbox" name="initiate_conversations" id="initiate_conversations">
<span class="checkmark"></span>
</span>
Initiate Conversations
</label>
</div>
<div class="mb-4">
<label for="can_stop_itself" class="checkbox-label">
<span class="checkbox-custom">
<input type="checkbox" name="can_stop_itself" id="can_stop_itself">
<span class="checkmark"></span>
</span>
Can Stop Itself
</label>
</div>
<div class="mb-4">
<label for="random_identity" class="checkbox-label">
<span class="checkbox-custom">
<input type="checkbox" name="random_identity" id="random_identity">
<span class="checkmark"></span>
</span>
Random Identity
</label>
</div>
<div class="mb-4">
<label for="long_term_memory" class="checkbox-label">
<span class="checkbox-custom">
<input type="checkbox" name="long_term_memory" id="long_term_memory">
<span class="checkmark"></span>
</span>
Long Term Memory
</label>
</div>
<div class="mb-4">
<label for="summary_long_term_memory" class="checkbox-label">
<span class="checkbox-custom">
<input type="checkbox" name="summary_long_term_memory" id="summary_long_term_memory">
<span class="checkmark"></span>
</span>
Long Term Memory (Summarize!)
</label>
</div>
<div class="mb-4">
<label for="identity_guidance">Identity Guidance</label>
<textarea name="identity_guidance" id="identity_guidance" placeholder="Identity Guidance"></textarea>
</div>
<div class="mb-4">
<label for="periodic_runs">Periodic Runs</label>
<input type="text" name="periodic_runs" id="periodic_runs" placeholder="Periodic Runs">
</div>
<div class="mb-4">
<label for="model">Model </label>
<input type="text" name="model" id="model" placeholder="Model name">
</div>
<div class="mb-4">
<label for="multimodal_model">Multimodal Model </label>
<input type="text" name="multimodal_model" id="multimodal_model" placeholder="Model name">
</div>
<div class="mb-4">
<label for="permanent_goal">Permanent Goal</label>
<textarea name="permanent_goal" id="permanent_goal" placeholder="Permanent goal"></textarea>
</div>
<div class="mb-4">
<label for="system_prompt">System Prompt</label>
<textarea name="system_prompt" id="system_prompt" placeholder="System prompt"></textarea>
</div>
<button type="submit" id="update-button" class="action-btn">
<i class="fas fa-save"></i> Update Agent
</button>
</form>
</div>
<div class="section-box"> <div class="section-box">
<h2>Export Data</h2> <h2>Export Data</h2>
<p class="mb-4">Export your agent configuration for backup or transfer.</p> <p class="mb-4">Export your agent configuration for backup or transfer.</p>
@@ -66,7 +233,15 @@
</div> </div>
<script> <script>
const actions = `{{ range .Actions }}<option value="{{.}}">{{.}}</option>{{ end }}`;
const connectors = `{{ range .Connectors }}<option value="{{.}}">{{.}}</option>{{ end }}`;
const promptBlocks = `{{ range .PromptBlocks }}<option value="{{.}}">{{.}}</option>{{ end }}`;
let agentConfig = null;
document.addEventListener('DOMContentLoaded', function() { document.addEventListener('DOMContentLoaded', function() {
// Load agent configuration when page loads
loadAgentConfig();
// Add event listener for delete button // Add event listener for delete button
document.querySelectorAll('[data-action="delete"]').forEach(button => { document.querySelectorAll('[data-action="delete"]').forEach(button => {
button.addEventListener('htmx:afterRequest', function(event) { button.addEventListener('htmx:afterRequest', function(event) {
@@ -142,8 +317,271 @@
} }
// Set current date for timestamp // Set current date for timestamp
const now = new Date("2025-03-04T20:52:02Z"); const now = new Date();
document.getElementById('current-date').textContent = now.toISOString().split('T')[0]; document.getElementById('current-date').textContent = now.toISOString().split('T')[0];
// Add connector functionality
document.getElementById('addConnectorButton').addEventListener('click', function() {
const connectorsSection = document.getElementById('connectorsSection');
const newConnectorIndex = connectorsSection.getElementsByClassName('connector').length;
const newConnectorHTML = `
<div class="connector mb-4 section-box" style="margin-top: 15px; padding: 15px;">
<h2>Connector ${newConnectorIndex + 1}</h2>
<div class="mb-4">
<label for="connectorType${newConnectorIndex}">Connector Type</label>
<select name="connectors[${newConnectorIndex}].type" id="connectorType${newConnectorIndex}">
${connectors}
</select>
</div>
<div class="mb-4">
<label for="connectorConfig${newConnectorIndex}">Connector Config (JSON)</label>
<textarea id="connectorConfig${newConnectorIndex}" name="connectors[${newConnectorIndex}].config" placeholder='{"token":"sk-mg3.."}'>{}</textarea>
</div>
</div>
`;
connectorsSection.insertAdjacentHTML('beforeend', newConnectorHTML);
});
// Add action functionality
document.getElementById('action_button').addEventListener('click', function() {
const actionsSection = document.getElementById('action_box');
const ii = actionsSection.getElementsByClassName('action').length;
const newActionHTML = `
<div class="action mb-4 section-box" style="margin-top: 15px; padding: 15px;">
<h2>Action ${ii + 1}</h2>
<div class="mb-4">
<label for="actionsName${ii}">Action</label>
<select name="actions[${ii}].name" id="actionsName${ii}">
${actions}
</select>
</div>
<div class="mb-4">
<label for="actionsConfig${ii}">Action Config (JSON)</label>
<textarea id="actionsConfig${ii}" name="actions[${ii}].config" placeholder='{"results":"5"}'>{}</textarea>
</div>
</div>
`;
actionsSection.insertAdjacentHTML('beforeend', newActionHTML);
});
// Add dynamic prompt functionality
document.getElementById('dynamic_button').addEventListener('click', function() {
const promptsSection = document.getElementById('dynamic_box');
const ii = promptsSection.getElementsByClassName('promptBlock').length;
const newPromptHTML = `
<div class="promptBlock mb-4 section-box" style="margin-top: 15px; padding: 15px;">
<h2>Prompt Block ${ii + 1}</h2>
<div class="mb-4">
<label for="promptName${ii}">Block Prompt</label>
<select name="promptblocks[${ii}].name" id="promptName${ii}">
${promptBlocks}
</select>
</div>
<div class="mb-4">
<label for="promptConfig${ii}">Prompt Config (JSON)</label>
<textarea id="promptConfig${ii}" name="promptblocks[${ii}].config" placeholder='{"results":"5"}'>{}</textarea>
</div>
</div>
`;
promptsSection.insertAdjacentHTML('beforeend', newPromptHTML);
});
// Form submission handling
const form = document.getElementById('edit-agent-form');
form.addEventListener('submit', function(e) {
e.preventDefault();
// Show a loading state
const updateButton = document.getElementById('update-button');
const originalButtonText = updateButton.innerHTML;
updateButton.innerHTML = '<i class="fas fa-spinner fa-spin"></i> Updating...';
updateButton.disabled = true;
// Build a structured data object
const formData = new FormData(form);
const jsonData = {};
// Process basic form fields
for (const [key, value] of formData.entries()) {
// Skip the array fields (connectors, actions, promptblocks) as they'll be processed separately
if (!key.includes('[') && !key.includes('].')) {
// Handle checkboxes
if (value === 'on') {
jsonData[key] = true;
}
// Handle numeric fields - specifically kb_results
else if (key === 'kb_results') {
// Convert to integer or default to 3 if empty
jsonData[key] = value ? parseInt(value, 10) : 3;
// Check if the parse was successful
if (isNaN(jsonData[key])) {
showToast('Knowledge Base Results must be a number', 'error');
updateButton.innerHTML = originalButtonText;
updateButton.disabled = false;
return; // Stop form submission
}
}
// Handle other numeric fields if needed
else if (key === 'periodic_runs' && value) {
// Try to parse as number if it looks like one
const numValue = parseInt(value, 10);
if (!isNaN(numValue) && String(numValue) === value) {
jsonData[key] = numValue;
} else {
jsonData[key] = value;
}
}
else {
jsonData[key] = value;
}
}
}
// Process connectors - KEEP CONFIG AS STRING
const connectors = [];
const connectorsElements = document.getElementsByClassName('connector');
for (let i = 0; i < connectorsElements.length; i++) {
const typeField = document.getElementById(`connectorType${i}`);
const configField = document.getElementById(`connectorConfig${i}`);
if (typeField && configField) {
try {
// Validate JSON but send as string
const configValue = configField.value.trim() || '{}';
// Parse to validate but don't use the parsed object
JSON.parse(configValue);
connectors.push({
type: typeField.value,
config: configValue // Send the raw string, not parsed JSON
});
} catch (err) {
console.error(`Error parsing connector ${i} config:`, err);
showToast(`Error in connector ${i+1} configuration: Invalid JSON`, 'error');
updateButton.innerHTML = originalButtonText;
updateButton.disabled = false;
return; // Stop form submission
}
}
}
jsonData.connectors = connectors;
// Process actions - KEEP CONFIG AS STRING
const actions = [];
const actionElements = document.getElementsByClassName('action');
for (let i = 0; i < actionElements.length; i++) {
const nameField = document.getElementById(`actionsName${i}`);
const configField = document.getElementById(`actionsConfig${i}`);
if (nameField && configField) {
try {
// Validate JSON but send as string
const configValue = configField.value.trim() || '{}';
// Parse to validate but don't use the parsed object
JSON.parse(configValue);
actions.push({
name: nameField.value,
config: configValue // Send the raw string, not parsed JSON
});
} catch (err) {
console.error(`Error parsing action ${i} config:`, err);
showToast(`Error in action ${i+1} configuration: Invalid JSON`, 'error');
updateButton.innerHTML = originalButtonText;
updateButton.disabled = false;
return; // Stop form submission
}
}
}
jsonData.actions = actions;
// Process prompt blocks - KEEP CONFIG AS STRING
const promptBlocks = [];
const promptBlockElements = document.getElementsByClassName('promptBlock');
for (let i = 0; i < promptBlockElements.length; i++) {
const nameField = document.getElementById(`promptName${i}`);
const configField = document.getElementById(`promptConfig${i}`);
if (nameField && configField) {
try {
// Validate JSON but send as string
const configValue = configField.value.trim() || '{}';
// Parse to validate but don't use the parsed object
JSON.parse(configValue);
promptBlocks.push({
name: nameField.value,
config: configValue // Send the raw string, not parsed JSON
});
} catch (err) {
console.error(`Error parsing prompt block ${i} config:`, err);
showToast(`Error in prompt block ${i+1} configuration: Invalid JSON`, 'error');
updateButton.innerHTML = originalButtonText;
updateButton.disabled = false;
return; // Stop form submission
}
}
}
jsonData.promptblocks = promptBlocks;
console.log('Sending data:', jsonData);
// Send the structured data as JSON
fetch(`/api/agent/${jsonData.name}/config`, {
method: 'PUT',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify(jsonData)
})
.then(response => {
if (!response.ok) {
return response.json().then(err => {
throw new Error(err.error || `Server error: ${response.status}`);
});
}
return response.json();
})
.then(data => {
// Restore button state
updateButton.innerHTML = originalButtonText;
updateButton.disabled = false;
if (data.status === "ok") {
// Show success toast
showToast('Agent updated successfully!', 'success');
// Reload agent config to get updated values
setTimeout(() => {
loadAgentConfig();
}, 500);
} else if (data.error) {
// Show error toast
showToast('Error: ' + data.error, 'error');
} else {
// Handle unexpected response format
showToast('Unexpected response format', 'error');
}
})
.catch(error => {
// Handle network or other errors
showToast('Error: ' + error.message, 'error');
console.error('Update error:', error);
// Restore button state
updateButton.innerHTML = originalButtonText;
updateButton.disabled = false;
});
});
}); });
// Function to handle API responses for delete action // Function to handle API responses for delete action
@@ -188,6 +626,176 @@
showToast(`Server error: ${xhr.status}`, 'error'); showToast(`Server error: ${xhr.status}`, 'error');
} }
} }
// Load agent configuration from server
function loadAgentConfig() {
const agentName = document.getElementById('name').value;
fetch(`/api/agent/${agentName}/config`)
.then(response => {
if (!response.ok) {
throw new Error(`Failed to load agent config: ${response.status}`);
}
return response.json();
})
.then(data => {
agentConfig = data;
populateFormWithConfig(data);
showToast('Agent configuration loaded', 'success');
})
.catch(error => {
console.error('Error loading agent config:', error);
showToast('Error loading agent configuration: ' + error.message, 'error');
});
}
function populateFormWithConfig(config) {
// Clear existing dynamic sections
document.getElementById('connectorsSection').innerHTML = '';
document.getElementById('action_box').innerHTML = '';
document.getElementById('dynamic_box').innerHTML = '';
// Populate simple fields
document.getElementById('hud').checked = config.hud || false;
document.getElementById('enable_kb').checked = config.enable_kb || false;
document.getElementById('enable_reasoning').checked = config.enable_reasoning || false;
document.getElementById('kb_results').value = config.kb_results || '';
document.getElementById('standalone_job').checked = config.standalone_job || false;
document.getElementById('initiate_conversations').checked = config.initiate_conversations || false;
document.getElementById('can_stop_itself').checked = config.can_stop_itself || false;
document.getElementById('random_identity').checked = config.random_identity || false;
document.getElementById('long_term_memory').checked = config.long_term_memory || false;
document.getElementById('summary_long_term_memory').checked = config.summary_long_term_memory || false;
document.getElementById('identity_guidance').value = config.identity_guidance || '';
document.getElementById('periodic_runs').value = config.periodic_runs || '';
document.getElementById('model').value = config.model || '';
document.getElementById('multimodal_model').value = config.multimodal_model || '';
document.getElementById('permanent_goal').value = config.permanent_goal || '';
document.getElementById('system_prompt').value = config.system_prompt || '';
// Populate connectors
if (config.connectors && Array.isArray(config.connectors)) {
config.connectors.forEach((connector, index) => {
// Add connector section
document.getElementById('addConnectorButton').click();
// Find the added connector elements
const connectorType = document.getElementById(`connectorType${index}`);
const connectorConfig = document.getElementById(`connectorConfig${index}`);
// Set values
if (connectorType) {
setSelectValue(connectorType, connector.type);
}
if (connectorConfig) {
// Format the config value
formatConfigValue(connectorConfig, connector.config);
}
});
}
// Populate actions
if (config.actions && Array.isArray(config.actions)) {
config.actions.forEach((action, index) => {
// Add action section
document.getElementById('action_button').click();
// Find the added action elements
const actionName = document.getElementById(`actionsName${index}`);
const actionConfig = document.getElementById(`actionsConfig${index}`);
// Set values
if (actionName) {
setSelectValue(actionName, action.name);
}
if (actionConfig) {
// Format the config value
formatConfigValue(actionConfig, action.config);
}
});
}
// Populate prompt blocks
if (config.promptblocks && Array.isArray(config.promptblocks)) {
config.promptblocks.forEach((block, index) => {
// Add prompt block section
document.getElementById('dynamic_button').click();
// Find the added prompt block elements
const promptName = document.getElementById(`promptName${index}`);
const promptConfig = document.getElementById(`promptConfig${index}`);
// Set values
if (promptName) {
setSelectValue(promptName, block.name);
}
if (promptConfig) {
// Format the config value
formatConfigValue(promptConfig, block.config);
}
});
}
}
// Helper function to format config values
function formatConfigValue(configElement, configValue) {
// If it's a string (already stringified JSON), try to parse it first
if (typeof configValue === 'string') {
try {
const parsed = JSON.parse(configValue);
configElement.value = JSON.stringify(parsed, null, 2);
} catch (e) {
console.warn("Failed to parse config JSON string:", e);
configElement.value = configValue; // Keep as is if parsing fails
}
} else if (configValue !== undefined && configValue !== null) {
// Direct object, just stringify with formatting
configElement.value = JSON.stringify(configValue, null, 2);
} else {
// Default to empty object
configElement.value = '{}';
}
}
// Helper function to set select value (with fallback if option doesn't exist)
function setSelectValue(selectElement, value) {
// Check if the option exists
const optionExists = Array.from(selectElement.options).some(option => option.value === value);
if (optionExists) {
selectElement.value = value;
} else if (value) {
// If value is provided but option doesn't exist, create a new option
const newOption = document.createElement('option');
newOption.value = value;
newOption.text = value + ' (custom)';
selectElement.add(newOption);
selectElement.value = value;
}
}
</script> </script>
<style>
@keyframes pulse {
0% { transform: scale(1); }
50% { transform: scale(1.05); }
100% { transform: scale(1); }
}
.checkbox-label {
display: flex;
align-items: center;
cursor: pointer;
margin-bottom: 10px;
}
.checkbox-label .checkbox-custom {
margin-right: 10px;
}
</style>
</body> </body>
</html> </html>