Ccmmutty logo
Commutty IT
2 min read

Gemini APIを触ってみた!

https://cdn.magicode.io/media/notebox/blob_o6nnjr5
最近話題のGeminiをAPIで利用できるとのことで試してみました。 以下のチュートリアルを進めていけば一通り遊べます
https://ai.google.dev/tutorials/python_quickstart

できること

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

チャット

Untitled
私の名前を覚えてもらい、それを答えさせる実験をしてみました。ちゃんと記憶してくれてますね!
コードはこちら
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)

Discussion

コメントにはログインが必要です。