diff --git a/mem0-ts/src/oss/src/index.ts b/mem0-ts/src/oss/src/index.ts index e9c9af7c7..68c98a90e 100644 --- a/mem0-ts/src/oss/src/index.ts +++ b/mem0-ts/src/oss/src/index.ts @@ -26,4 +26,5 @@ export * from "./vector_stores/supabase"; export * from "./vector_stores/langchain"; export * from "./vector_stores/vectorize"; export * from "./vector_stores/azure_ai_search"; +export * from "./vector_stores/pgvector"; export * from "./utils/factory"; diff --git a/mem0-ts/src/oss/src/utils/factory.ts b/mem0-ts/src/oss/src/utils/factory.ts index 208d31b09..706f38890 100644 --- a/mem0-ts/src/oss/src/utils/factory.ts +++ b/mem0-ts/src/oss/src/utils/factory.ts @@ -34,6 +34,7 @@ import { LangchainLLM } from "../llms/langchain"; import { LangchainEmbedder } from "../embeddings/langchain"; import { LangchainVectorStore } from "../vector_stores/langchain"; import { AzureAISearch } from "../vector_stores/azure_ai_search"; +import { PGVector } from "../vector_stores/pgvector"; export class EmbedderFactory { static create(provider: string, config: EmbeddingConfig): Embedder { @@ -104,6 +105,8 @@ export class VectorStoreFactory { return new VectorizeDB(config as any); case "azure-ai-search": return new AzureAISearch(config as any); + case "pgvector": + return new PGVector(config as any); default: throw new Error(`Unsupported vector store provider: ${provider}`); }