From a06f22aeaa34c1382bd33b46627910d5d48de26d Mon Sep 17 00:00:00 2001
From: Sotiris Dragonas <36576941+BazookaMusic@users.noreply.github.com>
Date: Mon, 29 Jun 2026 10:40:04 +0300
Subject: [PATCH] Python: apply codeql query format to prompt-injection
 framework files

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
---
 python/ql/lib/semmle/python/frameworks/OpenAI.qll     | 6 +++++-
 python/ql/lib/semmle/python/frameworks/OpenRouter.qll | 6 +++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/python/ql/lib/semmle/python/frameworks/OpenAI.qll b/python/ql/lib/semmle/python/frameworks/OpenAI.qll
index b56c868ba8f..3909d591b3b 100644
--- a/python/ql/lib/semmle/python/frameworks/OpenAI.qll
+++ b/python/ql/lib/semmle/python/frameworks/OpenAI.qll
@@ -93,7 +93,11 @@ module OpenAI {
   /** Gets the message dictionaries passed as a list to `responses.create`. */
   private API::Node responsesMessage() {
     result =
-      classRef().getMember("responses").getMember("create").getKeywordParameter("input").getASubscript()
+      classRef()
+          .getMember("responses")
+          .getMember("create")
+          .getKeywordParameter("input")
+          .getASubscript()
   }
 
   /** Gets the content sink of a message dictionary, including the `text` of structured content. */
diff --git a/python/ql/lib/semmle/python/frameworks/OpenRouter.qll b/python/ql/lib/semmle/python/frameworks/OpenRouter.qll
index 3b43770ec1c..bc795ef5b15 100644
--- a/python/ql/lib/semmle/python/frameworks/OpenRouter.qll
+++ b/python/ql/lib/semmle/python/frameworks/OpenRouter.qll
@@ -25,7 +25,11 @@ module OpenRouter {
   /** Gets the message dictionaries passed to `chat.send`. */
   private API::Node chatMessage() {
     result =
-      clientRef().getMember("chat").getMember("send").getKeywordParameter("messages").getASubscript()
+      clientRef()
+          .getMember("chat")
+          .getMember("send")
+          .getKeywordParameter("messages")
+          .getASubscript()
   }
 
   /** Gets the content sink of a message dictionary, including the `text` of structured content. */