import os

from google import genai
from google.genai import types
import base64

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

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

  tools = [
    types.Tool(google_search=types.GoogleSearch()),
  ]

  generate_content_config = types.GenerateContentConfig(
    temperature = 1,
    top_p = 0.95,
    max_output_tokens = 8192,
    response_modalities = ["TEXT"],
    tools=tools,
  )

  for chunk in client.models.generate_content_stream(
    model = model,
    contents = contents,
    config = generate_content_config,
    ):
    print(chunk.text, end="")


# os.environ["http_proxy"] = "http://127.0.0.1:7890"
# os.environ["https_proxy"] = "http://127.0.0.1:7890"
# os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "/Users/huangjiajun/Project/baidao-test-666808-4b2251497ed0.json"
os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = r"E:\Users\tianx\intelli-zen\OpenGeminiAPI\dotenv\potent-veld-462405-t3-8091a29b2894.json"
generate()