@avee1234/agent-kit 0.2.0 → 0.3.1

package/README.md

@@ -1,5 +1,9 @@
 # agent-kit
 
+[![CI](https://github.com/abhid1234/agent-kit/actions/workflows/ci.yml/badge.svg)](https://github.com/abhid1234/agent-kit/actions/workflows/ci.yml)
+[![npm version](https://img.shields.io/npm/v/@avee1234/agent-kit)](https://www.npmjs.com/package/@avee1234/agent-kit)
+[![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg)](LICENSE)
+
 TypeScript-first library for building stateful, persistent AI agents.
 
 <p align="center">

package/dist/cli.cjs

@@ -0,0 +1,100 @@
+#!/usr/bin/env node
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+
+// src/cli.ts
+var import_fs = __toESM(require("fs"), 1);
+var import_path = __toESM(require("path"), 1);
+var args = process.argv.slice(2);
+var command = args[0];
+var projectName = args[1] ?? "my-agent";
+if (command !== "init") {
+  console.error(`Unknown command: ${command}`);
+  console.error("Usage: agent-kit init [project-name]");
+  process.exit(1);
+}
+var targetDir = import_path.default.resolve(process.cwd(), projectName);
+if (import_fs.default.existsSync(targetDir)) {
+  console.error(`Directory already exists: ${targetDir}`);
+  process.exit(1);
+}
+import_fs.default.mkdirSync(targetDir, { recursive: true });
+var packageJson = {
+  name: projectName,
+  version: "1.0.0",
+  private: true,
+  type: "module",
+  scripts: {
+    start: "npx tsx agent.ts"
+  },
+  dependencies: {
+    "@avee1234/agent-kit": "^0.2.0"
+  },
+  devDependencies: {
+    tsx: "^4.0.0"
+  }
+};
+var agentTs = `import { Agent, Tool, Memory } from '@avee1234/agent-kit';
+
+const greet = Tool.create({
+  name: 'greet',
+  description: 'Greet someone by name',
+  parameters: {
+    name: { type: 'string', description: 'Name to greet' },
+  },
+  execute: async ({ name }) => \`Hello, \${name}!\`,
+});
+
+const agent = new Agent({
+  name: 'my-agent',
+  memory: new Memory({ store: 'sqlite', path: './memory.db' }),
+  tools: [greet],
+  system: 'You are a helpful assistant.',
+});
+
+const response = await agent.chat(process.argv[2] ?? 'Hello!');
+console.log(response.content);
+`;
+var readmeMd = `# ${projectName}
+
+Built with [agent-kit](https://github.com/abhid1234/agent-kit).
+
+## Quick Start
+
+\`\`\`bash
+npm install
+npm start "Hello, what can you do?"
+\`\`\`
+`;
+import_fs.default.writeFileSync(import_path.default.join(targetDir, "package.json"), JSON.stringify(packageJson, null, 2) + "\n");
+import_fs.default.writeFileSync(import_path.default.join(targetDir, "agent.ts"), agentTs);
+import_fs.default.writeFileSync(import_path.default.join(targetDir, "README.md"), readmeMd);
+console.log(`
+Scaffolded project: ${projectName}
+`);
+console.log("Next steps:\n");
+console.log(`  cd ${projectName}`);
+console.log("  npm install");
+console.log('  npm start "Hello, what can you do?"\n');
+//# sourceMappingURL=cli.cjs.map

package/dist/cli.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/cli.ts"],"sourcesContent":["#!/usr/bin/env node\n\nimport fs from 'fs';\nimport path from 'path';\n\nconst args = process.argv.slice(2);\nconst command = args[0];\nconst projectName = args[1] ?? 'my-agent';\n\nif (command !== 'init') {\n  console.error(`Unknown command: ${command}`);\n  console.error('Usage: agent-kit init [project-name]');\n  process.exit(1);\n}\n\nconst targetDir = path.resolve(process.cwd(), projectName);\n\nif (fs.existsSync(targetDir)) {\n  console.error(`Directory already exists: ${targetDir}`);\n  process.exit(1);\n}\n\nfs.mkdirSync(targetDir, { recursive: true });\n\nconst packageJson = {\n  name: projectName,\n  version: '1.0.0',\n  private: true,\n  type: 'module',\n  scripts: {\n    start: 'npx tsx agent.ts',\n  },\n  dependencies: {\n    '@avee1234/agent-kit': '^0.2.0',\n  },\n  devDependencies: {\n    tsx: '^4.0.0',\n  },\n};\n\nconst agentTs = `import { Agent, Tool, Memory } from '@avee1234/agent-kit';\n\nconst greet = Tool.create({\n  name: 'greet',\n  description: 'Greet someone by name',\n  parameters: {\n    name: { type: 'string', description: 'Name to greet' },\n  },\n  execute: async ({ name }) => \\`Hello, \\${name}!\\`,\n});\n\nconst agent = new Agent({\n  name: 'my-agent',\n  memory: new Memory({ store: 'sqlite', path: './memory.db' }),\n  tools: [greet],\n  system: 'You are a helpful assistant.',\n});\n\nconst response = await agent.chat(process.argv[2] ?? 'Hello!');\nconsole.log(response.content);\n`;\n\nconst readmeMd = `# ${projectName}\n\nBuilt with [agent-kit](https://github.com/abhid1234/agent-kit).\n\n## Quick Start\n\n\\`\\`\\`bash\nnpm install\nnpm start \"Hello, what can you do?\"\n\\`\\`\\`\n`;\n\nfs.writeFileSync(path.join(targetDir, 'package.json'), JSON.stringify(packageJson, null, 2) + '\\n');\nfs.writeFileSync(path.join(targetDir, 'agent.ts'), agentTs);\nfs.writeFileSync(path.join(targetDir, 'README.md'), readmeMd);\n\nconsole.log(`\\nScaffolded project: ${projectName}\\n`);\nconsole.log('Next steps:\\n');\nconsole.log(`  cd ${projectName}`);\nconsole.log('  npm install');\nconsole.log('  npm start \"Hello, what can you do?\"\\n');\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAEA,gBAAe;AACf,kBAAiB;AAEjB,IAAM,OAAO,QAAQ,KAAK,MAAM,CAAC;AACjC,IAAM,UAAU,KAAK,CAAC;AACtB,IAAM,cAAc,KAAK,CAAC,KAAK;AAE/B,IAAI,YAAY,QAAQ;AACtB,UAAQ,MAAM,oBAAoB,OAAO,EAAE;AAC3C,UAAQ,MAAM,sCAAsC;AACpD,UAAQ,KAAK,CAAC;AAChB;AAEA,IAAM,YAAY,YAAAA,QAAK,QAAQ,QAAQ,IAAI,GAAG,WAAW;AAEzD,IAAI,UAAAC,QAAG,WAAW,SAAS,GAAG;AAC5B,UAAQ,MAAM,6BAA6B,SAAS,EAAE;AACtD,UAAQ,KAAK,CAAC;AAChB;AAEA,UAAAA,QAAG,UAAU,WAAW,EAAE,WAAW,KAAK,CAAC;AAE3C,IAAM,cAAc;AAAA,EAClB,MAAM;AAAA,EACN,SAAS;AAAA,EACT,SAAS;AAAA,EACT,MAAM;AAAA,EACN,SAAS;AAAA,IACP,OAAO;AAAA,EACT;AAAA,EACA,cAAc;AAAA,IACZ,uBAAuB;AAAA,EACzB;AAAA,EACA,iBAAiB;AAAA,IACf,KAAK;AAAA,EACP;AACF;AAEA,IAAM,UAAU;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAsBhB,IAAM,WAAW,KAAK,WAAW;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAYjC,UAAAA,QAAG,cAAc,YAAAD,QAAK,KAAK,WAAW,cAAc,GAAG,KAAK,UAAU,aAAa,MAAM,CAAC,IAAI,IAAI;AAClG,UAAAC,QAAG,cAAc,YAAAD,QAAK,KAAK,WAAW,UAAU,GAAG,OAAO;AAC1D,UAAAC,QAAG,cAAc,YAAAD,QAAK,KAAK,WAAW,WAAW,GAAG,QAAQ;AAE5D,QAAQ,IAAI;AAAA,sBAAyB,WAAW;AAAA,CAAI;AACpD,QAAQ,IAAI,eAAe;AAC3B,QAAQ,IAAI,QAAQ,WAAW,EAAE;AACjC,QAAQ,IAAI,eAAe;AAC3B,QAAQ,IAAI,yCAAyC;","names":["path","fs"]}

package/dist/cli.d.cts

@@ -0,0 +1 @@
+#!/usr/bin/env node

package/dist/cli.d.ts

@@ -0,0 +1 @@
+#!/usr/bin/env node

package/dist/cli.js

@@ -0,0 +1,77 @@
+#!/usr/bin/env node
+
+// src/cli.ts
+import fs from "fs";
+import path from "path";
+var args = process.argv.slice(2);
+var command = args[0];
+var projectName = args[1] ?? "my-agent";
+if (command !== "init") {
+  console.error(`Unknown command: ${command}`);
+  console.error("Usage: agent-kit init [project-name]");
+  process.exit(1);
+}
+var targetDir = path.resolve(process.cwd(), projectName);
+if (fs.existsSync(targetDir)) {
+  console.error(`Directory already exists: ${targetDir}`);
+  process.exit(1);
+}
+fs.mkdirSync(targetDir, { recursive: true });
+var packageJson = {
+  name: projectName,
+  version: "1.0.0",
+  private: true,
+  type: "module",
+  scripts: {
+    start: "npx tsx agent.ts"
+  },
+  dependencies: {
+    "@avee1234/agent-kit": "^0.2.0"
+  },
+  devDependencies: {
+    tsx: "^4.0.0"
+  }
+};
+var agentTs = `import { Agent, Tool, Memory } from '@avee1234/agent-kit';
+
+const greet = Tool.create({
+  name: 'greet',
+  description: 'Greet someone by name',
+  parameters: {
+    name: { type: 'string', description: 'Name to greet' },
+  },
+  execute: async ({ name }) => \`Hello, \${name}!\`,
+});
+
+const agent = new Agent({
+  name: 'my-agent',
+  memory: new Memory({ store: 'sqlite', path: './memory.db' }),
+  tools: [greet],
+  system: 'You are a helpful assistant.',
+});
+
+const response = await agent.chat(process.argv[2] ?? 'Hello!');
+console.log(response.content);
+`;
+var readmeMd = `# ${projectName}
+
+Built with [agent-kit](https://github.com/abhid1234/agent-kit).
+
+## Quick Start
+
+\`\`\`bash
+npm install
+npm start "Hello, what can you do?"
+\`\`\`
+`;
+fs.writeFileSync(path.join(targetDir, "package.json"), JSON.stringify(packageJson, null, 2) + "\n");
+fs.writeFileSync(path.join(targetDir, "agent.ts"), agentTs);
+fs.writeFileSync(path.join(targetDir, "README.md"), readmeMd);
+console.log(`
+Scaffolded project: ${projectName}
+`);
+console.log("Next steps:\n");
+console.log(`  cd ${projectName}`);
+console.log("  npm install");
+console.log('  npm start "Hello, what can you do?"\n');
+//# sourceMappingURL=cli.js.map

package/dist/cli.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/cli.ts"],"sourcesContent":["#!/usr/bin/env node\n\nimport fs from 'fs';\nimport path from 'path';\n\nconst args = process.argv.slice(2);\nconst command = args[0];\nconst projectName = args[1] ?? 'my-agent';\n\nif (command !== 'init') {\n  console.error(`Unknown command: ${command}`);\n  console.error('Usage: agent-kit init [project-name]');\n  process.exit(1);\n}\n\nconst targetDir = path.resolve(process.cwd(), projectName);\n\nif (fs.existsSync(targetDir)) {\n  console.error(`Directory already exists: ${targetDir}`);\n  process.exit(1);\n}\n\nfs.mkdirSync(targetDir, { recursive: true });\n\nconst packageJson = {\n  name: projectName,\n  version: '1.0.0',\n  private: true,\n  type: 'module',\n  scripts: {\n    start: 'npx tsx agent.ts',\n  },\n  dependencies: {\n    '@avee1234/agent-kit': '^0.2.0',\n  },\n  devDependencies: {\n    tsx: '^4.0.0',\n  },\n};\n\nconst agentTs = `import { Agent, Tool, Memory } from '@avee1234/agent-kit';\n\nconst greet = Tool.create({\n  name: 'greet',\n  description: 'Greet someone by name',\n  parameters: {\n    name: { type: 'string', description: 'Name to greet' },\n  },\n  execute: async ({ name }) => \\`Hello, \\${name}!\\`,\n});\n\nconst agent = new Agent({\n  name: 'my-agent',\n  memory: new Memory({ store: 'sqlite', path: './memory.db' }),\n  tools: [greet],\n  system: 'You are a helpful assistant.',\n});\n\nconst response = await agent.chat(process.argv[2] ?? 'Hello!');\nconsole.log(response.content);\n`;\n\nconst readmeMd = `# ${projectName}\n\nBuilt with [agent-kit](https://github.com/abhid1234/agent-kit).\n\n## Quick Start\n\n\\`\\`\\`bash\nnpm install\nnpm start \"Hello, what can you do?\"\n\\`\\`\\`\n`;\n\nfs.writeFileSync(path.join(targetDir, 'package.json'), JSON.stringify(packageJson, null, 2) + '\\n');\nfs.writeFileSync(path.join(targetDir, 'agent.ts'), agentTs);\nfs.writeFileSync(path.join(targetDir, 'README.md'), readmeMd);\n\nconsole.log(`\\nScaffolded project: ${projectName}\\n`);\nconsole.log('Next steps:\\n');\nconsole.log(`  cd ${projectName}`);\nconsole.log('  npm install');\nconsole.log('  npm start \"Hello, what can you do?\"\\n');\n"],"mappings":";;;AAEA,OAAO,QAAQ;AACf,OAAO,UAAU;AAEjB,IAAM,OAAO,QAAQ,KAAK,MAAM,CAAC;AACjC,IAAM,UAAU,KAAK,CAAC;AACtB,IAAM,cAAc,KAAK,CAAC,KAAK;AAE/B,IAAI,YAAY,QAAQ;AACtB,UAAQ,MAAM,oBAAoB,OAAO,EAAE;AAC3C,UAAQ,MAAM,sCAAsC;AACpD,UAAQ,KAAK,CAAC;AAChB;AAEA,IAAM,YAAY,KAAK,QAAQ,QAAQ,IAAI,GAAG,WAAW;AAEzD,IAAI,GAAG,WAAW,SAAS,GAAG;AAC5B,UAAQ,MAAM,6BAA6B,SAAS,EAAE;AACtD,UAAQ,KAAK,CAAC;AAChB;AAEA,GAAG,UAAU,WAAW,EAAE,WAAW,KAAK,CAAC;AAE3C,IAAM,cAAc;AAAA,EAClB,MAAM;AAAA,EACN,SAAS;AAAA,EACT,SAAS;AAAA,EACT,MAAM;AAAA,EACN,SAAS;AAAA,IACP,OAAO;AAAA,EACT;AAAA,EACA,cAAc;AAAA,IACZ,uBAAuB;AAAA,EACzB;AAAA,EACA,iBAAiB;AAAA,IACf,KAAK;AAAA,EACP;AACF;AAEA,IAAM,UAAU;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAsBhB,IAAM,WAAW,KAAK,WAAW;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAYjC,GAAG,cAAc,KAAK,KAAK,WAAW,cAAc,GAAG,KAAK,UAAU,aAAa,MAAM,CAAC,IAAI,IAAI;AAClG,GAAG,cAAc,KAAK,KAAK,WAAW,UAAU,GAAG,OAAO;AAC1D,GAAG,cAAc,KAAK,KAAK,WAAW,WAAW,GAAG,QAAQ;AAE5D,QAAQ,IAAI;AAAA,sBAAyB,WAAW;AAAA,CAAI;AACpD,QAAQ,IAAI,eAAe;AAC3B,QAAQ,IAAI,QAAQ,WAAW,EAAE;AACjC,QAAQ,IAAI,eAAe;AAC3B,QAAQ,IAAI,yCAAyC;","names":[]}

package/dist/index.cjs

@@ -25,7 +25,9 @@ __export(index_exports, {
   InMemoryStore: () => InMemoryStore,
   Memory: () => Memory,
   MockAdapter: () => MockAdapter,
+  OllamaEmbeddingAdapter: () => OllamaEmbeddingAdapter,
   OpenAICompatibleAdapter: () => OpenAICompatibleAdapter,
+  PostgresStore: () => PostgresStore,
   SQLiteStore: () => SQLiteStore,
   Team: () => Team,
   Tool: () => Tool,
@@ -101,7 +103,15 @@ var OpenAICompatibleAdapter = class {
         function: {
           name: t.name,
           description: t.description,
-          parameters: { type: "object", properties: t.parameters }
+          parameters: {
+            type: "object",
+            properties: Object.fromEntries(
+              Object.entries(t.parameters).map(([key, val]) => {
+                const { required: _req, ...rest } = val;
+                return [key, rest];
+              })
+            )
+          }
         }
       }));
     }
@@ -147,7 +157,15 @@ var OpenAICompatibleAdapter = class {
         function: {
           name: t.name,
           description: t.description,
-          parameters: { type: "object", properties: t.parameters }
+          parameters: {
+            type: "object",
+            properties: Object.fromEntries(
+              Object.entries(t.parameters).map(([key, val]) => {
+                const { required: _req, ...rest } = val;
+                return [key, rest];
+              })
+            )
+          }
         }
       }));
     }
