mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-05-23 04:17:34 +00:00
8601f0d10c
- rename document_artifact to office_artifact and remove retired shims/facades - make text_editor own Markdown saves, canvas-open intent, refresh, and stale-save protection - keep Office artifacts Desktop-only with Office formats and update skills/tests
56 lines
2.1 KiB
Python
56 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from plugins._desktop.helpers import desktop_state
|
|
from plugins._office.helpers import document_store
|
|
|
|
OFFICE_EXTENSIONS = document_store.OPEN_DOCUMENT_EXTENSIONS | document_store.OOXML_EXTENSIONS
|
|
|
|
|
|
def build_context(max_items: int = 6) -> str:
|
|
documents = [
|
|
doc
|
|
for doc in document_store.get_open_documents(limit=max(max_items * 4, 20))
|
|
if str(doc.get("extension") or "").lower() in OFFICE_EXTENSIONS
|
|
][:max_items]
|
|
desktop_context = build_desktop_context()
|
|
if not documents:
|
|
return desktop_context
|
|
|
|
lines = [
|
|
"These Office artifacts have active document sessions. Content is omitted; use `office_artifact` with action `read` before content-sensitive edits.",
|
|
]
|
|
for doc in documents:
|
|
lines.append(format_document_line(doc))
|
|
lines.append(
|
|
"Use `office_artifact` with action `edit` and file_id or path for saved Office edits; tool results refresh already-open document canvases automatically."
|
|
)
|
|
if desktop_context:
|
|
lines.extend(["", desktop_context])
|
|
return "\n".join(lines)
|
|
|
|
|
|
def format_document_line(doc: dict[str, Any]) -> str:
|
|
return (
|
|
f"- {doc.get('basename', 'Untitled')} "
|
|
f"(.{doc.get('extension', '')}, file_id={doc.get('file_id', '')}, "
|
|
f"path={document_store.display_path(doc.get('path', ''))}, version={document_store.item_version(doc)}, "
|
|
f"size={doc.get('size', 0)} bytes, last_modified={doc.get('last_modified', '')}, "
|
|
f"open_sessions={doc.get('open_sessions', 1)})"
|
|
)
|
|
|
|
|
|
def build_desktop_context() -> str:
|
|
if not desktop_state.session_manifest_exists():
|
|
return ""
|
|
try:
|
|
return desktop_state.compact_prompt_context(
|
|
desktop_state.collect_state(include_screenshot=False),
|
|
)
|
|
except Exception as exc:
|
|
return (
|
|
"[DESKTOP STATE]\n"
|
|
f"- unavailable={exc}\n"
|
|
"- next=Open the Desktop surface manually, then run plugins/_desktop/skills/linux-desktop/scripts/desktopctl.sh observe --json."
|
|
)
|