@mastra/pg 1.0.0-beta.5 → 1.0.0-beta.7

package/dist/index.js

@@ -1,5 +1,5 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { createVectorErrorId, MastraStorage, createStorageErrorId, StoreOperations, TABLE_SCHEMAS, TABLE_WORKFLOW_SNAPSHOT, TABLE_SPANS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_SCORERS, ScoresStorage, normalizePerPage, calculatePagination, WorkflowsStorage, MemoryStorage, TABLE_RESOURCES, ObservabilityStorage, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
+import { createVectorErrorId, MastraStorage, createStorageErrorId, StoreOperations, TABLE_SCHEMAS, TABLE_WORKFLOW_SNAPSHOT, TABLE_SPANS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_SCORERS, ScoresStorage, normalizePerPage, calculatePagination, WorkflowsStorage, MemoryStorage, TABLE_RESOURCES, ObservabilityStorage, AgentsStorage, transformScoreRow as transformScoreRow$1, TABLE_AGENTS } from '@mastra/core/storage';
 import { parseSqlIdentifier, parseFieldKey } from '@mastra/core/utils';
 import { MastraVector } from '@mastra/core/vector';
 import { Mutex } from 'async-mutex';
@@ -480,12 +480,14 @@ var PgVector = class extends MastraVector {
   pool;
   describeIndexCache = /* @__PURE__ */ new Map();
   createdIndexes = /* @__PURE__ */ new Map();
+  indexVectorTypes = /* @__PURE__ */ new Map();
   mutexesByName = /* @__PURE__ */ new Map();
   schema;
   setupSchemaPromise = null;
   installVectorExtensionPromise = null;
   vectorExtensionInstalled = void 0;
   vectorExtensionSchema = null;
+  vectorExtensionVersion = null;
   schemaSetupComplete = void 0;
   cacheWarmupPromise = null;
   constructor(config) {
@@ -538,9 +540,11 @@ var PgVector = class extends MastraVector {
                 indexName,
                 metric: info.metric,
                 dimension: info.dimension,
-                type: info.type
+                type: info.type,
+                vectorType: info.vectorType
               });
               this.createdIndexes.set(indexName, key);
+              this.indexVectorTypes.set(indexName, info.vectorType);
             })
           );
         } catch (error) {
@@ -566,12 +570,12 @@ var PgVector = class extends MastraVector {
     return this.mutexesByName.get(indexName);
   }
   /**
-   * Detects which schema contains the vector extension
+   * Detects which schema contains the vector extension and its version
    */
   async detectVectorExtensionSchema(client) {
     try {
       const result = await client.query(`
-        SELECT n.nspname as schema_name
+        SELECT n.nspname as schema_name, e.extversion as version
         FROM pg_extension e
         JOIN pg_namespace n ON e.extnamespace = n.oid
         WHERE e.extname = 'vector'
@@ -579,7 +583,11 @@ var PgVector = class extends MastraVector {
       `);
       if (result.rows.length > 0) {
         this.vectorExtensionSchema = result.rows[0].schema_name;
-        this.logger.debug("Vector extension found in schema", { schema: this.vectorExtensionSchema });
+        this.vectorExtensionVersion = result.rows[0].version;
+        this.logger.debug("Vector extension found", {
+          schema: this.vectorExtensionSchema,
+          version: this.vectorExtensionVersion
+        });
         return this.vectorExtensionSchema;
       }
       return null;
@@ -588,18 +596,52 @@ var PgVector = class extends MastraVector {
       return null;
     }
   }
+  /**
+   * Checks if the installed pgvector version supports halfvec type.
+   * halfvec was introduced in pgvector 0.7.0.
+   */
+  supportsHalfvec() {
+    if (!this.vectorExtensionVersion) {
+      return false;
+    }
+    const parts = this.vectorExtensionVersion.split(".");
+    const major = parseInt(parts[0] ?? "", 10);
+    const minor = parseInt(parts[1] ?? "", 10);
+    if (isNaN(major) || isNaN(minor)) {
+      return false;
+    }
+    return major > 0 || major === 0 && minor >= 7;
+  }
   /**
    * Gets the properly qualified vector type name
+   * @param vectorType - The type of vector storage ('vector' or 'halfvec')
    */
-  getVectorTypeName() {
+  getVectorTypeName(vectorType = "vector") {
     if (this.vectorExtensionSchema) {
       if (this.vectorExtensionSchema === "pg_catalog") {
-        return "vector";
+        return vectorType;
       }
       const validatedSchema = parseSqlIdentifier(this.vectorExtensionSchema, "vector extension schema");
-      return `${validatedSchema}.vector`;
+      return `${validatedSchema}.${vectorType}`;
+    }
+    return vectorType;
+  }
+  /**
+   * Gets the operator class for index creation based on metric and vector type.
+   * pgvector uses different operator classes for vector vs halfvec types.
+   */
+  getMetricOperatorClass(metric, vectorType) {
+    const prefix = vectorType === "halfvec" ? "halfvec" : "vector";
+    switch (metric) {
+      case "cosine":
+        return `${prefix}_cosine_ops`;
+      case "euclidean":
+        return `${prefix}_l2_ops`;
+      case "dotproduct":
+        return `${prefix}_ip_ops`;
+      default:
+        return `${prefix}_cosine_ops`;
     }
-    return "vector";
   }
   getTableName(indexName) {
     const parsedIndexName = parseSqlIdentifier(indexName, "index name");
@@ -672,12 +714,12 @@ var PgVector = class extends MastraVector {
         await client.query(`SET LOCAL ivfflat.probes = ${probes}`);
       }
       const { tableName } = this.getTableName(indexName);
-      const vectorType = this.getVectorTypeName();
+      const qualifiedVectorType = this.getVectorTypeName(indexInfo.vectorType);
       const query = `
         WITH vector_scores AS (
           SELECT
             vector_id as id,
-            1 - (embedding <=> '${vectorStr}'::${vectorType}) as score,
+            1 - (embedding <=> '${vectorStr}'::${qualifiedVectorType}) as score,
             metadata
             ${includeVector ? ", embedding" : ""}
           FROM ${tableName}
@@ -741,14 +783,15 @@ var PgVector = class extends MastraVector {
         }
       }
       const vectorIds = ids || vectors.map(() => crypto.randomUUID());
-      const vectorType = this.getVectorTypeName();
+      const indexInfo = await this.getIndexInfo({ indexName });
+      const qualifiedVectorType = this.getVectorTypeName(indexInfo.vectorType);
       for (let i = 0; i < vectors.length; i++) {
         const query = `
           INSERT INTO ${tableName} (vector_id, embedding, metadata)
-          VALUES ($1, $2::${vectorType}, $3::jsonb)
+          VALUES ($1, $2::${qualifiedVectorType}, $3::jsonb)
           ON CONFLICT (vector_id)
           DO UPDATE SET
-            embedding = $2::${vectorType},
+            embedding = $2::${qualifiedVectorType},
             metadata = $3::jsonb
           RETURNING embedding::text
         `;
@@ -807,9 +850,10 @@ var PgVector = class extends MastraVector {
     indexName,
     dimension,
     metric,
-    type
+    type,
+    vectorType = "vector"
   }) {
-    const input = indexName + dimension + metric + (type || "ivfflat");
+    const input = indexName + dimension + metric + (type || "ivfflat") + vectorType;
     return (await this.hasher).h32(input);
   }
   cachedIndexExists(indexName, newKey) {
@@ -862,7 +906,8 @@ var PgVector = class extends MastraVector {
     dimension,
     metric = "cosine",
     indexConfig = {},
-    buildIndex = true
+    buildIndex = true,
+    vectorType = "vector"
   }) {
     const { tableName } = this.getTableName(indexName);
     try {
@@ -872,6 +917,9 @@ var PgVector = class extends MastraVector {
       if (!Number.isInteger(dimension) || dimension <= 0) {
         throw new Error("Dimension must be a positive integer");
       }
+      if (vectorType !== "vector" && vectorType !== "halfvec") {
+        throw new Error('vectorType must be "vector" or "halfvec"');
+      }
     } catch (error) {
       const mastraError = new MastraError(
         {
@@ -887,7 +935,13 @@ var PgVector = class extends MastraVector {
       this.logger?.trackException(mastraError);
       throw mastraError;
     }
-    const indexCacheKey = await this.getIndexCacheKey({ indexName, dimension, type: indexConfig.type, metric });
+    const indexCacheKey = await this.getIndexCacheKey({
+      indexName,
+      dimension,
+      type: indexConfig.type,
+      metric,
+      vectorType
+    });
     if (this.cachedIndexExists(indexName, indexCacheKey)) {
       return;
     }
@@ -900,24 +954,40 @@ var PgVector = class extends MastraVector {
       try {
         await this.setupSchema(client);
         await this.installVectorExtension(client);
+        if (vectorType === "halfvec" && !this.supportsHalfvec()) {
+          throw new MastraError({
+            id: createVectorErrorId("PG", "CREATE_INDEX", "HALFVEC_NOT_SUPPORTED"),
+            text: `halfvec type requires pgvector >= 0.7.0, but version ${this.vectorExtensionVersion || "unknown"} is installed. Either upgrade pgvector or use vectorType: 'vector' (which supports up to 2000 dimensions for indexes).`,
+            domain: ErrorDomain.MASTRA_VECTOR,
+            category: ErrorCategory.USER,
+            details: {
+              indexName,
+              requestedVectorType: vectorType,
+              pgvectorVersion: this.vectorExtensionVersion || "unknown",
+              requiredVersion: "0.7.0"
+            }
+          });
+        }
         if (this.schema && this.vectorExtensionSchema && this.schema !== this.vectorExtensionSchema && this.vectorExtensionSchema !== "pg_catalog") {
           await client.query(`SET search_path TO ${this.getSchemaName()}, "${this.vectorExtensionSchema}"`);
         }
-        const vectorType = this.getVectorTypeName();
+        const qualifiedVectorType = this.getVectorTypeName(vectorType);
         await client.query(`
           CREATE TABLE IF NOT EXISTS ${tableName} (
             id SERIAL PRIMARY KEY,
             vector_id TEXT UNIQUE NOT NULL,
-            embedding ${vectorType}(${dimension}),
+            embedding ${qualifiedVectorType}(${dimension}),
             metadata JSONB DEFAULT '{}'::jsonb
           );
         `);
         this.createdIndexes.set(indexName, indexCacheKey);
+        this.indexVectorTypes.set(indexName, vectorType);
         if (buildIndex) {
-          await this.setupIndex({ indexName, metric, indexConfig }, client);
+          await this.setupIndex({ indexName, metric, indexConfig, vectorType }, client);
         }
       } catch (error) {
         this.createdIndexes.delete(indexName);
+        this.indexVectorTypes.delete(indexName);
         throw error;
       } finally {
         client.release();
@@ -960,7 +1030,7 @@ var PgVector = class extends MastraVector {
       client.release();
     }
   }
-  async setupIndex({ indexName, metric, indexConfig }, client) {
+  async setupIndex({ indexName, metric, indexConfig, vectorType = "vector" }, client) {
     const mutex = this.getMutexByName(`build-${indexName}`);
     await mutex.runExclusive(async () => {
       const isConfigEmpty = !indexConfig || Object.keys(indexConfig).length === 0 || !indexConfig.type && !indexConfig.ivf && !indexConfig.hnsw;
@@ -982,9 +1052,11 @@ var PgVector = class extends MastraVector {
               indexName,
               dimension,
               type: existingIndexInfo.type,
-              metric: existingIndexInfo.metric
+              metric: existingIndexInfo.metric,
+              vectorType: existingIndexInfo.vectorType
             });
             this.createdIndexes.set(indexName, cacheKey);
+            this.indexVectorTypes.set(indexName, existingIndexInfo.vectorType);
             return;
           }
         }
@@ -1002,9 +1074,11 @@ var PgVector = class extends MastraVector {
             indexName,
             dimension,
             type: existingIndexInfo.type,
-            metric: existingIndexInfo.metric
+            metric: existingIndexInfo.metric,
+            vectorType: existingIndexInfo.vectorType
           });
           this.createdIndexes.set(indexName, cacheKey);
+          this.indexVectorTypes.set(indexName, existingIndexInfo.vectorType);
           return;
         }
         this.logger?.info(`Index ${vectorIndexName} configuration changed, rebuilding index`);
@@ -1017,7 +1091,8 @@ var PgVector = class extends MastraVector {
         this.describeIndexCache.delete(indexName);
         return;
       }
-      const metricOp = metric === "cosine" ? "vector_cosine_ops" : metric === "euclidean" ? "vector_l2_ops" : "vector_ip_ops";
+      const effectiveVectorType = existingIndexInfo?.vectorType ?? vectorType;
+      const metricOp = this.getMetricOperatorClass(metric, effectiveVectorType);
       let indexSQL;
       if (indexType === "hnsw") {
         const m = indexConfig.hnsw?.m ?? 8;
@@ -1067,6 +1142,12 @@ var PgVector = class extends MastraVector {
             if (this.schema && this.schema !== "public") {
               try {
                 await client.query(`CREATE EXTENSION IF NOT EXISTS vector SCHEMA ${this.getSchemaName()}`);
+                const installedSchema2 = await this.detectVectorExtensionSchema(client);
+                if (installedSchema2) {
+                  this.vectorExtensionInstalled = true;
+                  this.logger.info(`Vector extension installed in schema: ${installedSchema2}`);
+                  return;
+                }
                 this.vectorExtensionInstalled = true;
                 this.vectorExtensionSchema = this.schema;
                 this.logger.info(`Vector extension installed in schema: ${this.schema}`);
@@ -1129,7 +1210,7 @@ var PgVector = class extends MastraVector {
           WHERE c.table_schema = t.table_schema
           AND c.table_name = t.table_name
           AND c.column_name = 'embedding'
-          AND c.udt_name = 'vector'
+          AND c.udt_name IN ('vector', 'halfvec')
         )
         AND EXISTS (
           SELECT 1
@@ -1168,17 +1249,18 @@ var PgVector = class extends MastraVector {
     try {
       const { tableName } = this.getTableName(indexName);
       const tableExistsQuery = `
-        SELECT 1
+        SELECT udt_name
         FROM information_schema.columns
         WHERE table_schema = $1
           AND table_name = $2
-          AND udt_name = 'vector'
+          AND udt_name IN ('vector', 'halfvec')
         LIMIT 1;
       `;
       const tableExists = await client.query(tableExistsQuery, [this.schema || "public", indexName]);
       if (tableExists.rows.length === 0) {
         throw new Error(`Vector table ${tableName} does not exist`);
       }
+      const vectorType = tableExists.rows[0].udt_name === "halfvec" ? "halfvec" : "vector";
       const dimensionQuery = `
                 SELECT atttypmod as dimension
                 FROM pg_attribute
@@ -1228,6 +1310,7 @@ var PgVector = class extends MastraVector {
         count: parseInt(countResult.rows[0].count),
         metric,
         type: index_method,
+        vectorType,
         config
       };
     } catch (e) {
@@ -1255,6 +1338,8 @@ var PgVector = class extends MastraVector {
       const { tableName } = this.getTableName(indexName);
       await client.query(`DROP TABLE IF EXISTS ${tableName} CASCADE`);
       this.createdIndexes.delete(indexName);
+      this.indexVectorTypes.delete(indexName);
+      this.describeIndexCache.delete(indexName);
     } catch (error) {
       await client.query("ROLLBACK");
       const mastraError = new MastraError(
@@ -1343,12 +1428,13 @@ var PgVector = class extends MastraVector {
       }
       client = await this.pool.connect();
       const { tableName } = this.getTableName(indexName);
-      const vectorType = this.getVectorTypeName();
+      const indexInfo = await this.getIndexInfo({ indexName });
+      const qualifiedVectorType = this.getVectorTypeName(indexInfo.vectorType);
       let updateParts = [];
       let values = [];
       let valueIndex = 1;
       if (update.vector) {
-        updateParts.push(`embedding = $${valueIndex}::${vectorType}`);
+        updateParts.push(`embedding = $${valueIndex}::${qualifiedVectorType}`);
         values.push(`[${update.vector.join(",")}]`);
         valueIndex++;
       }
@@ -1637,7 +1723,302 @@ function transformFromSqlRow({
   return result;
 }
 
-// src/storage/domains/memory/index.ts
+// src/storage/domains/agents/index.ts
+var AgentsPG = class extends AgentsStorage {
+  client;
+  schema;
+  constructor({ client, schema }) {
+    super();
+    this.client = client;
+    this.schema = schema;
+  }
+  parseJson(value, fieldName) {
+    if (!value) return void 0;
+    if (typeof value !== "string") return value;
+    try {
+      return JSON.parse(value);
+    } catch (error) {
+      const details = {
+        value: value.length > 100 ? value.substring(0, 100) + "..." : value
+      };
+      if (fieldName) {
+        details.field = fieldName;
+      }
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "PARSE_JSON", "INVALID_JSON"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.SYSTEM,
+          text: `Failed to parse JSON${fieldName ? ` for field "${fieldName}"` : ""}: ${error instanceof Error ? error.message : "Unknown error"}`,
+          details
+        },
+        error
+      );
+    }
+  }
+  parseRow(row) {
+    return {
+      id: row.id,
+      name: row.name,
+      description: row.description,
+      instructions: row.instructions,
+      model: this.parseJson(row.model, "model"),
+      tools: this.parseJson(row.tools, "tools"),
+      defaultOptions: this.parseJson(row.defaultOptions, "defaultOptions"),
+      workflows: this.parseJson(row.workflows, "workflows"),
+      agents: this.parseJson(row.agents, "agents"),
+      inputProcessors: this.parseJson(row.inputProcessors, "inputProcessors"),
+      outputProcessors: this.parseJson(row.outputProcessors, "outputProcessors"),
+      memory: this.parseJson(row.memory, "memory"),
+      scorers: this.parseJson(row.scorers, "scorers"),
+      metadata: this.parseJson(row.metadata, "metadata"),
+      createdAt: row.createdAtZ || row.createdAt,
+      updatedAt: row.updatedAtZ || row.updatedAt
+    };
+  }
+  async getAgentById({ id }) {
+    try {
+      const tableName = getTableName({ indexName: TABLE_AGENTS, schemaName: getSchemaName(this.schema) });
+      const result = await this.client.oneOrNone(`SELECT * FROM ${tableName} WHERE id = $1`, [id]);
+      if (!result) {
+        return null;
+      }
+      return this.parseRow(result);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "GET_AGENT_BY_ID", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId: id }
+        },
+        error
+      );
+    }
+  }
+  async createAgent({ agent }) {
+    try {
+      const tableName = getTableName({ indexName: TABLE_AGENTS, schemaName: getSchemaName(this.schema) });
+      const now = /* @__PURE__ */ new Date();
+      const nowIso = now.toISOString();
+      await this.client.none(
+        `INSERT INTO ${tableName} (
+          id, name, description, instructions, model, tools, 
+          "defaultOptions", workflows, agents, "inputProcessors", "outputProcessors", memory, scorers, metadata, 
+          "createdAt", "createdAtZ", "updatedAt", "updatedAtZ"
+        ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18)`,
+        [
+          agent.id,
+          agent.name,
+          agent.description ?? null,
+          agent.instructions,
+          JSON.stringify(agent.model),
+          agent.tools ? JSON.stringify(agent.tools) : null,
+          agent.defaultOptions ? JSON.stringify(agent.defaultOptions) : null,
+          agent.workflows ? JSON.stringify(agent.workflows) : null,
+          agent.agents ? JSON.stringify(agent.agents) : null,
+          agent.inputProcessors ? JSON.stringify(agent.inputProcessors) : null,
+          agent.outputProcessors ? JSON.stringify(agent.outputProcessors) : null,
+          agent.memory ? JSON.stringify(agent.memory) : null,
+          agent.scorers ? JSON.stringify(agent.scorers) : null,
+          agent.metadata ? JSON.stringify(agent.metadata) : null,
+          nowIso,
+          nowIso,
+          nowIso,
+          nowIso
+        ]
+      );
+      return {
+        ...agent,
+        createdAt: now,
+        updatedAt: now
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "CREATE_AGENT", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId: agent.id }
+        },
+        error
+      );
+    }
+  }
+  async updateAgent({ id, ...updates }) {
+    try {
+      const tableName = getTableName({ indexName: TABLE_AGENTS, schemaName: getSchemaName(this.schema) });
+      const existingAgent = await this.getAgentById({ id });
+      if (!existingAgent) {
+        throw new MastraError({
+          id: createStorageErrorId("PG", "UPDATE_AGENT", "NOT_FOUND"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          text: `Agent ${id} not found`,
+          details: { agentId: id }
+        });
+      }
+      const setClauses = [];
+      const values = [];
+      let paramIndex = 1;
+      if (updates.name !== void 0) {
+        setClauses.push(`name = $${paramIndex++}`);
+        values.push(updates.name);
+      }
+      if (updates.description !== void 0) {
+        setClauses.push(`description = $${paramIndex++}`);
+        values.push(updates.description);
+      }
+      if (updates.instructions !== void 0) {
+        setClauses.push(`instructions = $${paramIndex++}`);
+        values.push(updates.instructions);
+      }
+      if (updates.model !== void 0) {
+        setClauses.push(`model = $${paramIndex++}`);
+        values.push(JSON.stringify(updates.model));
+      }
+      if (updates.tools !== void 0) {
+        setClauses.push(`tools = $${paramIndex++}`);
+        values.push(JSON.stringify(updates.tools));
+      }
+      if (updates.defaultOptions !== void 0) {
+        setClauses.push(`"defaultOptions" = $${paramIndex++}`);
+        values.push(JSON.stringify(updates.defaultOptions));
+      }
+      if (updates.workflows !== void 0) {
+        setClauses.push(`workflows = $${paramIndex++}`);
+        values.push(JSON.stringify(updates.workflows));
+      }
+      if (updates.agents !== void 0) {
+        setClauses.push(`agents = $${paramIndex++}`);
+        values.push(JSON.stringify(updates.agents));
+      }
+      if (updates.inputProcessors !== void 0) {
+        setClauses.push(`"inputProcessors" = $${paramIndex++}`);
+        values.push(JSON.stringify(updates.inputProcessors));
+      }
+      if (updates.outputProcessors !== void 0) {
+        setClauses.push(`"outputProcessors" = $${paramIndex++}`);
+        values.push(JSON.stringify(updates.outputProcessors));
+      }
+      if (updates.memory !== void 0) {
+        setClauses.push(`memory = $${paramIndex++}`);
+        values.push(JSON.stringify(updates.memory));
+      }
+      if (updates.scorers !== void 0) {
+        setClauses.push(`scorers = $${paramIndex++}`);
+        values.push(JSON.stringify(updates.scorers));
+      }
+      if (updates.metadata !== void 0) {
+        const mergedMetadata = { ...existingAgent.metadata, ...updates.metadata };
+        setClauses.push(`metadata = $${paramIndex++}`);
+        values.push(JSON.stringify(mergedMetadata));
+      }
+      const now = (/* @__PURE__ */ new Date()).toISOString();
+      setClauses.push(`"updatedAt" = $${paramIndex++}`);
+      values.push(now);
+      setClauses.push(`"updatedAtZ" = $${paramIndex++}`);
+      values.push(now);
+      values.push(id);
+      if (setClauses.length > 2) {
+        await this.client.none(`UPDATE ${tableName} SET ${setClauses.join(", ")} WHERE id = $${paramIndex}`, values);
+      }
+      const updatedAgent = await this.getAgentById({ id });
+      if (!updatedAgent) {
+        throw new MastraError({
+          id: createStorageErrorId("PG", "UPDATE_AGENT", "NOT_FOUND_AFTER_UPDATE"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.SYSTEM,
+          text: `Agent ${id} not found after update`,
+          details: { agentId: id }
+        });
+      }
+      return updatedAgent;
+    } catch (error) {
+      if (error instanceof MastraError) {
+        throw error;
+      }
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "UPDATE_AGENT", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId: id }
+        },
+        error
+      );
+    }
+  }
+  async deleteAgent({ id }) {
+    try {
+      const tableName = getTableName({ indexName: TABLE_AGENTS, schemaName: getSchemaName(this.schema) });
+      await this.client.none(`DELETE FROM ${tableName} WHERE id = $1`, [id]);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "DELETE_AGENT", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId: id }
+        },
+        error
+      );
+    }
+  }
+  async listAgents(args) {
+    const { page = 0, perPage: perPageInput, orderBy } = args || {};
+    const { field, direction } = this.parseOrderBy(orderBy);
+    if (page < 0) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "LIST_AGENTS", "INVALID_PAGE"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { page }
+        },
+        new Error("page must be >= 0")
+      );
+    }
+    const perPage = normalizePerPage(perPageInput, 100);
+    const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
+    try {
+      const tableName = getTableName({ indexName: TABLE_AGENTS, schemaName: getSchemaName(this.schema) });
+      const countResult = await this.client.one(`SELECT COUNT(*) as count FROM ${tableName}`);
+      const total = parseInt(countResult.count, 10);
+      if (total === 0) {
+        return {
+          agents: [],
+          total: 0,
+          page,
+          perPage: perPageForResponse,
+          hasMore: false
+        };
+      }
+      const limitValue = perPageInput === false ? total : perPage;
+      const dataResult = await this.client.manyOrNone(
+        `SELECT * FROM ${tableName} ORDER BY "${field}" ${direction} LIMIT $1 OFFSET $2`,
+        [limitValue, offset]
+      );
+      const agents = (dataResult || []).map((row) => this.parseRow(row));
+      return {
+        agents,
+        total,
+        page,
+        perPage: perPageForResponse,
+        hasMore: perPageInput === false ? false : offset + perPage < total
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "LIST_AGENTS", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+};
 var MemoryPG = class extends MemoryStorage {
   client;
   schema;
@@ -4053,6 +4434,27 @@ var WorkflowsPG = class extends WorkflowsStorage {
       );
     }
   }
