refactor: add sensitive to prompt_queue

2025-11-08 02:50:27 +08:00
parent a4e3bc2571
commit 0ff751b60f
1 changed files with 5 additions and 1 deletions
@@ -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,