google-genai SDK

pip install google-genai

from google import genai

client = genai.Client(
    api_key="sk-XXXXXXXXXXXXXXXX",
    http_options={"base_url": "https://openp.ai"},
)

resp = client.models.generate_content(
    model="gemini-3.1-pro-preview",
    contents="用一句话介绍 OpenPAI",
)
print(resp.text)

npm install @google/genai

import { GoogleGenAI } from "@google/genai";

const client = new GoogleGenAI({
  apiKey: process.env.OPENPAI_API_KEY!,
  httpOptions: { baseUrl: "https://openp.ai" },
});

const resp = await client.models.generateContent({
  model: "gemini-3.1-pro-preview",
  contents: "hello",
});
console.log(resp.text);

stream = client.models.generate_content_stream(
    model="gemini-3.1-pro-preview",
    contents="讲个 100 字的故事",
)
for chunk in stream:
    print(chunk.text, end="", flush=True)

from google.genai import types

resp = client.models.generate_content(
    model="gemini-3.1-pro-preview",
    contents=[
        types.Part.from_uri(file_uri="https://example.com/x.png", mime_type="image/png"),
        "描述这张图。",
    ],
)

weather_tool = types.Tool(function_declarations=[
    types.FunctionDeclaration(
        name="get_weather",
        description="Get weather of a city",
        parameters={
            "type": "object",
            "properties": {"city": {"type": "string"}},
            "required": ["city"],
        },
    )
])

resp = client.models.generate_content(
    model="gemini-3.1-pro-preview",
    contents="北京天气",
    config=types.GenerateContentConfig(tools=[weather_tool]),
)

resp = client.models.generate_content(
    model="gemini-3.1-pro-preview",
    contents="证明哥德巴赫猜想",
    config=types.GenerateContentConfig(
        thinking_config=types.ThinkingConfig(thinking_budget=8000),
    ),
)

resp = client.models.generate_content(
    model="gemini-2.5-pro-thinking",
    contents="...",
)

resp = client.models.embed_content(
    model="text-embedding-004",
    contents=["hello world"],
)
print(resp.embeddings[0].values[:5])