rice-node-sdk 1.0.6 → 1.0.8

package/README.md

@@ -106,19 +106,31 @@ You can load a specific config file by passing the path to the `Client` construc
 const client = new Client({ configPath: "./config/prod.rice.config.js" });
 ```
 
-#### Managing Run IDs (Multi-User/Multi-Agent)
+#### Managing Run IDs (State Isolation Boundary)
 
-For State memory, the `runId` determines the session or agent context. You can switch run IDs dynamically to manage memory for different users or agents.
+For State memory, `runId` is the isolation boundary. Use one unique run ID per execution/session/test, perform all state operations with that run, then clean up with `deleteRun()`.
 
 ```typescript
-// Option A: Set globally in constructor
-const client = new Client({ runId: "user-123-session" });
+const runId = `session-${Date.now()}`;
+const client = new Client({ runId });
+await client.connect();
+
+await client.state.focus("working memory item");
+await client.state.commit("input", "outcome");
+await client.state.setVariable("k", { v: 1 });
+await client.state.addGoal("Do X", "high");
+await client.state.defineConcept("Profile", { type: "object" });
 
-// Option B: Switch dynamically
-client.state.setRunId("user-456-session");
-await client.state.focus("New task for user 456");
+// cleanup this run
+await client.state.deleteRun();
 ```
 
+Practical rule: use `runId` for isolation. Do not emulate run isolation with metadata filters.
+
+Reference examples:
+- `examples/check_state_run_isolation.ts`
+- `examples/check_state_all_memory_isolation.ts`
+
 ## State Features
 
 The State service provides comprehensive AI agent memory and cognition capabilities.
@@ -266,6 +278,25 @@ console.log(cycleResult.planningTimeMs);
 const history = await client.state.getCycleHistory(10);
 ```
 
+### Pub/Sub (Real-time Events)
+
+Subscribe to real-time events from the State service, such as variable updates or memory commits. This is useful for multi-agent coordination.
+
+```typescript
+// Subscribe to variable updates
+const stream = client.state.subscribe(["VariableUpdate"]);
+
+stream.on("data", (event) => {
+  console.log("Received event:", event.type);
+  if (event.type === "VariableUpdate") {
+    const variable = JSON.parse(event.payload);
+    console.log(`Variable ${variable.name} updated to:`, variable.value_json);
+  }
+});
+
+stream.on("error", (err) => console.error("Stream error:", err));
+```
+
 ## AI Tool Definitions
 
 The SDK provides pre-built tool definitions tailored for popular LLM providers. These tools map directly to State memory operations.
