highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-06-22 05:50:57 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add backwards compatibility for users who don't have groups:read

Browse Source
This commit is contained in:
Weves 2023-09-10 10:54:43 -07:00 committed by Chris Weaver
parent e72f26ef53
commit 05a5419c8e
1 changed files with 26 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
backend/danswer/connectors/slack/connector.py
View File

@ -7,6 +7,7 @@ from typing import Any
from typing import cast from typing import cast
from slack_sdk import WebClient from slack_sdk import WebClient
from slack_sdk.errors import SlackApiError
from slack_sdk.web import SlackResponse from slack_sdk.web import SlackResponse
from danswer.configs.app_configs import INDEX_BATCH_SIZE from danswer.configs.app_configs import INDEX_BATCH_SIZE
@ -55,22 +56,43 @@ def get_channel_info(client: WebClient, channel_id: str) -> ChannelType:
] ]
def get_channels( def _get_channels(
client: WebClient, client: WebClient,
exclude_archived: bool = True, exclude_archived: bool,
get_private: bool,
) -> list[ChannelType]: ) -> list[ChannelType]:
"""Get all channels in the workspace"""
channels: list[dict[str, Any]] = [] channels: list[dict[str, Any]] = []
for result in _make_paginated_slack_api_call( for result in _make_paginated_slack_api_call(
client.conversations_list, client.conversations_list,
exclude_archived=exclude_archived, exclude_archived=exclude_archived,
# also get private channels the bot is added to # also get private channels the bot is added to
types=["public_channel", "private_channel"], types=["public_channel", "private_channel"]
if get_private
else ["public_channel"],
): ):
channels.extend(result["channels"]) channels.extend(result["channels"])
return channels return channels
def get_channels(
client: WebClient,
exclude_archived: bool = True,
) -> list[ChannelType]:
"""Get all channels in the workspace"""
# try getting private channels as well at first
try:
return _get_channels(
client=client, exclude_archived=exclude_archived, get_private=True
)
except SlackApiError as e:
logger.info(f"Unable to fetch private channels due to - {e}")
return _get_channels(
client=client, exclude_archived=exclude_archived, get_private=False
)
def get_channel_messages( def get_channel_messages(
client: WebClient, client: WebClient,
channel: dict[str, Any], channel: dict[str, Any],