From 0ff751b60fdd11ddd0e465689f89b0ba38d499cf Mon Sep 17 00:00:00 2001 From: KarrixLee Date: Sat, 8 Nov 2025 02:50:27 +0800 Subject: [PATCH] refactor: add sensitive to prompt_queue --- custom_routes.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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,