From 7e3d526b274196428d9d7c0047d29fa29e0956b4 Mon Sep 17 00:00:00 2001 From: Kazuhiro Sera Date: Tue, 17 Sep 2024 09:48:18 +0900 Subject: [PATCH] Add assistant.threads.* APIs (#1563) --- slack_sdk/web/async_client.py | 43 +++++++++++++++++++ slack_sdk/web/client.py | 43 +++++++++++++++++++ slack_sdk/web/legacy_client.py | 43 +++++++++++++++++++ .../web/test_web_client_coverage.py | 15 ++++++- 4 files changed, 143 insertions(+), 1 deletion(-) diff --git a/slack_sdk/web/async_client.py b/slack_sdk/web/async_client.py index 3bd6eed7..b33c68c3 100644 --- a/slack_sdk/web/async_client.py +++ b/slack_sdk/web/async_client.py @@ -2013,6 +2013,49 @@ async def tooling_tokens_rotate( kwargs.update({"refresh_token": refresh_token}) return await self.api_call("tooling.tokens.rotate", params=kwargs) + async def assistant_threads_setStatus( + self, + *, + channel_id: str, + thread_ts: str, + status: str, + **kwargs, + ) -> AsyncSlackResponse: + """Revokes a token. + https://api.slack.com/methods/assistant.threads.setStatus + """ + kwargs.update({"channel_id": channel_id, "thread_ts": thread_ts, "status": status}) + return await self.api_call("assistant.threads.setStatus", params=kwargs) + + async def assistant_threads_setTitle( + self, + *, + channel_id: str, + thread_ts: str, + title: str, + **kwargs, + ) -> AsyncSlackResponse: + """Revokes a token. + https://api.slack.com/methods/assistant.threads.setTitle + """ + kwargs.update({"channel_id": channel_id, "thread_ts": thread_ts, "title": title}) + return await self.api_call("assistant.threads.setTitle", params=kwargs) + + async def assistant_threads_setSuggestedPrompts( + self, + *, + channel_id: str, + thread_ts: str, + title: Optional[str] = None, + prompts: List[Dict[str, str]], + **kwargs, + ) -> AsyncSlackResponse: + """Revokes a token. + https://api.slack.com/methods/assistant.threads.setSuggestedPrompts + """ + kwargs.update({"channel_id": channel_id, "thread_ts": thread_ts, "title": title, "prompts": prompts}) + return await self.api_call("assistant.threads.setSuggestedPrompts", json=kwargs) + async def auth_revoke( self, *, diff --git a/slack_sdk/web/client.py b/slack_sdk/web/client.py index b9ce20b9..e42efef7 100644 --- a/slack_sdk/web/client.py +++ b/slack_sdk/web/client.py @@ -2004,6 +2004,49 @@ def tooling_tokens_rotate( kwargs.update({"refresh_token": refresh_token}) return self.api_call("tooling.tokens.rotate", params=kwargs) + def assistant_threads_setStatus( + self, + *, + channel_id: str, + thread_ts: str, + status: str, + **kwargs, + ) -> SlackResponse: + """Revokes a token. + https://api.slack.com/methods/assistant.threads.setStatus + """ + kwargs.update({"channel_id": channel_id, "thread_ts": thread_ts, "status": status}) + return self.api_call("assistant.threads.setStatus", params=kwargs) + + def assistant_threads_setTitle( + self, + *, + channel_id: str, + thread_ts: str, + title: str, + **kwargs, + ) -> SlackResponse: + """Revokes a token. + https://api.slack.com/methods/assistant.threads.setTitle + """ + kwargs.update({"channel_id": channel_id, "thread_ts": thread_ts, "title": title}) + return self.api_call("assistant.threads.setTitle", params=kwargs) + + def assistant_threads_setSuggestedPrompts( + self, + *, + channel_id: str, + thread_ts: str, + title: Optional[str] = None, + prompts: List[Dict[str, str]], + **kwargs, + ) -> SlackResponse: + """Revokes a token. + https://api.slack.com/methods/assistant.threads.setSuggestedPrompts + """ + kwargs.update({"channel_id": channel_id, "thread_ts": thread_ts, "title": title, "prompts": prompts}) + return self.api_call("assistant.threads.setSuggestedPrompts", json=kwargs) + def auth_revoke( self, *, diff --git a/slack_sdk/web/legacy_client.py b/slack_sdk/web/legacy_client.py index 8618b5a4..d93c3924 100644 --- a/slack_sdk/web/legacy_client.py +++ b/slack_sdk/web/legacy_client.py @@ -2015,6 +2015,49 @@ def tooling_tokens_rotate( kwargs.update({"refresh_token": refresh_token}) return self.api_call("tooling.tokens.rotate", params=kwargs) + def assistant_threads_setStatus( + self, + *, + channel_id: str, + thread_ts: str, + status: str, + **kwargs, + ) -> Union[Future, SlackResponse]: + """Revokes a token. + https://api.slack.com/methods/assistant.threads.setStatus + """ + kwargs.update({"channel_id": channel_id, "thread_ts": thread_ts, "status": status}) + return self.api_call("assistant.threads.setStatus", params=kwargs) + + def assistant_threads_setTitle( + self, + *, + channel_id: str, + thread_ts: str, + title: str, + **kwargs, + ) -> Union[Future, SlackResponse]: + """Revokes a token. + https://api.slack.com/methods/assistant.threads.setTitle + """ + kwargs.update({"channel_id": channel_id, "thread_ts": thread_ts, "title": title}) + return self.api_call("assistant.threads.setTitle", params=kwargs) + + def assistant_threads_setSuggestedPrompts( + self, + *, + channel_id: str, + thread_ts: str, + title: Optional[str] = None, + prompts: List[Dict[str, str]], + **kwargs, + ) -> Union[Future, SlackResponse]: + """Revokes a token. + https://api.slack.com/methods/assistant.threads.setSuggestedPrompts + """ + kwargs.update({"channel_id": channel_id, "thread_ts": thread_ts, "title": title, "prompts": prompts}) + return self.api_call("assistant.threads.setSuggestedPrompts", json=kwargs) + def auth_revoke( self, *, diff --git a/tests/slack_sdk_async/web/test_web_client_coverage.py b/tests/slack_sdk_async/web/test_web_client_coverage.py index d56dbf7f..885acdb0 100644 --- a/tests/slack_sdk_async/web/test_web_client_coverage.py +++ b/tests/slack_sdk_async/web/test_web_client_coverage.py @@ -17,7 +17,7 @@ class TestWebClientCoverage(unittest.TestCase): # 289 endpoints as of Sept 4, 2024 # Can be fetched by running `var methodNames = [].slice.call(document.getElementsByClassName('apiReferenceFilterableList__listItemLink')).map(e => e.href.replace("https://api.slack.com/methods/", ""));console.log(methodNames.toString());console.log(methodNames.length);` on https://api.slack.com/methods - all_api_methods = "admin.analytics.getFile,admin.apps.activities.list,admin.apps.approve,admin.apps.clearResolution,admin.apps.restrict,admin.apps.uninstall,admin.apps.approved.list,admin.apps.config.lookup,admin.apps.config.set,admin.apps.requests.cancel,admin.apps.requests.list,admin.apps.restricted.list,admin.audit.anomaly.allow.getItem,admin.audit.anomaly.allow.updateItem,admin.auth.policy.assignEntities,admin.auth.policy.getEntities,admin.auth.policy.removeEntities,admin.barriers.create,admin.barriers.delete,admin.barriers.list,admin.barriers.update,admin.conversations.archive,admin.conversations.bulkArchive,admin.conversations.bulkDelete,admin.conversations.bulkMove,admin.conversations.convertToPrivate,admin.conversations.convertToPublic,admin.conversations.create,admin.conversations.delete,admin.conversations.disconnectShared,admin.conversations.getConversationPrefs,admin.conversations.getCustomRetention,admin.conversations.getTeams,admin.conversations.invite,admin.conversations.lookup,admin.conversations.removeCustomRetention,admin.conversations.rename,admin.conversations.search,admin.conversations.setConversationPrefs,admin.conversations.setCustomRetention,admin.conversations.setTeams,admin.conversations.unarchive,admin.conversations.ekm.listOriginalConnectedChannelInfo,admin.conversations.restrictAccess.addGroup,admin.conversations.restrictAccess.listGroups,admin.conversations.restrictAccess.removeGroup,admin.emoji.add,admin.emoji.addAlias,admin.emoji.list,admin.emoji.remove,admin.emoji.rename,admin.functions.list,admin.functions.permissions.lookup,admin.functions.permissions.set,admin.inviteRequests.approve,admin.inviteRequests.deny,admin.inviteRequests.list,admin.inviteRequests.approved.list,admin.inviteRequests.denied.list,admin.roles.addAssignments,admin.roles.listAssignments,admin.roles.removeAssignments,admin.teams.admins.list,admin.teams.create,admin.teams.list,admin.teams.owners.list,admin.teams.settings.info,admin.teams.settings.setDefaultChannels,admin.teams.settings.setDescription,admin.teams.settings.setDiscoverability,admin.teams.settings.setIcon,admin.teams.settings.setName,admin.usergroups.addChannels,admin.usergroups.addTeams,admin.usergroups.listChannels,admin.usergroups.removeChannels,admin.users.assign,admin.users.invite,admin.users.list,admin.users.remove,admin.users.setAdmin,admin.users.setExpiration,admin.users.setOwner,admin.users.setRegular,admin.users.session.clearSettings,admin.users.session.getSettings,admin.users.session.invalidate,admin.users.session.list,admin.users.session.reset,admin.users.session.resetBulk,admin.users.session.setSettings,admin.users.unsupportedVersions.export,admin.workflows.collaborators.add,admin.workflows.collaborators.remove,admin.workflows.permissions.lookup,admin.workflows.search,admin.workflows.unpublish,admin.workflows.triggers.types.permissions.lookup,admin.workflows.triggers.types.permissions.set,api.test,apps.activities.list,apps.auth.external.delete,apps.auth.external.get,apps.connections.open,apps.uninstall,apps.datastore.bulkDelete,apps.datastore.bulkGet,apps.datastore.bulkPut,apps.datastore.count,apps.datastore.delete,apps.datastore.get,apps.datastore.put,apps.datastore.query,apps.datastore.update,apps.event.authorizations.list,apps.manifest.create,apps.manifest.delete,apps.manifest.export,apps.manifest.update,apps.manifest.validate,auth.revoke,auth.test,auth.teams.list,bookmarks.add,bookmarks.edit,bookmarks.list,bookmarks.remove,bots.info,calls.add,calls.end,calls.info,calls.update,calls.participants.add,calls.participants.remove,canvases.access.delete,canvases.access.set,canvases.create,canvases.delete,canvases.edit,canvases.sections.lookup,chat.delete,chat.deleteScheduledMessage,chat.getPermalink,chat.meMessage,chat.postEphemeral,chat.postMessage,chat.scheduleMessage,chat.unfurl,chat.update,chat.scheduledMessages.list,conversations.acceptSharedInvite,conversations.approveSharedInvite,conversations.archive,conversations.close,conversations.create,conversations.declineSharedInvite,conversations.history,conversations.info,conversations.invite,conversations.inviteShared,conversations.join,conversations.kick,conversations.leave,conversations.list,conversations.listConnectInvites,conversations.mark,conversations.members,conversations.open,conversations.rename,conversations.replies,conversations.setPurpose,conversations.setTopic,conversations.unarchive,conversations.canvases.create,conversations.externalInvitePermissions.set,conversations.requestSharedInvite.approve,conversations.requestSharedInvite.deny,dialog.open,dnd.endDnd,dnd.endSnooze,dnd.info,dnd.setSnooze,dnd.teamInfo,emoji.list,files.completeUploadExternal,files.delete,files.getUploadURLExternal,files.info,files.list,files.revokePublicURL,files.sharedPublicURL,files.upload,files.comments.delete,files.remote.add,files.remote.info,files.remote.list,files.remote.remove,files.remote.share,files.remote.update,functions.completeError,functions.completeSuccess,functions.distributions.permissions.add,functions.distributions.permissions.list,functions.distributions.permissions.remove,functions.distributions.permissions.set,functions.workflows.steps.list,functions.workflows.steps.responses.export,migration.exchange,oauth.access,oauth.v2.access,oauth.v2.exchange,openid.connect.token,openid.connect.userInfo,pins.add,pins.list,pins.remove,reactions.add,reactions.get,reactions.list,reactions.remove,reminders.add,reminders.complete,reminders.delete,reminders.info,reminders.list,rtm.connect,rtm.start,search.all,search.files,search.messages,stars.add,stars.list,stars.remove,team.accessLogs,team.billableInfo,team.info,team.integrationLogs,team.billing.info,team.externalTeams.disconnect,team.externalTeams.list,team.preferences.list,team.profile.get,tooling.tokens.rotate,usergroups.create,usergroups.disable,usergroups.enable,usergroups.list,usergroups.update,usergroups.users.list,usergroups.users.update,users.conversations,users.deletePhoto,users.getPresence,users.identity,users.info,users.list,users.lookupByEmail,users.setActive,users.setPhoto,users.setPresence,users.discoverableContacts.lookup,users.profile.get,users.profile.set,views.open,views.publish,views.push,views.update,workflows.stepCompleted,workflows.stepFailed,workflows.updateStep,workflows.triggers.permissions.add,workflows.triggers.permissions.list,workflows.triggers.permissions.remove,workflows.triggers.permissions.set,channels.create,channels.info,channels.invite,channels.mark,groups.create,groups.info,groups.invite,groups.mark,groups.open,im.list,im.mark,im.open,mpim.list,mpim.mark,mpim.open".split( + all_api_methods = "admin.analytics.getFile,admin.apps.activities.list,admin.apps.approve,admin.apps.clearResolution,admin.apps.restrict,admin.apps.uninstall,admin.apps.approved.list,admin.apps.config.lookup,admin.apps.config.set,admin.apps.requests.cancel,admin.apps.requests.list,admin.apps.restricted.list,admin.audit.anomaly.allow.getItem,admin.audit.anomaly.allow.updateItem,admin.auth.policy.assignEntities,admin.auth.policy.getEntities,admin.auth.policy.removeEntities,admin.barriers.create,admin.barriers.delete,admin.barriers.list,admin.barriers.update,admin.conversations.archive,admin.conversations.bulkArchive,admin.conversations.bulkDelete,admin.conversations.bulkMove,admin.conversations.convertToPrivate,admin.conversations.convertToPublic,admin.conversations.create,admin.conversations.delete,admin.conversations.disconnectShared,admin.conversations.getConversationPrefs,admin.conversations.getCustomRetention,admin.conversations.getTeams,admin.conversations.invite,admin.conversations.lookup,admin.conversations.removeCustomRetention,admin.conversations.rename,admin.conversations.search,admin.conversations.setConversationPrefs,admin.conversations.setCustomRetention,admin.conversations.setTeams,admin.conversations.unarchive,admin.conversations.ekm.listOriginalConnectedChannelInfo,admin.conversations.restrictAccess.addGroup,admin.conversations.restrictAccess.listGroups,admin.conversations.restrictAccess.removeGroup,admin.emoji.add,admin.emoji.addAlias,admin.emoji.list,admin.emoji.remove,admin.emoji.rename,admin.functions.list,admin.functions.permissions.lookup,admin.functions.permissions.set,admin.inviteRequests.approve,admin.inviteRequests.deny,admin.inviteRequests.list,admin.inviteRequests.approved.list,admin.inviteRequests.denied.list,admin.roles.addAssignments,admin.roles.listAssignments,admin.roles.removeAssignments,admin.teams.admins.list,admin.teams.create,admin.teams.list,admin.teams.owners.list,admin.teams.settings.info,admin.teams.settings.setDefaultChannels,admin.teams.settings.setDescription,admin.teams.settings.setDiscoverability,admin.teams.settings.setIcon,admin.teams.settings.setName,admin.usergroups.addChannels,admin.usergroups.addTeams,admin.usergroups.listChannels,admin.usergroups.removeChannels,admin.users.assign,admin.users.invite,admin.users.list,admin.users.remove,admin.users.setAdmin,admin.users.setExpiration,admin.users.setOwner,admin.users.setRegular,admin.users.session.clearSettings,admin.users.session.getSettings,admin.users.session.invalidate,admin.users.session.list,admin.users.session.reset,admin.users.session.resetBulk,admin.users.session.setSettings,admin.users.unsupportedVersions.export,admin.workflows.collaborators.add,admin.workflows.collaborators.remove,admin.workflows.permissions.lookup,admin.workflows.search,admin.workflows.unpublish,admin.workflows.triggers.types.permissions.lookup,admin.workflows.triggers.types.permissions.set,api.test,apps.activities.list,apps.auth.external.delete,apps.auth.external.get,apps.connections.open,apps.uninstall,apps.datastore.bulkDelete,apps.datastore.bulkGet,apps.datastore.bulkPut,apps.datastore.count,apps.datastore.delete,apps.datastore.get,apps.datastore.put,apps.datastore.query,apps.datastore.update,apps.event.authorizations.list,apps.manifest.create,apps.manifest.delete,apps.manifest.export,apps.manifest.update,apps.manifest.validate,auth.revoke,auth.test,auth.teams.list,bookmarks.add,bookmarks.edit,bookmarks.list,bookmarks.remove,bots.info,calls.add,calls.end,calls.info,calls.update,calls.participants.add,calls.participants.remove,canvases.access.delete,canvases.access.set,canvases.create,canvases.delete,canvases.edit,canvases.sections.lookup,chat.delete,chat.deleteScheduledMessage,chat.getPermalink,chat.meMessage,chat.postEphemeral,chat.postMessage,chat.scheduleMessage,chat.unfurl,chat.update,chat.scheduledMessages.list,conversations.acceptSharedInvite,conversations.approveSharedInvite,conversations.archive,conversations.close,conversations.create,conversations.declineSharedInvite,conversations.history,conversations.info,conversations.invite,conversations.inviteShared,conversations.join,conversations.kick,conversations.leave,conversations.list,conversations.listConnectInvites,conversations.mark,conversations.members,conversations.open,conversations.rename,conversations.replies,conversations.setPurpose,conversations.setTopic,conversations.unarchive,conversations.canvases.create,conversations.externalInvitePermissions.set,conversations.requestSharedInvite.approve,conversations.requestSharedInvite.deny,dialog.open,dnd.endDnd,dnd.endSnooze,dnd.info,dnd.setSnooze,dnd.teamInfo,emoji.list,files.completeUploadExternal,files.delete,files.getUploadURLExternal,files.info,files.list,files.revokePublicURL,files.sharedPublicURL,files.upload,files.comments.delete,files.remote.add,files.remote.info,files.remote.list,files.remote.remove,files.remote.share,files.remote.update,functions.completeError,functions.completeSuccess,functions.distributions.permissions.add,functions.distributions.permissions.list,functions.distributions.permissions.remove,functions.distributions.permissions.set,functions.workflows.steps.list,functions.workflows.steps.responses.export,migration.exchange,oauth.access,oauth.v2.access,oauth.v2.exchange,openid.connect.token,openid.connect.userInfo,pins.add,pins.list,pins.remove,reactions.add,reactions.get,reactions.list,reactions.remove,reminders.add,reminders.complete,reminders.delete,reminders.info,reminders.list,rtm.connect,rtm.start,search.all,search.files,search.messages,stars.add,stars.list,stars.remove,team.accessLogs,team.billableInfo,team.info,team.integrationLogs,team.billing.info,team.externalTeams.disconnect,team.externalTeams.list,team.preferences.list,team.profile.get,tooling.tokens.rotate,usergroups.create,usergroups.disable,usergroups.enable,usergroups.list,usergroups.update,usergroups.users.list,usergroups.users.update,users.conversations,users.deletePhoto,users.getPresence,users.identity,users.info,users.list,users.lookupByEmail,users.setActive,users.setPhoto,users.setPresence,users.discoverableContacts.lookup,users.profile.get,users.profile.set,views.open,views.publish,views.push,views.update,workflows.stepCompleted,workflows.stepFailed,workflows.updateStep,workflows.triggers.permissions.add,workflows.triggers.permissions.list,workflows.triggers.permissions.remove,workflows.triggers.permissions.set,channels.create,channels.info,channels.invite,channels.mark,groups.create,groups.info,groups.invite,groups.mark,groups.open,im.list,im.mark,im.open,mpim.list,mpim.mark,mpim.open,assistant.threads.setStatus,assistant.threads.setTitle,assistant.threads.setSuggestedPrompts".split( "," ) @@ -394,6 +394,19 @@ async def run_method(self, method_name, method, async_method): elif method_name == "apps_uninstall": self.api_methods_to_call.remove(method(client_id="111.222", client_secret="xxx")["method"]) await async_method(client_id="111.222", client_secret="xxx") + elif method_name == "assistant_threads_setStatus": + self.api_methods_to_call.remove( + method(channel_id="D111", thread_ts="111.222", status="is typing...")["method"] + ) + await async_method(channel_id="D111", thread_ts="111.222", status="is typing...") + elif method_name == "assistant_threads_setTitle": + self.api_methods_to_call.remove(method(channel_id="D111", thread_ts="111.222", title="New chat")["method"]) + await async_method(channel_id="D111", thread_ts="111.222", title="New chat") + elif method_name == "assistant_threads_setSuggestedPrompts": + self.api_methods_to_call.remove( + method(channel_id="D111", thread_ts="111.222", prompts=[{"title": "X", "message": "Y"}])["method"] + ) + await async_method(channel_id="D111", thread_ts="111.222", prompts=[{"title": "X", "message": "Y"}]) elif method_name == "bookmarks_add": self.api_methods_to_call.remove(method(channel_id="C1234", title="bedtime story", type="article")["method"]) await async_method(channel_id="C1234", title="bedtime story", type="article")