Skip to content

Commit

Permalink
chore: black formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
@meetbryce
meetbryce committed Aug 23, 2024
1 parent abb6f84 commit 320a678
Show file tree
Hide file tree
Showing 10 changed files with 781 additions and 459 deletions.
198 changes: 124 additions & 74 deletions ossai/handlers.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -26,9 +26,9 @@ def handler_feedback(body):
Handler for the feedback buttons that passes the feedback to Langsmith.
"""
client = Client()
actions_data = body.get('actions')[0]
run_id = actions_data.get('value')
action_id = actions_data.get('action_id')
actions_data = body.get("actions")[0]
run_id = actions_data.get("value")
action_id = actions_data.get("action_id")

score = 0.0

Expand All @@ -44,75 +44,110 @@ def handler_feedback(body):
project_id=os.environ.get("LANGSMITH_PROJECT_ID"),
key="user_feedback",
score=score,
comment=f"Feedback from action: {action_id}"
comment=f"Feedback from action: {action_id}",
)


async def handler_shortcuts(client: WebClient, is_private: bool, payload, say, user_id: str):
channel_id = payload['channel']['id'] if payload['channel']['id'] else payload['channel_id']

async def handler_shortcuts(
client: WebClient, is_private: bool, payload, say, user_id: str
):
channel_id = (
payload["channel"]["id"] if payload["channel"]["id"] else payload["channel_id"]
)
dm_channel_id = await get_direct_message_channel_id(client, user_id)
channel_id_for_say = dm_channel_id if is_private else channel_id
await say(channel=channel_id_for_say, text='...')
await say(channel=channel_id_for_say, text="...")

try:
response = client.conversations_replies(channel=channel_id, ts=payload['message_ts'])
if response['ok']:
messages = response['messages']
original_message = messages[0]['text']
response = client.conversations_replies(
channel=channel_id, ts=payload["message_ts"]
)
if response["ok"]:
messages = response["messages"]
original_message = messages[0]["text"]
workspace_name = get_workspace_name(client)
link = f"https://{workspace_name}.slack.com/archives/{channel_id}/p{payload['message_ts'].replace('.', '')}"

# truncate the message if it's longer than 120 characters &/or it contains \n and append the link
original_message = original_message.split('\n')
thread_hint = original_message[0] if len(original_message) == 1 else f'{original_message[0]}...'
thread_hint = thread_hint if len(thread_hint) <= 120 else thread_hint[:120] + '...'
original_message = original_message.split("\n")
thread_hint = (
original_message[0]
if len(original_message) == 1
else f"{original_message[0]}..."
)
thread_hint = (
thread_hint if len(thread_hint) <= 120 else thread_hint[:120] + "..."
)

title = f'*Summary of <{link}|{"thread" if len(messages) > 1 else "message"}>:*\n>{thread_hint}\n'
user = await get_user_context(client, user_id)
summary, run_id = summarize_slack_messages(client, messages, channel_id, feature_name="summarize_thread", user=user)
text, blocks = get_text_and_blocks_for_say(title=title, run_id=run_id, messages=summary)
summary, run_id = summarize_slack_messages(
client, messages, channel_id, feature_name="summarize_thread", user=user
)
text, blocks = get_text_and_blocks_for_say(
title=title, run_id=run_id, messages=summary
)
return await say(channel=channel_id_for_say, text=text, blocks=blocks)
else:
return await say(channel=channel_id_for_say, text="Sorry, couldn't fetch the message and its replies.")
return await say(
channel=channel_id_for_say,
text="Sorry, couldn't fetch the message and its replies.",
)
except SlackApiError as e:
return await handle_slack_api_error_with_say(client, e, dm_channel_id, say)


async def handler_tldr_slash_command(client: WebClient, ack, payload, say, user_id: str):
async def handler_tldr_slash_command(
client: WebClient, ack, payload, say, user_id: str
):
await ack() # fixme: this seemingly does nothing
text = payload.get("text", None)
channel_name = payload["channel_name"]
channel_id = payload["channel_id"]
dm_channel_id = None

if text == 'public':
await say('...') # hack to get the bot to not show an error message but works fine
if text == "public":
await say(
"..."
) # hack to get the bot to not show an error message but works fine
else:
dm_channel_id = await get_direct_message_channel_id(client, user_id)
await say(channel=dm_channel_id, text='...') # hack to get the bot to not show an error message but works fine
await say(
channel=dm_channel_id, text="..."
) # hack to get the bot to not show an error message but works fine

if text and text != 'public':
if text and text != "public":
return await say("ERROR: Invalid command. Try /tldr or /tldr public.")
try:
history = await get_channel_history(client, channel_id)
history.reverse()
user = await get_user_context(client, user_id)
title = f'*Summary of #{channel_name}* (last {len(history)} messages)\n'
summary, run_id = summarize_slack_messages(client, history, channel_id, feature_name="summarize_channel_messages", user=user)
text, blocks = get_text_and_blocks_for_say(title=title, run_id=run_id, messages=summary)
if text == 'public':
title = f"*Summary of #{channel_name}* (last {len(history)} messages)\n"
summary, run_id = summarize_slack_messages(
client,
history,
channel_id,
feature_name="summarize_channel_messages",
user=user,
)
text, blocks = get_text_and_blocks_for_say(
title=title, run_id=run_id, messages=summary
)
if text == "public":
return await say(text=text, blocks=blocks)
return await say(channel=dm_channel_id, text=text, blocks=blocks)
except SlackApiError as e:
return await handle_slack_api_error_with_say(client, e, dm_channel_id, say)


async def handler_topics_slash_command(client: WebClient, ack, payload, say, user_id: str):
async def handler_topics_slash_command(
client: WebClient, ack, payload, say, user_id: str
):
# START boilerplate
await ack()
channel_id = payload["channel_id"]
dm_channel_id = await get_direct_message_channel_id(client, user_id)
await say(channel=dm_channel_id, text='...')
await say(channel=dm_channel_id, text="...")

try:
history = await get_channel_history(client, channel_id)
Expand All @@ -124,78 +159,93 @@ async def handler_topics_slash_command(client: WebClient, ack, payload, say, use
user = await get_user_context(client, user_id)
is_private, channel_name = get_is_private_and_channel_name(client, channel_id)
# fixme: give the user an error if not enough messages (>=6)
topic_overview, run_id = await analyze_topics_of_history(channel_name, messages, user=user, is_private=is_private)
topic_overview, run_id = await analyze_topics_of_history(
channel_name, messages, user=user, is_private=is_private
)
title = f"*Channel Overview: #{channel_name}*\n\n"
text, blocks = get_text_and_blocks_for_say(title=title, run_id=run_id, messages=[topic_overview])
text, blocks = get_text_and_blocks_for_say(
title=title, run_id=run_id, messages=[topic_overview]
)
return await say(channel=dm_channel_id, text=text, blocks=blocks)


async def handler_tldr_since_slash_command(client: WebClient, payload, say):
title = "Choose your summary timeframe."
dm_channel_id = await get_direct_message_channel_id(client, payload['user_id'])
dm_channel_id = await get_direct_message_channel_id(client, payload["user_id"])

try:
client.chat_postEphemeral(channel=payload['channel_id'], user=payload['user_id'], text=title, blocks=[
{
"type": "actions",
"elements": [
get_since_timeframe_presets(),
{
"type": "datepicker",
"placeholder": {
"type": "plain_text",
"text": "Select a date",
"emoji": True
},
"action_id": "summarize_since"
},
]
}
])
client.chat_postEphemeral(
channel=payload["channel_id"],
user=payload["user_id"],
text=title,
blocks=[
{
"type": "actions",
"elements": [
get_since_timeframe_presets(),
{
"type": "datepicker",
"placeholder": {
"type": "plain_text",
"text": "Select a date",
"emoji": True,
},
"action_id": "summarize_since",
},
],
}
],
)
except SlackApiError as e:
return await handle_slack_api_error_with_say(client, e, dm_channel_id, say)

await say(channel=dm_channel_id, text=f'In #{payload["channel_name"]}, choose a date or timeframe to get your summary')

await say(
channel=dm_channel_id,
text=f'In #{payload["channel_name"]}, choose a date or timeframe to get your summary',
)


async def handler_action_summarize_since_date(client: WebClient, body):
"""
Provide a message summary of the channel since a given date.
"""
channel_name = body['channel']['name']
channel_id = body['channel']['id']
user_id = body['user']['id']
feature_name = body['actions'][0]['action_id']
# todo: make util function for testability
channel_name = body["channel"]["name"]
channel_id = body["channel"]["id"]
user_id = body["user"]["id"]
feature_name = body["actions"][0]["action_id"]

# todo: make util function for testability
# todo: account for the user's timezone :melt:
if feature_name == 'summarize_since_preset':
since_datetime: datetime = datetime.fromtimestamp(int(body['actions'][0]['selected_option']['value'])).date()
if feature_name == "summarize_since_preset":
since_datetime: datetime = datetime.fromtimestamp(
int(body["actions"][0]["selected_option"]["value"])
).date()
else:
since_date = body['actions'][0]['selected_date']
since_datetime: datetime = datetime.strptime(since_date, '%Y-%m-%d').date()
since_date = body["actions"][0]["selected_date"]
since_datetime: datetime = datetime.strptime(since_date, "%Y-%m-%d").date()

dm_channel_id = await get_direct_message_channel_id(client, user_id)
client.chat_postMessage(channel=dm_channel_id, text='...')
client.chat_postMessage(channel=dm_channel_id, text="...")

async with ClientSession() as session:
await session.post(
body['response_url'],
json={"delete_original": "true"}
)

await session.post(body["response_url"], json={"delete_original": "true"})

try:
history = await get_channel_history(client, channel_id, since=since_datetime)
history.reverse()
user = await get_user_context(client, user_id)
summary, run_id = summarize_slack_messages(client, history, channel_id, feature_name=feature_name, user=user)
summary, run_id = summarize_slack_messages(
client, history, channel_id, feature_name=feature_name, user=user
)
text, blocks = get_text_and_blocks_for_say(
title=f'*Summary of #{channel_name}* since {since_datetime.strftime("%A %b %-d, %Y")} ({len(history)} messages)\n',
run_id=run_id,
messages=summary
run_id=run_id,
messages=summary,
)
# todo: somehow add date/preset choice to langsmith metadata
# feature_name: str -> feature: str || Tuple[str, List(Tuple[str, str])]
return client.chat_postMessage(channel=dm_channel_id, text=text, blocks=blocks)
except SlackApiError as e:
return client.chat_postMessage(channel=dm_channel_id, text=f"Encountered an error: {e.response['error']}")
return client.chat_postMessage(
channel=dm_channel_id, text=f"Encountered an error: {e.response['error']}"
)
Loading

0 comments on commit 320a678

Please sign in to comment.