thaloco/pyro-mcp
1
Fork 0
Code Pull requests Releases Packages Activity Actions

Refactor public API around environments

Browse source
This commit is contained in:
Thales Maciel 2026-03-08 16:02:02 -03:00
parent 57dae52cc2
commit 5d5243df23
41 changed files with 1301 additions and 459 deletions
Download patch file Download diff file Expand all files Collapse all files

10
examples/langchain_vm_run.py
View file

@ -19,7 +19,7 @@ F = TypeVar("F", bound=Callable[..., Any])
def run_vm_run_tool(
*,
profile: str,
environment: str,
command: str,
vcpu_count: int,
mem_mib: int,
@ -29,7 +29,7 @@ def run_vm_run_tool(
) -> str:
pyro = Pyro()
result = pyro.run_in_vm(
profile=profile,
environment=environment,
command=command,
vcpu_count=vcpu_count,
mem_mib=mem_mib,
@ -53,17 +53,17 @@ def build_langchain_vm_run_tool() -> Any:
@decorator
def vm_run(
profile: str,
environment: str,
command: str,
vcpu_count: int,
mem_mib: int,
timeout_seconds: int = 30,
ttl_seconds: int = 600,
network: bool = False,
) -> str:
) -> str:
"""Run one command in an ephemeral Firecracker VM and clean it up."""
return run_vm_run_tool(
profile=profile,
environment=environment,
command=command,
vcpu_count=vcpu_count,
mem_mib=mem_mib,