mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-08-25 16:25:34 +00:00
9e5fd188e2
Add per-scope availability, duplication, active-profile status, and inline project selection while keeping the profile switcher synchronized with runtime state. Reconcile unavailable profiles at explicit disable, deletion, project-transition, and context-creation boundaries; polish validation, prompt layout, deletion, and completion states; and extend focused backend and WebUI coverage.
45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
from agent import AgentContext
|
|
from helpers import projects, subagents
|
|
from helpers.api import ApiHandler, Request, Response
|
|
from helpers.persist_chat import save_tmp_chat
|
|
from helpers.state_monitor_integration import mark_dirty_for_context
|
|
from initialize import initialize_agent
|
|
|
|
|
|
class SetAgentProfile(ApiHandler):
|
|
async def process(self, input: dict, request: Request) -> dict | Response:
|
|
context_id = str(input.get("context_id", "") or "").strip()
|
|
profile = str(input.get("agent_profile", "") or "").strip()
|
|
|
|
if not context_id:
|
|
return Response(status=400, response="Missing context_id")
|
|
if not profile:
|
|
return Response(status=400, response="Missing agent_profile")
|
|
|
|
context = AgentContext.get(context_id)
|
|
if not context:
|
|
return Response(status=404, response="Context not found")
|
|
if context.is_running():
|
|
return Response(
|
|
status=409,
|
|
response="Agent profile can be changed after the current run finishes.",
|
|
)
|
|
|
|
profiles = subagents.get_available_agents_dict(
|
|
projects.get_context_project_name(context)
|
|
)
|
|
selected_profile = profiles.get(profile)
|
|
if selected_profile is None:
|
|
return Response(status=404, response=f"Agent profile '{profile}' not found")
|
|
|
|
config = initialize_agent(override_settings={"agent_profile": profile})
|
|
context.config = config
|
|
context.agent0.config = config
|
|
|
|
save_tmp_chat(context)
|
|
mark_dirty_for_context(context.id, reason="agent_profile_change")
|
|
return {
|
|
"ok": True,
|
|
"agent_profile": profile,
|
|
"agent_profile_label": selected_profile.title or profile,
|
|
}
|