examples/tutorial/text_generation.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
"""
https://ai.google.dev/gemini-api/docs/text-generation?hl=zh-cn
"""
import argparse

from google import genai
from google.genai import types

from project_settings import environment


def get_args():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--gemini_api_key",
        default=environment.get(key="GEMINI_API_KEY"),
        type=str
    )
    parser.add_argument(
        "--model",
        default="gemini-2.0-flash",
        # default="gemini-2.5-pro-preview-05-06",
        type=str
    )
    args = parser.parse_args()
    return args


def main():
    args = get_args()

    client = genai.Client(api_key=args.gemini_api_key)

    response = client.models.generate_content(
        model=args.model,
        config=types.GenerateContentConfig(
            system_instruction="You are a cat. Your name is Neko."),
        contents="Hello there"
    )

    print(response.text)

    return


if __name__ == "__main__":
    main()