diff --git a/llm_server/database/log_to_db.py b/llm_server/database/log_to_db.py index 76712ff..8e27055 100644 --- a/llm_server/database/log_to_db.py +++ b/llm_server/database/log_to_db.py @@ -16,7 +16,7 @@ def log_to_db(ip: str, token: str, prompt: str, response: Union[str, None], gen_ 'response': response, 'gen_time': gen_time, 'parameters': parameters, - 'headers': headers, + 'headers': dict(headers) if headers else headers, 'backend_response_code': backend_response_code, 'request_url': request_url, 'backend_url': backend_url,