From e86e6634c5098b8efe7039942de26ea9c61cbf66 Mon Sep 17 00:00:00 2001
From: Kazuhiro Sera <seratch@gmail.com>
Date: Mon, 28 Oct 2024 09:36:21 +0900
Subject: [PATCH] Add title argument to SetSuggestedPrompts arguments

---
 .../set_suggested_prompts/async_set_suggested_prompts.py | 9 +++++++--
 .../set_suggested_prompts/set_suggested_prompts.py       | 9 +++++++--
 tests/scenario_tests/test_events_assistant.py            | 3 +++
 tests/scenario_tests_async/test_events_assistant.py      | 4 ++++
 4 files changed, 21 insertions(+), 4 deletions(-)

diff --git a/slack_bolt/context/set_suggested_prompts/async_set_suggested_prompts.py b/slack_bolt/context/set_suggested_prompts/async_set_suggested_prompts.py
index 76f827732..aeeb244d7 100644
--- a/slack_bolt/context/set_suggested_prompts/async_set_suggested_prompts.py
+++ b/slack_bolt/context/set_suggested_prompts/async_set_suggested_prompts.py
@@ -1,4 +1,4 @@
-from typing import List, Dict, Union
+from typing import List, Dict, Union, Optional
 
 from slack_sdk.web.async_client import AsyncWebClient
 from slack_sdk.web.async_slack_response import AsyncSlackResponse
@@ -19,7 +19,11 @@ def __init__(
         self.channel_id = channel_id
         self.thread_ts = thread_ts
 
-    async def __call__(self, prompts: List[Union[str, Dict[str, str]]]) -> AsyncSlackResponse:
+    async def __call__(
+        self,
+        prompts: List[Union[str, Dict[str, str]]],
+        title: Optional[str] = None,
+    ) -> AsyncSlackResponse:
         prompts_arg: List[Dict[str, str]] = []
         for prompt in prompts:
             if isinstance(prompt, str):
@@ -31,4 +35,5 @@ async def __call__(self, prompts: List[Union[str, Dict[str, str]]]) -> AsyncSlac
             channel_id=self.channel_id,
             thread_ts=self.thread_ts,
             prompts=prompts_arg,
+            title=title,
         )
diff --git a/slack_bolt/context/set_suggested_prompts/set_suggested_prompts.py b/slack_bolt/context/set_suggested_prompts/set_suggested_prompts.py
index 3714f4830..fc9304b17 100644
--- a/slack_bolt/context/set_suggested_prompts/set_suggested_prompts.py
+++ b/slack_bolt/context/set_suggested_prompts/set_suggested_prompts.py
@@ -1,4 +1,4 @@
-from typing import List, Dict, Union
+from typing import List, Dict, Union, Optional
 
 from slack_sdk import WebClient
 from slack_sdk.web import SlackResponse
@@ -19,7 +19,11 @@ def __init__(
         self.channel_id = channel_id
         self.thread_ts = thread_ts
 
-    def __call__(self, prompts: List[Union[str, Dict[str, str]]]) -> SlackResponse:
+    def __call__(
+        self,
+        prompts: List[Union[str, Dict[str, str]]],
+        title: Optional[str] = None,
+    ) -> SlackResponse:
         prompts_arg: List[Dict[str, str]] = []
         for prompt in prompts:
             if isinstance(prompt, str):
@@ -31,4 +35,5 @@ def __call__(self, prompts: List[Union[str, Dict[str, str]]]) -> SlackResponse:
             channel_id=self.channel_id,
             thread_ts=self.thread_ts,
             prompts=prompts_arg,
+            title=title,
         )
diff --git a/tests/scenario_tests/test_events_assistant.py b/tests/scenario_tests/test_events_assistant.py
index f5ccbb565..07f7ede53 100644
--- a/tests/scenario_tests/test_events_assistant.py
+++ b/tests/scenario_tests/test_events_assistant.py
@@ -46,6 +46,9 @@ def start_thread(say: Say, set_suggested_prompts: SetSuggestedPrompts, context:
             assert context.thread_ts == "1726133698.626339"
             say("Hi, how can I help you today?")
             set_suggested_prompts(prompts=[{"title": "What does SLACK stand for?", "message": "What does SLACK stand for?"}])
+            set_suggested_prompts(
+                prompts=[{"title": "What does SLACK stand for?", "message": "What does SLACK stand for?"}], title="foo"
+            )
             state["called"] = True
 
         @assistant.thread_context_changed
diff --git a/tests/scenario_tests_async/test_events_assistant.py b/tests/scenario_tests_async/test_events_assistant.py
index bf71e914b..ac2c734c5 100644
--- a/tests/scenario_tests_async/test_events_assistant.py
+++ b/tests/scenario_tests_async/test_events_assistant.py
@@ -61,6 +61,10 @@ async def start_thread(say: AsyncSay, set_suggested_prompts: AsyncSetSuggestedPr
             await set_suggested_prompts(
                 prompts=[{"title": "What does SLACK stand for?", "message": "What does SLACK stand for?"}]
             )
+            await set_suggested_prompts(
+                prompts=[{"title": "What does SLACK stand for?", "message": "What does SLACK stand for?"}],
+                title="foo",
+            )
             state["called"] = True
 
         @assistant.thread_context_changed