server/rtc: fix topic output

This commit is contained in:
Mathieu Virbel
2023-08-01 19:12:51 +02:00
parent 99c9ba3e6b
commit d320558cc9
4 changed files with 57 additions and 13 deletions

View File

@@ -21,9 +21,10 @@ class TranscriptTopicDetectorProcessor(Processor):
{input_text}
### Assistant:
"""
def __init__(self, min_transcript_length=25, **kwargs):
def __init__(self, min_transcript_length=100, **kwargs):
super().__init__(**kwargs)
self.transcript = None
self.min_transcript_length = min_transcript_length
@@ -43,6 +44,12 @@ class TranscriptTopicDetectorProcessor(Processor):
return
prompt = self.PROMPT.format(input_text=self.transcript.text)
result = await self.llm.generate(prompt=prompt)
summary = TitleSummary(title=result["title"], summary=result["summary"])
summary = TitleSummary(
title=result["title"],
summary=result["summary"],
timestamp=self.transcript.timestamp,
duration=self.transcript.duration,
transcript=self.transcript,
)
self.transcript = None
await self.emit(summary)

View File

@@ -21,12 +21,6 @@ class Word:
end: float
@dataclass
class TitleSummary:
title: str
summary: str
@dataclass
class Transcript:
text: str = ""
@@ -63,3 +57,12 @@ class Transcript:
Word(text=word.text, start=word.start, end=word.end) for word in self.words
]
return Transcript(text=self.text, words=words)
@dataclass
class TitleSummary:
title: str
summary: str
timestamp: float
duration: float
transcript: Transcript