From f424891d7626577a8acb20908f7f1fe35a47adb4 Mon Sep 17 00:00:00 2001 From: Yujong Lee Date: Sat, 5 Sep 2026 17:34:50 -0700 Subject: [PATCH] fix(rust): restore callback handle typing --- litellm/rust_bridge/request.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/litellm/rust_bridge/request.py b/litellm/rust_bridge/request.py index 7dc70110ab8..daf991037b5 100644 --- a/litellm/rust_bridge/request.py +++ b/litellm/rust_bridge/request.py @@ -5,6 +5,8 @@ from dataclasses import dataclass, replace from types import MappingProxyType from typing import Generic, Protocol, TypeVar +from .callbacks import OneShotCallbackHandle + @dataclass(frozen=True, slots=True) class NativeBedrockOptions: @@ -72,8 +74,7 @@ def vertex_options(params: Mapping[str, object]) -> NativeVertexOptions: location=location if isinstance(location, str) else None, ) - -from typing_extensions import ReadOnly, TypedDict +from typing_extensions import ReadOnly, TypedDict, TypeVar class NativePreCallDetails(TypedDict):