@xdarkicex/openclaw-memory-libravdb 1.4.8 → 1.4.10

package/dist/context-engine.d.ts

@@ -1,6 +1,42 @@
 import type { PluginRuntime } from "./plugin-runtime.js";
 import type { LoggerLike, PluginConfig, RecallCache, SearchResult } from "./types.js";
-import { AssembleContextInternalResponse } from "@xdarkicex/libravdb-contracts";
+import { AssembleContextInternalResponse } from "./generated/libravdb/ipc/v1/rpc_pb.js";
+type KernelCompatibleMessage = {
+    role: string;
+    content: string;
+    id?: string;
+};
+type OpenClawCompatibleMessage = {
+    role: string;
+    content: string;
+    id?: string;
+};
+type OpenClawCompatibleAssembleResult = {
+    messages: OpenClawCompatibleMessage[];
+    estimatedTokens: number;
+    systemPromptAddition: string;
+    debug?: AssembleContextInternalResponse["debug"];
+};
+export declare function normalizeKernelMessage(message: {
+    role: string;
+    content: unknown;
+    id?: string;
+}): KernelCompatibleMessage;
+export declare function normalizeKernelMessages(messages: Array<{
+    role: string;
+    content: unknown;
+    id?: string;
+}>): KernelCompatibleMessage[];
+export declare function normalizeAssembleResult(result: {
+    messages?: Array<{
+        role: string;
+        content?: unknown;
+        id?: string;
+    }>;
+    estimatedTokens?: number;
+    systemPromptAddition?: string;
+    debug?: AssembleContextInternalResponse["debug"];
+}): OpenClawCompatibleAssembleResult;
 export declare function buildContextEngineFactory(runtime: PluginRuntime, cfg: PluginConfig, recallCache: RecallCache<SearchResult>, logger?: LoggerLike): {
     info: {
         id: string;
@@ -19,7 +55,7 @@ export declare function buildContextEngineFactory(runtime: PluginRuntime, cfg: P
         userId?: string;
         message: {
             role: string;
-            content: string;
+            content: unknown;
             id?: string;
         };
         isHeartbeat?: boolean;
@@ -30,12 +66,12 @@ export declare function buildContextEngineFactory(runtime: PluginRuntime, cfg: P
         userId?: string;
         messages: Array<{
             role: string;
-            content: string;
+            content: unknown;
             id?: string;
         }>;
         tokenBudget: number;
         prompt?: string;
-    }): Promise<AssembleContextInternalResponse>;
+    }): Promise<OpenClawCompatibleAssembleResult>;
     compact(args: {
         sessionId: string;
         force?: boolean;
@@ -47,10 +83,11 @@ export declare function buildContextEngineFactory(runtime: PluginRuntime, cfg: P
         userId?: string;
         messages: Array<{
             role: string;
-            content: string;
+            content: unknown;
             id?: string;
         }>;
         prePromptMessageCount?: number;
         isHeartbeat?: boolean;
     }): Promise<any>;
 };
+export {};

package/dist/context-engine.js

