trying to narrow down error

This commit is contained in:
Cyberes 2023-10-02 21:43:36 -06:00
parent 94141b8ecf
commit aed5db4968
1 changed files with 3 additions and 0 deletions

View File

@ -11,6 +11,9 @@ from llm_server.llm import get_token_count
def log_prompt(ip: str, token: str, prompt: str, response: Union[str, None], gen_time: Union[int, float, None], parameters: dict, headers: dict, backend_response_code: int, request_url: str, backend_url: str, response_tokens: int = None, is_error: bool = False):
assert isinstance(prompt, str)
assert isinstance(backend_url, str)
def background_task():
nonlocal ip, token, prompt, response, gen_time, parameters, headers, backend_response_code, request_url, backend_url, response_tokens, is_error
# Try not to shove JSON into the database.