update tests: LLM mock to return LLM TITLE for all cases

This commit is contained in:
Gokul Mohanarangan
2023-09-25 10:22:41 +05:30
parent 3a374ea593
commit 0b00881ce4
2 changed files with 3 additions and 3 deletions

View File

@@ -30,7 +30,7 @@ def dummy_processors():
"reflector.processors.transcript_final_short_summary.TranscriptFinalShortSummaryProcessor.get_short_summary"
) as mock_short_summary:
mock_topic.return_value = {"title": "LLM TITLE", "summary": "LLM SUMMARY"}
mock_title.return_value = {"title": "LLM FINAL TITLE"}
mock_title.return_value = {"title": "LLM TITLE"}
mock_long_summary.return_value = "LLM LONG SUMMARY"
mock_short_summary.return_value = {"short_summary": "LLM SHORT SUMMARY"}