mem0ai/mem0

feat(ts-sdk): add Vertex AI embedding provider

Open

#5.768 aberto em 23 de jun. de 2026

Ver no GitHub
 (1 comment) (0 reactions) (0 assignees)Python (6.343 forks)batch import
enhancementhelp wantedsdk-typescriptsize:M

Métricas do repositório

Stars
 (55.741 stars)
Métricas de merge de PR
 (Mesclagem média 6d 11h) (208 fundiu PRs em 30d)

Description

Summary

The Python SDK supports Vertex AI as an embedding provider, but the TypeScript OSS SDK (mem0ai/oss) does not. Add it to bring the TS SDK to parity.

Python reference mem0/embeddings/vertexai.py
Registered in (Python) mem0/utils/factory.py (EmbedderFactory)
Target file (TypeScript) mem0-ts/src/oss/src/embeddings/vertexai.ts
Suggested implementation Use @google-cloud/aiplatform (PredictionServiceClient) text-embedding models.

Requirements

  • Implement VertexAIEmbedder in mem0-ts/src/oss/src/embeddings/vertexai.ts, extending Embedder (mem0-ts/src/oss/src/embeddings/base.ts) and mirroring the Python provider's behavior (embed / embedBatch).
  • Register the "vertexai" provider in mem0-ts/src/oss/src/utils/factory.ts (EmbedderFactory).
  • Add config typing in mem0-ts/src/oss/src/types/.
  • Add a unit test under mem0-ts/src/oss/src/tests/.
  • Add @google-cloud/aiplatform to mem0-ts/package.json (optional/peer dependency, lazy-imported like other providers).
  • Update docs under docs/ if this provider is user-facing.

Reference pattern

Mirror an existing TS provider: embeddings/openai.ts.

Notes

GCP auth (ADC / service account). Mirror mem0/embeddings/vertexai.py.


Part of the TypeScript ↔ Python SDK provider-parity effort. One provider per issue (atomic).

Guia do colaborador