selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-21 04:39:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

PR review comments

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

5
server/reflector/llm/base.py
View File

@@ -2,7 +2,6 @@ import importlib
import json
import re
from time import monotonic
from typing import Union
from reflector.logger import logger as reflector_logger
from reflector.settings import settings
@@ -47,7 +46,7 @@ class LLM:
pass
async def generate(
self, prompt: str, logger: reflector_logger, schema: str = None, **kwargs
self, prompt: str, logger: reflector_logger, schema: str | None = None, **kwargs
) -> dict:
logger.info("LLM generate", prompt=repr(prompt))
try:
@@ -63,7 +62,7 @@ class LLM:
return result
async def _generate(self, prompt: str, schema: Union[str | None], **kwargs) -> str:
async def _generate(self, prompt: str, schema: str | None, **kwargs) -> str:
raise NotImplementedError
def _parse_json(self, result: str) -> dict: