highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-03-30 04:31:49 +02:00
Code Issues Packages Projects Releases Wiki Activity

Consolidate blocks into ee/handle_standard_answers

Browse Source
This commit is contained in:
danswer-trial 2024-09-13 18:41:35 -07:00
parent 62ce65b8dc
commit 6acb9d5611
2 changed files with 21 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
backend/ee/danswer/danswerbot/slack/blocks.py
View File

@ -1,22 +0,0 @@
from slack_sdk.models.blocks import ActionsBlock
from slack_sdk.models.blocks import Block
from slack_sdk.models.blocks import ButtonElement
from slack_sdk.models.blocks import SectionBlock
from danswer.danswerbot.slack.constants import GENERATE_ANSWER_BUTTON_ACTION_ID
def build_standard_answer_blocks(
answer_message: str,
) -> list[Block]:
generate_button_block = ButtonElement(
action_id=GENERATE_ANSWER_BUTTON_ACTION_ID,
text="Generate Full Answer",
)
answer_block = SectionBlock(text=answer_message)
return [
answer_block,
ActionsBlock(
elements=[generate_button_block],
),
]

22
backend/ee/danswer/danswerbot/slack/handlers/handle_standard_answers.py
View File

@ -1,9 +1,14 @@
from slack_sdk import WebClient from slack_sdk import WebClient
from slack_sdk.models.blocks import ActionsBlock
from slack_sdk.models.blocks import Block
from slack_sdk.models.blocks import ButtonElement
from slack_sdk.models.blocks import SectionBlock
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from danswer.configs.constants import MessageType from danswer.configs.constants import MessageType
from danswer.configs.danswerbot_configs import DANSWER_REACT_EMOJI from danswer.configs.danswerbot_configs import DANSWER_REACT_EMOJI
from danswer.danswerbot.slack.blocks import get_restate_blocks from danswer.danswerbot.slack.blocks import get_restate_blocks
from danswer.danswerbot.slack.constants import GENERATE_ANSWER_BUTTON_ACTION_ID
from danswer.danswerbot.slack.handlers.utils import send_team_member_message from danswer.danswerbot.slack.handlers.utils import send_team_member_message
from danswer.danswerbot.slack.models import SlackMessageInfo from danswer.danswerbot.slack.models import SlackMessageInfo
from danswer.danswerbot.slack.utils import respond_in_thread from danswer.danswerbot.slack.utils import respond_in_thread
@ -19,13 +24,28 @@ from danswer.db.models import StandardAnswer as StandardAnswerModel
from danswer.server.manage.models import StandardAnswer as PydanticStandardAnswer from danswer.server.manage.models import StandardAnswer as PydanticStandardAnswer
from danswer.utils.logger import DanswerLoggingAdapter from danswer.utils.logger import DanswerLoggingAdapter
from danswer.utils.logger import setup_logger from danswer.utils.logger import setup_logger
from ee.danswer.danswerbot.slack.blocks import build_standard_answer_blocks
from ee.danswer.db.standard_answer import fetch_standard_answer_categories_by_names from ee.danswer.db.standard_answer import fetch_standard_answer_categories_by_names
from ee.danswer.db.standard_answer import find_matching_standard_answers from ee.danswer.db.standard_answer import find_matching_standard_answers
logger = setup_logger() logger = setup_logger()
def build_standard_answer_blocks(
answer_message: str,
) -> list[Block]:
generate_button_block = ButtonElement(
action_id=GENERATE_ANSWER_BUTTON_ACTION_ID,
text="Generate Full Answer",
)
answer_block = SectionBlock(text=answer_message)
return [
answer_block,
ActionsBlock(
elements=[generate_button_block],
),
]
def oneoff_standard_answers( def oneoff_standard_answers(
message: str, message: str,
slack_bot_categories: list[str], slack_bot_categories: list[str],