@@ -609,13 +627,11 @@ var HierarchicalStrategy = class {
       parameters: {
         agentName: {
           type: "string",
-          description: "The name of the agent to delegate to",
-          required: true
+          description: `The name of the agent to delegate to. Available: ${[...agents.map((a) => a.name)].join(", ")}`
         },
         task: {
           type: "string",
-          description: "The task to assign to the agent",
-          required: true
+          description: "The task to assign to the agent"
         }
       },
       execute: async (params) => {
@@ -724,10 +740,27 @@ var Team = class {
   }
 };
 
+// src/store/cosine.ts
+function cosineSimilarity(a, b) {
+  let dot = 0;
+  let normA = 0;
+  let normB = 0;
+  for (let i = 0; i < a.length; i++) {
+    dot += a[i] * b[i];
+    normA += a[i] * a[i];
+    normB += b[i] * b[i];
+  }
+  const denom = Math.sqrt(normA) * Math.sqrt(normB);
+  if (denom === 0) return 0;
+  return dot / denom;
+}
+
 // src/store/in-memory.ts
 var InMemoryStore = class {
   messages = /* @__PURE__ */ new Map();
   summaries = /* @__PURE__ */ new Map();
+  // agentId -> (summaryId -> embedding vector)
+  embeddings = /* @__PURE__ */ new Map();
   async saveMessages(agentId, messages) {
     const existing = this.messages.get(agentId) ?? [];
     existing.push(...messages);
@@ -748,6 +781,28 @@ var InMemoryStore = class {
     const matches = all.filter((s) => s.content.toLowerCase().includes(queryLower)).sort((a, b) => b.timestamp - a.timestamp).slice(0, limit);
     return matches;
   }
+  async saveEmbedding(agentId, summaryId, embedding) {
+    let agentMap = this.embeddings.get(agentId);
+    if (!agentMap) {
+      agentMap = /* @__PURE__ */ new Map();
+      this.embeddings.set(agentId, agentMap);
+    }
+    agentMap.set(summaryId, embedding);
+  }
+  async searchByEmbedding(agentId, embedding, limit) {
+    const agentMap = this.embeddings.get(agentId);
+    if (!agentMap || agentMap.size === 0) return [];
+    const summaries = this.summaries.get(agentId) ?? [];
+    const summaryById = new Map(summaries.map((s) => [s.id, s]));
+    const scored = [];
+    for (const [summaryId, storedEmbedding] of agentMap) {
+      const summary = summaryById.get(summaryId);
+      if (!summary) continue;
+      const score = cosineSimilarity(embedding, storedEmbedding);
+      scored.push({ summary, score });
+    }
+    return scored.sort((a, b) => b.score - a.score).slice(0, limit).map((entry) => entry.summary);
+  }
 };
 
 // src/store/sqlite.ts
@@ -781,8 +836,14 @@ var SQLiteStore = class {
         message_range_from INTEGER NOT NULL,
         message_range_to INTEGER NOT NULL
       );
+      CREATE TABLE IF NOT EXISTS embeddings (
+        summary_id TEXT PRIMARY KEY,
+        agent_id TEXT NOT NULL,
+        embedding BLOB NOT NULL
+      );
       CREATE INDEX IF NOT EXISTS idx_messages_agent ON messages(agent_id);
       CREATE INDEX IF NOT EXISTS idx_summaries_agent ON summaries(agent_id);
+      CREATE INDEX IF NOT EXISTS idx_embeddings_agent ON embeddings(agent_id);
     `);
   }
   async saveMessages(agentId, messages) {
@@ -842,25 +903,209 @@ var SQLiteStore = class {
       messageRange: { from: row.message_range_from, to: row.message_range_to }
     }));
   }
+  async saveEmbedding(agentId, summaryId, embedding) {
+    this.db.prepare(
+      "INSERT OR REPLACE INTO embeddings (summary_id, agent_id, embedding) VALUES (?, ?, ?)"
+    ).run(summaryId, agentId, JSON.stringify(embedding));
+  }
+  async searchByEmbedding(agentId, embedding, limit) {
+    const rows = this.db.prepare(
+      `SELECT e.summary_id, s.content, s.timestamp, s.message_range_from, s.message_range_to, e.embedding
+         FROM embeddings e
+         JOIN summaries s ON e.summary_id = s.id
+         WHERE e.agent_id = ?`
+    ).all(agentId);
+    if (rows.length === 0) return [];
+    const scored = rows.map((row) => {
+      const storedEmbedding = JSON.parse(row.embedding);
+      const score = cosineSimilarity(embedding, storedEmbedding);
+      return {
+        summary: {
+          id: row.summary_id,
+          content: row.content,
+          timestamp: row.timestamp,
+          messageRange: { from: row.message_range_from, to: row.message_range_to }
+        },
+        score
+      };
+    });
+    return scored.sort((a, b) => b.score - a.score).slice(0, limit).map((entry) => entry.summary);
+  }
   close() {
     this.db.close();
   }
 };
 
+// src/store/postgres.ts
+var import_module2 = require("module");
+var import_meta2 = {};
+var require3 = (0, import_module2.createRequire)(import_meta2.url);
+var SCHEMA_SQL = `
+CREATE EXTENSION IF NOT EXISTS vector;
+
+CREATE TABLE IF NOT EXISTS agent_messages (
+  id SERIAL PRIMARY KEY,
+  agent_id TEXT NOT NULL,
+  role TEXT NOT NULL,
+  content TEXT NOT NULL,
+  timestamp BIGINT NOT NULL,
+  tool_calls JSONB,
+  tool_call_id TEXT
+);
+
+CREATE TABLE IF NOT EXISTS agent_summaries (
+  id TEXT PRIMARY KEY,
+  agent_id TEXT NOT NULL,
+  content TEXT NOT NULL,
+  timestamp BIGINT NOT NULL,
+  message_range_from BIGINT NOT NULL,
+  message_range_to BIGINT NOT NULL,
+  embedding vector(768)
+);
+
+CREATE INDEX IF NOT EXISTS idx_agent_messages_agent ON agent_messages(agent_id);
+CREATE INDEX IF NOT EXISTS idx_agent_summaries_agent ON agent_summaries(agent_id);
+`;
+var PostgresStore = class _PostgresStore {
+  pool;
+  constructor(config) {
+    const { Pool } = require3("pg");
+    this.pool = new Pool({ connectionString: config.connectionString });
+    void this.init();
+  }
+  async init() {
+    await this.pool.query(SCHEMA_SQL);
+  }
+  /** Factory that ensures schema is ready before returning the store. */
+  static async create(config) {
+    const store = new _PostgresStore(config);
+    await store.init();
+    return store;
+  }
+  async saveMessages(agentId, messages) {
+    for (const msg of messages) {
+      await this.pool.query(
+        `INSERT INTO agent_messages (agent_id, role, content, timestamp, tool_calls, tool_call_id)
+         VALUES ($1, $2, $3, $4, $5, $6)`,
+        [
+          agentId,
+          msg.role,
+          msg.content,
+          msg.timestamp,
+          msg.toolCalls ? JSON.stringify(msg.toolCalls) : null,
+          msg.toolCallId ?? null
+        ]
+      );
+    }
+  }
+  async getRecentMessages(agentId, limit) {
+    const result = await this.pool.query(
+      `SELECT role, content, timestamp, tool_calls, tool_call_id
+       FROM agent_messages
+       WHERE agent_id = $1
+       ORDER BY id DESC
+       LIMIT $2`,
+      [agentId, limit]
+    );
+    return result.rows.reverse().map((row) => ({
+      role: row.role,
+      content: row.content,
+      timestamp: Number(row.timestamp),
+      ...row.tool_calls ? {
+        toolCalls: typeof row.tool_calls === "string" ? JSON.parse(row.tool_calls) : row.tool_calls
+      } : {},
+      ...row.tool_call_id ? { toolCallId: row.tool_call_id } : {}
+    }));
+  }
+  async saveSummary(agentId, summary) {
+    await this.pool.query(
+      `INSERT INTO agent_summaries (id, agent_id, content, timestamp, message_range_from, message_range_to)
+       VALUES ($1, $2, $3, $4, $5, $6)`,
+      [
+        summary.id,
+        agentId,
+        summary.content,
+        summary.timestamp,
+        summary.messageRange.from,
+        summary.messageRange.to
+      ]
+    );
+  }
+  async searchSummaries(agentId, query, limit) {
+    const result = await this.pool.query(
+      `SELECT id, content, timestamp, message_range_from, message_range_to
+       FROM agent_summaries
+       WHERE agent_id = $1 AND content ILIKE $2
+       ORDER BY timestamp DESC
+       LIMIT $3`,
+      [agentId, `%${query}%`, limit]
+    );
+    return result.rows.map((row) => ({
+      id: row.id,
+      content: row.content,
+      timestamp: Number(row.timestamp),
+      messageRange: {
+        from: Number(row.message_range_from),
+        to: Number(row.message_range_to)
+      }
+    }));
+  }
+  async saveEmbedding(agentId, summaryId, embedding) {
+    const vectorLiteral = `[${embedding.join(",")}]`;
+    await this.pool.query(
+      `UPDATE agent_summaries SET embedding = $1 WHERE id = $2 AND agent_id = $3`,
+      [vectorLiteral, summaryId, agentId]
+    );
+  }
+  async searchByEmbedding(agentId, embedding, limit) {
+    const vectorLiteral = `[${embedding.join(",")}]`;
+    const result = await this.pool.query(
+      `SELECT id, content, timestamp, message_range_from, message_range_to
+       FROM agent_summaries
+       WHERE agent_id = $1 AND embedding IS NOT NULL
+       ORDER BY embedding <=> $2
+       LIMIT $3`,
+      [agentId, vectorLiteral, limit]
+    );
+    return result.rows.map((row) => ({
+      id: row.id,
+      content: row.content,
+      timestamp: Number(row.timestamp),
+      messageRange: {
+        from: Number(row.message_range_from),
+        to: Number(row.message_range_to)
+      }
+    }));
+  }
+  async close() {
+    await this.pool.end();
+  }
+};
+
 // src/memory.ts
 var Memory = class {
   store;
   windowSize;
   summarizeAfter;
   model;
+  embeddingAdapter;
   messageCount = /* @__PURE__ */ new Map();
   constructor(config = {}) {
     this.windowSize = config.windowSize ?? 20;
     this.summarizeAfter = config.summarizeAfter ?? 20;
+    this.embeddingAdapter = config.embedding;
     if (!config.store || config.store === "memory") {
       this.store = new InMemoryStore();
     } else if (config.store === "sqlite") {
       this.store = new SQLiteStore(config.path ?? "./agent-memory.db");
+    } else if (config.store === "postgres") {
+      const connectionString = config.url ?? config.path;
+      if (!connectionString) {
+        throw new Error(
+          "Memory store 'postgres' requires a connection string via the 'url' config option."
+        );
+      }
+      this.store = new PostgresStore({ connectionString });
     } else {
       this.store = config.store;
     }
@@ -882,7 +1127,13 @@ var Memory = class {
   }
   async getContext(agentId, query) {
     const recentMessages = await this.store.getRecentMessages(agentId, this.windowSize);
-    const relevantSummaries = await this.store.searchSummaries(agentId, query, 3);
+    let relevantSummaries;
+    if (this.embeddingAdapter && this.store.searchByEmbedding) {
+      const queryEmbedding = await this.embeddingAdapter.embed(query);
+      relevantSummaries = await this.store.searchByEmbedding(agentId, queryEmbedding, 3);
+    } else {
+      relevantSummaries = await this.store.searchSummaries(agentId, query, 3);
+    }
     return { recentMessages, relevantSummaries };
   }
   async summarize(agentId) {
@@ -905,6 +1156,10 @@ var Memory = class {
       }
     });
     await this.store.saveSummary(agentId, summary);
+    if (this.embeddingAdapter && this.store.saveEmbedding) {
+      const embedding = await this.embeddingAdapter.embed(summary.content);
+      await this.store.saveEmbedding(agentId, summary.id, embedding);
+    }
   }
   close() {
     if ("close" in this.store && typeof this.store.close === "function") {
@@ -912,6 +1167,36 @@ var Memory = class {
     }
   }
 };
+
+// src/model/ollama-embedding.ts
+var OllamaEmbeddingAdapter = class {
+  baseURL;
+  model;
+  constructor(config = {}) {
+    this.baseURL = config.baseURL ?? "http://localhost:11434";
+    this.model = config.model ?? "nomic-embed-text";
+  }
+  async embed(text) {
+    const [vector] = await this.request(text);
+    return vector;
+  }
+  async embedBatch(texts) {
+    if (texts.length === 0) return [];
+    return this.request(texts);
+  }
+  async request(input) {
+    const response = await fetch(`${this.baseURL}/api/embed`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ model: this.model, input })
+    });
+    if (!response.ok) {
+      throw new Error(`Ollama embedding request failed: ${response.status} ${response.statusText}`);
+    }
+    const data = await response.json();
+    return data.embeddings;
+  }
+};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   Agent,
@@ -919,7 +1204,9 @@ var Memory = class {
   InMemoryStore,
   Memory,
   MockAdapter,
+  OllamaEmbeddingAdapter,
   OpenAICompatibleAdapter,
+  PostgresStore,
   SQLiteStore,
   Team,
   Tool,