memory-search-plugin 0.3.0 → 0.5.0

package/gateway-client.ts

@@ -57,6 +57,31 @@ export interface MemoryGetResponse {
   path: string;
 }
 
+export interface MemoryRawSnapshotsRequest {
+  release_name: string;
+  start_time: string;   // ISO 8601
+  end_time: string;      // ISO 8601
+  path_prefix?: string;  // e.g. "memory/"
+  page?: number;
+  page_size?: number;
+}
+
+export interface MemoryRawSnapshot {
+  id: string;
+  path: string;
+  content: string;
+  release_name: string;
+  synced_at: string;
+  [key: string]: unknown;
+}
+
+export interface MemoryRawSnapshotsResponse {
+  records: MemoryRawSnapshot[];
+  total: number;
+  page: number;
+  page_size: number;
+}
+
 // ── Gateway Client ─────────────────────────────────────
 
 export interface GatewayClientConfig {
@@ -67,6 +92,7 @@ export interface GatewayClientConfig {
 export interface GatewayClient {
   callGatewaySearch(options: MemorySearchRequest): Promise<MemorySearchResponse>;
   callGatewayGet(options: MemoryGetRequest): Promise<MemoryGetResponse>;
+  callGatewayRawSnapshots(options: MemoryRawSnapshotsRequest): Promise<MemoryRawSnapshotsResponse>;
 }
 
 export function createGatewayClient(config: GatewayClientConfig): GatewayClient {
@@ -109,5 +135,21 @@ export function createGatewayClient(config: GatewayClientConfig): GatewayClient
       }
       return response.json() as Promise<MemoryGetResponse>;
     },
+
+    async callGatewayRawSnapshots(options) {
+      const url = `${baseUrl}/api/memory/raw_snapshots`;
+      const response = await fetch(url, {
+        method: "POST",
+        headers: { "Content-Type": "application/json", ...authHeaders },
+        body: JSON.stringify(options),
+      });
+      if (!response.ok) {
+        const errorBody = await response.text().catch(() => "");
+        throw new Error(
+          `Gateway raw_snapshots failed: ${response.status} ${response.statusText} ${errorBody}`
+        );
+      }
+      return response.json() as Promise<MemoryRawSnapshotsResponse>;
+    },
   };
 }

package/index.ts

@@ -28,7 +28,7 @@ import { definePluginEntry } from "openclaw/plugin-sdk/plugin-entry";
 import { createGatewayClient } from "./gateway-client";
 import { resolveIdentityFromParams, resolveIdentity, extractAgentId } from "./identity";
 import type { IdentityParams } from "./identity";
-import type { MemorySearchResult } from "./gateway-client";
+import type { MemorySearchResult, MemoryRawSnapshot } from "./gateway-client";
 
 // ── Prompt Section ──────────────────────────────────────
 // 告诉 LLM 如何使用 memory 工具（和 memory-core 的逻辑一致）
