diff --git a/sdk/packages/python/iii/src/iii/iii.py b/sdk/packages/python/iii/src/iii/iii.py index 4bfc77edf..b748fad0e 100644 --- a/sdk/packages/python/iii/src/iii/iii.py +++ b/sdk/packages/python/iii/src/iii/iii.py @@ -57,6 +57,7 @@ StreamListGroupsInput, StreamListInput, StreamSetInput, + StreamUpdateInput, ) from .telemetry_types import OtelConfig from .triggers import Trigger, TriggerConfig, TriggerHandler, TriggerTypeRef @@ -1422,6 +1423,10 @@ async def list_groups_handler(data: Any) -> list[str]: ) return await stream.list_groups(input_data) + async def update_handler(data: Any) -> list[Any]: + input_data = StreamUpdateInput(**data) if isinstance(data, dict) else data + return await stream.update(input_data) + self.register_function({"id": f"stream::get({stream_name})"}, get_handler) self.register_function({"id": f"stream::set({stream_name})"}, set_handler) self.register_function({"id": f"stream::delete({stream_name})"}, delete_handler) @@ -1429,6 +1434,7 @@ async def list_groups_handler(data: Any) -> list[str]: self.register_function( {"id": f"stream::list_groups({stream_name})"}, list_groups_handler ) + self.register_function({"id": f"stream::update({stream_name})"}, update_handler) class TriggerAction: