【ITニュース解説】Using Idempotency Keys for Safe AI Agent Tool Execution

AI技術の進化は目覚ましく、最近では単に質問に答えるだけでなく、ビジネスの現場で具体的な操作を実行するAIエージェントが登場している。例えば、顧客への払い戻し処理、顧客情報の更新、注文の作成、クラウドリソースのプロビジョニングといった業務をAIエージェントが担当する場面が増えてきている。これは非常に便利である一方で、システム設計においては新たな課題も生じさせている。

その一つが、「操作の重複実行」という問題である。AIエージェントが何らかのツール（API呼び出しなど）を実行する際、処理が成功したにもかかわらず、その応答がエージェントに届く前にネットワークのタイムアウトなどが発生することがある。するとエージェントは、操作が失敗したと誤解し、同じ処理をもう一度試みてしまう可能性がある。このとき、もしシステムに重複実行を防ぐ仕組みがなければ、顧客に二重に払い戻しが行われたり、同じ注文が二度作成されたりといった深刻な問題が発生しかねない。このような問題は、実は大規模なシステムを構築する際に常に考慮すべき「分散システム」特有の課題であり、AIエージェントに限った話ではない。ネットワークの不具合、処理を担当するワーカーのクラッシュ、システムによる自動的な再試行、ワークフローの中断からの再開、あるいは同時に複数のワーカーが同じリクエストを処理しようとする場合など、さまざまなシナリオで重複実行のリスクは存在する。

この重複実行の問題を安全に解決するための重要な概念が「べき等性キー」である。べき等性キーとは、ある論理的な操作を一意に識別するための特別なコードのことだ。システムがこのキーを受け取ると、それが新しい操作なのか、それとも以前に一度試みられた操作の再試行なのかを判断できるようになる。例えば、払い戻し処理の際に「refund_4821」というべき等性キーを割り当てたとしよう。もし何らかの理由でこの払い戻し処理が再試行された場合、システムは同じキー「refund_4821」が再度送られてきたことを認識し、新しい払い戻し処理としてではなく、以前の操作の繰り返しとして扱う。これにより、既に完了している操作であればその結果を返す、現在処理中であればその旨を伝える、あるいは結果が不確かな場合は状態を確認する、といった安全な対応が可能になる。

べき等性キーを生成する上で大切なのは、AIモデル自身にその判断を任せないことである。AIモデルがその都度新しいキーを生成したり、同じ会話内で異なる目的の操作に同じキーを使ってしまったりすると、重複排除の仕組みが正しく機能しなくなる。代わりに、システムの「オーケストレーション層」、つまりワークフロー全体の流れを管理する部分が、永続的に保存されたワークフロー識別子やビジネス操作識別子に基づいてキーを生成するのが望ましい。例えば、ワークフローの実行IDとツール操作IDを組み合わせることで、ユニークかつ安定したべき等性キーを作成できる。このキーは、操作が完了するまでのライフサイクル全体で一貫して再利用されるべきだ。

本番環境で安全にAIエージェントを運用するためには、以下のような階層的なアーキテクチャの導入が推奨される。まず、「ツールゲートウェイ」がエージェントからのリクエストを受け取り、引数の検証、ユーザーの権限確認、ビジネスルールの適用を行う。必要であれば人間の承認を求めることもある。次に、「べき等性コーディネーター」がべき等性キーを解決し、リクエストのパラメータが以前と同じであるかを確認する「リクエストのフィンガープリント」を使って、重複した操作の作成を防ぐ。これらの情報は「永続操作ストア」というデータベースに保存され、操作の状態や実行メタデータ、結果の参照が管理される。実際の操作は「実行ワーカー」が担当し、外部サービスに接続する際には、可能な限り外部サービスが提供するべき等性メカニズムにも同じキーを伝播させる。最後に、タイムアウトやワーカーのクラッシュなどで操作の結果が不明になった場合は、「調整サービス」が外部システムと連携して結果を解決する。

このように操作の状態を明確に定義することも重要だ。例えば、操作が受け付けられた「保留中（PENDING）」、ワーカーが処理中の「進行中（IN_PROGRESS）」、成功が確認された「成功（SUCCEEDED）」、失敗が確認された「失敗（FAILED）」、そして外部への影響はあったかもしれないが結果が不明な「不明（UNKNOWN）」といった状態が考えられる。「不明」の状態にある操作は、自動的に再試行してはならない。必ず外部システムに問い合わせて、実際に操作が完了したかどうかを確認する必要がある。