@@ -1,3 +1,91 @@
+function describeUnexpectedContent(value) {
+    try {
+        const serialized = JSON.stringify(value);
+        return serialized === undefined ? String(value) : serialized;
+    }
+    catch {
+        return String(value);
+    }
+}
+function stringifyKernelBlock(block) {
+    if (!block || typeof block !== "object") {
+        return "";
+    }
+    const record = block;
+    switch (record.type) {
+        case "text":
+            return typeof record.text === "string" ? record.text : "";
+        case "thinking":
+            return typeof record.thinking === "string" ? record.thinking : "";
+        case "toolCall": {
+            const name = typeof record.name === "string" ? record.name : "tool";
+            const args = record.arguments;
+            let renderedArgs = "";
+            if (typeof args === "string") {
+                renderedArgs = args;
+            }
+            else if (args !== undefined) {
+                try {
+                    renderedArgs = JSON.stringify(args);
+                }
+                catch {
+                    renderedArgs = String(args);
+                }
+            }
+            return renderedArgs ? `[tool:${name}] ${renderedArgs}` : `[tool:${name}]`;
+        }
+        case "image":
+            return "[image omitted]";
+        default:
+            console.warn("[libravdb] unsupported kernel content block", {
+                type: record.type,
+                block: describeUnexpectedContent(record),
+            });
+            return typeof record.text === "string" ? record.text : "";
+    }
+}
+function normalizeKernelContent(content) {
+    if (typeof content === "string") {
+        return content;
+    }
+    if (!Array.isArray(content)) {
+        console.warn("[libravdb] unexpected kernel content shape", {
+            kind: typeof content,
+            value: describeUnexpectedContent(content),
+        });
+        return "";
+    }
+    return content.map(stringifyKernelBlock).filter((part) => part.length > 0).join("\n");
+}
+export function normalizeKernelMessage(message) {
+    return {
+        role: message.role,
+        content: normalizeKernelContent(message.content),
+        ...(typeof message.id === "string" ? { id: message.id } : {}),
+    };
+}
+export function normalizeKernelMessages(messages) {
+    return messages.map((message) => normalizeKernelMessage(message));
+}
+export function normalizeAssembleResult(result) {
+    const messages = Array.isArray(result.messages)
+        ? result.messages.map((message) => ({
+            // OpenClaw replay only expects conversational turns here, so assemble output
+            // is collapsed to user/assistant even though normalizeKernelMessage preserves
+            // richer inbound roles. If kernel.assembleContext starts emitting other roles,
+            // this coercion point is where that contract needs to be revisited.
+            role: message.role === "user" ? "user" : "assistant",
+            content: normalizeKernelContent(message.content),
+            ...(typeof message.id === "string" ? { id: message.id } : {}),
+        }))
+        : [];
+    return {
+        messages,
+        estimatedTokens: typeof result.estimatedTokens === "number" ? result.estimatedTokens : 0,
+        systemPromptAddition: typeof result.systemPromptAddition === "string" ? result.systemPromptAddition : "",
+        ...(result.debug != null ? { debug: result.debug } : {}),
+    };
+}
 export function buildContextEngineFactory(runtime, cfg, recallCache, logger = console) {
     return {
         info: { id: "libravdb-memory", name: "LibraVDB Memory", ownsCompaction: true },
@@ -24,39 +112,44 @@ export function buildContextEngineFactory(runtime, cfg, recallCache, logger = co
             return await rpc.call("bootstrap_session_kernel", args);
         },
         async ingest(args) {
+            const message = normalizeKernelMessage(args.message);
             const kernel = runtime.getKernel();
             if (kernel) {
                 return await kernel.ingestMessage({
                     sessionId: args.sessionId,
                     sessionKey: args.sessionKey,
                     userId: args.userId,
-                    message: args.message,
+                    message,
                     isHeartbeat: args.isHeartbeat,
                 });
             }
             const rpc = await runtime.getRpc();
-            return await rpc.call("ingest_message_kernel", args);
+            return await rpc.call("ingest_message_kernel", {
+                ...args,
+                message,
+            });
         },
         async assemble(args) {
+            const messages = normalizeKernelMessages(args.messages);
             const kernel = runtime.getKernel();
             if (kernel) {
-                return await kernel.assembleContext({
+                return normalizeAssembleResult(await kernel.assembleContext({
                     sessionId: args.sessionId,
                     sessionKey: args.sessionKey,
                     userId: args.userId,
                     queryText: args.prompt ?? "",
-                    visibleMessages: args.messages,
+                    visibleMessages: messages,
                     tokenBudget: args.tokenBudget,
                     config: {},
                     emitDebug: true
-                });
+                }));
             }
             const rpc = await runtime.getRpc();
             const resp = await rpc.call("assemble_context_internal", {
                 sessionId: args.sessionId,
                 sessionKey: args.sessionKey,
                 userId: args.userId,
-                messages: args.messages,
+                messages,
                 tokenBudget: args.tokenBudget,
                 prompt: args.prompt,
                 emitDebug: true,
@@ -92,7 +185,7 @@ export function buildContextEngineFactory(runtime, cfg, recallCache, logger = co
                     ingestionGateThreshold: cfg.ingestionGateThreshold,
                 },
             });
-            return resp;
+            return normalizeAssembleResult(resp);
         },
         async compact(args) {
             const kernel = runtime.getKernel();
@@ -107,19 +200,23 @@ export function buildContextEngineFactory(runtime, cfg, recallCache, logger = co
             return await rpc.call("compact_session", args);
         },
         async afterTurn(args) {
+            const messages = normalizeKernelMessages(args.messages);
             const kernel = runtime.getKernel();
             if (kernel) {
                 return await kernel.afterTurn({
                     sessionId: args.sessionId,
                     sessionKey: args.sessionKey,
                     userId: args.userId,
-                    messages: args.messages,
+                    messages,
                     prePromptMessageCount: args.prePromptMessageCount,
                     isHeartbeat: args.isHeartbeat,
                 });
             }
             const rpc = await runtime.getRpc();
-            return await rpc.call("after_turn_kernel", args);
+            return await rpc.call("after_turn_kernel", {
+                ...args,
+                messages,
+            });
         }
     };
 }