@kernl-sdk/pg 0.1.22 → 0.1.24

package/.turbo/turbo-build.log

@@ -1,4 +1,4 @@
 
-> @kernl-sdk/pg@0.1.22 build /home/runner/work/kernl/kernl/packages/storage/pg
+> @kernl-sdk/pg@0.1.24 build /home/runner/work/kernl/kernl/packages/storage/pg
 > tsc && tsc-alias --resolve-full-paths
 

package/CHANGELOG.md

@@ -1,5 +1,22 @@
 # @kernl/pg
 
+## 0.1.24
+
+### Patch Changes
+
+- Updated dependencies [572ae80]
+  - kernl@0.9.0
+  - @kernl-sdk/retrieval@0.1.4
+  - @kernl-sdk/storage@0.1.24
+
+## 0.1.23
+
+### Patch Changes
+
+- Updated dependencies [e90b227]
+  - kernl@0.8.4
+  - @kernl-sdk/storage@0.1.23
+
 ## 0.1.22
 
 ### Patch Changes

package/dist/__tests__/memory-integration.test.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"memory-integration.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/memory-integration.test.ts"],"names":[],"mappings":"AAIA,OAAO,sBAAsB,CAAC"}
+{"version":3,"file":"memory-integration.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/memory-integration.test.ts"],"names":[],"mappings":"AAKA,OAAO,sBAAsB,CAAC"}

package/dist/__tests__/memory-integration.test.js

@@ -26,7 +26,7 @@ describe.sequential("Memory Integration with PGVector", { timeout: 30000 }, () =
                 vector: pgvector({ pool }),
             },
             memory: {
-                embeddingModel: "openai/text-embedding-3-small",
+                embedding: "openai/text-embedding-3-small",
                 dimensions: 1536,
             },
         });
@@ -204,7 +204,7 @@ describe.sequential("Memory Integration with PGVector", { timeout: 30000 }, () =
                 vector: pgvector({ pool }),
             },
             memory: {
-                embeddingModel: "openai/text-embedding-3-small",
+                embedding: "openai/text-embedding-3-small",
                 dimensions: 1536,
             },
         });
@@ -253,13 +253,62 @@ describe.sequential("Memory Integration with PGVector", { timeout: 30000 }, () =
             id: "m1",
             metadata: { version: 2, updated: true },
         });
-        // Verify metadata updated in vector index
-        const vectorResult = await pool.query('SELECT metadata FROM "kernl"."memories_sindex" WHERE id = $1', ["m1"]);
-        expect(vectorResult.rows[0].metadata).toEqual({
+        // Verify metadata updated in primary DB (metadata is not indexed)
+        const dbResult = await pool.query('SELECT metadata FROM "kernl"."memories" WHERE id = $1', ["m1"]);
+        expect(dbResult.rows[0].metadata).toEqual({
             version: 2,
             updated: true,
         });
     });
+    it("isolates memories by agentId", async () => {
+        // Register two agents
+        const model = {
+            spec: "1.0",
+            provider: "test",
+            modelId: "test-model",
+        };
+        const agent1 = new Agent({
+            id: "test-agent-1",
+            name: "Test Agent 1",
+            instructions: () => "test instructions",
+            model,
+        });
+        const agent2 = new Agent({
+            id: "test-agent-2",
+            name: "Test Agent 2",
+            instructions: () => "test instructions",
+            model,
+        });
+        kernl.register(agent1);
+        kernl.register(agent2);
+        // Create memories for each agent in the same namespace
+        await agent1.memories.create({
+            namespace: "shared-ns",
+            collection: "facts",
+            content: { text: "Agent 1 knows about cats" },
+        });
+        await agent2.memories.create({
+            namespace: "shared-ns",
+            collection: "facts",
+            content: { text: "Agent 2 knows about cats" },
+        });
+        // Search from agent1 - should only see agent1's memory
+        const results1 = await agent1.memories.search({
+            query: "cats",
+            filter: { scope: { namespace: "shared-ns" } },
+            limit: 10,
+        });
+        // Search from agent2 - should only see agent2's memory
+        const results2 = await agent2.memories.search({
+            query: "cats",
+            filter: { scope: { namespace: "shared-ns" } },
+            limit: 10,
+        });
+        expect(results1).toHaveLength(1);
+        expect(results1[0].document?.agentId).toBe("test-agent-1");
+        expect(results2).toHaveLength(1);
+        expect(results2[0].document?.agentId).toBe("test-agent-2");
+    });
     it("creates memories with multimodal content", async () => {
         await kernl.memories.create({
             id: "m1",

package/dist/thread/store.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"store.d.ts","sourceRoot":"","sources":["../../src/thread/store.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAE3C,OAAO,EAEL,KAAK,aAAa,EAClB,KAAK,aAAa,EAClB,KAAK,WAAW,EAChB,KAAK,SAAS,EACd,KAAK,YAAY,EACjB,KAAK,aAAa,EAClB,KAAK,iBAAiB,EACtB,KAAK,oBAAoB,EAC1B,MAAM,OAAO,CAAC;AACf,OAAO,EAAE,MAAM,EAAE,KAAK,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC1D,OAAO,EAIL,KAAK,YAAY,EAElB,MAAM,oBAAoB,CAAC;AAI5B;;;;;GAKG;AACH,qBAAa,aAAc,YAAW,WAAW;IAC/C,OAAO,CAAC,EAAE,CAAoB;IAC9B,OAAO,CAAC,UAAU,CAA0D;IAC5E,OAAO,CAAC,UAAU,CAAsB;gBAE5B,EAAE,EAAE,IAAI,GAAG,UAAU,EAAE,UAAU,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC;IAMlE;;;;OAIG;IACH,IAAI,CAAC,UAAU,EAAE;QAAE,MAAM,EAAE,aAAa,CAAC;QAAC,MAAM,EAAE,aAAa,CAAA;KAAE,GAAG,IAAI;IAIxE;;OAEG;IACG,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAuGvE;;OAEG;IACG,IAAI,CAAC,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAuC1D;;OAEG;IACG,MAAM,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC;IA4BhD;;OAEG;IACG,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC;IAkB/D;;OAEG;IACG,MAAM,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQxC;;OAEG;IACG,OAAO,CACX,GAAG,EAAE,MAAM,EACX,IAAI,CAAC,EAAE,oBAAoB,GAC1B,OAAO,CAAC,WAAW,EAAE,CAAC;IAuCzB;;;;;;;;;OASG;IACG,MAAM,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAmClD;;OAEG;IACH,OAAO,CAAC,MAAM,EAAE;QAAE,MAAM,EAAE,YAAY,CAAC;QAAC,MAAM,CAAC,EAAE,WAAW,EAAE,CAAA;KAAE,GAAG,MAAM;CAmC1E"}
+{"version":3,"file":"store.d.ts","sourceRoot":"","sources":["../../src/thread/store.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAE3C,OAAO,EAGL,KAAK,aAAa,EAClB,KAAK,aAAa,EAClB,KAAK,WAAW,EAChB,KAAK,SAAS,EACd,KAAK,YAAY,EACjB,KAAK,aAAa,EAClB,KAAK,iBAAiB,EACtB,KAAK,oBAAoB,EAC1B,MAAM,OAAO,CAAC;AACf,OAAO,EAAE,MAAM,EAAE,KAAK,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC1D,OAAO,EAIL,KAAK,YAAY,EAElB,MAAM,oBAAoB,CAAC;AAI5B;;;;;GAKG;AACH,qBAAa,aAAc,YAAW,WAAW;IAC/C,OAAO,CAAC,EAAE,CAAoB;IAC9B,OAAO,CAAC,UAAU,CAA0D;IAC5E,OAAO,CAAC,UAAU,CAAsB;gBAE5B,EAAE,EAAE,IAAI,GAAG,UAAU,EAAE,UAAU,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC;IAMlE;;;;OAIG;IACH,IAAI,CAAC,UAAU,EAAE;QAAE,MAAM,EAAE,aAAa,CAAC;QAAC,MAAM,EAAE,aAAa,CAAA;KAAE,GAAG,IAAI;IAIxE;;OAEG;IACG,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAuGvE;;OAEG;IACG,IAAI,CAAC,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAuC1D;;OAEG;IACG,MAAM,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC;IA4BhD;;OAEG;IACG,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC;IAkB/D;;OAEG;IACG,MAAM,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQxC;;OAEG;IACG,OAAO,CACX,GAAG,EAAE,MAAM,EACX,IAAI,CAAC,EAAE,oBAAoB,GAC1B,OAAO,CAAC,WAAW,EAAE,CAAC;IAuCzB;;;;;;;;;OASG;IACG,MAAM,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAmClD;;OAEG;IACH,OAAO,CAAC,MAAM,EAAE;QAAE,MAAM,EAAE,YAAY,CAAC;QAAC,MAAM,CAAC,EAAE,WAAW,EAAE,CAAA;KAAE,GAAG,MAAM;CA0C1E"}

package/dist/thread/store.js

@@ -267,8 +267,12 @@ export class PGThreadStore {
         if (!agent || !model) {
             throw new Error(`Thread ${record.id} references non-existent agent/model (agent: ${record.agent_id}, model: ${record.model})`);
         }
+        // safety: threads only exist for llm agents
+        if (agent.kind !== "llm") {
+            throw new Error(`Thread ${record.id} references non-llm agent ${record.agent_id} (kind: ${agent.kind})`);
+        }
         return new Thread({
-            agent,
+            agent: agent,
             history: events,
             context: new Context(record.namespace, record.context),
             model,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kernl-sdk/pg",
-  "version": "0.1.22",
+  "version": "0.1.24",
   "description": "PostgreSQL storage adapter for kernl",
   "keywords": [
     "kernl",
@@ -35,15 +35,15 @@
     "tsc-alias": "^1.8.10",
     "typescript": "5.9.2",
     "vitest": "^4.0.8",
-    "@kernl-sdk/protocol": "^0.2.8",
-    "@kernl-sdk/ai": "^0.3.0"
+    "@kernl-sdk/ai": "^0.3.1",
+    "@kernl-sdk/protocol": "^0.3.0"
   },
   "dependencies": {
     "pg": "^8.16.3",
-    "kernl": "^0.8.3",
-    "@kernl-sdk/retrieval": "^0.1.3",
+    "kernl": "^0.9.0",
+    "@kernl-sdk/retrieval": "^0.1.4",
     "@kernl-sdk/shared": "^0.3.0",
-    "@kernl-sdk/storage": "0.1.22"
+    "@kernl-sdk/storage": "0.1.24"
   },
   "scripts": {
     "clean": "rm -rf dist",

package/src/__tests__/memory-integration.test.ts

@@ -2,6 +2,7 @@ import { describe, it, expect, beforeAll, afterAll, beforeEach } from "vitest";
 import { Pool } from "pg";
 import { Kernl, Agent } from "kernl";
 import type { LanguageModel } from "@kernl-sdk/protocol";
+import type { SearchHit } from "@kernl-sdk/retrieval";
 import "@kernl-sdk/ai/openai"; // Register OpenAI embedding provider
 
 import { postgres, pgvector } from "../index";
@@ -38,7 +39,7 @@ describe.sequential(
           vector: pgvector({ pool }),
         },
         memory: {
-          embeddingModel: "openai/text-embedding-3-small",
+          embedding: "openai/text-embedding-3-small",
           dimensions: 1536,
         },
       });
@@ -149,7 +150,7 @@ describe.sequential(
       expect(results.length).toBeGreaterThan(0);
 
       // Should find TypeScript-related memories with higher scores
-      const ids = results.map((r) => r.document?.id);
+      const ids = results.map((r: SearchHit) => r.document?.id);
       expect(ids).toContain("m1"); // Direct match
       expect(ids).toContain("m3"); // Related to TypeScript
     });
@@ -255,7 +256,7 @@ describe.sequential(
           vector: pgvector({ pool }),
         },
         memory: {
-          embeddingModel: "openai/text-embedding-3-small",
+          embedding: "openai/text-embedding-3-small",
           dimensions: 1536,
         },
       });
@@ -292,7 +293,7 @@ describe.sequential(
       });
 
       expect(results.length).toBeGreaterThan(0);
-      const match = results.find((r) => r.document?.id === "m1");
+      const match = results.find((r: SearchHit) => r.document?.id === "m1");
       expect(match).toBeDefined();
       expect(match?.document?.text).toBe("Updated content about cats");
     });
@@ -313,18 +314,75 @@ describe.sequential(
         metadata: { version: 2, updated: true },
       });
 
