From 355805d3b6b4b2f803a34f7f514f6b16d622300f Mon Sep 17 00:00:00 2001 From: nightcityblade Date: Tue, 30 Jun 2026 23:17:26 +0800 Subject: [PATCH] fix: handle bare dict in transform --- src/openai/_utils/_transform.py | 12 ++++++++++-- tests/test_transform.py | 8 ++++++++ 2 files changed, 18 insertions(+), 2 deletions(-) diff --git a/src/openai/_utils/_transform.py b/src/openai/_utils/_transform.py index 414f38c340..cd3084a3bb 100644 --- a/src/openai/_utils/_transform.py +++ b/src/openai/_utils/_transform.py @@ -180,7 +180,11 @@ def _transform_recursive( return _transform_typeddict(data, stripped_type) if origin == dict and is_mapping(data): - items_type = get_args(stripped_type)[1] + args = get_args(stripped_type) + if len(args) < 2: + return data + + items_type = args[1] return {key: _transform_recursive(value, annotation=items_type) for key, value in data.items()} if ( @@ -346,7 +350,11 @@ async def _async_transform_recursive( return await _async_transform_typeddict(data, stripped_type) if origin == dict and is_mapping(data): - items_type = get_args(stripped_type)[1] + args = get_args(stripped_type) + if len(args) < 2: + return data + + items_type = args[1] return {key: _transform_recursive(value, annotation=items_type) for key, value in data.items()} if ( diff --git a/tests/test_transform.py b/tests/test_transform.py index bece75dfc7..f31bf74951 100644 --- a/tests/test_transform.py +++ b/tests/test_transform.py @@ -458,3 +458,11 @@ async def test_strips_notgiven(use_async: bool) -> None: async def test_strips_omit(use_async: bool) -> None: assert await transform({"foo_bar": "bar"}, Foo1, use_async) == {"fooBar": "bar"} assert await transform({"foo_bar": omit}, Foo1, use_async) == {} + + +@parametrize +@pytest.mark.asyncio +async def test_transform_with_bare_dict_annotation(use_async: bool) -> None: + data = {"key": "value"} + + assert await transform(data, dict, use_async) == data