Google Gemini

Memori integrates with Google Gemini via the google-genai SDK (Python) and @google/genai SDK (TypeScript). Register the client instance and all calls are automatically captured and stored.

Want a zero-setup option? The Memori Cloud at app.memorilabs.ai.

Quick Start

Gemini Integration
import os
from google import genai
from memori import Memori
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

engine = create_engine("sqlite:///memori.db")
SessionLocal = sessionmaker(bind=engine)

client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])

mem = Memori(conn=SessionLocal).llm.register(client)
mem.config.storage.build()
mem.attribution(entity_id="user_123", process_id="gemini_assistant")

response = client.models.generate_content(
    model="gemini-2.5-flash",
    contents="Hello!"
)
print(response.text)

Supported Modes

ModePythonTypeScript
Syncclient.models.generate_content()
Asyncawait client.aio.models.generate_content()await client.models.generateContent()
Streamedclient.models.generate_content_stream()await client.models.generateContentStream()