refactor: consolidate agent/editor plugins under integrations/ (#5491)
Co-authored-by: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,104 @@
|
||||
"""Resolve mem0 identity: API key, user_id, and settings.
|
||||
|
||||
API key resolution (first non-empty wins):
|
||||
1. MEM0_API_KEY env var (explicit / shell profile)
|
||||
2. CLAUDE_PLUGIN_OPTION_API_KEY (injected by Claude Code userConfig)
|
||||
3. CLAUDE_PLUGIN_OPTION_MEM0_API_KEY (legacy userConfig)
|
||||
4. Extract from shell profile files (~/.zshrc, ~/.bashrc, etc.)
|
||||
Desktop app doesn't inherit shell env — this covers users who
|
||||
set MEM0_API_KEY in their profile but use the Desktop app.
|
||||
|
||||
User ID resolution:
|
||||
1. MEM0_USER_ID env var (explicit override)
|
||||
2. $USER, else "default"
|
||||
|
||||
Settings resolution:
|
||||
~/.mem0/settings.json (user-editable, falls back to defaults)
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
import re
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
def _extract_key_from_shell_profiles() -> str:
|
||||
"""Extract MEM0_API_KEY from shell profile files.
|
||||
|
||||
The Desktop app only reads PATH from shell profiles — env vars like
|
||||
MEM0_API_KEY are not inherited. This handles the common
|
||||
``export MEM0_API_KEY=...`` pattern without sourcing the full profile.
|
||||
"""
|
||||
profiles = [".zshrc", ".bashrc", ".zprofile", ".bash_profile", ".profile"]
|
||||
pattern = re.compile(r'^\s*(?:export\s+)?MEM0_API_KEY=(.+)$')
|
||||
|
||||
for name in profiles:
|
||||
path = Path.home() / name
|
||||
if not path.is_file():
|
||||
continue
|
||||
try:
|
||||
for line in path.read_text(encoding="utf-8", errors="replace").splitlines():
|
||||
m = pattern.match(line)
|
||||
if not m:
|
||||
continue
|
||||
value = m.group(1).strip()
|
||||
value = re.sub(r'#.*$', '', value).strip()
|
||||
value = value.strip("\"'")
|
||||
if value and not value.startswith("$"):
|
||||
return value
|
||||
except OSError:
|
||||
continue
|
||||
return ""
|
||||
|
||||
|
||||
def resolve_api_key() -> str:
|
||||
key = os.environ.get("MEM0_API_KEY", "").strip()
|
||||
if key:
|
||||
return key
|
||||
key = os.environ.get("CLAUDE_PLUGIN_OPTION_API_KEY", "").strip()
|
||||
if key:
|
||||
return key
|
||||
key = os.environ.get("CLAUDE_PLUGIN_OPTION_MEM0_API_KEY", "").strip()
|
||||
if key:
|
||||
return key
|
||||
key = _extract_key_from_shell_profiles()
|
||||
if key:
|
||||
return key
|
||||
return ""
|
||||
|
||||
|
||||
def resolve_user_id() -> str:
|
||||
explicit = os.environ.get("MEM0_USER_ID", "").strip()
|
||||
if explicit:
|
||||
return explicit
|
||||
return os.environ.get("USER") or "default"
|
||||
|
||||
|
||||
def resolve_config() -> dict:
|
||||
"""Resolve settings from ~/.mem0/settings.json (primary) with env var overrides."""
|
||||
try:
|
||||
from load_settings import load_settings
|
||||
return load_settings()
|
||||
except ImportError:
|
||||
return {
|
||||
"auto_save": True,
|
||||
"auto_search": True,
|
||||
"search_limit": 10,
|
||||
"retention_session_days": 90,
|
||||
"confidence_threshold": 0.3,
|
||||
"debug": False,
|
||||
}
|
||||
|
||||
|
||||
try:
|
||||
from _project import resolve_branch, resolve_project_id, save_project_mapping
|
||||
except ImportError:
|
||||
def resolve_project_id(cwd: str | None = None) -> str:
|
||||
return os.path.basename(cwd or os.getcwd())
|
||||
|
||||
def resolve_branch(cwd: str | None = None) -> str:
|
||||
return "unknown"
|
||||
|
||||
def save_project_mapping(cwd: str, project_id: str) -> None:
|
||||
pass
|
||||
Reference in New Issue
Block a user