Fix gate issues

Signed-off-by: Lucas Alvares Gomes <lucasagomes@gmail.com>

Author: umago

scripts/query_rag.py

@@ -42,8 +42,8 @@ def _llama_index_query(args: argparse.Namespace) -> None:
             "node": {
                 "id": node.node_id,
                 "text": node.text,
-                "metadata": node.metadata if hasattr(node, 'metadata') else {}
-            }
+                "metadata": node.metadata if hasattr(node, "metadata") else {},
+            },
         }
         if args.json:
             print(json.dumps(result, indent=2))
@@ -60,7 +60,7 @@ def _llama_index_query(args: argparse.Namespace) -> None:
                     "query": args.query,
                     "top_k": args.top_k,
                     "threshold": args.threshold,
-                    "nodes": []
+                    "nodes": [],
                 }
                 print(json.dumps(result, indent=2))
             exit(1)
@@ -75,7 +75,7 @@ def _llama_index_query(args: argparse.Namespace) -> None:
                     "query": args.query,
                     "top_k": args.top_k,
                     "threshold": args.threshold,
-                    "nodes": []
+                    "nodes": [],
                 }
                 print(json.dumps(result, indent=2))
             exit(1)
@@ -85,14 +85,14 @@ def _llama_index_query(args: argparse.Namespace) -> None:
             "query": args.query,
             "top_k": args.top_k,
             "threshold": args.threshold,
-            "nodes": []
+            "nodes": [],
         }
-        for node in nodes:
+        for node in nodes:  # type: ignore
             node_data = {
                 "id": node.node_id,
                 "score": node.score,
                 "text": node.text,
-                "metadata": node.metadata if hasattr(node, 'metadata') else {}
+                "metadata": node.metadata if hasattr(node, "metadata") else {},
             }
             result["nodes"].append(node_data)
 
@@ -169,7 +169,7 @@ def _llama_stack_query(args: argparse.Namespace) -> None:
                 "query": args.query,
                 "top_k": args.top_k,
                 "threshold": args.threshold,
-                "nodes": []
+                "nodes": [],
             }
             print(json.dumps(result, indent=2))
         exit(1)
@@ -185,7 +185,7 @@ def _llama_stack_query(args: argparse.Namespace) -> None:
                 "query": args.query,
                 "top_k": args.top_k,
                 "threshold": args.threshold,
-                "nodes": []
+                "nodes": [],
             }
             print(json.dumps(result, indent=2))
         exit(1)
@@ -195,15 +195,15 @@ def _llama_stack_query(args: argparse.Namespace) -> None:
         "query": args.query,
         "top_k": args.top_k,
         "threshold": args.threshold,
-        "nodes": []
+        "nodes": [],
     }
 
     for _id, chunk, score in zip(md["document_ids"], md["chunks"], md["scores"]):
         node_data = {
             "id": _id,
-            "score": score.score if hasattr(score, 'score') else score,
+            "score": score.score if hasattr(score, "score") else score,
             "text": chunk,
-            "metadata": {}
+            "metadata": {},
         }
         result["nodes"].append(node_data)
 
@@ -221,6 +221,7 @@ def _llama_stack_query(args: argparse.Namespace) -> None:
     #     else:
     #         print(content)
 
+
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(
         description="Utility script for querying RAG database"
@@ -263,15 +264,12 @@ def _llama_stack_query(args: argparse.Namespace) -> None:
         # In JSON mode, only show ERROR or higher to avoid polluting JSON output
         logging.basicConfig(
             level=logging.ERROR,
-            format='%(levelname)s: %(message)s',
-            stream=sys.stderr  # Send logs to stderr to keep stdout clean for JSON
+            format="%(levelname)s: %(message)s",
+            stream=sys.stderr,  # Send logs to stderr to keep stdout clean for JSON
         )
     else:
         # In normal mode, show info and above
-        logging.basicConfig(
-            level=logging.INFO,
-            format='%(message)s'
-        )
+        logging.basicConfig(level=logging.INFO, format="%(message)s")
 
     if not args.json:
         logging.info("Command line used: " + " ".join(sys.argv))