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

ChatGPTのようなAI(人工知能)の機能を自分のアプリに組み込みたいとき、多くの場合は「AI API」と呼ばれる仕組みを使います。APIとは「Application Programming Interface」の略です。簡単に言えば、「あるサービスの機能を、プログラムから呼び出せるようにした窓口」のことです。この記事では、Rustから外部のAI APIをHTTPリクエストで呼び出す基本的な流れを、初心者向けに解説します。

AI APIを呼び出すとはどういうことか

ChatGPTを提供するOpenAI社などは、インターネット経由でAIに質問を送り、回答を受け取れる「AI API」を公開しています。これはWebブラウザで検索するときのように、決まった宛先(URL)にデータを送信し、返ってきた結果を受け取る仕組みです。この「データを送って結果を受け取る」通信方式を、HTTPリクエストと呼びます。Rustのプログラムからこの通信を行うことで、自分のアプリの中でAIに質問し、回答を取得できるようになります。

reqwestクレートを使う

Rustには標準でHTTP通信の機能は含まれていません。そこで使うのが「クレート」と呼ばれる外部ライブラリです。クレートはPythonのライブラリやJavaScriptのパッケージに近いもので、Cargo.tomlというファイルに追記するだけで簡単に導入できます。HTTP通信でよく使われるのがreqwestというクレートです。Cargo.tomlに次のように追記します(バージョン番号は執筆時点の例なので、実際に使う際は最新版を確認してください)。

[dependencies] reqwest = { version = "0.11", features = ["json"] } tokio = { version = "1", features = ["full"] } serde_json = "1.0"

ここでtokioという見慣れないクレートも一緒に入れています。これは「非同期処理」と呼ばれる、通信の待ち時間の間に他の処理を進められる仕組みを扱うためのものです。ネットワーク通信は結果が返ってくるまで時間がかかるため、Rustで通信処理を書く場合はこの非同期処理とセットで使うのが一般的です。

APIを呼び出すコードの流れ

実際にAI APIへリクエストを送るコードの骨格はおおよそ次のようになります。

#[tokio::main] async fn main() -> Result<(), Box> { let api_key = std::env::var("OPENAI_API_KEY")?; let client = reqwest::Client::new(); let res = client .post("https://api.openai.com/v1/chat/completions") .bearer_auth(api_key) .json(&serde_json::json!({ "model": "gpt-4o-mini", "messages": [{"role": "user", "content": "こんにちは"}] })) .send() .await?; let body = res.text().await?; println!("{}", body); Ok(()) }

ポイントは3つです。まず#[tokio::main]という目印(属性)を付けることで、非同期処理を扱えるmain関数になります。次にbearer_authで「このAPIキーを使う権限がある」ことをサーバーに伝えます。最後に.json(...)で送信したい内容をJSON形式(データをやり取りするための共通フォーマット)に変換して送っています。

APIキーの扱いに注意する

AI APIを使うには「APIキー」と呼ばれる、いわばパスワードのような文字列が必要です。このAPIキーをソースコードに直接書き込んでしまうと、GitHubなどで公開したときに誰でも使えてしまいます。勝手に使われて、高額な請求が発生する恐れもあります。上のコード例のように、環境変数(OSに設定しておく変数)からAPIキーを読み込む方法が基本です。加えて、.gitignoreにAPIキーを含むファイルを登録し、うっかり公開しないよう気を付けましょう。

Rustならではの注意点

PythonやJavaScriptでは、非同期処理を使わなくても比較的手軽にHTTP通信を書けることが多いです。ですがRustでは、async/awaitという非同期処理の書き方とほぼセットで通信処理を扱います。最初はこのResult型(成功か失敗かを表す型)や?演算子(エラーが起きたらすぐ処理を中断して呼び出し元に返す記法)に戸惑うかもしれません。ですが、通信エラーやAPIからの異常な応答を見逃しにくくなるという利点があります。エラー処理を後回しにできない分、最初の学習コストはやや高めですが、安全性の高いコードを書きやすいのがRustらしいところです。

まとめ

RustからAI APIを呼び出すには、reqwestクレートでHTTPリクエストを送り、tokioで非同期処理を扱うのが基本的な流れです。APIキーはコードに直接書かず、環境変数から読み込みましょう。さらに、通信結果はResult型でしっかりエラー処理することを意識すれば、Rustらしい安全なAI連携プログラムの第一歩を踏み出せます。

関連記事

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

参考書籍(PR)

Rustをもっと学びたい方には「実践Rust入門[言語仕様から開発手法まで]」がおすすめです。

コメント

コメントを投稿

このブログの人気の投稿

#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,...
続きを読む