diff --git a/skynet/modules/ttt/summaries/processor.py b/skynet/modules/ttt/summaries/processor.py index 861128d..a4b7d75 100644 --- a/skynet/modules/ttt/summaries/processor.py +++ b/skynet/modules/ttt/summaries/processor.py @@ -7,8 +7,13 @@ from skynet.env import app_uuid, azure_openai_api_version, llama_n_ctx, llama_path, openai_api_base_url from skynet.logs import get_logger -from .prompts.action_items import action_items_conversation, action_items_emails, action_items_text -from .prompts.summary import summary_conversation, summary_emails, summary_text +from .prompts.action_items import ( + action_items_conversation, + action_items_emails, + action_items_meeting, + action_items_text, +) +from .prompts.summary import summary_conversation, summary_emails, summary_meeting, summary_text from .v1.models import DocumentPayload, HintType, JobType llm = None @@ -19,11 +24,13 @@ JobType.SUMMARY: { HintType.CONVERSATION: summary_conversation, HintType.EMAILS: summary_emails, + HintType.MEETING: summary_meeting, HintType.TEXT: summary_text, }, JobType.ACTION_ITEMS: { HintType.CONVERSATION: action_items_conversation, HintType.EMAILS: action_items_emails, + HintType.MEETING: action_items_meeting, HintType.TEXT: action_items_text, }, } diff --git a/skynet/modules/ttt/summaries/prompts/action_items.py b/skynet/modules/ttt/summaries/prompts/action_items.py index db52ab9..f7a449b 100644 --- a/skynet/modules/ttt/summaries/prompts/action_items.py +++ b/skynet/modules/ttt/summaries/prompts/action_items.py @@ -18,6 +18,16 @@ If there are no action items, respond just with "No action items", else start your response with "Response:", followed by the list of action items. """ +action_items_meeting = """ + You are an AI assistant that will be provided with a text transcript of a meeting. You will extract a short list of specific, unique action items from that transcript. + An action item can be defined when someone commits to doing something in the future, or when someone charges someone one else to do something in the future. + + Example 1: If Andrew says "I will send you the report by tomorrow", then the action item would be "- Andrew will send the report by tomorrow". + Example 2: If George says "As an action item for Michael, don't forget to send the report by tomorrow", then the action item would be "- Michael will send the report by tomorrow". + + If there are no action items, respond just with "No action items", else start your response with "Response:", followed by the list of action items. +""" + action_items_text = """ You are an AI assistant that will be provided with a text transcript. You will extract a short list of specific, unique action items from that transcript. An action item can be defined when someone commits to doing something in the future, or when someone charges someone one else to do something in the future. diff --git a/skynet/modules/ttt/summaries/prompts/summary.py b/skynet/modules/ttt/summaries/prompts/summary.py index 61e93c5..4c69cdb 100644 --- a/skynet/modules/ttt/summaries/prompts/summary.py +++ b/skynet/modules/ttt/summaries/prompts/summary.py @@ -10,6 +10,12 @@ Start your response with "Response:". """ +summary_meeting = """ + You are an AI assistant that will be provided a text transcript of a meeting. You will extract a summary of that transcript. + The response should be plain text, without the use of any formatting like bullet points, numbering, or asterisks. + Start your response with "Response:". +""" + summary_text = """ You are an AI assistant that will be provided a text transcript of a meeting. You will extract a summary of that transcript. The response should be plain text, without the use of any formatting like bullet points, numbering, or asterisks. diff --git a/skynet/modules/ttt/summaries/v1/models.py b/skynet/modules/ttt/summaries/v1/models.py index ff74852..a9316d7 100644 --- a/skynet/modules/ttt/summaries/v1/models.py +++ b/skynet/modules/ttt/summaries/v1/models.py @@ -7,6 +7,7 @@ class HintType(Enum): CONVERSATION = 'conversation' EMAILS = 'emails' + MEETING = 'meeting' TEXT = 'text' @@ -17,7 +18,7 @@ class Priority(Enum): class DocumentPayload(BaseModel): text: str - hint: HintType = HintType.TEXT + hint: HintType = HintType.MEETING priority: Priority = Priority.NORMAL prompt: str | None = None