mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-08-26 16:55:43 +00:00
b6be345a56
Refactor of messages.js to support lazy rendering. Move chat rename functionality out of core and into a new _chat_naming plugin, and add a built-in _pin_to_top plugin for sidebar pinning. The chat-naming plugin includes API handlers, prompts, helper logic for selecting user messages and budgeted Utility Model calls, a python monologue_start extension, web UI (modal, store, config, sidebar action), and comprehensive tests. Removed the old core renaming extension and deprecated prompts/tests. Also added pin-to-top plugin files (API, helpers, webui store, tests/docs) and updated various AGENTS.md docs to reflect the new plugins and lifecycle clarifications. Minor UI/js tweak: reorder speak/copy buttons in browser tool handler and add a new webui message-window.js along with related webui/store/component updates and test adjustments.
68 lines
2.8 KiB
Python
68 lines
2.8 KiB
Python
from __future__ import annotations
|
|
|
|
from agent import AgentContext
|
|
from helpers.api import ApiHandler, Input, Output, Request, Response
|
|
from helpers.persist_chat import save_tmp_chat
|
|
from helpers.state_monitor_integration import mark_dirty_all
|
|
from helpers.task_scheduler import TaskScheduler
|
|
from plugins._chat_naming.helpers import naming
|
|
|
|
|
|
class ChatName(ApiHandler):
|
|
async def process(self, input: Input, request: Request) -> Output:
|
|
del request
|
|
action = str(input.get("action", "get") or "get").strip().lower()
|
|
kind = str(input.get("kind", "chat") or "chat").strip().lower()
|
|
item_id = str(input.get("item_id", "") or "").strip()
|
|
if kind not in {"chat", "task"}:
|
|
return Response("Invalid row kind.", 400)
|
|
if not item_id:
|
|
return Response("Missing chat or task ID.", 400)
|
|
|
|
context = AgentContext.get(item_id)
|
|
if not context:
|
|
return Response("Chat context not found.", 404)
|
|
|
|
try:
|
|
if action == "get":
|
|
current_name = await self._current_name(kind, item_id, context.name or "")
|
|
return {"ok": True, "name": current_name}
|
|
if action == "generate":
|
|
current_name = await self._current_name(kind, item_id, context.name or "")
|
|
name = await naming.generate_name(
|
|
context.agent0,
|
|
current_name=current_name,
|
|
)
|
|
return {"ok": True, "name": name}
|
|
if action == "save":
|
|
name = naming.normalize_manual_name(input.get("name", ""))
|
|
if kind == "task":
|
|
await self._save_task_name(item_id, name)
|
|
context.name = name
|
|
save_tmp_chat(context)
|
|
mark_dirty_all(reason="plugins._chat_naming.save_task_name")
|
|
else:
|
|
naming.save_context_name(context.agent0, name)
|
|
return {"ok": True, "name": name}
|
|
return Response(f"Unknown action: {action}", 400)
|
|
except ValueError as error:
|
|
return Response(str(error), 400)
|
|
except Exception as error:
|
|
return Response(str(error), 500)
|
|
|
|
async def _current_name(self, kind: str, item_id: str, fallback: str) -> str:
|
|
if kind != "task":
|
|
return fallback
|
|
scheduler = TaskScheduler.get()
|
|
await scheduler.reload()
|
|
task = scheduler.get_task_by_uuid(item_id)
|
|
if not task:
|
|
raise ValueError("Scheduled task not found.")
|
|
return str(task.name or fallback)
|
|
|
|
async def _save_task_name(self, item_id: str, name: str) -> None:
|
|
scheduler = TaskScheduler.get()
|
|
await scheduler.reload()
|
|
task = await scheduler.update_task(item_id, name=name)
|
|
if not task:
|
|
raise ValueError("Scheduled task not found.")
|