Googleの大規模言語モデル「PaLM」のAPIを使ってみた

今回はGoogleの大規模言語モデルである「PaLM」のAPIを使ってみました。
APIキーの発行は下記を記事を参考にしました。
https://biztechdx.com/%E3%80%90google-ai%E3%80%91google%E3%81%AE%E5%A4%A7%E8%A6%8F%E6%A8%A1%E8%A8%80%E8%AA%9E%E3%83%A2%E3%83%87%E3%83%ABpalm-api%E4%BD%BF%E3%81%A3%E3%81%A6%E3%81%BF%E3%81%9F/

2023年10月現在、下記の3つのモデルが利用可能です。

1. text-bison-001:
   • 更新日: 2023年5月
   • モデルサイズ: Bison
   • 機能:
     • テキストの生成
     • 以下のような言語タスクの最適化：
       • コード生成
       • テキスト生成
       • テキスト編集
       • 問題解決
       • 推薦生成
       • 情報抽出
       • データ抽出または生成
       • AIエージェント
   • ゼロショット、ワンショット、およびフューショットタスクを処理可能
   • 安全設定: 開発者は6次元の害のための調整可能な安全設定を利用できます
2. chat-bison-001:
   • 更新日: 2023年5月
   • モデルサイズ: Bison
   • 機能:
     • 対話形式でテキストを生成
     • チャットボットまたはAIエージェントの実装など、対話言語タスクの最適化
   • ゼロショット、ワンショット、およびフューショットタスクを処理可能
   • 安全設定はありません
3. embedding-gecko-001:
   • 更新日: 2023年5月
   • モデルサイズ: Gecko
   • 機能:
     • 入力テキストのテキスト埋め込みを生成
     • 最大1024トークンのテキストの埋め込みを作成するための最適化
   • 安全設定はありません​​。

今回はその中でも、対話形式でテキストを生成できるchat-bison-001を利用します。

実際のコードは以下から

import google.generativeai as palm

# APIキーの設定
palm.configure(api_key='API-Key')

def chat_with_model():
    # 新しい対話を作成
    user_message = input("You: ")
    response = palm.chat(messages=user_message)

    while True:
        # モデルのレスポンスを取得および表示
        print(f'Model: {response.last}')

        # 既存の対話に返信を送信
        user_message = input("You: ")
        response = response.reply(user_message)

# chat_with_model関数を呼び出して対話を開始
chat_with_model()

API-Keyには取得したAPIキーに置き換えてください。
このコードを実行するとターミナル上でAIチャットボットとして機能します。

実行結果

書き起こし

You: Hello
Model: Hello! How can I help you today?
Google was founded in 1998 by Larry Page and Sergey Brin, two Stanford University students. The company's mission is to "organize the world's information and make it universally accessible and useful." Google has grown rapidly since its founding, and it is now one of the most valuable companies in the world.
Google was founded in 1998 by Larry Page and Sergey Brin, two Stanford University students. The company's mission is to "organize the world's information and make it universally accessible and useful." Google has grown rapidly since its founding, and it is now one of the most valuable companies in the world.

Google's search engine is the most popular in the world. It allows users to search for information on the Internet. Google's search engine is based on a complex algorithm that ranks websites based on their relevance to the user's search query.

Google also offers a variety of other services, including email, maps, and document storage. Google's email service is called Gmail. Google Maps is a web mapping service that provides detailed maps and satellite imagery of locations around the world. Google Drive is a cloud storage service that allows users to store files online and access them from anywhere.

Google is a major player in the technology industry. The company's products and services are used by billions of people around the world. Google is constantly innovating and expanding its reach. It is likely that Google will continue to play a major role in the technology industry for many years to come.

Googleのことを聞いたら饒舌に