agentenv-framework-protocol 0.1.386__tar.gz → 0.1.387__tar.gz

{agentenv_framework_protocol-0.1.386 → agentenv_framework_protocol-0.1.387}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: agentenv-framework-protocol
-Version: 0.1.386
+Version: 0.1.387
 Summary: Open data-plane protocol and server SDK for agent environments.
 Project-URL: Homepage, https://www.agentenvframework.com
 Project-URL: Documentation, https://www.agentenvframework.com/docs

{agentenv_framework_protocol-0.1.386 → agentenv_framework_protocol-0.1.387}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentenv-framework-protocol"
-version = "0.1.386"
+version = "0.1.387"
 description = "Open data-plane protocol and server SDK for agent environments."
 readme = "README.md"
 license = "Apache-2.0"

agentenv_framework_protocol-0.1.387/src/agentenv_protocol/computer/serve.py

@@ -0,0 +1,38 @@
+"""Serve a computer driver from its file, as a composed computer's program starts it:
+``python -m agentenv_protocol.computer.serve <file>:<Class>``, run in the driver's folder, serves the class on
+``MCP_PORT``. The file's folder is importable, so it may import its own modules."""
+from __future__ import annotations
+
+import importlib.util
+import sys
+from pathlib import Path
+
+from .driver import ComputerDriverEnvironment
+
+USAGE = "usage: python -m agentenv_protocol.computer.serve <file>:<Class>"
+
+
+def main(argv: list[str]) -> None:
+    if len(argv) != 1 or ":" not in argv[0]:
+        raise SystemExit(USAGE)
+    file, _, name = argv[0].partition(":")
+    driver_class(Path(file), name)().serve()
+
+
+def driver_class(file: Path, name: str) -> type[ComputerDriverEnvironment]:
+    """The class ``name`` in ``file``, imported as a module of its own; exits saying why when it isn't a driver."""
+    if not file.is_file():
+        raise SystemExit(f"{file} isn't a file")
+    sys.path.insert(0, str(file.resolve().parent))
+    spec = importlib.util.spec_from_file_location(file.stem, file)
+    module = importlib.util.module_from_spec(spec)
+    sys.modules[file.stem] = module
+    spec.loader.exec_module(module)
+    cls = getattr(module, name, None)
+    if not (isinstance(cls, type) and issubclass(cls, ComputerDriverEnvironment)):
+        raise SystemExit(f"{file} has no ComputerDriverEnvironment subclass named {name}")
+    return cls
+
+
+if __name__ == "__main__":
+    main(sys.argv[1:])

agentenv_framework_protocol-0.1.387/tests/test_computer_serve.py

@@ -0,0 +1,75 @@
+"""Serving a driver from its file: the class is imported with its folder importable and served; anything that isn't
+a driver class exits saying why."""
+from __future__ import annotations
+
+import sys
+
+import pytest
+
+pytest.importorskip("mcp.server.fastmcp")
+pytest.importorskip("PIL")
+
+from agentenv_protocol.computer import ComputerDriverEnvironment  # noqa: E402
+from agentenv_protocol.computer.serve import USAGE, main  # noqa: E402
+
+DRIVER = '''from desk_helpers import SIZE
+
+from agentenv_protocol.computer import (ComputerDriverEnvironment, Display, capture, click, computer_driver, display,
+                                        drag, press, scroll, type_text)
+
+
+@computer_driver(name="desk", supports=[{"os": {"type": "linux", "display": ["x11"]}, "arch": ["amd64"]}])
+class Desk(ComputerDriverEnvironment):
+    @display
+    def size(self): return Display(*SIZE)
+
+    @capture
+    def grab(self): ...
+
+    @click
+    def click(self, x, y, button, count): ...
+
+    @drag
+    def drag(self, path, button): ...
+
+    @scroll
+    def scroll(self, x, y, dx, dy): ...
+
+    @type_text
+    def type(self, text): ...
+
+    @press
+    def press(self, keys): ...
+'''
+
+
+@pytest.fixture
+def served(monkeypatch):
+    calls = []
+    monkeypatch.setattr(ComputerDriverEnvironment, "serve", lambda self: calls.append(type(self).__name__))
+    monkeypatch.setattr(sys, "path", list(sys.path))
+    monkeypatch.setattr(sys, "modules", dict(sys.modules))
+    return calls
+
+
+def test_the_class_is_imported_from_its_file_with_its_folder_importable_and_served(tmp_path, served):
+    (tmp_path / "driver.py").write_text(DRIVER)
+    (tmp_path / "desk_helpers.py").write_text("SIZE = (10, 10)\n")
+
+    main([f"{tmp_path / 'driver.py'}:Desk"])
+
+    assert served == ["Desk"]
+
+
+@pytest.mark.parametrize("argv, message", [
+    ([], USAGE), (["driver.py"], USAGE), (["missing.py:Desk"], "missing.py isn't a file"),
+    (["driver.py:Helper"], "has no ComputerDriverEnvironment subclass named Helper"),
+], ids=["nothing", "no-class", "no-file", "not-a-driver"])
+def test_what_isnt_a_driver_class_exits_saying_why(tmp_path, served, monkeypatch, argv, message):
+    (tmp_path / "driver.py").write_text("class Helper:\n    pass\n")
+    monkeypatch.chdir(tmp_path)
+
+    with pytest.raises(SystemExit) as stopped:
+        main(argv)
+
+    assert message in str(stopped.value) and served == []