From ab0250e0a6e6bfb1344fb3e51e7fc167cafe22ff Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=9A=D0=B8=D1=80=D0=B8=D0=BB=D0=BB=20=D0=9A=D1=83=D1=82?= =?UTF-8?q?=D0=BB=D0=B0=D1=85=D0=BC=D0=B5=D1=82=D0=BE=D0=B2?= Date: Mon, 25 May 2026 23:25:44 +0000 Subject: [PATCH] Restore solution for 69f8e929da860fb4533faa2a --- memory_server.py | 85 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 85 insertions(+) create mode 100644 memory_server.py diff --git a/memory_server.py b/memory_server.py new file mode 100644 index 0000000..6e78113 --- /dev/null +++ b/memory_server.py @@ -0,0 +1,85 @@ +from fastmcp import FastMCP +import json +from datetime import datetime +from pathlib import Path +from typing import Any, Optional, List, Dict + +class MemoryServer: + def __init__(self): + self.mcp = FastMCP("Memory-Server") + self.storage_path = Path("./memory_data.json") + self._register_tools() + + def _load_memory(self) -> dict: + if not self.storage_path.exists(): + return {} + with open(self.storage_path, 'r', encoding='utf-8') as f: + return json.load(f) + + def _save_memory(self, data: dict): + self.storage_path.parent.mkdir(parents=True, exist_ok=True) + with open(self.storage_path, 'w', encoding='utf-8') as f: + json.dump(data, f, indent=2, ensure_ascii=False) + + def _register_tools(self): + @self.mcp.tool() + def save(key: str, value: Any) -> bool: + """Сохраняет значение по ключу в память сервера.""" + data = self._load_memory() + data[key] = {"value": value, "timestamp": datetime.utcnow().isoformat()} + self._save_memory(data) + return True + + @self.mcp.tool() + def get(key: str) -> Optional[Dict[str, Any]]: + """Возвращает значение по ключу с метаданными.""" + data = self._load_memory() + entry = data.get(key) + if entry is None: + return None + return {"key": key, "value": entry["value"], "timestamp": entry["timestamp"]} + + @self.mcp.tool() + def delete(key: str) -> bool: + """Удаляет ключ из памяти сервера.""" + data = self._load_memory() + if key in data: + del data[key] + self._save_memory(data) + return True + return False + + @self.mcp.tool() + def list_keys(pattern: str = "*") -> List[str]: + """Возвращает список всех ключей с поддержкой wildcard-паттерна.""" + import fnmatch + data = self._load_memory() + return [k for k in data.keys() if fnmatch.fnmatch(k, pattern)] + + @self.mcp.tool() + def save_with_namespace(key: str, value: Any, namespace: str = "default") -> bool: + """Сохраняет значение с указанием пространства имён.""" + full_key = f"{namespace}:{key}" + data = self._load_memory() + data[full_key] = {"value": value, "timestamp": datetime.utcnow().isoformat()} + self._save_memory(data) + return True + + @self.mcp.tool() + def get_by_namespace(namespace: str = "default") -> List[Dict[str, Any]]: + """Возвращает все ключи из указанного namespace.""" + prefix = f"{namespace}:" + data = self._load_memory() + result = [] + for k, v in data.items(): + if k.startswith(prefix): + result.append({"key": k, "value": v["value"], "timestamp": v["timestamp"]}) + return result + +if __name__ == "__main__": + server = MemoryServer() + server.mcp.run( + transport="stdio", + show_banner=False, + log_level="ERROR" + ) \ No newline at end of file