Spaces:
Running
Running
| 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.") | |