#25 Goで学ぶAI API(ChatGPT等)の呼び出し方の基本

はじめに

ChatGPTのようなAI(人工知能)を使ったサービスは、プログラムから「API(エーピーアイ)」と呼ばれる窓口を通して呼び出すことができます。APIとは、あるプログラムが別のプログラムの機能を使うための「決められたやり取りのルール」のことです。難しく聞こえますが、実はやっていることはシンプルで、「インターネット越しにデータを送って、返ってきたデータを受け取る」だけです。

この記事では、Go言語を使ってAI APIのようなWebサービスを呼び出す際の「基本の型」を解説します。実際のサービスに接続するコードそのものというより、どんな流れで書けば良いのかという考え方を身につけることが目的です。

AI APIとのやり取りの流れ

ChatGPTのようなAI APIの多くは、次のような流れで動いています。

  1. 「こう質問したい」という内容をJSON(Webでよく使われるデータの形式)にまとめる
  2. そのJSONをHTTP(インターネット通信の約束事)のPOSTリクエストとしてAPIのサーバーに送る
  3. サーバーがAIに問い合わせて、答えをJSONで返してくる
  4. 返ってきたJSONを読み取り、必要な部分(AIの回答文など)を取り出す

つまり「JSONを作る → 送る → JSONを受け取って読む」という3ステップが基本の流れです。Goでこれをどう書くか、順番に見ていきましょう。

net/httpでPOSTリクエストを送る

GoにはHTTP通信を行うための標準ライブラリnet/httpが最初から用意されています。PythonのrequestsやJavaScriptのfetchのような外部ライブラリを別途インストールしなくても、標準機能だけでAPI通信ができるのがGoの特徴の一つです。

POSTリクエスト(サーバーにデータを送るタイプのリクエスト)を送る基本形は次のようになります。

req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonData))
if err != nil {
    log.Fatal(err)
}
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", "Bearer " + apiKey)

client := &http.Client{}
resp, err := client.Do(req)

http.NewRequestでリクエストの中身(送り先のURL、送るデータ)を組み立て、client.Doで実際に送信します。ヘッダー(通信の付加情報)には、送るデータの種類(Content-Type)と、APIキー(利用者を識別するための認証用の文字列)を設定するのが一般的です。

APIキーの扱い方に注意

APIキーはいわば「サービスを使うためのパスワード」です。コードに直接書き込んでGitHubなどに公開してしまうと、第三者に不正利用されてしまう危険があります。実際の開発では、環境変数(OSに設定しておく値)から読み込むのが基本です。Goではos.Getenv("API_KEY")のように書くことで、コードにキーを直書きせずに済みます。

JSONのエンコードとデコード

Goは静的型付け言語(変数の型をあらかじめ決めておく言語)です。JSONを扱うときも「構造体(struct)」と呼ばれる型を定義してから変換するのが一般的なやり方です。PythonやJavaScriptのように辞書やオブジェクトへ自由に詰め込むのとは少し感覚が異なります。

type RequestBody struct {
    Model    string    `json:"model"`
    Messages []Message `json:"messages"`
}

type Message struct {
    Role    string `json:"role"`
    Content string `json:"content"`
}

reqBody := RequestBody{
    Model: "gpt-example",
    Messages: []Message{
        {Role: "user", Content: "こんにちは"},
    },
}
jsonData, err := json.Marshal(reqBody)

json:"model"のような部分は「タグ」と呼ばれ、Goの構造体のフィールド名とJSONのキー名を対応付けるものです。json.Marshalは構造体をJSONの文字列(正確にはバイト列)に変換する関数で、これを「エンコード」と呼びます。

反対に、受け取ったJSONをGoの構造体に変換することを「デコード」と呼び、json.Unmarshalを使います。ここでのResponseBodyも、先ほどのRequestBodyと同じように、受け取るJSONの形に合わせて自分で定義しておく構造体です。

var result ResponseBody
err = json.Unmarshal(body, &result)

「どんな形式のデータが返ってくるか」をAPIのドキュメントで確認し、それに沿った構造体を用意するのがGoでのAPI連携の基本です。

レスポンスを読み取る

サーバーから返ってきたレスポンス(応答)の本文は、resp.Bodyから読み出します。読み終わったら必ず閉じる必要があるため、deferという「関数が終わるときに実行する処理」を使うのが定番です。

defer resp.Body.Close()
body, err := io.ReadAll(resp.Body)

こうして取得したバイト列を先ほどのjson.Unmarshalでデコードすれば、AIの回答文などを構造体のフィールドとして取り出せるようになります。

まとめ

GoでAI APIを呼び出す基本は、次の流れでした。「構造体でリクエストの形を定義する → JSONにエンコードする → net/httpでPOST送信する → 返ってきたJSONを構造体にデコードする」という流れです。動的型付け言語に比べると事前に型を定義する手間はありますが、その分「どんなデータをやり取りしているか」がコード上で明確になるのがGoらしい書き方だと言えます。まずはこの基本の型を押さえておけば、どんなAI APIやWeb APIを相手にしても応用が利くはずです。

