Class VertexAiRagMemoryService

A BaseMemoryService backed by a Vertex AI RAG Engine corpus.

A finished session is uploaded as one RAG file holding its transcript, and a search retrieves the chunks of those transcripts that match the query. Only addSessionToMemory and searchMemory are supported, matching adk-python.

Example

const memoryService = new VertexAiRagMemoryService({
  ragCorpus: 'projects/my-project/locations/us-central1/ragCorpora/12345',
  similarityTopK: 5,
});
await memoryService.addSessionToMemory(session);

Implements

BaseMemoryService

Constructors

constructor

new VertexAiRagMemoryService(
options: VertexAiRagMemoryServiceOptions,
ragApiClient?: RagApiClient,
): VertexAiRagMemoryService
Internal
Parameters
- options: VertexAiRagMemoryServiceOptions
- OptionalragApiClient: RagApiClient
  An internal seam for tests to substitute a fake REST client. Production callers omit it and get a client for the resolved location.
Returns VertexAiRagMemoryService
- Defined in core/src/memory/vertex_ai_rag_memory_service.ts:213

Methods

addSessionToMemory

addSessionToMemory(session: Session): Promise<void>
Uploads the session's transcript into the corpus as one RAG file.
Parameters
- session: Session
Returns Promise<void>
Implementation of BaseMemoryService.addSessionToMemory
- Defined in core/src/memory/vertex_ai_rag_memory_service.ts:229

searchMemory

searchMemory(request: SearchMemoryRequest): Promise<SearchMemoryResponse>
Searches for sessions that match the query.
Parameters
- request: SearchMemoryRequest
  The request to search memory.
Returns Promise<SearchMemoryResponse>
A promise that resolves to SearchMemoryResponse containing the matching memories.
Implementation of BaseMemoryService.searchMemory
- Defined in core/src/memory/vertex_ai_rag_memory_service.ts:241

Class VertexAiRagMemoryService

Example

Implements

Constructors

constructor

Parameters

Returns VertexAiRagMemoryService

Methods

addSessionToMemory

Parameters

Returns Promise<void>

searchMemory

Parameters

Returns Promise<SearchMemoryResponse>