diff --git a/custom_routes.py b/custom_routes.py index e04317f..8cf7e81 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -334,8 +334,12 @@ async def post_prompt(json_data): extra_data["client_id"] = json_data["client_id"] if valid[0]: outputs_to_execute = valid[2] + sensitive = {} + for sensitive_val in execution.SENSITIVE_EXTRA_DATA_KEYS: + if sensitive_val in extra_data: + sensitive[sensitive_val] = extra_data.pop(sensitive_val) prompt_server.prompt_queue.put( - (number, prompt_id, prompt, extra_data, outputs_to_execute) + (number, prompt_id, prompt, extra_data, outputs_to_execute, sensitive) ) response = { "prompt_id": prompt_id,