具体的な実装例として、PostgreSQLのようなリレーショナルデータベースは、べき等性レコードの保存に非常に適している。PostgreSQLはトランザクション機能や一意制約を提供し、複数のリクエストが同時に同じ操作を作成しようとした際の競合処理もサポートしている。例えば、agent_tool_operationsというテーブルを作成し、テナントID（システムの利用者単位の識別子）とべき等性キーの組み合わせを主キーに設定することで、同じテナント内で重複するキーが作成されるのを防ぐことができる。また、リクエストのフィンガープリントを生成するために、ツール名や引数などのリクエストパラメータを正規化し、ハッシュ関数（例えばSHA256）で一意の文字列を生成する方法がある。これにより、同じキーが誤って異なる操作に再利用されることを防ぐ。操作をデータベースに記録する際は、INSERT ... ON CONFLICT DO NOTHINGのようなデータベースの機能を利用して、並行するリクエストが安全に操作を「予約」し、重複せずに処理を進めることができる。

最も複雑なケースは、外部サービスでの処理は成功したが、その結果をローカルシステムが保存する前にワーカーがクラッシュしてしまった場合だ。この場合、ローカルデータベースだけを見ても操作が完了したかどうかは分からない。このような状況では、まず操作を「不明」状態としてマークする。次に、外部サービス（例えば支払いプロバイダー）に直接問い合わせて、その操作のトランザクションが実際に完了したかを確認する。もし成功していたらその結果を保存し、完了していなければ同じキーで安全に再試行する。それでも結果が不明なままであれば、システムが盲目的に再試行するのではなく、人間の介入を促すなどして状況をエスカレートさせるべきである。

もし外部APIがべき等性をサポートしていない場合、話はさらに複雑になる。ローカルのデータベースで重複を排除できても、外部サービスとの連携で正確に一度だけ処理を実行する「アトミックなコミット」（分割できない一連の処理）は保証できない。この場合は、可能な限り外部サービスのユニークな識別子や条件付き書き込み機能を利用するか、再試行前に必ず外部レコードと照合する「調整」のプロセスが必要となる。どうしても信頼できる重複排除や調整の仕組みがない場合、特に影響の大きい操作については、手動レビューや、万が一重複が発生した場合にそれを打ち消すような補償処理の導入を検討する必要があるだろう。

べき等性を実現するための技術選択肢はいくつか存在する。PostgreSQLのようなリレーショナルデータベースは、ビジネス操作の永続的な記録やトランザクション管理に優れている。Redisのようなインメモリデータベースは、高速な調整や短期間の重複排除に役立つが、データの永続性やフェイルオーバー時の挙動に注意が必要だ。また、外部サービス自体がべき等性キーをサポートしている場合は、それを最大限に活用することが最も効果的だ。

AIエージェントのシステムを構築する際には、べき等性メカニズムだけでなく、セキュリティ対策も不可欠である。AIモデルが生成した引数は信頼できない入力とみなし、厳格なスキーマ検証、リソース所有権の確認、ビジネス制約の適用、適切な権限チェックを常に行う必要がある。機密性の高い操作には、人間による最終承認を求めることも重要だ。また、システム全体でテナントごとに操作レコードを分離し、あるテナントのキーが他のテナントの操作に誤って影響を与えないようにする。リクエストのフィンガープリントや操作結果のような機密情報は、アクセス制御、暗号化、適切な保持ポリシー、匿名化によって保護されるべきである。

最後に、本番環境での運用を始める前には、徹底した監視とテストが不可欠だ。重複リクエストの発生率、フィンガープリントの競合、結果が不明になった操作の数、調整処理の遅延などを常に監視する仕組みが必要となる。テストでは、同じリクエストの繰り返しが既存の結果を返すか、異なるパラメータでキーを再利用した場合に拒否されるか、並行するリクエストが安全に一度だけ処理されるか、ワーカーのクラッシュからの安全な回復、タイムアウト後の盲目的な再試行が起きないことなど、様々なシナリオを網羅的に検証しなければならない。

AIエージェントを本番環境で活用する際には、複数のワーカー、非同期キュー、長時間実行されるワークフローなど、システムが複雑になるため、べき等性はシステム全体のアーキテクチャの一部として最初から組み込むべきだ。まず影響の大きいツールから始め、その操作の識別子、フィンガープリントのルール、状態の遷移、外部サービスとの保証内容、そして故障時のリカバリ動作を明確に定義し、徹底的にテストと監視を行った上で、他のツールへと適用範囲を広げていくのが賢明なアプローチだ。

このように、べき等性キーは、永続的な状態管理、アトミックな操作の確保、下流サービスでの重複排除、そして明示的なリカバリ手順と組み合わせることで、AIエージェントのツール実行をはるかに安全にする。システムエンジニアを目指す皆さんにとって、これらの概念は、現代の複雑なシステムを安全に構築するために不可欠な知識となるだろう。