Gemini APIを試してみた

カテゴリ無し

2024.02.042024.02.15

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

できること

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

チャット

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

コードはこちら

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)