Gemini Provider

Wrapper around google-genai's Client. Provides a model-centric interface matching how GenerativeModel was used in the old SDK.

Usage

# Before
from google import genai
client = genai.Client(api_key="...")
response = client.models.generate_content(
    model="gemini-2.0-flash", contents="Hello"
)

# After
from kostrack import GenerativeModel

model = GenerativeModel(
    "gemini-2.0-flash",
    tags={"project": "openmanagr", "feature": "document-ocr"},
)
response = model.generate_content("Extract text from this document...")

API key

Pass api_key= to the constructor, or set GEMINI_API_KEY (or GOOGLE_API_KEY) as an environment variable.

Supported models