update casing and trimming

This commit is contained in:
Gokul Mohanarangan
2023-09-22 07:29:01 +05:30
parent ab41ce90e8
commit 009d52ea23
3 changed files with 9 additions and 8 deletions

View File

@@ -60,8 +60,8 @@ class TranscriptFinalTitleProcessor(Processor):
accumulated_titles = ".".join([chunk.title for chunk in self.chunks])
title_result = await self.get_title(accumulated_titles)
final_title = self.llm.ensure_casing(title_result["title"])
final_title = self.llm.trim_title(final_title)
final_title = self.llm.trim_title(title_result["title"])
final_title = self.llm.ensure_casing(final_title)
final_title = FinalTitle(title=final_title)
await self.emit(final_title)

View File

@@ -54,9 +54,8 @@ class TranscriptTopicDetectorProcessor(Processor):
text = self.transcript.text
self.logger.info(f"Topic detector got {len(text)} length transcript")
topic_result = await self.get_topic(text=text)
title = self.llm.ensure_casing(topic_result["title"])
title = self.llm.trim_title(title)
title = self.llm.trim_title(topic_result["title"])
title = self.llm.ensure_casing(title)
summary = TitleSummary(
title=title,