fix: codegen async

Author: zimeg

scripts/codegen.py

@@ -1,5 +1,5 @@
-import sys
 import argparse
+import sys
 
 parser = argparse.ArgumentParser()
 parser.add_argument("-p", "--path", help="Path to the project source code.", type=str)
@@ -47,6 +47,23 @@
         async_source,
     )
     async_source = re.sub(r"= WebClient\(", "= AsyncWebClient(", async_source)
+    async_source = re.sub(
+        "from slack_sdk.web.chat_stream import ChatStream",
+        "from slack_sdk.web.async_chat_stream import AsyncChatStream",
+        async_source,
+    )
+    async_source = re.sub(r"ChatStream:", "AsyncChatStream:", async_source)
+    async_source = re.sub(r"ChatStream\(", "AsyncChatStream(", async_source)
+    async_source = re.sub(
+        r" streamer.append\(",
+        " await streamer.append(",
+        async_source,
+    )
+    async_source = re.sub(
+        r" streamer.stop\(",
+        " await streamer.stop(",
+        async_source,
+    )
     async_source = re.sub(
         r" self.files_getUploadURLExternal\(",
         " await self.files_getUploadURLExternal(",

slack_sdk/web/async_client.py

@@ -18,7 +18,7 @@
 
 import slack_sdk.errors as e
 from slack_sdk.models.views import View
-from slack_sdk.web.chat_stream import ChatStream
+from slack_sdk.web.async_chat_stream import AsyncChatStream
 
 from ..models.attachments import Attachment
 from ..models.blocks import Block
@@ -2916,7 +2916,7 @@ async def chat_stream(
         unfurl_links: Optional[bool] = None,
         unfurl_media: Optional[bool] = None,
         **kwargs,
-    ) -> ChatStream:
+    ) -> AsyncChatStream:
         """Stream markdown text into a conversation.
 
         This method provides an easy way to stream markdown text using the following endpoints:
@@ -2945,12 +2945,12 @@ async def chat_stream(
                 recipient_team_id="T0123456789",
                 recipient_user_id="U0123456789",
             )
-            streamer.append(markdown_text="**hello wo")
-            streamer.append(markdown_text="rld!**")
-            streamer.stop()
+            await streamer.append(markdown_text="**hello wo")
+            await streamer.append(markdown_text="rld!**")
+            await streamer.stop()
             ```
         """
-        return ChatStream(
+        return AsyncChatStream(
             self,
             logger=self._logger,
             channel=channel,