#!/usr/bin/python3
# -*- coding: utf-8 -*-
import os

from google import genai
from google.genai import types

os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = r"E:\Users\tianx\intelli-zen\OpenGeminiAPI\dotenv\potent-veld-462405-t3-8091a29b2894.json"


client = genai.Client(
    vertexai=True,
    project="potent-veld-462405-t3",
    location="global",
)

# model = "gemini-2.0-flash-001"
model = "gemini-2.5-flash"
contents = [
    types.Content(
        role="user",
        parts=[
            types.Part.from_text(text="""今天深圳的天气怎么样？""")
        ]
    )
]

generate_content_config = types.GenerateContentConfig(
    temperature=1,
    top_p=0.95,
    max_output_tokens=8192,
    response_modalities=["TEXT"],
)
response: types.GenerateContentResponse = client.models.generate_content(
    model=model,
    contents=contents,
    config=generate_content_config,
)
text = response.candidates[0].content.parts[0].text
print(text)


if __name__ == "__main__":
    pass