ジェネレーター関数をトレースする

from typing import Generator
import weave

weave.init("my-project")

# この関数はシンプルな sync ジェネレーターを使用します。
# Weave はこの call とその入力 (`x`) をトレースしますが、
# 出力値が取得されるのは、ジェネレーターが消費された後だけです（たとえば、`list()` を使用した場合）。
@weave.op
def basic_gen(x: int) -> Generator[int, None, None]:
    yield from range(x)

# ジェネレーターのパイプライン内で使用される通常の sync 関数です。
# この関数への calls も Weave によって個別にトレースされます。
@weave.op
def inner(x: int) -> int:
    return x + 1

# 別のトレース対象関数 (`inner`) を呼び出す sync ジェネレーターです。
# `yield` される各値は、`inner` への個別のトレース済み call から生成されます。
@weave.op
def nested_generator(x: int) -> Generator[int, None, None]:
    for i in range(x):
        yield inner(i)

# 上記のジェネレーターを組み合わせた、より複雑なジェネレーターです。
# ここでのトレースでは、階層的な call ツリーが生成されます:
# - `deeply_nested_generator` (親)
#   - `nested_generator` (子)
#     - `inner` (孫)
@weave.op
def deeply_nested_generator(x: int) -> Generator[int, None, None]:
    for i in range(x):
        for j in nested_generator(i):
            yield j

# Weave が出力を取得するには、ジェネレーターを*消費*する必要があります。
# これは sync ジェネレーターと async ジェネレーターの両方に当てはまります。
res = deeply_nested_generator(4)
list(res)  # ネストされたすべての call と yield のトレースをトリガーします