mirror of
https://github.com/Monadical-SAS/reflector.git
synced 2025-12-21 04:39:06 +00:00
Mathieu Virbel
28ac031ff6
* feat: use llamaindex for transcript final title too * refactor: removed llm backend, replaced with one single class+llamaindex * refactor: self-review * fix: typing * fix: tests * refactor: extract clean_title and add tests * test: fix * test: remove ensure_casing/nltk * fix: tiny mistake
73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
from textwrap import dedent
|
|
|
|
from reflector.llm import LLM
|
|
from reflector.processors.base import Processor
|
|
from reflector.processors.types import FinalTitle, TitleSummary
|
|
from reflector.settings import settings
|
|
from reflector.utils.text import clean_title
|
|
|
|
TITLE_PROMPT = dedent(
|
|
"""
|
|
Generate a concise title for this meeting based on the following topic titles.
|
|
Ignore casual conversation, greetings, or administrative matters.
|
|
|
|
The title must:
|
|
- Be maximum 10 words
|
|
- Use noun phrases when possible (e.g., "Q1 Budget Review" not "Reviewing the Q1 Budget")
|
|
- Avoid generic terms like "Team Meeting" or "Discussion"
|
|
|
|
If multiple unrelated topics were discussed, prioritize the most significant one.
|
|
or create a compound title (e.g., "Product Launch and Budget Planning").
|
|
|
|
<topics_discussed>
|
|
{titles}
|
|
</topics_discussed>
|
|
|
|
Do not explain, just output the meeting title as a single line.
|
|
"""
|
|
).strip()
|
|
|
|
|
|
class TranscriptFinalTitleProcessor(Processor):
|
|
"""
|
|
Generate a final title from topic titles using LlamaIndex
|
|
"""
|
|
|
|
INPUT_TYPE = TitleSummary
|
|
OUTPUT_TYPE = FinalTitle
|
|
|
|
def __init__(self, **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.chunks: list[TitleSummary] = []
|
|
self.llm = LLM(settings=settings, temperature=0.5, max_tokens=200)
|
|
|
|
async def _push(self, data: TitleSummary):
|
|
self.chunks.append(data)
|
|
|
|
async def get_title(self, accumulated_titles: str) -> str:
|
|
"""
|
|
Generate a title for the whole recording using LLM
|
|
"""
|
|
prompt = TITLE_PROMPT.format(titles=accumulated_titles)
|
|
response = await self.llm.get_response(
|
|
prompt,
|
|
[accumulated_titles],
|
|
tone_name="Title generator",
|
|
)
|
|
|
|
self.logger.info(f"Generated title response: {response}")
|
|
|
|
return response
|
|
|
|
async def _flush(self):
|
|
if not self.chunks:
|
|
self.logger.warning("No summary to output")
|
|
return
|
|
|
|
accumulated_titles = "\n".join([f"- {chunk.title}" for chunk in self.chunks])
|
|
title = await self.get_title(accumulated_titles)
|
|
title = clean_title(title)
|
|
|
|
final_title = FinalTitle(title=title)
|
|
await self.emit(final_title)
|