Spaces:
Running
Running
| from __future__ import annotations | |
| from agent import AgentContext | |
| from helpers.api import ApiHandler, Request, Response | |
| class ApiPoll(ApiHandler): | |
| """API-key protected status for chats started through ``api_message``.""" | |
| def requires_auth(cls) -> bool: | |
| return False | |
| def requires_csrf(cls) -> bool: | |
| return False | |
| def requires_api_key(cls) -> bool: | |
| return True | |
| async def process(self, input: dict, request: Request) -> dict | Response: | |
| context_id = str(input.get("context_id", "") or "").strip() | |
| if not context_id: | |
| return Response("Missing context_id", status=400) | |
| context = AgentContext.get(context_id) | |
| if context is None: | |
| return Response("Context not found", status=404) | |
| output = context.log.output(start=max(0, int(input.get("log_from", 0) or 0))) | |
| result = None | |
| for item in reversed(output.items): | |
| if item.get("type") in {"agent", "assistant", "response"}: | |
| result = item.get("content") | |
| break | |
| running = context.is_running() | |
| status = "running" if running else "completed" | |
| if not running and context.task and context.task.is_ready(): | |
| try: | |
| result = context.task.result_sync(timeout=0) | |
| except Exception: | |
| status = "failed" | |
| goal_status = None | |
| goal_error = "" | |
| try: | |
| import plugins._goal.tools.goal as goal | |
| goal_status = goal.get_goal(context_id) | |
| if goal_status and goal_status.get("status") == "active" and not running: | |
| goal_status = goal.reconcile_goal(context_id, running=False) | |
| except Exception as error: | |
| goal_status = None | |
| goal_error = type(error).__name__ | |
| semantic_status = status | |
| if goal_status: | |
| semantic_status = ( | |
| "working" | |
| if running and goal_status["status"] == "active" | |
| else goal_status["status"] | |
| ) | |
| response = { | |
| "context_id": context.id, | |
| "status": semantic_status, | |
| "running": running, | |
| "log_progress": context.log.progress, | |
| "log_from": output.end, | |
| "result": result if not running else None, | |
| "goal_id": goal_status.get("goal_id") if goal_status else None, | |
| "milestone": goal_status.get("current_milestone") if goal_status else "", | |
| "progress_revision": goal_status.get("progress_revision") | |
| if goal_status | |
| else 0, | |
| "requires_attention": bool( | |
| goal_status and goal_status.get("requires_attention") | |
| ), | |
| "attention_reason": goal_status.get("attention_reason") | |
| if goal_status | |
| else "", | |
| "attention_message": goal_status.get("attention_message") | |
| if goal_status | |
| else "", | |
| "next_poll_after": 2, | |
| "goal_error": goal_error or None, | |
| } | |
| if input.get("include_logs") is True: | |
| response["logs"] = output.items | |
| return response | |