トレースユーティリティモジュールは、高度なトレースシナリオのためのヘルパー関数とクラスを提供します。

​
ThreadPoolExecutor

スレッド境界を越えてトレースコンテキストを保持するコンテキスト対応の ThreadPoolExecutor です。 
from weave import ThreadPoolExecutor

@weave.op
def process_item(item: str) -> str:
    return item.upper()

@weave.op
def process_batch(items: List[str]) -> List[str]:
    with ThreadPoolExecutor(max_workers=4) as executor:
        # Each item is processed in parallel while maintaining trace context
        results = list(executor.map(process_item, items))
    return results

​
主な特徴

  • トレース内の親子関係を保持
  • スレッド境界を越えてトレースコンテキストを維持
  • 標準の ThreadPoolExecutor API と互換性あり

​
ネストされたトレースの例

@weave.op
def fetch_data(url: str) -> dict:
    # Simulated API call
    return {"data": f"from {url}"}

@weave.op
def process_urls(urls: List[str]) -> List[dict]:
    with ThreadPoolExecutor(max_workers=10) as executor:
        # All fetch_data calls will be properly nested under process_urls
        results = list(executor.map(fetch_data, urls))
    return results

# Usage
urls = ["http://api1.com", "http://api2.com", "http://api3.com"]
data = process_urls(urls)

​
attributes

呼び出しに属性を追加するためのコンテキストマネージャ。 
from weave import attributes

@weave.op
def my_function():
    return "result"

# Add metadata to the call
with attributes({'environment': 'production', 'version': '1.0.0'}):
    my_function()

​
ネストされた属性

属性はネストでき、マージされます： 
with attributes({'env': 'prod'}):
    with attributes({'region': 'us-east-1'}):
        # Call will have both attributes
        my_function()

​
set_tracing_enabled

トレースを一時的に無効にするためのコンテキストマネージャ。 
from weave.trace.context.call_context import set_tracing_enabled

@weave.op
def traced_function():
    return "This will be traced"

# Temporarily disable tracing
with set_tracing_enabled(False):
    traced_function()  # This call won't be traced

# Tracing is automatically re-enabled outside the context
traced_function()  # This will be traced

​
get_current_call

op 内で現在実行中の呼び出しを取得します。 
from weave.trace.context import get_current_call

@weave.op
def my_function():
    call = get_current_call()
    if call:
        print(f"Current call ID: {call.id}")
        print(f"Current call inputs: {call.inputs}")

​
サマリーユーティリティ

​
add_summary

呼び出しにカスタムサマリーメトリクスを追加します。 
@weave.op
def evaluate_model(prompt: str) -> str:
    result = model.generate(prompt)
    
    # Add custom metrics to the call summary
    call = get_current_call()
    if call:
        call.summary.update({
            'token_count': len(result.split()),
            'confidence': 0.95
        })
    
    return result
これらのユーティリティは Weave のトレースシステムとシームレスに連携するように設計されています。これらの機能を使用する前に、必ず weave.init() が呼び出されていることを確認してください。