Google GenAI

pip install 'niadra[google-genai]'   # google-genai 2.25 or newer, below 3

npm install @niadra/sdk @google/genai   # @niadra/sdk/google-genai

"""Gemini through the Google GenAI SDK with the customer's context."""

from google import genai

from niadra import Niadra, phone
from niadra.integrations.google_genai import wrap

niadra = Niadra(channel="chat")
gemini = wrap(genai.Client())

with niadra.conversation("thread-81", subject=phone("+5511912345678")) as conversation:
    conversation.customer("Where is my replacement lid?")
    response = gemini.models.generate_content(
        model="gemini-2.5-flash",
        contents="Where is my replacement lid?",
        config={"system_instruction": "You are Acme's agent."},
    )
    print(response.text)

import { GoogleGenAI } from "@google/genai";
import { Niadra, handles } from "@niadra/sdk";
import { wrapGoogleGenAI } from "@niadra/sdk/google-genai";

const niadra = new Niadra();
const genai = new GoogleGenAI({});

/** One customer message in; `userId` comes from your session, never from the model. */
export async function reply(userId: string, chatId: string, text: string): Promise<string> {
  const convo = niadra.conversation({ subject: handles.appUserId(userId), channel: "web_chat", conversation_id: chatId });
  const ai = wrapGoogleGenAI(genai, convo);
  const response = await ai.models.generateContent({
    model: "gemini-2.5-flash",
    contents: text,
    config: { systemInstruction: "You are Acme's support agent. Be brief." },
  });
  return response.text ?? "";
}