3764c30628
Stub implementations for: - base.py: Tool, ToolResult, @tool decorator, registry - calculate.py: Math expressions (simpleeval) - fetch.py: HTTP requests (aiohttp) - files.py: read_file, write_file, list_dir - shell.py: run_command (sandboxed) - search.py: web_search - keyvalue.py: key_value_get/set/delete (in-memory stub) - librarian.py: exist-db integration (store, get, query, search) All stubs return "Not implemented" - ready for real implementation. Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
78 lines
1.8 KiB
Python
78 lines
1.8 KiB
Python
"""
|
|
Key-value store tool - persistent agent state.
|
|
"""
|
|
|
|
from typing import Any, Optional
|
|
from .base import tool, ToolResult
|
|
|
|
|
|
# TODO: Implement backend (Redis, SQLite, or in-memory)
|
|
_store: dict = {} # Temporary in-memory store
|
|
|
|
|
|
@tool
|
|
async def key_value_get(
|
|
key: str,
|
|
namespace: Optional[str] = None,
|
|
) -> ToolResult:
|
|
"""
|
|
Get a value from the key-value store.
|
|
|
|
Args:
|
|
key: Key to retrieve
|
|
namespace: Namespace for isolation (default: agent name)
|
|
|
|
Returns:
|
|
Stored value, or null if not found
|
|
"""
|
|
# TODO: Implement with Redis/SQLite
|
|
ns_key = f"{namespace or 'default'}:{key}"
|
|
value = _store.get(ns_key)
|
|
return ToolResult(success=True, data=value)
|
|
|
|
|
|
@tool
|
|
async def key_value_set(
|
|
key: str,
|
|
value: Any,
|
|
namespace: Optional[str] = None,
|
|
ttl: Optional[int] = None,
|
|
) -> ToolResult:
|
|
"""
|
|
Set a value in the key-value store.
|
|
|
|
Args:
|
|
key: Key to store
|
|
value: Value to store (JSON-serializable)
|
|
namespace: Namespace for isolation (default: agent name)
|
|
ttl: Time-to-live in seconds (optional)
|
|
|
|
Returns:
|
|
success (bool)
|
|
"""
|
|
# TODO: Implement with Redis/SQLite, handle TTL
|
|
ns_key = f"{namespace or 'default'}:{key}"
|
|
_store[ns_key] = value
|
|
return ToolResult(success=True, data=True)
|
|
|
|
|
|
@tool
|
|
async def key_value_delete(
|
|
key: str,
|
|
namespace: Optional[str] = None,
|
|
) -> ToolResult:
|
|
"""
|
|
Delete a key from the key-value store.
|
|
|
|
Args:
|
|
key: Key to delete
|
|
namespace: Namespace for isolation
|
|
|
|
Returns:
|
|
deleted (bool)
|
|
"""
|
|
# TODO: Implement with Redis/SQLite
|
|
ns_key = f"{namespace or 'default'}:{key}"
|
|
deleted = ns_key in _store
|
|
_store.pop(ns_key, None)
|
|
return ToolResult(success=True, data=deleted)
|