仕組みからわかる大規模言語モデル 生成AI時代のソフトウェア開発入門

6

す。

total

は現在の貯金額、

count

は入金回数を表します。また、

last_deposit

は最後に入金

した金額を表します。

total

と

count

は、

Annotated

で演算子

add

を使って更新されることを示

しています。これにより、

total

と

count

は更新時にそれぞれ自動的に加算されます。一方、

last_deposit

は更新時に上書きされます。このため、

Annotated

による修飾は不要です。

ノード関数の定義

次に、各ノードの処理を関数として定義します。LangGraph では、各ノードの処理を

Python の関数または

Runnable

オブジェクトとして定義します。この関数または

Runnable

オブ

ジェクトの引数は、状態の定義で指定した型と一致する必要があります。さらに、戻り値は状

態を更新するための辞書を返す必要があります。それでは、各処理を定義することで実際に確

認していきます。

Deposit

ノードでは、ユーザからの入金を受け付け、

total

、

count

、

last_deposit

を更新し

ます。次に実装を示します。

def deposit(state: PiggyBankState) -> dict:

amount = int(input("Enter the amount to deposit: "))

return {"total": amount, "count": 1, "last_deposit": amount}

deposit

関数は、

PiggyBankState