genai_pydantic_ai.py

# /// script
# requires-python = ">=3.12"
# dependencies = [
#     "logfire",
#     "pydantic-ai-slim[google]",
# ]
# ///
import logfire

from pydantic_ai import Agent

logfire.configure()
logfire.instrument_pydantic_ai()

agent = Agent('google-vertex:gemini-2.5-flash')
r = agent.run_sync('what is 1 + 1')
print(r.output)

genai_vanilla.py

# /// script
# requires-python = ">=3.12"
# dependencies = [
#     "google-genai",
#     "logfire[httpx]",
#     "opentelemetry-instrumentation-google-genai",
# ]
# ///
import os

import logfire
from google import genai
from opentelemetry.instrumentation.google_genai import GoogleGenAiSdkInstrumentor

logfire.configure()
# to instrument the raw http requests, uncomment this line:
# logfire.instrument_httpx(capture_all=True)

os.environ['OTEL_GOOGLE_GENAI_GENERATE_CONTENT_CONFIG_INCLUDES'] = '*'
GoogleGenAiSdkInstrumentor().instrument()

client = genai.Client(vertexai=True, location='global')

response = client.models.generate_content(model='gemini-2.5-flash', contents='what is 1 + 1')
print(response.text)