agent returns success and message (#44)
taylor_socfortress committed
Jul 17, 2023 at 14:23 UTC
f9a897c13f23fa4b2977e4e1b8611693173152db
2 files changed
+13
-4
backend/app/routes/agents.py
+7
-2
@@ -2,6 +2,7 @@ from typing import Any
2
3
from flask import Blueprint
4
from flask import jsonify
5
+from loguru import logger
6
7
from app.models.agents import agent_metadata_schema
8
from app.services.agents.agents import AgentService
@@ -39,8 +40,12 @@ def get_agent(agent_id: str) -> Any:
40
if agent is None:
41
return jsonify({"message": "Agent not found", "success": False}), 404
42
else:
42
- agent_dict = agent_metadata_schema.dump(agent)
43
- return jsonify(agent_dict)
43
+ try:
44
+ agent_dict = agent_metadata_schema.dump(agent)
45
+ return {"agent": agent_dict, "success": True, "message": "Agent found"}
46
+ except Exception as e:
47
+ logger.error(f"Error returning agent: {e}")
48
+ return jsonify({"message": "Error returning agent", "success": False}), 500
49
50
51
@bp.route("/agents/<agent_id>/critical", methods=["POST"])
backend/app/services/agents/agents.py
+6
-2
@@ -46,8 +46,12 @@ class AgentService:
46
Returns:
47
List[dict]: A list of dictionaries where each dictionary represents the serialized data of an agent.
48
"""
49
- agents = db.session.query(AgentMetadata).all()
50
- return agent_metadatas_schema.dump(agents)
49
+ try:
50
+ agents = db.session.query(AgentMetadata).all()
51
+ return {"message": "Agents retrieved successfully", "success": True, "agents": agent_metadatas_schema.dump(agents)}
52
+ except Exception as e:
53
+ logger.error(f"Failed to retrieve agents: {e}")
54
+ return {"message": "Failed to retrieve agents.", "success": False}
55
56
def get_agent(self, agent_id: str) -> Optional[AgentMetadata]:
57
"""