93 lines
2.6 KiB
Python
93 lines
2.6 KiB
Python
"""Robot Framework helper for actor loader smoke tests.
|
|
|
|
Provides a CLI-style interface for Robot to invoke actor discovery
|
|
and loading. Exit code 0 = success, 1 = failure.
|
|
|
|
Usage:
|
|
python robot/helper_actor_loading.py discover <directory>
|
|
python robot/helper_actor_loading.py discover-examples
|
|
python robot/helper_actor_loading.py discover-invalid <directory>
|
|
python robot/helper_actor_loading.py namespace-default <directory>
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
_SRC = str(Path(__file__).resolve().parents[1] / "src")
|
|
if _SRC not in sys.path:
|
|
sys.path.insert(0, _SRC)
|
|
|
|
from cleveragents.actor.loader import ActorLoader # noqa: E402
|
|
|
|
|
|
def main() -> int:
|
|
if len(sys.argv) < 2:
|
|
print("Usage: helper_actor_loading.py <command> [args...]")
|
|
return 1
|
|
|
|
command = sys.argv[1]
|
|
|
|
if command == "discover":
|
|
directory = sys.argv[2]
|
|
return _discover(Path(directory))
|
|
|
|
if command == "discover-examples":
|
|
project_root = Path(__file__).resolve().parents[1]
|
|
return _discover(project_root / "examples" / "actors")
|
|
|
|
if command == "discover-invalid":
|
|
directory = sys.argv[2]
|
|
return _discover_invalid(Path(directory))
|
|
|
|
if command == "namespace-default":
|
|
return _namespace_default()
|
|
|
|
print(f"Unknown command: {command}")
|
|
return 1
|
|
|
|
|
|
def _discover(directory: Path) -> int:
|
|
loader = ActorLoader(search_roots=[directory])
|
|
actors = loader.discover()
|
|
for actor in actors:
|
|
print(f"actor-loaded: {actor.name} type={actor.type.value}")
|
|
print(f"actor-count: {len(actors)}")
|
|
return 0
|
|
|
|
|
|
def _discover_invalid(directory: Path) -> int:
|
|
loader = ActorLoader(search_roots=[directory])
|
|
try:
|
|
loader.discover()
|
|
print("actor-loading-unexpected-success")
|
|
return 1
|
|
except Exception as exc:
|
|
print(f"actor-loading-expected-fail: {exc}")
|
|
return 0
|
|
|
|
|
|
def _namespace_default() -> int:
|
|
tmp = Path(tempfile.mkdtemp(prefix="actor_ns_"))
|
|
yaml_content = (
|
|
"name: bare-name\n"
|
|
"type: llm\n"
|
|
"description: Test namespace defaulting\n"
|
|
'version: "1.0"\n'
|
|
"model: gpt-4\n"
|
|
)
|
|
(tmp / "bare.yaml").write_text(yaml_content)
|
|
loader = ActorLoader(search_roots=[tmp])
|
|
actors = loader.discover()
|
|
if len(actors) == 1 and actors[0].name == "local/bare-name":
|
|
print("namespace-default-ok: local/bare-name")
|
|
return 0
|
|
print(f"namespace-default-fail: {[a.name for a in actors]}")
|
|
return 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|