diff --git a/llm_server/database/log_to_db.py b/llm_server/database/log_to_db.py index fa97ad7..e971747 100644 --- a/llm_server/database/log_to_db.py +++ b/llm_server/database/log_to_db.py @@ -24,4 +24,5 @@ def log_to_db(ip: str, token: str, prompt: str, response: Union[str, None], gen_ 'is_error': is_error } } + print(data) r.publish('database-logger', pickle.dumps(data))