2023-12-16

Gemini APIを試してみた

最近話題のGeminiをAPIで利用できるとのことで試してみました。

できること

  • テキスト生成
  • 画像とテキストからテキスト生成
  • チャットでの会話

チャット
Image in a image block

私の名前を覚えてもらい、それを答えさせる実験をしてみました。ちゃんと記憶してくれてますね!

コードはこちら

import google.generativeai as genai

def main():
    genai.configure(api_key='API_KEY')
    model = genai.GenerativeModel('gemini-pro')
    chat = model.start_chat(history=[])
    while True:
        prompt = input('あなた: \n')
        if prompt == 'exit':
            break
        print("_"*80)
        response = chat.send_message(prompt, stream=True)
        print('Gemini: ')
        for chunk in response:
            print(chunk.text)
        print("_"*80)

モデルの種類

使用例