Spaces:
Sleeping
Sleeping
File size: 1,011 Bytes
0ec61d2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
#!/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()
|