PR review comments

This commit is contained in:
Gokul Mohanarangan
2023-08-17 14:42:45 +05:30
parent 235ee73f46
commit a98a9853be
9 changed files with 17 additions and 20 deletions

View File

@@ -9,16 +9,13 @@ async def test_basic_process(event_loop):
from reflector.settings import settings
from reflector.llm.base import LLM
from pathlib import Path
from typing import Union
# use an LLM test backend
settings.LLM_BACKEND = "test"
settings.TRANSCRIPT_BACKEND = "whisper"
class LLMTest(LLM):
async def _generate(
self, prompt: str, schema: Union[str | None], **kwargs
) -> str:
async def _generate(self, prompt: str, schema: str | None, **kwargs) -> str:
return {
"title": "TITLE",
"summary": "SUMMARY",