fix(mem0-ts): support pgvector connection strings and ssl (#5789)

mem0-ts/src/oss/tests/pgvector.unit.test.ts

@@ -1,5 +1,3 @@
-/// <reference types="jest" />
-
 const searchRows = [
   {
     id: "a",
@@ -23,9 +21,13 @@ const searchRows = [
   },
 ];
 
+const mockState = {
+  databaseExists: true,
+};
+
 function mockPgQuery(sql: string) {
   if (sql.includes("SELECT 1 FROM pg_database")) {
-    return { rows: [{ "?column?": 1 }] };
+    return { rows: mockState.databaseExists ? [{ "?column?": 1 }] : [] };
   }
 
   if (sql.includes("FROM information_schema.tables")) {
@@ -69,13 +71,129 @@ jest.mock("pg", () => {
 
 import { PGVector } from "../src/vector_stores/pgvector";
 
-describe("PGVector - search()", () => {
+function getClientQueries(client: { query: jest.Mock }) {
+  return client.query.mock.calls.map(([sql]) => sql as string);
+}
+
+describe("PGVector", () => {
   beforeEach(() => {
     const pg = require("pg");
+    mockState.databaseExists = true;
     pg.__mock.Client.mockClear();
     pg.__mock.clients.length = 0;
   });
 
+  test("uses one direct client for connectionString mode and skips bootstrap database creation", async () => {
+    mockState.databaseExists = false;
+
+    const ssl = { rejectUnauthorized: false };
+    const store = new PGVector({
+      collectionName: "memories",
+      connectionString:
+        "postgresql://postgres:postgres@db.example.com:5432/neondb",
+      ssl,
+      embeddingModelDims: 3,
+      dimension: 3,
+    } as any);
+
+    await store.initialize();
+
+    const pg = require("pg");
+    expect(pg.__mock.Client).toHaveBeenCalledTimes(1);
+    expect(pg.__mock.Client).toHaveBeenCalledWith({
+      connectionString:
+        "postgresql://postgres:postgres@db.example.com:5432/neondb",
+      ssl,
+    });
+
+    const directClient = pg.__mock.clients[0];
+    const queries = getClientQueries(directClient);
+
+    expect(queries).not.toEqual(
+      expect.arrayContaining([
+        expect.stringContaining("SELECT 1 FROM pg_database"),
+      ]),
+    );
+    expect(queries).not.toEqual(
+      expect.arrayContaining([expect.stringContaining("CREATE DATABASE")]),
+    );
+    expect(queries).toEqual(
+      expect.arrayContaining([
+        "CREATE EXTENSION IF NOT EXISTS vector",
+        expect.stringContaining("FROM information_schema.tables"),
+      ]),
+    );
+  });
+
+  test("keeps the split-field bootstrap flow when connectionString is absent", async () => {
+    mockState.databaseExists = false;
+    const ssl = { rejectUnauthorized: false };
+
+    const store = new PGVector({
+      collectionName: "memories",
+      user: "postgres",
+      password: "postgres",
+      host: "localhost",
+      port: 5432,
+      dbname: "vector_store",
+      ssl,
+      embeddingModelDims: 3,
+      dimension: 3,
+    } as any);
+
+    await store.initialize();
+
+    const pg = require("pg");
+    expect(pg.__mock.Client).toHaveBeenCalledTimes(2);
+    expect(pg.__mock.Client).toHaveBeenNthCalledWith(1, {
+      database: "postgres",
+      user: "postgres",
+      password: "postgres",
+      host: "localhost",
+      port: 5432,
+      ssl,
+    });
+    expect(pg.__mock.Client).toHaveBeenNthCalledWith(2, {
+      database: "vector_store",
+      user: "postgres",
+      password: "postgres",
+      host: "localhost",
+      port: 5432,
+      ssl,
+    });
+
+    const bootstrapClient = pg.__mock.clients[0];
+    const activeClient = pg.__mock.clients[1];
+    const bootstrapQueries = getClientQueries(bootstrapClient);
+
+    expect(bootstrapQueries).toEqual(
+      expect.arrayContaining([
+        "SELECT 1 FROM pg_database WHERE datname = $1",
+        'CREATE DATABASE "vector_store"',
+      ]),
+    );
+    expect(bootstrapClient.end).toHaveBeenCalledTimes(1);
+    expect(getClientQueries(activeClient)).toEqual(
+      expect.arrayContaining([
+        "CREATE EXTENSION IF NOT EXISTS vector",
+        expect.stringContaining("FROM information_schema.tables"),
+      ]),
+    );
+  });
+
+  test("throws when connectionString is absent and split-field params are missing", () => {
+    expect(
+      () =>
+        new PGVector({
+          collectionName: "memories",
+          embeddingModelDims: 3,
+          dimension: 3,
+        } as any),
+    ).toThrow(
+      "PGVector requires either connectionString or user, password, host, port",
+    );
+  });
+
   test("returns similarity score (1 - distance) clamped to [0, 1]", async () => {
     const store = new PGVector({
       collectionName: "memories",