@@ -408,6 +439,7 @@ if (call) {
 | `getActionLog`    | Get the action log for the current run           | `client.state.getActionLog()`    |
 | `runCycle`        | Run a decision cycle with action candidates      | `client.state.runCycle()`        |
 | `getCycleHistory` | Get history of decision cycles                   | `client.state.getCycleHistory()` |
+| `subscribe`       | Subscribe to real-time state events              | `client.state.subscribe()`       |
 
 ## API Reference
 
@@ -415,7 +447,12 @@ if (call) {
 
 ```typescript
 class Client {
-  constructor(options?: { configPath?: string; runId?: string });
+  constructor(options?: {
+    configPath?: string;
+    runId?: string;
+    stateRunId?: string;
+    storageRunId?: string;
+  });
   async connect(): Promise<void>;
   get storage(): StorageClient;
   get state(): StateClient;
@@ -489,8 +526,12 @@ interface StateClient {
   ): Promise<CycleResult>;
   getCycleHistory(limit?: number): Promise<CycleResult[]>;
 
+  // Events
+  subscribe(eventTypes?: string[]): NodeJS.EventEmitter;
+
   // Session Management
   setRunId(runId: string): void;
+
   deleteRun(): Promise<boolean>;
 
   // Skills

package/dist/Client.d.ts

@@ -3,6 +3,8 @@ import { StateClient } from "./state";
 export interface ClientOptions {
     configPath?: string;
     runId?: string;
+    stateRunId?: string;
+    storageRunId?: string;
 }
 export declare class Client {
     private _config;

package/dist/Client.js

@@ -77,9 +77,12 @@ class Client {
             const httpPort = process.env.STORAGE_HTTP_PORT
                 ? parseInt(process.env.STORAGE_HTTP_PORT, 10)
                 : 3000;
-            // We assume STORAGE_INSTANCE_URL points to the gRPC port
-            // Pass token to constructor initially (if it's a valid token, it works; if it's a password, we login)
-            this._storage = new RiceDBClient_1.RiceDBClient(host, "auto", port, httpPort, token);
+            const storageRunId = this.options.storageRunId ||
+                this.options.runId ||
+                process.env.STORAGE_RUN_ID;
+            // We assume STORAGE_INSTANCE_URL points to the gRPC port.
+            // Pass token to constructor initially (if it's a valid token, it works; if it's a password, we login).
+            this._storage = new RiceDBClient_1.RiceDBClient(host, "auto", port, httpPort, token, storageRunId);
             await this._storage.connect();
             if (token) {
                 try {
@@ -100,7 +103,10 @@ class Client {
         if (this._config.state?.enabled) {
             const address = process.env.STATE_INSTANCE_URL || "localhost:50051";
             const token = process.env.STATE_AUTH_TOKEN;
-            const runId = this.options.runId || process.env.STATE_RUN_ID || "default";
+            const runId = this.options.stateRunId ||
+                this.options.runId ||
+                process.env.STATE_RUN_ID ||
+                "default";
             this._state = new state_1.StateClient(address, token, runId);
         }
     }

package/dist/state/index.d.ts

@@ -148,4 +148,13 @@ export declare class StateClient {
      * @returns A promise that resolves to an array of cycle records.
      */
     getCycleHistory(limit?: number): Promise<any[]>;
+    /**
+     * Subscribe to real-time events for the current run_id.
+     * Returns a gRPC ClientReadableStream that emits 'data', 'error', and 'end' events.
+     *
+     * usage:
+     * const stream = client.subscribe();
+     * stream.on('data', (event: any) => { ... });
+     */
+    subscribe(eventTypes?: string[]): any;
 }

package/dist/state/index.js

@@ -506,5 +506,25 @@ class StateClient {
             });
         });
     }
+    // ==========================================================================
+    // Events (Pub/Sub)
+    // ==========================================================================
+    /**
+     * Subscribe to real-time events for the current run_id.
+     * Returns a gRPC ClientReadableStream that emits 'data', 'error', and 'end' events.
+     *
+     * usage:
+     * const stream = client.subscribe();
+     * stream.on('data', (event: any) => { ... });
+     */
+    subscribe(eventTypes = []) {
+        const request = {
+            run_id: this.runId,
+            event_types: eventTypes,
+        };
+        // For streaming RPCs, we don't provide a callback.
+        // The call object itself is the stream.
+        return this.client.Subscribe(request, this.metadata);
+    }
 }
 exports.StateClient = StateClient;

package/dist/state/proto/state.proto

@@ -41,6 +41,9 @@ service Cortex {
 
   // Management
   rpc DeleteRun(RunRequest) returns (Ack);
+
+  // Events
+  rpc Subscribe(SubscribeRequest) returns (stream Event);
 }
 
 message FocusRequest {
@@ -292,3 +295,17 @@ message CycleHistoryRequest {
 message CycleHistoryResponse {
   repeated CycleResponse cycles = 1;
 }
+
+message SubscribeRequest {
+  string run_id = 1;
+  repeated string event_types = 2; // Optional filter e.g. ["Focus", "Commit"]
+}
+
+message Event {
+  string id = 1;
+  string type = 2; // "Focus", "Commit", "VariableUpdate", "GoalUpdate"
+  string run_id = 3;
+  string agent_id = 4;
+  string payload = 5; // JSON string of the object involved
+  string created_at = 6;
+}

package/dist/storage/client/BaseClient.d.ts

@@ -41,11 +41,16 @@ export declare abstract class BaseRiceDBClient {
     abstract deleteUser(username: string): Promise<boolean>;
     abstract getUser(username: string): Promise<any>;
     abstract listUsers(): Promise<any[]>;
-    abstract insert(nodeId: Long | number | string, text: string, metadata: any, userId?: Long | number | string, sessionId?: string, embedding?: number[]): Promise<InsertResult>;
+    abstract insert(nodeId: Long | number | string, text: string, metadata: any, userId?: Long | number | string, sessionId?: string, embedding?: number[], runId?: string): Promise<InsertResult>;
     abstract search(query: string, userId: Long | number | string, k?: number, sessionId?: string, filter?: {
         [key: string]: any;
-    }, queryEmbedding?: number[]): Promise<SearchResultItem[]>;
+    }, queryEmbedding?: number[], runId?: string): Promise<SearchResultItem[]>;
     abstract delete(nodeId: Long | number | string, sessionId?: string): Promise<boolean>;
+    abstract deleteRun(runId?: string): Promise<{
+        success: boolean;
+        message: string;
+        count: Long;
+    }>;
     abstract createSession(parentSessionId?: string): Promise<string>;
     abstract snapshotSession(sessionId: string, path: string): Promise<boolean>;
     abstract loadSession(path: string): Promise<string>;

package/dist/storage/client/GrpcClient.d.ts

@@ -15,11 +15,16 @@ export declare class GrpcClient extends BaseRiceDBClient {
     deleteUser(username: string): Promise<boolean>;
     getUser(username: string): Promise<any>;
     listUsers(): Promise<any[]>;
-    insert(nodeId: Long | number | string, text: string, metadata: any, userId?: Long | number | string, sessionId?: string, embedding?: number[]): Promise<InsertResult>;
+    insert(nodeId: Long | number | string, text: string, metadata: any, userId?: Long | number | string, sessionId?: string, embedding?: number[], runId?: string): Promise<InsertResult>;
     search(query: string, userId: Long | number | string, k?: number, sessionId?: string, filter?: {
         [key: string]: any;
-    }, queryEmbedding?: number[]): Promise<SearchResultItem[]>;
+    }, queryEmbedding?: number[], runId?: string): Promise<SearchResultItem[]>;
     delete(nodeId: Long | number | string, sessionId?: string): Promise<boolean>;
+    deleteRun(runId?: string): Promise<{
+        success: boolean;
+        message: string;
+        count: Long;
+    }>;
     createSession(parentSessionId?: string): Promise<string>;
     snapshotSession(sessionId: string, path: string): Promise<boolean>;
     loadSession(path: string): Promise<string>;

package/dist/storage/client/GrpcClient.js

@@ -95,7 +95,7 @@ class GrpcClient extends BaseClient_1.BaseRiceDBClient {
     async listUsers() {
         throw new Error("List users is not supported via gRPC transport. Use HTTP.");
     }
-    async insert(nodeId, text, metadata, userId = 1, sessionId, embedding) {
+    async insert(nodeId, text, metadata, userId = 1, sessionId, embedding, runId) {
         if (!this.client)
             throw new Error("Not connected");
         // Automatically store text in metadata so it can be retrieved
@@ -110,11 +110,12 @@ class GrpcClient extends BaseClient_1.BaseRiceDBClient {
             userId: this.toLong(userId),
             sessionId,
             embedding: embedding || [],
+            runId,
         };
         const res = await this.promisify(this.client.insert, req);
         return res;
     }
-    async search(query, userId, k = 10, sessionId, filter, queryEmbedding) {
+    async search(query, userId, k = 10, sessionId, filter, queryEmbedding, runId) {
         if (!this.client)
             throw new Error("Not connected");
         const req = {
@@ -124,6 +125,7 @@ class GrpcClient extends BaseClient_1.BaseRiceDBClient {
             sessionId,
             filter: filter ? JSON.stringify(filter) : "",
             queryEmbedding: queryEmbedding || [],
+            runId,
         };
         const res = await this.promisify(this.client.search, req);
         return res.results.map((r) => {
@@ -145,6 +147,21 @@ class GrpcClient extends BaseClient_1.BaseRiceDBClient {
         });
         return res.success;
     }
+    async deleteRun(runId) {
+        if (!this.client)
+            throw new Error("Not connected");
+        if (!runId)
+            throw new Error("runId is required for deleteRun");
+        const res = await this.promisify(this.client.deleteRun, {
+            runId,
+            userId: this.toLong(1),
+        });
+        return {
+            success: res.success,
+            message: res.message,
+            count: res.count,
+        };
+    }
     async createSession(parentSessionId) {
         if (!this.client)
             throw new Error("Not connected");
@@ -316,6 +333,7 @@ class GrpcClient extends BaseClient_1.BaseRiceDBClient {
                     metadata,
                     userId: docUserId,
                     sessionId: undefined,
+                    embedding: [],
                 });
             }
             stream.end();

package/dist/storage/client/HttpClient.d.ts

@@ -15,11 +15,16 @@ export declare class HttpClient extends BaseRiceDBClient {
     deleteUser(username: string): Promise<boolean>;
     getUser(username: string): Promise<any>;
     listUsers(): Promise<any[]>;
-    insert(nodeId: Long | number | string, text: string, metadata: any, userId?: Long | number | string, sessionId?: string, embedding?: number[]): Promise<InsertResult>;
+    insert(nodeId: Long | number | string, text: string, metadata: any, userId?: Long | number | string, sessionId?: string, embedding?: number[], runId?: string): Promise<InsertResult>;
     search(query: string, userId: Long | number | string, k?: number, sessionId?: string, filter?: {
         [key: string]: any;
-    }, queryEmbedding?: number[]): Promise<SearchResultItem[]>;
+    }, queryEmbedding?: number[], runId?: string): Promise<SearchResultItem[]>;
     delete(nodeId: Long | number | string, sessionId?: string): Promise<boolean>;
+    deleteRun(runId?: string): Promise<{
+        success: boolean;
+        message: string;
+        count: Long;
+    }>;
     createSession(parentSessionId?: string): Promise<string>;
     snapshotSession(sessionId: string, path: string): Promise<boolean>;
     loadSession(path: string): Promise<string>;

package/dist/storage/client/HttpClient.js

@@ -79,7 +79,7 @@ class HttpClient extends BaseClient_1.BaseRiceDBClient {
     async listUsers() {
         return this.request("GET", "/auth/users");
     }
-    async insert(nodeId, text, metadata, userId = 1, sessionId, embedding) {
+    async insert(nodeId, text, metadata, userId = 1, sessionId, embedding, runId) {
         // Automatically store text in metadata so it can be retrieved
         const meta = { ...metadata };
         if (text && !meta.stored_text) {
@@ -95,6 +95,8 @@ class HttpClient extends BaseClient_1.BaseRiceDBClient {
             payload.session_id = sessionId;
         if (embedding)
             payload.embedding = embedding;
+        if (runId)
+            payload.run_id = runId;
         const res = await this.request("POST", "/insert", payload);
         if (!res.success) {
             throw new Error(res.message || "Insert failed");
@@ -105,7 +107,7 @@ class HttpClient extends BaseClient_1.BaseRiceDBClient {
             message: res.message,
         };
     }
-    async search(query, userId, k = 10, sessionId, filter, queryEmbedding) {
+    async search(query, userId, k = 10, sessionId, filter, queryEmbedding, runId) {
         const payload = {
             query,
             user_id: this.toLong(userId).toNumber(),
@@ -117,6 +119,8 @@ class HttpClient extends BaseClient_1.BaseRiceDBClient {
             payload.filter = filter;
         if (queryEmbedding)
             payload.query_embedding = queryEmbedding;
+        if (runId)
+            payload.run_id = runId;
         const res = await this.request("POST", "/search", payload);
         const results = Array.isArray(res) ? res : res.results || [];
         return results.map((r) => ({
@@ -136,6 +140,16 @@ class HttpClient extends BaseClient_1.BaseRiceDBClient {
         await this.request("DELETE", `/node/${this.toLong(nodeId).toString()}`, undefined, params);
         return true;
     }
+    async deleteRun(runId) {
+        if (!runId)
+            throw new Error("runId is required for deleteRun");
+        const res = await this.request("DELETE", `/run/${runId}`);
+        return {
+            success: res.success,
+            message: res.message,
+            count: long_1.default.fromValue(res.count),
+        };
+    }
     // Cortex
     async createSession(parentSessionId) {
         const payload = {};

package/dist/storage/client/RiceDBClient.d.ts

@@ -11,6 +11,7 @@ export declare class RiceDBClient extends BaseRiceDBClient {
     private _grpcPort;
     private _httpPort;
     private token;
+    private runId;
     /**
      * Creates a new RiceDB client instance.
      * @param host - The hostname of the RiceDB server (default: "localhost").
@@ -18,8 +19,10 @@ export declare class RiceDBClient extends BaseRiceDBClient {
      * @param grpcPort - The port for gRPC connections (default: 50051).
      * @param httpPort - The port for HTTP connections (default: 3000).
      * @param token - Optional authentication token.
+     * @param runId - Optional run identifier for storage data isolation.
      */
-    constructor(host?: string, transport?: "grpc" | "http" | "auto", grpcPort?: number, httpPort?: number, token?: string);
+    constructor(host?: string, transport?: "grpc" | "http" | "auto", grpcPort?: number, httpPort?: number, token?: string, runId?: string);
+    setRunId(runId: string): void;
     /**
      * Connects to the RiceDB server.
      * If transport is "auto", tries gRPC first, then falls back to HTTP.
@@ -78,7 +81,7 @@ export declare class RiceDBClient extends BaseRiceDBClient {
      * @param embedding - Optional pre-computed embedding vector.
      * @returns Result of the insertion.
      */
-    insert(nodeId: Long | number | string, text: string, metadata: any, userId?: Long | number | string, sessionId?: string, embedding?: number[]): Promise<InsertResult>;
+    insert(nodeId: Long | number | string, text: string, metadata: any, userId?: Long | number | string, sessionId?: string, embedding?: number[], runId?: string): Promise<InsertResult>;
     /**
      * Searches for similar documents/nodes.
      * @param query - The query text.
@@ -91,7 +94,7 @@ export declare class RiceDBClient extends BaseRiceDBClient {
      */
     search(query: string, userId: Long | number | string, k?: number, sessionId?: string, filter?: {
         [key: string]: any;
-    }, queryEmbedding?: number[]): Promise<SearchResultItem[]>;
+    }, queryEmbedding?: number[], runId?: string): Promise<SearchResultItem[]>;
     /**
      * Deletes a document/node.
      * @param nodeId - The ID of the node to delete.
@@ -99,6 +102,11 @@ export declare class RiceDBClient extends BaseRiceDBClient {
      * @returns True if successful.
      */
     delete(nodeId: Long | number | string, sessionId?: string): Promise<boolean>;
+    deleteRun(runId?: string): Promise<{
+        success: boolean;
+        message: string;
+        count: Long;
+    }>;
     /**
      * Creates a new session.
      * @param parentSessionId - Optional parent session ID to fork from.

package/dist/storage/client/RiceDBClient.js

@@ -20,14 +20,19 @@ class RiceDBClient extends BaseClient_1.BaseRiceDBClient {
      * @param grpcPort - The port for gRPC connections (default: 50051).
      * @param httpPort - The port for HTTP connections (default: 3000).
      * @param token - Optional authentication token.
+     * @param runId - Optional run identifier for storage data isolation.
      */
-    constructor(host = "localhost", transport = "auto", grpcPort = 50051, httpPort = 3000, token) {
+    constructor(host = "localhost", transport = "auto", grpcPort = 50051, httpPort = 3000, token, runId) {
         super(host, 0);
         this.client = null;
         this.transport = transport;
         this._grpcPort = grpcPort;
         this._httpPort = httpPort;
         this.token = token;
+        this.runId = runId;
+    }
+    setRunId(runId) {
+        this.runId = runId;
     }
     /**
      * Connects to the RiceDB server.
@@ -145,9 +150,9 @@ class RiceDBClient extends BaseClient_1.BaseRiceDBClient {
      * @param embedding - Optional pre-computed embedding vector.
      * @returns Result of the insertion.
      */
-    async insert(nodeId, text, metadata, userId = 1, sessionId, embedding) {
+    async insert(nodeId, text, metadata, userId = 1, sessionId, embedding, runId) {
         this.checkConnected();
-        return this.client.insert(nodeId, text, metadata, userId, sessionId, embedding);
+        return this.client.insert(nodeId, text, metadata, userId, sessionId, embedding, runId || this.runId);
     }
     /**
      * Searches for similar documents/nodes.
@@ -159,9 +164,9 @@ class RiceDBClient extends BaseClient_1.BaseRiceDBClient {
      * @param queryEmbedding - Optional query embedding vector.
      * @returns An array of search results.
      */
-    async search(query, userId, k = 10, sessionId, filter, queryEmbedding) {
+    async search(query, userId, k = 10, sessionId, filter, queryEmbedding, runId) {
         this.checkConnected();
-        return this.client.search(query, userId, k, sessionId, filter, queryEmbedding);
+        return this.client.search(query, userId, k, sessionId, filter, queryEmbedding, runId || this.runId);
     }
     /**
      * Deletes a document/node.
@@ -173,6 +178,14 @@ class RiceDBClient extends BaseClient_1.BaseRiceDBClient {
         this.checkConnected();
         return this.client.delete(nodeId, sessionId);
     }
+    async deleteRun(runId) {
+        this.checkConnected();
+        const targetRunId = runId || this.runId;
+        if (!targetRunId) {
+            throw new Error("runId is required for deleteRun");
+        }
+        return this.client.deleteRun(targetRunId);
+    }
     /**
      * Creates a new session.
      * @param parentSessionId - Optional parent session ID to fork from.

package/dist/storage/proto/ricedb.d.ts

@@ -44,6 +44,7 @@ export interface Node {
      * repeated float vector = 2; // Deprecated
      */
     metadata: Buffer;
+    runId?: string | undefined;
 }
 /** Insert operations */
 export interface InsertRequest {
@@ -54,6 +55,9 @@ export interface InsertRequest {
     metadata: Buffer;
     userId: Long;
     sessionId?: string | undefined;
+    /** Pre-computed embedding (optional) */
+    embedding: number[];
+    runId?: string | undefined;
 }
 export interface InsertResponse {
     success: boolean;
@@ -79,6 +83,15 @@ export interface DeleteNodeResponse {
     success: boolean;
     message: string;
 }
+export interface DeleteRunRequest {
+    runId: string;
+    userId: Long;
+}
+export interface DeleteRunResponse {
+    success: boolean;
+    message: string;
+    count: Long;
+}
 /** Search operations */
 export interface SearchRequest {
     /** Raw text query for HDC encoding */
@@ -88,6 +101,9 @@ export interface SearchRequest {
     sessionId?: string | undefined;
     /** JSON-encoded metadata filter */
     filter: string;
+    /** Pre-computed embedding (optional) */
+    queryEmbedding: number[];
+    runId?: string | undefined;
 }
 export interface SearchResponse {
     results: SearchResult[];
@@ -413,6 +429,22 @@ export declare const DeleteNodeResponse: {
     create<I extends Exact<DeepPartial<DeleteNodeResponse>, I>>(base?: I): DeleteNodeResponse;
     fromPartial<I extends Exact<DeepPartial<DeleteNodeResponse>, I>>(object: I): DeleteNodeResponse;
 };
+export declare const DeleteRunRequest: {
+    encode(message: DeleteRunRequest, writer?: _m0.Writer): _m0.Writer;
+    decode(input: _m0.Reader | Uint8Array, length?: number): DeleteRunRequest;
+    fromJSON(object: any): DeleteRunRequest;
+    toJSON(message: DeleteRunRequest): unknown;
+    create<I extends Exact<DeepPartial<DeleteRunRequest>, I>>(base?: I): DeleteRunRequest;
+    fromPartial<I extends Exact<DeepPartial<DeleteRunRequest>, I>>(object: I): DeleteRunRequest;
+};
+export declare const DeleteRunResponse: {
+    encode(message: DeleteRunResponse, writer?: _m0.Writer): _m0.Writer;
+    decode(input: _m0.Reader | Uint8Array, length?: number): DeleteRunResponse;
+    fromJSON(object: any): DeleteRunResponse;
+    toJSON(message: DeleteRunResponse): unknown;
+    create<I extends Exact<DeepPartial<DeleteRunResponse>, I>>(base?: I): DeleteRunResponse;
+    fromPartial<I extends Exact<DeepPartial<DeleteRunResponse>, I>>(object: I): DeleteRunResponse;
+};
 export declare const SearchRequest: {
     encode(message: SearchRequest, writer?: _m0.Writer): _m0.Writer;
     decode(input: _m0.Reader | Uint8Array, length?: number): SearchRequest;
@@ -830,6 +862,16 @@ export declare const RiceDBService: {
         readonly responseSerialize: (value: DeleteNodeResponse) => Buffer<ArrayBuffer>;
         readonly responseDeserialize: (value: Buffer) => DeleteNodeResponse;
     };
+    /** Delete all documents in a run */
+    readonly deleteRun: {
+        readonly path: "/ricedb.RiceDB/DeleteRun";
+        readonly requestStream: false;
+        readonly responseStream: false;
+        readonly requestSerialize: (value: DeleteRunRequest) => Buffer<ArrayBuffer>;
+        readonly requestDeserialize: (value: Buffer) => DeleteRunRequest;
+        readonly responseSerialize: (value: DeleteRunResponse) => Buffer<ArrayBuffer>;
+        readonly responseDeserialize: (value: Buffer) => DeleteRunResponse;
+    };
     /** Search for similar documents */
     readonly search: {
         readonly path: "/ricedb.RiceDB/Search";
@@ -1036,6 +1078,8 @@ export interface RiceDBServer extends UntypedServiceImplementation {
     getNode: handleUnaryCall<GetNodeRequest, GetNodeResponse>;
     /** Delete a document */
     deleteNode: handleUnaryCall<DeleteNodeRequest, DeleteNodeResponse>;
+    /** Delete all documents in a run */
+    deleteRun: handleUnaryCall<DeleteRunRequest, DeleteRunResponse>;
     /** Search for similar documents */
     search: handleUnaryCall<SearchRequest, SearchResponse>;
     /** Batch insert (streaming) */
@@ -1096,6 +1140,10 @@ export interface RiceDBClient extends Client {
     deleteNode(request: DeleteNodeRequest, callback: (error: ServiceError | null, response: DeleteNodeResponse) => void): ClientUnaryCall;
     deleteNode(request: DeleteNodeRequest, metadata: Metadata, callback: (error: ServiceError | null, response: DeleteNodeResponse) => void): ClientUnaryCall;
     deleteNode(request: DeleteNodeRequest, metadata: Metadata, options: Partial<CallOptions>, callback: (error: ServiceError | null, response: DeleteNodeResponse) => void): ClientUnaryCall;
+    /** Delete all documents in a run */
+    deleteRun(request: DeleteRunRequest, callback: (error: ServiceError | null, response: DeleteRunResponse) => void): ClientUnaryCall;
+    deleteRun(request: DeleteRunRequest, metadata: Metadata, callback: (error: ServiceError | null, response: DeleteRunResponse) => void): ClientUnaryCall;
+    deleteRun(request: DeleteRunRequest, metadata: Metadata, options: Partial<CallOptions>, callback: (error: ServiceError | null, response: DeleteRunResponse) => void): ClientUnaryCall;
     /** Search for similar documents */
     search(request: SearchRequest, callback: (error: ServiceError | null, response: SearchResponse) => void): ClientUnaryCall;
     search(request: SearchRequest, metadata: Metadata, callback: (error: ServiceError | null, response: SearchResponse) => void): ClientUnaryCall;

package/dist/storage/proto/ricedb.js

@@ -2,14 +2,14 @@
 // Code generated by protoc-gen-ts_proto. DO NOT EDIT.
 // versions:
 //   protoc-gen-ts_proto  v1.181.2
-//   protoc               v6.33.2
+//   protoc               v6.33.4
 // source: ricedb.proto
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.MemoryEvent = exports.WatchMemoryRequest = exports.ClearMemoryResponse = exports.ClearMemoryRequest = exports.GetMemoryResponse = exports.GetMemoryRequest_FilterEntry = exports.GetMemoryRequest = exports.AddMemoryResponse = exports.AddMemoryRequest_MetadataEntry = exports.AddMemoryRequest = exports.MemoryEntry_MetadataEntry = exports.MemoryEntry = exports.ReadMemoryResponse = exports.ReadMemoryRequest = exports.WriteMemoryResponse = exports.WriteMemoryRequest = exports.BitVector = exports.PubSubEvent = exports.SubscribeRequest = exports.TraverseGraphResponse = exports.TraverseGraphRequest = exports.GetNeighborsResponse = exports.GetNeighborsRequest = exports.AddEdgeResponse = exports.AddEdgeRequest = exports.RevokePermissionResponse = exports.RevokePermissionRequest = exports.GrantPermissionResponse = exports.GrantPermissionRequest = exports.Permissions = exports.SearchResult = exports.SearchResponse = exports.SearchRequest = exports.DeleteNodeResponse = exports.DeleteNodeRequest = exports.GetNodeResponse = exports.GetNodeRequest = exports.BatchInsertResponse = exports.InsertResponse = exports.InsertRequest = exports.Node = exports.HealthResponse = exports.HealthRequest = exports.DeleteUserResponse = exports.DeleteUserRequest = exports.CreateUserResponse = exports.CreateUserRequest = exports.LoginResponse = exports.LoginRequest = exports.protobufPackage = void 0;
-exports.RiceDBClient = exports.RiceDBService = exports.DropSessionResponse = exports.DropSessionRequest = exports.CommitSessionResponse = exports.CommitSessionRequest = exports.LoadSessionResponse = exports.LoadSessionRequest = exports.SnapshotSessionResponse = exports.SnapshotSessionRequest = exports.CreateSessionResponse = exports.CreateSessionRequest = void 0;
+exports.ClearMemoryResponse = exports.ClearMemoryRequest = exports.GetMemoryResponse = exports.GetMemoryRequest_FilterEntry = exports.GetMemoryRequest = exports.AddMemoryResponse = exports.AddMemoryRequest_MetadataEntry = exports.AddMemoryRequest = exports.MemoryEntry_MetadataEntry = exports.MemoryEntry = exports.ReadMemoryResponse = exports.ReadMemoryRequest = exports.WriteMemoryResponse = exports.WriteMemoryRequest = exports.BitVector = exports.PubSubEvent = exports.SubscribeRequest = exports.TraverseGraphResponse = exports.TraverseGraphRequest = exports.GetNeighborsResponse = exports.GetNeighborsRequest = exports.AddEdgeResponse = exports.AddEdgeRequest = exports.RevokePermissionResponse = exports.RevokePermissionRequest = exports.GrantPermissionResponse = exports.GrantPermissionRequest = exports.Permissions = exports.SearchResult = exports.SearchResponse = exports.SearchRequest = exports.DeleteRunResponse = exports.DeleteRunRequest = exports.DeleteNodeResponse = exports.DeleteNodeRequest = exports.GetNodeResponse = exports.GetNodeRequest = exports.BatchInsertResponse = exports.InsertResponse = exports.InsertRequest = exports.Node = exports.HealthResponse = exports.HealthRequest = exports.DeleteUserResponse = exports.DeleteUserRequest = exports.CreateUserResponse = exports.CreateUserRequest = exports.LoginResponse = exports.LoginRequest = exports.protobufPackage = void 0;
+exports.RiceDBClient = exports.RiceDBService = exports.DropSessionResponse = exports.DropSessionRequest = exports.CommitSessionResponse = exports.CommitSessionRequest = exports.LoadSessionResponse = exports.LoadSessionRequest = exports.SnapshotSessionResponse = exports.SnapshotSessionRequest = exports.CreateSessionResponse = exports.CreateSessionRequest = exports.MemoryEvent = exports.WatchMemoryRequest = void 0;
 /* eslint-disable */
 const grpc_js_1 = require("@grpc/grpc-js");
 const long_1 = __importDefault(require("long"));
@@ -531,7 +531,7 @@ exports.HealthResponse = {
     },
 };
 function createBaseNode() {
-    return { id: long_1.default.UZERO, metadata: Buffer.alloc(0) };
+    return { id: long_1.default.UZERO, metadata: Buffer.alloc(0), runId: undefined };
 }
 exports.Node = {
     encode(message, writer = minimal_1.default.Writer.create()) {
@@ -541,6 +541,9 @@ exports.Node = {
         if (message.metadata.length !== 0) {
             writer.uint32(26).bytes(message.metadata);
         }
+        if (message.runId !== undefined) {
+            writer.uint32(34).string(message.runId);
+        }
         return writer;
     },
     decode(input, length) {
@@ -562,6 +565,12 @@ exports.Node = {
                     }
                     message.metadata = reader.bytes();
                     continue;
+                case 4:
+                    if (tag !== 34) {
+                        break;
+                    }
+                    message.runId = reader.string();
+                    continue;
             }
             if ((tag & 7) === 4 || tag === 0) {
                 break;
@@ -574,6 +583,7 @@ exports.Node = {
         return {
             id: isSet(object.id) ? long_1.default.fromValue(object.id) : long_1.default.UZERO,
             metadata: isSet(object.metadata) ? Buffer.from(bytesFromBase64(object.metadata)) : Buffer.alloc(0),
+            runId: isSet(object.runId) ? globalThis.String(object.runId) : undefined,
         };
     },
     toJSON(message) {
@@ -584,6 +594,9 @@ exports.Node = {
         if (message.metadata.length !== 0) {
             obj.metadata = base64FromBytes(message.metadata);
         }
+        if (message.runId !== undefined) {
+            obj.runId = message.runId;
+        }
         return obj;
     },
     create(base) {
@@ -593,11 +606,20 @@ exports.Node = {
         const message = createBaseNode();
         message.id = (object.id !== undefined && object.id !== null) ? long_1.default.fromValue(object.id) : long_1.default.UZERO;
         message.metadata = object.metadata ?? Buffer.alloc(0);
+        message.runId = object.runId ?? undefined;
         return message;
     },
 };
 function createBaseInsertRequest() {
-    return { id: long_1.default.UZERO, text: "", metadata: Buffer.alloc(0), userId: long_1.default.UZERO, sessionId: undefined };
+    return {
+        id: long_1.default.UZERO,
+        text: "",
+        metadata: Buffer.alloc(0),
+        userId: long_1.default.UZERO,
+        sessionId: undefined,
+        embedding: [],
+        runId: undefined,
+    };
 }
 exports.InsertRequest = {
     encode(message, writer = minimal_1.default.Writer.create()) {
@@ -616,6 +638,14 @@ exports.InsertRequest = {
         if (message.sessionId !== undefined) {
             writer.uint32(42).string(message.sessionId);
         }
+        writer.uint32(50).fork();
+        for (const v of message.embedding) {
+            writer.float(v);
+        }
+        writer.ldelim();
+        if (message.runId !== undefined) {
+            writer.uint32(58).string(message.runId);
+        }
         return writer;
     },
     decode(input, length) {
@@ -655,6 +685,25 @@ exports.InsertRequest = {
                     }
                     message.sessionId = reader.string();
                     continue;
+                case 6:
+                    if (tag === 53) {
+                        message.embedding.push(reader.float());
+                        continue;
+                    }
+                    if (tag === 50) {
+                        const end2 = reader.uint32() + reader.pos;
+                        while (reader.pos < end2) {
+                            message.embedding.push(reader.float());
+                        }
+                        continue;
+                    }
+                    break;
+                case 7:
+                    if (tag !== 58) {
+                        break;
+                    }
+                    message.runId = reader.string();
+                    continue;
             }
             if ((tag & 7) === 4 || tag === 0) {
                 break;
@@ -670,6 +719,10 @@ exports.InsertRequest = {
             metadata: isSet(object.metadata) ? Buffer.from(bytesFromBase64(object.metadata)) : Buffer.alloc(0),
             userId: isSet(object.userId) ? long_1.default.fromValue(object.userId) : long_1.default.UZERO,
             sessionId: isSet(object.sessionId) ? globalThis.String(object.sessionId) : undefined,
+            embedding: globalThis.Array.isArray(object?.embedding)
+                ? object.embedding.map((e) => globalThis.Number(e))
+                : [],
+            runId: isSet(object.runId) ? globalThis.String(object.runId) : undefined,
         };
     },
     toJSON(message) {
@@ -689,6 +742,12 @@ exports.InsertRequest = {
         if (message.sessionId !== undefined) {
             obj.sessionId = message.sessionId;
         }
+        if (message.embedding?.length) {
+            obj.embedding = message.embedding;
+        }
+        if (message.runId !== undefined) {
+            obj.runId = message.runId;
+        }
         return obj;
     },
     create(base) {
@@ -703,6 +762,8 @@ exports.InsertRequest = {
             ? long_1.default.fromValue(object.userId)
             : long_1.default.UZERO;
         message.sessionId = object.sessionId ?? undefined;
+        message.embedding = object.embedding?.map((e) => e) || [];
+        message.runId = object.runId ?? undefined;
         return message;
     },
 };
@@ -1099,8 +1160,164 @@ exports.DeleteNodeResponse = {
         return message;
     },
 };
+function createBaseDeleteRunRequest() {
+    return { runId: "", userId: long_1.default.UZERO };
+}
+exports.DeleteRunRequest = {
+    encode(message, writer = minimal_1.default.Writer.create()) {
+        if (message.runId !== "") {
+            writer.uint32(10).string(message.runId);
+        }
+        if (!message.userId.equals(long_1.default.UZERO)) {
+            writer.uint32(16).uint64(message.userId);
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof minimal_1.default.Reader ? input : minimal_1.default.Reader.create(input);
+        let end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBaseDeleteRunRequest();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1:
+                    if (tag !== 10) {
+                        break;
+                    }
+                    message.runId = reader.string();
+                    continue;
+                case 2:
+                    if (tag !== 16) {
+                        break;
+                    }
+                    message.userId = reader.uint64();
+                    continue;
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skipType(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return {
+            runId: isSet(object.runId) ? globalThis.String(object.runId) : "",
+            userId: isSet(object.userId) ? long_1.default.fromValue(object.userId) : long_1.default.UZERO,
+        };
+    },
+    toJSON(message) {
+        const obj = {};
+        if (message.runId !== "") {
+            obj.runId = message.runId;
+        }
+        if (!message.userId.equals(long_1.default.UZERO)) {
+            obj.userId = (message.userId || long_1.default.UZERO).toString();
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.DeleteRunRequest.fromPartial(base ?? {});
+    },
+    fromPartial(object) {
+        const message = createBaseDeleteRunRequest();
+        message.runId = object.runId ?? "";
+        message.userId = (object.userId !== undefined && object.userId !== null)
+            ? long_1.default.fromValue(object.userId)
+            : long_1.default.UZERO;
+        return message;
+    },
+};
+function createBaseDeleteRunResponse() {
+    return { success: false, message: "", count: long_1.default.UZERO };
+}
+exports.DeleteRunResponse = {
+    encode(message, writer = minimal_1.default.Writer.create()) {
+        if (message.success !== false) {
+            writer.uint32(8).bool(message.success);
+        }
+        if (message.message !== "") {
+            writer.uint32(18).string(message.message);
+        }
+        if (!message.count.equals(long_1.default.UZERO)) {
+            writer.uint32(24).uint64(message.count);
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof minimal_1.default.Reader ? input : minimal_1.default.Reader.create(input);
+        let end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBaseDeleteRunResponse();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1:
+                    if (tag !== 8) {
+                        break;
+                    }
+                    message.success = reader.bool();
+                    continue;
+                case 2:
+                    if (tag !== 18) {
+                        break;
+                    }
+                    message.message = reader.string();
+                    continue;
+                case 3:
+                    if (tag !== 24) {
+                        break;
+                    }
+                    message.count = reader.uint64();
+                    continue;
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skipType(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return {
+            success: isSet(object.success) ? globalThis.Boolean(object.success) : false,
+            message: isSet(object.message) ? globalThis.String(object.message) : "",
+            count: isSet(object.count) ? long_1.default.fromValue(object.count) : long_1.default.UZERO,
+        };
+    },
+    toJSON(message) {
+        const obj = {};
+        if (message.success !== false) {
+            obj.success = message.success;
+        }
+        if (message.message !== "") {
+            obj.message = message.message;
+        }
+        if (!message.count.equals(long_1.default.UZERO)) {
+            obj.count = (message.count || long_1.default.UZERO).toString();
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.DeleteRunResponse.fromPartial(base ?? {});
+    },
+    fromPartial(object) {
+        const message = createBaseDeleteRunResponse();
+        message.success = object.success ?? false;
+        message.message = object.message ?? "";
+        message.count = (object.count !== undefined && object.count !== null) ? long_1.default.fromValue(object.count) : long_1.default.UZERO;
+        return message;
+    },
+};
 function createBaseSearchRequest() {
-    return { queryText: "", userId: long_1.default.UZERO, k: 0, sessionId: undefined, filter: "" };
+    return {
+        queryText: "",
+        userId: long_1.default.UZERO,
+        k: 0,
+        sessionId: undefined,
+        filter: "",
+        queryEmbedding: [],
+        runId: undefined,
+    };
 }
 exports.SearchRequest = {
     encode(message, writer = minimal_1.default.Writer.create()) {
@@ -1119,6 +1336,14 @@ exports.SearchRequest = {
         if (message.filter !== "") {
             writer.uint32(42).string(message.filter);
         }
+        writer.uint32(50).fork();
+        for (const v of message.queryEmbedding) {
+            writer.float(v);
+        }
+        writer.ldelim();
+        if (message.runId !== undefined) {
+            writer.uint32(58).string(message.runId);
+        }
         return writer;
     },
     decode(input, length) {
@@ -1158,6 +1383,25 @@ exports.SearchRequest = {
                     }
                     message.filter = reader.string();
                     continue;
+                case 6:
+                    if (tag === 53) {
+                        message.queryEmbedding.push(reader.float());
+                        continue;
+                    }
+                    if (tag === 50) {
+                        const end2 = reader.uint32() + reader.pos;
+                        while (reader.pos < end2) {
+                            message.queryEmbedding.push(reader.float());
+                        }
+                        continue;
+                    }
+                    break;
+                case 7:
+                    if (tag !== 58) {
+                        break;
+                    }
+                    message.runId = reader.string();
+                    continue;
             }
             if ((tag & 7) === 4 || tag === 0) {
                 break;
@@ -1173,6 +1417,10 @@ exports.SearchRequest = {
             k: isSet(object.k) ? globalThis.Number(object.k) : 0,
             sessionId: isSet(object.sessionId) ? globalThis.String(object.sessionId) : undefined,
             filter: isSet(object.filter) ? globalThis.String(object.filter) : "",
+            queryEmbedding: globalThis.Array.isArray(object?.queryEmbedding)
+                ? object.queryEmbedding.map((e) => globalThis.Number(e))
+                : [],
+            runId: isSet(object.runId) ? globalThis.String(object.runId) : undefined,
         };
     },
     toJSON(message) {
@@ -1192,6 +1440,12 @@ exports.SearchRequest = {
         if (message.filter !== "") {
             obj.filter = message.filter;
         }
+        if (message.queryEmbedding?.length) {
+            obj.queryEmbedding = message.queryEmbedding;
+        }
+        if (message.runId !== undefined) {
+            obj.runId = message.runId;
+        }
         return obj;
     },
     create(base) {
@@ -1206,6 +1460,8 @@ exports.SearchRequest = {
         message.k = object.k ?? 0;
         message.sessionId = object.sessionId ?? undefined;
         message.filter = object.filter ?? "";
+        message.queryEmbedding = object.queryEmbedding?.map((e) => e) || [];
+        message.runId = object.runId ?? undefined;
         return message;
     },
 };
@@ -4259,6 +4515,16 @@ exports.RiceDBService = {
         responseSerialize: (value) => Buffer.from(exports.DeleteNodeResponse.encode(value).finish()),
         responseDeserialize: (value) => exports.DeleteNodeResponse.decode(value),
     },
+    /** Delete all documents in a run */
+    deleteRun: {
+        path: "/ricedb.RiceDB/DeleteRun",
+        requestStream: false,
+        responseStream: false,
+        requestSerialize: (value) => Buffer.from(exports.DeleteRunRequest.encode(value).finish()),
+        requestDeserialize: (value) => exports.DeleteRunRequest.decode(value),
+        responseSerialize: (value) => Buffer.from(exports.DeleteRunResponse.encode(value).finish()),
+        responseDeserialize: (value) => exports.DeleteRunResponse.decode(value),
+    },
     /** Search for similar documents */
     search: {
         path: "/ricedb.RiceDB/Search",

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "rice-node-sdk",
-    "version": "1.0.6",
+    "version": "1.0.8",
     "description": "SDK for Rice",
     "repository": {
         "type": "git",