mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-09-24 16:05:44 +08:00
33 lines
1.3 KiB
Python
33 lines
1.3 KiB
Python
from google import genai
|
|
from google.genai import types
|
|
|
|
client = genai.Client(
|
|
api_key="sk-d0678689ab0000696efa7fc0679026bf7bedb954d98ef5da9aa8161f99227ef8",
|
|
http_options={"api_version": "v1beta", "base_url": "https://clicodeplus.com"}
|
|
)
|
|
|
|
try:
|
|
response = client.models.generate_content(
|
|
model="gemini-3.1-flash-image-preview",
|
|
contents="Draw a red circle",
|
|
config=types.GenerateContentConfig(
|
|
response_modalities=["TEXT", "IMAGE"]
|
|
)
|
|
)
|
|
print("Response received!")
|
|
for part in response.candidates[0].content.parts:
|
|
if part.text:
|
|
print("Text:", part.text[:200])
|
|
if part.inline_data:
|
|
print("Image: mime=%s, size=%d bytes" % (part.inline_data.mime_type, len(part.inline_data.data)))
|
|
um = response.usage_metadata
|
|
if um:
|
|
print("Prompt tokens:", um.prompt_token_count)
|
|
print("Candidates tokens:", um.candidates_token_count)
|
|
print("Total tokens:", um.total_token_count)
|
|
if hasattr(um, "candidates_tokens_details") and um.candidates_tokens_details:
|
|
for d in um.candidates_tokens_details:
|
|
print(" Detail: modality=%s tokens=%d" % (d.modality, d.token_count))
|
|
except Exception as e:
|
|
print("Error:", type(e).__name__, e)
|