mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 00:40:45 +08:00
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from oclaw.runtime.tools.base import ToolSpec
|
|
from oclaw.runtime.tools.public.local_sdk import get_local_adapter
|
|
|
|
|
|
def set_env_tool() -> ToolSpec:
|
|
def _handler(args: dict[str, Any]) -> dict[str, Any]:
|
|
key = str(args.get("key") or "").strip()
|
|
value = args.get("value")
|
|
return get_local_adapter().set_env(key=key, value=str(value) if value is not None else None)
|
|
|
|
return ToolSpec(
|
|
name="set_env",
|
|
description="Set or delete an environment variable in the current process.",
|
|
parameters={
|
|
"type": "object",
|
|
"properties": {
|
|
"key": {"type": "string", "description": "Environment variable key."},
|
|
"value": {"type": "string", "description": "Value to set. If omitted, deletes the key."},
|
|
},
|
|
"required": ["key"],
|
|
"additionalProperties": False,
|
|
},
|
|
handler=_handler,
|
|
tags=frozenset({"public", "local", "write"}),
|
|
risk_level="high",
|
|
timeout_s=2.0,
|
|
)
|
|
|
|
|
|
__all__ = ["set_env_tool"]
|