{createDropdown && (
{
return api.put(`/agentexecutions/update/${executionId}`, executionData);
};
+export const editAgentTemplate = (agentTemplateId, agentTemplateData) => {
+ return api.put(`/agent_templates/update_agent_template/${agentTemplateId}`, agentTemplateData)
+}
+
export const addExecution = (executionData) => {
return api.post(`/agentexecutions/add`, executionData);
};
diff --git a/superagi/controllers/agent_template.py b/superagi/controllers/agent_template.py
index 140622aee..100bee8d0 100644
--- a/superagi/controllers/agent_template.py
+++ b/superagi/controllers/agent_template.py
@@ -13,6 +13,7 @@
from superagi.models.agent_template_config import AgentTemplateConfig
from superagi.models.agent_workflow import AgentWorkflow
from superagi.models.tool import Tool
+import json
# from superagi.types.db import AgentTemplateIn, AgentTemplateOut
router = APIRouter()
@@ -144,6 +145,59 @@ def update_agent_template(agent_template_id: int,
return db_agent_template
+@router.put("/update_agent_template/{agent_template_id}", status_code=200)
+def edit_agent_template(agent_template_id: int,
+ updated_agent_configs: dict,
+ organisation=Depends(get_user_organisation)):
+
+ """
+ Update the details of an agent template.
+
+ Args:
+ agent_template_id (int): The ID of the agent template to update.
+ edited_agent_configs (dict): The updated agent configurations.
+ organisation (Depends): Dependency to get the user organisation.
+
+ Returns:
+ HTTPException (status_code=200): If the agent gets successfully edited.
+
+ Raises:
+ HTTPException (status_code=404): If the agent template is not found.
+ """
+
+ db_agent_template = db.session.query(AgentTemplate).filter(AgentTemplate.organisation_id == organisation.id,
+ AgentTemplate.id == agent_template_id).first()
+ if db_agent_template is None:
+ raise HTTPException(status_code=404, detail="Agent Template not found")
+
+ db_agent_template.name = updated_agent_configs["name"]
+ db_agent_template.description = updated_agent_configs["description"]
+
+ db.session.commit()
+
+ agent_config_values = updated_agent_configs.get('agent_configs', {})
+
+ for key, value in agent_config_values.items():
+ if isinstance(value, (list, dict)):
+ value = json.dumps(value)
+ config = db.session.query(AgentTemplateConfig).filter(
+ AgentTemplateConfig.agent_template_id == agent_template_id,
+ AgentTemplateConfig.key == key
+ ).first()
+
+ if config is not None:
+ config.value = value
+ else:
+ new_config = AgentTemplateConfig(
+ agent_template_id=agent_template_id,
+ key=key,
+ value= value
+ )
+ db.session.add(new_config)
+
+ db.session.commit()
+ db.session.flush()
+
@router.post("/save_agent_as_template/{agent_id}")
def save_agent_as_template(agent_id: str,
diff --git a/tests/unit_tests/controllers/test_agent_template.py b/tests/unit_tests/controllers/test_agent_template.py
new file mode 100644
index 000000000..050e9e216
--- /dev/null
+++ b/tests/unit_tests/controllers/test_agent_template.py
@@ -0,0 +1,127 @@
+from unittest.mock import patch, MagicMock
+from superagi.models.agent_template import AgentTemplate
+from superagi.models.agent_template_config import AgentTemplateConfig
+from fastapi.testclient import TestClient
+from main import app
+
+client = TestClient(app)
+
+@patch('superagi.controllers.agent_template.db')
+@patch('superagi.helper.auth.db')
+@patch('superagi.helper.auth.get_user_organisation')
+def test_edit_agent_template_success(mock_get_user_org, mock_auth_db, mock_db):
+ # Create a mock agent template
+ mock_agent_template = AgentTemplate(id=1, name="Test Agent Template", description="Test Description")
+ # mock_agent_goals = AgentTemplateConfig()
+
+ # Create a mock edited agent configuration
+ mock_updated_agent_configs = {
+ "name": "Updated Agent Template",
+ "description": "Updated Description",
+ "agent_configs": {
+ "goal": ["Create a simple pacman game for me.", "Write all files properly."],
+ "instruction": ["write spec","write code","improve the code","write test"],
+ "agent_type": "Don't Maintain Task Queue",
+ "constraints": ["If you are unsure how you previously did something or want to recall past events, thinking about similar events will help you remember.","Ensure the tool and args are as per current plan and reasoning","Exclusively use the tools listed under \"TOOLS\"","REMEMBER to format your response as JSON, using double quotes (\"\") around keys and string values, and commas (,) to separate items in arrays and objects. IMPORTANTLY, to use a JSON object as a string in another JSON object, you need to escape the double quotes."],
+ "tools": ["Read Email", "Send Email", "Write File"],
+ "exit": "No exit criterion",
+ "iteration_interval": 500,
+ "model": "gpt-4",
+ "max_iterations": 25,
+ "permission_type": "God Mode",
+ "LTM_DB": "Pinecone"
+ }
+ }
+
+ # Mocking the user organisation
+ mock_get_user_org.return_value = MagicMock(id=1)
+
+ # Create a session mock
+ session_mock = MagicMock()
+ mock_db.session = session_mock
+ mock_db.session.query.return_value.filter.return_value.first.return_value = mock_agent_template
+ mock_db.session.commit.return_value = None
+ mock_db.session.add.return_value = None
+ mock_db.session.flush.return_value = None
+
+ mock_agent_template_config = AgentTemplateConfig(agent_template_id = 1, key="goal", value=["Create a simple pacman game for me.", "Write all files properly."])
+
+
+ # Call the endpoint
+ response = client.put("agent_templates/update_agent_template/1", json=mock_updated_agent_configs)
+
+ assert response.status_code == 200
+
+ # Verify changes in the mock agent template
+ assert mock_agent_template.name == "Updated Agent Template"
+ assert mock_agent_template.description == "Updated Description"
+ assert mock_agent_template_config.key == "goal"
+ assert mock_agent_template_config.value == ["Create a simple pacman game for me.", "Write all files properly."]
+
+
+ session_mock.commit.assert_called()
+ session_mock.flush.assert_called()
+
+
+@patch('superagi.controllers.agent_template.db')
+@patch('superagi.helper.auth.db')
+@patch('superagi.helper.auth.get_user_organisation')
+def test_edit_agent_template_failure(mock_get_user_org, mock_auth_db, mock_db):
+ # Setup: The user organisation exists, but the agent template does not exist.
+ mock_get_user_org.return_value = MagicMock(id=1)
+
+ # Create a session mock
+ session_mock = MagicMock()
+ mock_db.session = session_mock
+ mock_db.session.query.return_value.filter.return_value.first.return_value = None
+
+ # Call the endpoint
+ response = client.put("agent_templates/update_agent_template/1", json={})
+
+ # Verify: The response status code should be 404, indicating that the agent template was not found.
+ assert response.status_code == 404
+ assert response.json() == {"detail": "Agent Template not found"}
+
+ # Verify: The database commit method should not have been called because the agent template was not found.
+ session_mock.commit.assert_not_called()
+ session_mock.flush.assert_not_called()
+
+
+@patch('superagi.controllers.agent_template.db')
+@patch('superagi.helper.auth.db')
+@patch('superagi.helper.auth.get_user_organisation')
+def test_edit_agent_template_with_new_config_success(mock_get_user_org, mock_auth_db, mock_db):
+ # Create a mock agent template
+ mock_agent_template = AgentTemplate(id=1, name="Test Agent Template", description="Test Description")
+
+ # Create a mock edited agent configuration
+ mock_updated_agent_configs = {
+ "name": "Updated Agent Template",
+ "description": "Updated Description",
+ "agent_configs": {
+ "new_config_key": "New config value" # This is a new config
+ }
+ }
+
+ # Mocking the user organisation
+ mock_get_user_org.return_value = MagicMock(id=1)
+
+ # Create a session mock
+ session_mock = MagicMock()
+ mock_db.session = session_mock
+ mock_db.session.query.return_value.filter.return_value.first.return_value = mock_agent_template
+ mock_db.session.commit.return_value = None
+ mock_db.session.add.return_value = None
+ mock_db.session.flush.return_value = None
+
+ # Call the endpoint
+ response = client.put("agent_templates/update_agent_template/1", json=mock_updated_agent_configs)
+
+ assert response.status_code == 200
+
+ # Verify changes in the mock agent template
+ assert mock_agent_template.name == "Updated Agent Template"
+ assert mock_agent_template.description == "Updated Description"
+
+ session_mock.commit.assert_called()
+ session_mock.flush.assert_called()
\ No newline at end of file