majiayu000 opened a new pull request, #612: URL: https://github.com/apache/burr/pull/612
## Summary - Updated type hints to accept Union types for `stream_type` parameter in `streaming_action.pydantic` decorator - This allows users to specify union types like `Union[Model1, Model2]` for `stream_type` ## Changes - Modified `PartialType` in `burr/integrations/pydantic.py` to include `type` for Union support - Updated `stream_type` parameter type in `streaming_action.pydantic` in `burr/core/action.py` - Added test case `test_streaming_pydantic_action_union_stream_type` to verify the fix ## Test plan - [x] Added unit test for union type stream_type - [x] All existing pydantic tests pass (31 tests) - [x] All core action tests pass (46 tests) - [x] Pre-commit checks pass Closes #607 -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