+  async deleteWorkflowRunById({ runId, workflowName }) {
+    try {
+      await this.client.none(
+        `DELETE FROM ${getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: this.schema })} WHERE run_id = $1 AND workflow_name = $2`,
+        [runId, workflowName]
+      );
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "DELETE_WORKFLOW_RUN_BY_ID", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            runId,
+            workflowName
+          }
+        },
+        error
+      );
+    }
+  }
   async listWorkflowRuns({
     workflowName,
     fromDate,
@@ -4188,12 +4590,14 @@ var PostgresStore = class extends MastraStorage {
       const workflows = new WorkflowsPG({ client: this.#db, operations, schema: this.schema });
       const memory = new MemoryPG({ client: this.#db, schema: this.schema, operations });
       const observability = new ObservabilityPG({ client: this.#db, operations, schema: this.schema });
+      const agents = new AgentsPG({ client: this.#db, schema: this.schema });
       this.stores = {
         operations,
         scores,
         workflows,
         memory,
-        observability
+        observability,
+        agents
       };
     } catch (e) {
       throw new MastraError(
@@ -4245,7 +4649,8 @@ var PostgresStore = class extends MastraStorage {
       deleteMessages: true,
       observabilityInstance: true,
       indexManagement: true,
-      listScoresBySpan: true
+      listScoresBySpan: true,
+      agents: true
     };
   }
   async createTable({
@@ -4364,6 +4769,9 @@ var PostgresStore = class extends MastraStorage {
   }) {
     return this.stores.workflows.getWorkflowRunById({ runId, workflowName });
   }
+  async deleteWorkflowRunById({ runId, workflowName }) {
+    return this.stores.workflows.deleteWorkflowRunById({ runId, workflowName });
+  }
   async close() {
     this.pgp.end();
   }