From 079164ccd781a1262c6126a0772cf36898edbfd3 Mon Sep 17 00:00:00 2001 From: lonpatovaadelina Date: Fri, 29 May 2026 06:26:57 +0000 Subject: [PATCH] =?UTF-8?q?MCP-=D1=81=D0=B5=D1=80=D0=B2=D0=B5=D1=80=20?= =?UTF-8?q?=D0=B4=D0=BB=D1=8F=20=D1=83=D0=BF=D1=80=D0=B0=D0=B2=D0=BB=D0=B5?= =?UTF-8?q?=D0=BD=D0=B8=D1=8F=20=D0=BF=D0=B0=D0=BC=D1=8F=D1=82=D1=8C=D1=8E?= =?UTF-8?q?=20=D0=B0=D0=B3=D0=B5=D0=BD=D1=82=D0=B0:=20client.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- client.py | 103 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 103 insertions(+) create mode 100644 client.py diff --git a/client.py b/client.py new file mode 100644 index 0000000..de9619c --- /dev/null +++ b/client.py @@ -0,0 +1,103 @@ +import argparse +import json +import sys +from pathlib import Path + +import requests +from rich import print +from rich.console import Console +from rich.table import Table + +console = Console() + + +def _request(method: str, url: str, **kwargs): + """ + Универсальная обёртка над requests для работы с MCP‑сервером. + """ + try: + resp = requests.request(method, url, timeout=10, **kwargs) + resp.raise_for_status() + except requests.RequestException as exc: + console.print(f"[red]Ошибка запроса: {exc}[/red]") + sys.exit(1) + try: + return resp.json() + except ValueError: + console.print("[red]Ответ не в формате JSON[/red]") + sys.exit(1) + + +def add_memory(server: str, key: str, value: str): + """ + Добавить запись в память агента. + """ + url = f"{server}/memory" + payload = {"key": key, "value": value} + result = _request("POST", url, json=payload) + console.print(f"[green]Запись добавлена: {result}[/green]") + + +def get_memory(server: str, key: str): + """ + Получить значение по ключу. + """ + url = f"{server}/memory/{key}" + result = _request("GET", url) + console.print(f"[cyan]Получено: {result}[/cyan]") + + +def delete_memory(server: str, key: str): + """ + Удалить запись из памяти. + """ + url = f"{server}/memory/{key}" + result = _request("DELETE", url) + console.print(f"[yellow]Удалено: {result}[/yellow]") + + +def list_memory(server: str): + """ + Вывести таблицу всех записей в памяти. + """ + url = f"{server}/memory" + result = _request("GET", url) + table = Table(title="Память агента") + table.add_column("Ключ", style="magenta") + table.add_column("Значение", style="green") + for item in result.get("items", []): + table.add_row(item.get("key", ""), item.get("value", "")) + console.print(table) + + +def main(): + parser = argparse.ArgumentParser(description="Клиент MCP‑сервера для управления памятью агента.") + parser.add_argument("--server", default="http://localhost:8000", help="URL сервера MCP (по умолчанию http://localhost:8000)") + sub = parser.add_subparsers(dest="command", required=True) + + add = sub.add_parser("add", help="Добавить запись") + add.add_argument("key", help="Ключ записи") + add.add_argument("value", help="Значение записи") + + get = sub.add_parser("get", help="Получить запись") + get.add_argument("key", help="Ключ записи") + + delete = sub.add_parser("delete", help="Удалить запись") + delete.add_argument("key", help="Ключ записи") + + list_ = sub.add_parser("list", help="Список всех записей") + + args = parser.parse_args() + + if args.command == "add": + add_memory(args.server, args.key, args.value) + elif args.command == "get": + get_memory(args.server, args.key) + elif args.command == "delete": + delete_memory(args.server, args.key) + elif args.command == "list": + list_memory(args.server) + + +if __name__ == "__main__": + main() \ No newline at end of file