Initial commit

skills/memory/memory_cli.py

@@ -0,0 +1,133 @@
+#!/usr/bin/env python3
+"""Memory Discovery CLI for AI Runtime
+
+提供基于 SQL 风格参数的情景记忆 (episodic) 查询能力。
+
+示例：
+    python3 .ai-runtime/memory/memory_cli.py query \
+        --select "id,timestamp,title" \
+        --where "date>='2025-11-14' AND tags CONTAINS 'architecture'" \
+        --order-by "timestamp desc" \
+        --limit 20
+"""
+
+from __future__ import annotations
+
+import argparse
+import sys
+from pathlib import Path
+
+# 确保可以从当前目录导入 memory_discovery
+CURRENT_DIR = Path(__file__).parent
+sys.path.insert(0, str(CURRENT_DIR))
+
+from memory_discovery import MemoryDiscovery  # type: ignore
+
+
+class MemoryCLI:
+    """Episodic 记忆查询 CLI 接口"""
+
+    def __init__(self, memory_root: Path) -> None:
+        self.memory_root = memory_root
+        self.discovery = MemoryDiscovery(memory_root)
+
+    # ------------------------------------------------------------------
+    # 外部入口
+    # ------------------------------------------------------------------
+    def run(self, args: None | list[str] = None) -> int:
+        parser = self._create_parser()
+        parsed = parser.parse_args(args)
+
+        if not parsed.command:
+            parser.print_help()
+            return 0
+
+        if parsed.command == "query":
+            return self._cmd_query(parsed)
+
+        parser.print_help()
+        return 0
+
+    # ------------------------------------------------------------------
+    # Parser 定义
+    # ------------------------------------------------------------------
+    def _create_parser(self) -> argparse.ArgumentParser:
+        parser = argparse.ArgumentParser(
+            description="Memory discovery and query tool (episodic)",
+            formatter_class=argparse.RawDescriptionHelpFormatter,
+            epilog="""示例:
+  python3 .ai-runtime/memory/memory_cli.py query \
+    --select "id,timestamp,title" \
+    --where "level='day' AND date>='2025-11-14'" \
+    --order-by "timestamp desc" \
+    --limit 20
+            """,
+        )
+
+        subparsers = parser.add_subparsers(dest="command", help="可用命令")
+        self._add_query_parser(subparsers)
+
+        return parser
+
+    def _add_query_parser(self, subparsers: argparse._SubParsersAction) -> None:
+        query = subparsers.add_parser("query", help="查询 episodic 记忆事件")
+        query.add_argument(
+            "--select",
+            help="SELECT 字段列表，逗号分隔 (默认: id,timestamp,title)",
+            default="id,timestamp,title",
+        )
+        query.add_argument(
+            "--where",
+            help="SQL 风格 WHERE 条件，仅支持 AND / = / != / >= / <= / tags CONTAINS",
+        )
+        query.add_argument(
+            "--order-by",
+            dest="order_by",
+            help="排序字段，如 'timestamp desc' 或 'date asc'",
+        )
+        query.add_argument(
+            "--limit",
+            type=int,
+            default=50,
+            help="LIMIT 结果数量 (默认 50)",
+        )
+        query.add_argument(
+            "--offset",
+            type=int,
+            default=0,
+            help="OFFSET 偏移量 (默认 0)",
+        )
+        query.add_argument(
+            "--format",
+            choices=["table", "json"],
+            default="table",
+            help="输出格式 (table/json)",
+        )
+
+    # ------------------------------------------------------------------
+    # 命令实现
+    # ------------------------------------------------------------------
+    def _cmd_query(self, args: argparse.Namespace) -> int:
+        # 解析 select 字段
+        select_fields = [f.strip() for f in (args.select or "").split(",") if f.strip()]
+
+        events = self.discovery.query(
+            where=args.where,
+            order_by=args.order_by,
+            limit=args.limit,
+            offset=args.offset,
+        )
+
+        output = self.discovery.format_events(events, select=select_fields, format_type=args.format)
+        print(output)
+        return 0
+
+
+def main() -> int:
+    memory_root = CURRENT_DIR  # .ai-runtime/memory
+    cli = MemoryCLI(memory_root)
+    return cli.run()
+
+
+if __name__ == "__main__":  # pragma: no cover
+    sys.exit(main())