diff --git a/dashscope/api_entities/api_request_factory.py b/dashscope/api_entities/api_request_factory.py index 3666ecd..9347a19 100644 --- a/dashscope/api_entities/api_request_factory.py +++ b/dashscope/api_entities/api_request_factory.py @@ -18,7 +18,7 @@ def _get_protocol_params(kwargs): ws_stream_mode = kwargs.pop('ws_stream_mode', WebsocketStreamingMode.OUT) is_binary_input = kwargs.pop('is_binary_input', False) http_method = kwargs.pop('http_method', HTTPMethod.POST) - stream = kwargs.pop('stream', False) + stream = kwargs.get('stream', False) if not stream and ws_stream_mode == WebsocketStreamingMode.OUT: ws_stream_mode = WebsocketStreamingMode.NONE diff --git a/samples/test_image_generation.py b/samples/test_image_generation.py index bfbbaa7..b97ef11 100644 --- a/samples/test_image_generation.py +++ b/samples/test_image_generation.py @@ -87,4 +87,27 @@ res = ImageGeneration.list() print("-----------async-task-list-res-----------") - print(res) \ No newline at end of file + print(res) + + print("-" * 100) + + image_message = Message( + role=Role.USER, + # 支持本地文件 如 "image": "file://umbrella1.png" + content=[ + { + "text": "给我一个3张图辣椒炒肉教程" + } + ] + ) + + image_stream_res = ImageGeneration.call( + model=image_model, + messages=[image_message], + stream=True, + enable_interleave=True, + max_images=3 + ) + print("-----------sync-image-stream-call-res-----------") + for stream_res in image_stream_res: + print(stream_res) \ No newline at end of file