mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
Paul Hernandez
55d675e278
Signed-off-by: phernandez <paul@basicmachines.co> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
88 lines
3.3 KiB
Python
88 lines
3.3 KiB
Python
"""CLI routing utilities for --local/--cloud flag handling.
|
|
|
|
This module provides utilities for CLI commands to override the default routing
|
|
behavior (determined by cloud_mode_enabled in config). This allows users to:
|
|
|
|
1. Use local MCP server even when cloud mode is enabled
|
|
2. Force local routing for specific CLI commands with --local flag
|
|
3. Force cloud routing with --cloud flag (requires authentication)
|
|
|
|
The routing is controlled via environment variables:
|
|
- BASIC_MEMORY_FORCE_LOCAL: When "true", forces local ASGI transport
|
|
- BASIC_MEMORY_EXPLICIT_ROUTING: When "true", signals that --local/--cloud
|
|
was explicitly passed, overriding per-project routing in get_client()
|
|
- These are checked in basic_memory.mcp.async_client.get_client()
|
|
"""
|
|
|
|
import os
|
|
from contextlib import contextmanager
|
|
from typing import Generator
|
|
|
|
|
|
@contextmanager
|
|
def force_routing(local: bool = False, cloud: bool = False) -> Generator[None, None, None]:
|
|
"""Context manager to temporarily override routing mode.
|
|
|
|
Sets environment variables that are checked by get_client() to determine
|
|
whether to use local ASGI transport or cloud proxy transport.
|
|
|
|
When either flag is set, BASIC_MEMORY_EXPLICIT_ROUTING is also set so
|
|
that get_client() skips per-project routing and honors the flag directly.
|
|
This only affects CLI commands — the MCP server sets FORCE_LOCAL directly
|
|
(without EXPLICIT_ROUTING), so per-project routing still works for MCP tools.
|
|
|
|
Args:
|
|
local: If True, force local ASGI transport (ignores cloud_mode_enabled)
|
|
cloud: If True, clear force_local to allow cloud routing
|
|
|
|
Usage:
|
|
with force_routing(local=True):
|
|
# All API calls will use local ASGI transport
|
|
await some_api_call()
|
|
|
|
Raises:
|
|
ValueError: If both local and cloud are True
|
|
"""
|
|
if local and cloud:
|
|
raise ValueError("Cannot specify both --local and --cloud")
|
|
|
|
# Save original values
|
|
original_force_local = os.environ.get("BASIC_MEMORY_FORCE_LOCAL")
|
|
original_explicit = os.environ.get("BASIC_MEMORY_EXPLICIT_ROUTING")
|
|
|
|
try:
|
|
if local:
|
|
os.environ["BASIC_MEMORY_FORCE_LOCAL"] = "true"
|
|
os.environ["BASIC_MEMORY_EXPLICIT_ROUTING"] = "true"
|
|
elif cloud:
|
|
# Ensure force_local is NOT set, let cloud_mode_enabled take effect
|
|
os.environ.pop("BASIC_MEMORY_FORCE_LOCAL", None)
|
|
os.environ["BASIC_MEMORY_EXPLICIT_ROUTING"] = "true"
|
|
# If neither is set, don't change anything (use default behavior)
|
|
yield
|
|
finally:
|
|
# Restore original values
|
|
if original_force_local is None:
|
|
os.environ.pop("BASIC_MEMORY_FORCE_LOCAL", None)
|
|
else:
|
|
os.environ["BASIC_MEMORY_FORCE_LOCAL"] = original_force_local
|
|
|
|
if original_explicit is None:
|
|
os.environ.pop("BASIC_MEMORY_EXPLICIT_ROUTING", None)
|
|
else:
|
|
os.environ["BASIC_MEMORY_EXPLICIT_ROUTING"] = original_explicit
|
|
|
|
|
|
def validate_routing_flags(local: bool, cloud: bool) -> None:
|
|
"""Validate that --local and --cloud flags are not both specified.
|
|
|
|
Args:
|
|
local: Value of --local flag
|
|
cloud: Value of --cloud flag
|
|
|
|
Raises:
|
|
ValueError: If both flags are True
|
|
"""
|
|
if local and cloud:
|
|
raise ValueError("Cannot specify both --local and --cloud flags")
|