diff --git a/llm_server/database/log_to_db.py b/llm_server/database/log_to_db.py index e971747..76712ff 100644 --- a/llm_server/database/log_to_db.py +++ b/llm_server/database/log_to_db.py @@ -24,5 +24,6 @@ def log_to_db(ip: str, token: str, prompt: str, response: Union[str, None], gen_ 'is_error': is_error } } - print(data) + for k, v in data['kwargs'].items(): + print(k, type(v)) r.publish('database-logger', pickle.dumps(data))