API 概要

`クラス` `DisplayNameFuncError`

`クラス` `OpCallError`

`クラス` `OpKwargs`

op() のキーワード引数用の TypedDict です。

`クラス` `Sentinel`

Sentinel(package: ‘str’, path: ‘str’, name: ‘str’)

`メソッド` `init`

__init__(package: 'str', path: 'str', name: 'str') → None

`クラス` `WeaveKwargs`

`関数` `as_op`

as_op(fn: 'Callable[P, R]') → Op[P, R]

@weave.op でデコレートされた関数を受け取り、その Op を返します。 @weave.op でデコレートされた関数はすでに Op のインスタンスであるため、この関数は実行時には no-op となります。ただし、OpDef の属性に型安全にアクセスする必要がある場合は、型チェッカーを満たす目的でこの関数を使用できます。 引数:

fn: weave.op でデコレートされた関数。 戻り値: 関数の Op。

`関数` `call`

call(
    op: 'Op',
    *args: 'Any',
    __weave: 'WeaveKwargs | None' = None,
    __should_raise: 'bool' = False,
    __require_explicit_finish: 'bool' = False,
    **kwargs: 'Any'
) → tuple[Any, Call] | Coroutine[Any, Any, tuple[Any, Call]]

op を実行し、結果とその実行を表す Call の両方を返します。この関数は例外を送出しません。すべてのエラーは Call オブジェクトに取得されます。このメソッドは @weave.op でデコレートされたすべての関数に自動的にバインドされ、次のように使用できます。

@weave.op
def add(a: int, b: int) -> int:
     return a + b

result, call = add.call(1, 2)

`関数` `calls`

calls(op: 'Op') → CallsIter

この op へのすべての Call を走査するイテレーターを取得します。このメソッドは、@weave.op でデコレートされたすべての関数に自動的にバインドされ、次のように使用できます。

@weave.op
def add(a: int, b: int) -> int:
     return a + b

calls = add.calls()
for call in calls:
     print(call)

`関数` `get_captured_code`

get_captured_code(op: 'Op') → str

op の取得されたコードを取得します。これは、ref から op を取得した場合にのみ機能します。使用パターンは次のとおりです。 ref = weave.publish(func) op = ref.get() captured_code = op.get_captured_code()

`関数` `is_op`

is_op(obj: 'Any') → TypeIs[Op]

オブジェクトが Op かどうかを確認します。

`関数` `is_placeholder_call`

is_placeholder_call(call: 'Call') → TypeIs[NoOpCall]

`関数` `is_tracing_setting_disabled`

is_tracing_setting_disabled() → bool

`関数` `maybe_bind_method`

maybe_bind_method(func: 'Callable', self: 'Any' = None) → Callable | MethodType

任意のオブジェクトに関数をバインドします (クラスでない場合でも) 。 self が None の場合、関数をそのまま返します。

`関数` `maybe_unbind_method`

maybe_unbind_method(oplike: 'Op | MethodType | partial') → Op

Op に類似したメソッドまたは部分適用の束縛を解除し、通常の Op 関数にします。対象ごとの処理:

メソッドの場合、設定済みの self パラメーターを削除します
部分適用の場合、事前設定されたすべてのパラメーターを削除します

`関数` `op`

op(
    func: 'Callable[P, R] | None' = None,
    name: 'str | None' = None,
    call_display_name: 'str | CallDisplayNameFunc | None' = None,
    postprocess_inputs: 'PostprocessInputsFunc | None' = None,
    postprocess_output: 'PostprocessOutputFunc | None' = None,
    tracing_sample_rate: 'float' = 1.0,
    enable_code_capture: 'bool' = True,
    accumulator: 'Callable[[Any | None, Any], Any] | None' = None,
    kind: 'OpKind | None' = None,
    color: 'OpColor | None' = None,
    attributes: 'dict[str, Any] | None' = None,
    eager_call_start: 'bool' = False
) → Callable[[Callable[P, R]], Op[P, R]] | Op[P, R]

関数またはメソッドを Weave op に変換するデコレーターです。同期と非同期の両方に対応しています。イテレーター関数を自動的に検出し、適切な動作を適用します。 引数:

`関数` `placeholder_call`

placeholder_call() → Call

`関数` `setup_dunder_weave_dict`

setup_dunder_weave_dict(op: 'Op', d: 'WeaveKwargs | None' = None) → WeaveKwargs

WeaveKwargs を op に渡すために使用される __weave dict を設定します。

func: デコレートする関数。
name: op のカスタム名。関数の名をデフォルトとします。
call_display_name: Call の表示名。string または callable を指定できます。
postprocess_inputs: ログする前に入力を変換する関数。
postprocess_output: ログする前に出力を変換する関数。
tracing_sample_rate: トレースする Call の割合 (0.0 から 1.0) 。
enable_code_capture: この op のソースコードを取得するかどうか。
accumulator: ストリーミング op の結果を蓄積する関数。
attributes: この op が作成するすべての Call にマージされるデフォルトの属性 (優先度が最も低い) 。weave.attributes() コンテキストと明示的な Call ごとの属性は、キーの競合時にこれらを上書きします。予約済みの “weave” キーはここでは設定できません。
eager_call_start: True の場合、Call の開始がバッチ処理ではなく即時に送信されます。評価などの長時間実行される操作で、UI にすぐに表示する必要がある場合に便利です。 引数:
d: 更新する既存の WeaveKwargs dict (オプション) 。
op: kind と color を抽出する Op。 戻り値: attributes、display_name、およびオプションで kind/color が設定された WeaveKwargs dict。

`関数` `should_skip_tracing_for_op`

should_skip_tracing_for_op(op: 'Op') → bool

​API 概要

​クラス DisplayNameFuncError

​クラス OpCallError

​クラス OpKwargs

​クラス Sentinel

​メソッド __init__

​クラス WeaveKwargs

​関数 as_op

​関数 call

​関数 calls

​関数 get_captured_code

​関数 is_op

​関数 is_placeholder_call

​関数 is_tracing_setting_disabled

​関数 maybe_bind_method

​関数 maybe_unbind_method

​関数 op

​関数 placeholder_call

​関数 setup_dunder_weave_dict

​関数 should_skip_tracing_for_op