mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-05-18 23:45:49 +00:00
8c5cf1f69f
Introduce the builtin `_a0_connector` plugin that lets the host-side A0 CLI connect to Agent Zero over authenticated HTTP and `/ws`. This adds connector capability discovery, chat/context lifecycle endpoints, log streaming, and the remote text editing, code execution, and file tree bridge used by the CLI workflow.
120 lines
4.4 KiB
Python
120 lines
4.4 KiB
Python
"""POST /api/plugins/_a0_connector/v1/message_send."""
|
|
from __future__ import annotations
|
|
|
|
import base64
|
|
import os
|
|
import uuid
|
|
|
|
from helpers.api import Request, Response
|
|
from helpers.print_style import PrintStyle
|
|
from helpers.security import safe_filename
|
|
import plugins._a0_connector.api.v1.base as connector_base
|
|
|
|
|
|
class MessageSend(connector_base.ProtectedConnectorApiHandler):
|
|
async def process(self, input: dict, request: Request) -> dict | Response:
|
|
from agent import UserMessage
|
|
from helpers import files
|
|
from plugins._a0_connector.helpers.chat_context import (
|
|
ConnectorContextError,
|
|
create_context,
|
|
get_existing_context,
|
|
)
|
|
|
|
message = str(input.get("message", "")).strip()
|
|
if not message:
|
|
return Response(
|
|
response='{"error": "message is required"}',
|
|
status=400,
|
|
mimetype="application/json",
|
|
)
|
|
|
|
context_id = str(input.get("context_id", "")).strip() or None
|
|
current_context_id = (
|
|
str(input.get("current_context", input.get("current_context_id", ""))).strip()
|
|
or None
|
|
)
|
|
project_name = str(input.get("project_name", "")).strip() or None
|
|
agent_profile = str(input.get("agent_profile", "")).strip() or None
|
|
attachments_data = input.get("attachments", [])
|
|
|
|
attachment_paths: list[str] = []
|
|
if isinstance(attachments_data, list) and attachments_data:
|
|
upload_folder_ext = files.get_abs_path("usr/uploads")
|
|
upload_folder_int = "/a0/usr/uploads"
|
|
os.makedirs(upload_folder_ext, exist_ok=True)
|
|
|
|
for attachment in attachments_data:
|
|
if not isinstance(attachment, dict):
|
|
continue
|
|
filename = str(attachment.get("filename", "")).strip()
|
|
b64_content = str(attachment.get("base64", "")).strip()
|
|
if not filename or not b64_content:
|
|
continue
|
|
|
|
try:
|
|
safe_name = safe_filename(filename)
|
|
if not safe_name:
|
|
continue
|
|
save_path = os.path.join(upload_folder_ext, safe_name)
|
|
with open(save_path, "wb") as handle:
|
|
handle.write(base64.b64decode(b64_content))
|
|
attachment_paths.append(os.path.join(upload_folder_int, safe_name))
|
|
except Exception as exc:
|
|
PrintStyle.error(f"[a0-connector] attachment error: {exc}")
|
|
|
|
try:
|
|
if context_id:
|
|
context = get_existing_context(
|
|
context_id,
|
|
agent_profile=agent_profile,
|
|
project_name=project_name,
|
|
)
|
|
else:
|
|
context = create_context(
|
|
lock=self.thread_lock,
|
|
current_context_id=current_context_id,
|
|
agent_profile=agent_profile,
|
|
project_name=project_name,
|
|
)
|
|
context_id = context.id
|
|
except ConnectorContextError as exc:
|
|
return Response(
|
|
response=f'{{"error": "{str(exc)}"}}',
|
|
status=exc.status_code,
|
|
mimetype="application/json",
|
|
)
|
|
except Exception as exc:
|
|
return Response(
|
|
response=f'{{"error": "Failed to activate project: {str(exc)}"}}',
|
|
status=400,
|
|
mimetype="application/json",
|
|
)
|
|
|
|
attachment_names = [os.path.basename(path) for path in attachment_paths]
|
|
message_id = str(uuid.uuid4())
|
|
context.log.log(
|
|
type="user",
|
|
heading="",
|
|
content=message,
|
|
kvps={"attachments": attachment_names},
|
|
id=message_id,
|
|
)
|
|
|
|
try:
|
|
task = context.communicate(
|
|
UserMessage(message=message, attachments=attachment_paths, id=message_id)
|
|
)
|
|
result = await task.result()
|
|
return {
|
|
"context_id": context_id,
|
|
"status": "completed",
|
|
"response": result,
|
|
}
|
|
except Exception as exc:
|
|
PrintStyle.error(f"[a0-connector] message_send error: {exc}")
|
|
return Response(
|
|
response=f'{{"error": "{str(exc)}"}}',
|
|
status=500,
|
|
mimetype="application/json",
|
|
)
|