제너레이터 함수 트레이스

from typing import Generator
import weave

weave.init("my-project")

# 이 함수는 단순한 동기 제너레이터를 사용합니다.
# Weave는 Call과 해당 입력(`x`)을 트레이스하지만,
# 출력 값은 제너레이터가 소비된 후에만 캡처됩니다(예: `list()` 사용).
@weave.op
def basic_gen(x: int) -> Generator[int, None, None]:
    yield from range(x)

# 제너레이터 파이프라인 내에서 사용되는 일반 동기 함수입니다.
# 이 함수의 Call도 Weave가 별도로 트레이스합니다.
@weave.op
def inner(x: int) -> int:
    return x + 1

# 다른 트레이스된 함수(`inner`)를 호출하는 동기 제너레이터입니다.
# 각 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가 출력을 캡처하려면 제너레이터를 반드시 *소비*해야 합니다.
# 이는 동기 및 비동기 제너레이터 모두에 해당합니다.
res = deeply_nested_generator(4)
list(res)  # 중첩된 모든 Call과 yield의 트레이싱을 트리거합니다