@@ -87,7 +87,7 @@ const buildPromptSection = ({
   return lines;
 };
 
-// ── 路径判断 ───────────────────────────────────────────���
+// ── 路径判断 ────────────────────────────────────────────
 
 /**
  * 判断是否是 MEMORY 路径
@@ -136,6 +136,23 @@ function formatSearchResults(results: MemorySearchResult[]): string {
     .join("\n\n");
 }
 
+/**
+ * 将 raw_snapshots 查询结果格式化为可读文本
+ * 包含 path、synced_at、release_name 和内容摘要
+ */
+function formatRawSnapshots(records: MemoryRawSnapshot[]): string {
+  if (records.length === 0) {
+    return "No matching raw snapshot records found.";
+  }
+
+  return records
+    .map((r, i) => {
+      const header = `[${i + 1}] path: ${r.path} | release: ${r.release_name} | synced_at: ${r.synced_at}`;
+      return `${header}\n${r.content}`;
+    })
+    .join("\n\n");
+}
+
 // ── 插件定义 ────────────────────────────────────────────
 
 export default definePluginEntry({
@@ -270,7 +287,7 @@ export default definePluginEntry({
 
               try {
                 const fallbackTool =
-                  api.runtime.tools.createMemorySearchTool({
+                  api.runtime?.tools?.createMemorySearchTool({
                     config: ctx.config,
                     agentSessionKey: ctx.sessionKey,
                   });
@@ -309,12 +326,6 @@ export default definePluginEntry({
         const fallbackIdentity = ctx.sessionKey ? resolveIdentity(ctx.sessionKey) : null;
         const fallbackAgentId = ctx.sessionKey ? extractAgentId(ctx.sessionKey) : null;
 
-        // 获取原始 memory_get 实现，用于非 MEMORY 路径的 passthrough
-        const originalGetTool = api.runtime.tools.createMemoryGetTool({
-          config: ctx.config,
-          agentSessionKey: ctx.sessionKey,
-        });
-
         return {
           name: "memory_get",
           description:
@@ -376,6 +387,17 @@ export default definePluginEntry({
               };
             }
 
+            // 懒加载原始 memory_get 实现，用于非 MEMORY 路径的 passthrough
+            let originalGetTool: any = null;
+            try {
+              originalGetTool = api.runtime?.tools?.createMemoryGetTool({
+                config: ctx.config,
+                agentSessionKey: ctx.sessionKey,
+              });
+            } catch {
+              // memory-core 被替换后 api.runtime.tools 不可用，忽略
+            }
+
             // ── 路径分流 ──
             if (isMemoryPath(path)) {
               const agentId = params.agent_id?.trim() || fallbackAgentId || "main";
@@ -460,11 +482,123 @@ export default definePluginEntry({
     );
 
     // ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
-    // 4. 保留 CLI 命令（和 memory-core 一致）
+    // 4. 注册 memory_raw_snapshots_query → 查询 memory_raw_snapshots 表
+    //    按 synced_at 时间段 + path 为 memory/xxx + release_name 过滤
+    // ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+    api.registerTool(
+      (_ctx: any) => {
+        return {
+          name: "memory_raw_snapshots_query",
+          description:
+            "Query raw memory snapshot records from the memory_raw_snapshots table. " +
+            "Filters by synced_at time range, release_name, and only returns records whose path starts with 'memory/' " +
+            "(excluding MEMORY.md, USER.md, etc.).",
+          parameters: {
+            type: "object" as const,
+            properties: {
+              release_name: {
+                type: "string" as const,
+                description: "The release_name to filter by (e.g. 'lobster_xxx')",
+              },
+              start_time: {
+                type: "string" as const,
+                description: "Start time in ISO 8601 format (e.g. '2025-05-01T00:00:00Z')",
+              },
+              end_time: {
+                type: "string" as const,
+                description: "End time in ISO 8601 format (e.g. '2025-05-07T23:59:59Z')",
+              },
+              page: {
+                type: "number" as const,
+                description: "Page number for pagination (default 1)",
+              },
+              page_size: {
+                type: "number" as const,
+                description: "Number of records per page (default 50, max 200)",
+              },
+            },
+            required: ["release_name", "start_time", "end_time"],
+          },
+
+          async execute(
+            _toolCallId: string,
+            params: {
+              release_name?: string;
+              start_time?: string;
+              end_time?: string;
+              page?: number;
+              page_size?: number;
+            }
+          ) {
+            const releaseName = params.release_name?.trim() || "";
+            const startTime = params.start_time?.trim() || "";
+            const endTime = params.end_time?.trim() || "";
+
+            if (!releaseName || !startTime || !endTime) {
+              return {
+                content: [
+                  {
+                    type: "text" as const,
+                    text: "Missing required parameters: release_name, start_time, and end_time are all required.",
+                  },
+                ],
+              };
+            }
+
+            console.log(
+              `[memory-search-plugin] raw_snapshots_query: release_name=${releaseName} ` +
+              `start_time=${startTime} end_time=${endTime} page=${params.page || 1} page_size=${params.page_size || 50}`
+            );
+
+            try {
+              const data = await gateway.callGatewayRawSnapshots({
+                release_name: releaseName,
+                start_time: startTime,
+                end_time: endTime,
+                path_prefix: "memory/",
+                page: params.page || 1,
+                page_size: params.page_size || 50,
+              });
+
+              const text = formatRawSnapshots(data.records);
+              const summary = `Found ${data.total} record(s) (page ${data.page}, page_size ${data.page_size}).\n\n${text}`;
+
+              return {
+                content: [{ type: "text" as const, text: summary }],
+                details: {
+                  records: data.records,
+                  total: data.total,
+                  page: data.page,
+                  page_size: data.page_size,
+                },
+              };
+            } catch (err: any) {
+              console.error("[memory-search-plugin] raw_snapshots_query failed:", err.message);
+              return {
+                content: [
+                  {
+                    type: "text" as const,
+                    text: `Raw snapshots query failed: ${err.message}`,
+                  },
+                ],
+              };
+            }
+          },
+        };
+      },
+      { names: ["memory_raw_snapshots_query"] }
+    );
+
+    // ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+    // 5. 保留 CLI 命令（和 memory-core 一致）
     // ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
     api.registerCli(
       ({ program }: any) => {
-        api.runtime.tools.registerMemoryCli(program);
+        try {
+          api.runtime?.tools?.registerMemoryCli(program);
+        } catch {
+          // memory-core 被替换后 api.runtime.tools 不可用，忽略
+        }
       },
       { commands: ["memory"] }
     );

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "memory-search-plugin",
-  "version": "0.3.0",
+  "version": "0.5.0",
   "type": "module",
   "main": "index.ts",
   "files": [