File size: 1,360 Bytes
be22d4d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | """Shim for tool discovery. Registers `computer_use` with tools.registry.
The real implementation lives in the `tools/computer_use/` package to keep
the file structure clean. This shim exists because tools.registry auto-imports
`tools/*.py` — we need a top-level module to trigger the registration.
"""
from __future__ import annotations
from tools.computer_use.schema import COMPUTER_USE_SCHEMA
from tools.computer_use.tool import (
check_computer_use_requirements,
handle_computer_use,
release_computer_use_session,
set_approval_callback,
)
from tools.registry import registry
registry.register(
name="computer_use",
toolset="computer_use",
schema=COMPUTER_USE_SCHEMA,
handler=lambda args, **kw: handle_computer_use(args, **kw),
check_fn=check_computer_use_requirements,
requires_env=[],
description=(
"Universal desktop control via cua-driver (macOS, Windows, Linux). Works with any "
"tool-capable model (Anthropic, OpenAI, OpenRouter, local vLLM, "
"etc.). Background computer-use: does NOT steal the user's cursor "
"or keyboard focus."
),
)
__all__ = [
"handle_computer_use",
"release_computer_use_session",
"set_approval_callback",
"check_computer_use_requirements",
"release_computer_use_session",
]
|