-      // Verify metadata updated in vector index
-      const vectorResult = await pool.query(
-        'SELECT metadata FROM "kernl"."memories_sindex" WHERE id = $1',
+      // Verify metadata updated in primary DB (metadata is not indexed)
+      const dbResult = await pool.query(
+        'SELECT metadata FROM "kernl"."memories" WHERE id = $1',
         ["m1"],
       );
 
-      expect(vectorResult.rows[0].metadata).toEqual({
+      expect(dbResult.rows[0].metadata).toEqual({
         version: 2,
         updated: true,
       });
     });
 
+    it("isolates memories by agentId", async () => {
+      // Register two agents
+      const model = {
+        spec: "1.0" as const,
+        provider: "test",
+        modelId: "test-model",
+      } as unknown as LanguageModel;
+
+      const agent1 = new Agent({
+        id: "test-agent-1",
+        name: "Test Agent 1",
+        instructions: () => "test instructions",
+        model,
+      });
+      const agent2 = new Agent({
+        id: "test-agent-2",
+        name: "Test Agent 2",
+        instructions: () => "test instructions",
+        model,
+      });
+      kernl.register(agent1);
+      kernl.register(agent2);
+
+      // Create memories for each agent in the same namespace
+      await agent1.memories.create({
+        namespace: "shared-ns",
+        collection: "facts",
+        content: { text: "Agent 1 knows about cats" },
+      });
+
+      await agent2.memories.create({
+        namespace: "shared-ns",
+        collection: "facts",
+        content: { text: "Agent 2 knows about cats" },
+      });
+
+      // Search from agent1 - should only see agent1's memory
+      const results1 = await agent1.memories.search({
+        query: "cats",
+        filter: { scope: { namespace: "shared-ns" } },
+        limit: 10,
+      });
+
+      // Search from agent2 - should only see agent2's memory
+      const results2 = await agent2.memories.search({
+        query: "cats",
+        filter: { scope: { namespace: "shared-ns" } },
+        limit: 10,
+      });
+
+      expect(results1).toHaveLength(1);
+      expect(results1[0].document?.agentId).toBe("test-agent-1");
+
+      expect(results2).toHaveLength(1);
+      expect(results2[0].document?.agentId).toBe("test-agent-2");
+    });
+
     it("creates memories with multimodal content", async () => {
       await kernl.memories.create({
         id: "m1",
@@ -348,7 +406,7 @@ describe.sequential(
       });
 
       expect(results.length).toBeGreaterThan(0);
-      const match = results.find((r) => r.document?.id === "m1");
+      const match = results.find((r: SearchHit) => r.document?.id === "m1");
       expect(match).toBeDefined();
     });
   },

package/src/thread/store.ts

@@ -2,6 +2,7 @@ import assert from "assert";
 import type { Pool, PoolClient } from "pg";
 
 import {
+  Agent,
   Context,
   type AgentRegistry,
   type ModelRegistry,
@@ -371,8 +372,15 @@ export class PGThreadStore implements ThreadStore {
       );
     }
 
+    // safety: threads only exist for llm agents
+    if (agent.kind !== "llm") {
+      throw new Error(
+        `Thread ${record.id} references non-llm agent ${record.agent_id} (kind: ${agent.kind})`,
+      );
+    }
+
     return new Thread({
-      agent,
+      agent: agent as Agent,
       history: events,
       context: new Context(record.namespace, record.context),
       model,