関連記事

本ページはプロモーションを含みます。

参考書籍(PR)

Goをもっと学びたい方には「Go言語入門」がおすすめです。

コメント

コメントを投稿

このブログの人気の投稿

#11 Rustで学ぶエラーハンドリングの基本

Rustのエラーハンドリングは、他の言語のような「例外」という仕組みを持たず、`Result`という型を使った独自の考え方が中心になります。この記事では、Rustにおけるエラーハンドリングの基本を解説します。 Result型という考え方 失敗する可能性がある処理は、`Result`という「成功か失敗かを表す型」を返します。 fn divide(a: i32, b: i32) -> Result<i32, String> { if b == 0 { return Err(String::from("0では割れません")); } Ok(a / b) } 成功時は`Ok`、失敗時は`Err`という形で結果を包んで返します。 matchで結果を扱う match divide(10, 0) { Ok(result) => println!("結果: {}", result), Err(message) => println!("エラー: {}", message), } ?という便利な記号 Rustには、`Result`がエラーだった場合に、その場で関数の呼び出し元にエラーを伝えて処理を中断できる`?`という記号があります。 fn calculate() -> Result<i32, String> { let result = divide(10, 0)?; Ok(result) } Rustならではの注意点 他の言語の「エラーが起きても気づかずに進んでしまう」という問題を防ぐため、Rustは`Result`の中身を確認しない限り、警告を出す仕組みになっています。エラーの可能性を無視できない設計こそが、Rustの安全性を支えています。 まとめ Rustのエラーハンドリングは`Result`型が中心で、`match`や`?`を使って結果を扱います。エラーの可能性を無視できない設計に慣れていきましょう。 本ページはプロモーションを含みます。 参考書籍(PR) Rustをもっと学びたい方には「 実践Rust入門[言語仕様から開発手法まで] 」がおすすめです。...
続きを読む

#16 TypeScriptで学ぶ辞書型(マップ)の基本

TypeScriptの辞書に相当する仕組みは、JavaScriptと同じですが、キーと値の型をあらかじめ決めておけるのが特徴です。この記事では、TypeScriptにおける辞書の扱い方を解説します。 型付きのオブジェクト interface Person { name: string; age: number; } const person: Person = { name: "太郎", age: 20 }; キーの型が決まっていない辞書 キーの名前が事前に決まっていない辞書のような使い方をしたい場合は、次のように書きます。 const scores: { [key: string]: number } = {}; scores["太郎"] = 90; Mapを使う場合 const person = new Map<string, string>(); person.set("name", "太郎"); `Map<string, string>`のように、キーと値それぞれの型を指定できます。 TypeScriptならではの注意点 型付きのオブジェクト(`interface`)を使うと、存在しないキーにアクセスしようとした際に、実行前の段階でエラーとして教えてもらえます。データの形が決まっている場合は、キーが自由な辞書よりも`interface`を使うほうが安全です。 まとめ TypeScriptの辞書は、形が決まっているなら`interface`、自由なキーを扱うなら`{ [key: string]: 型 }`や`Map`を使います。用途に応じて型の付け方を選びましょう。 本ページはプロモーションを含みます。 参考書籍(PR) TypeScriptをもっと学びたい方には「 プロを目指す人のためのTypeScript入門 」がおすすめです。
続きを読む

#19 Goで学ぶテスト(単体テスト)の基本

Goはシンプルな文法が魅力ですが、書いたコードが正しく動くかどうかは別問題です。この記事では、Goで書いたコードを自動でチェックする「テスト」の基本を解説します。 なぜテストを書くのか コードを修正するたびに、関連する機能が壊れていないかを手作業で確認するのは大変です。テストを用意しておけば、コマンド一つで動作確認ができ、安心してコードを直せるようになります。 Go標準のテスト機能 Goには追加のライブラリを入れなくても使える testing パッケージが標準で備わっています。テスト対象のファイルと同じディレクトリに、名前の末尾が _test.go のファイルを作ります。 // add.go package main func add(a int, b int) int { return a + b } // add_test.go package main import "testing" func TestAdd(t *testing.T) { result := add(2, 3) if result != 5 { t.Errorf("add(2, 3) = %d; want 5", result) } } テスト関数の名前は Test から始める決まりです。ターミナルで go test と実行すると、 _test.go 内の Test から始まる関数がすべて自動で実行されます。 複数パターンをまとめてテストする Goでは、複数の入力パターンをスライス(配列のようなもの)にまとめて、まとめてテストする書き方(テーブル駆動テスト)がよく使われます。 func TestAddCases(t *testing.T) { cases := []struct{ a, b, want int }{ {2, 3, 5}, {0, 0, 0}, {-1, 1, 0}, } for _, c := range cases { if got := add(c.a, c.b); got != c.want { t.Errorf("add(%d,...
続きを読む