lint

Signed-off-by: Anxhela Coba <acoba@redhat.com>

Author: Anxhela21

tests/unit/app/endpoints/test_query.py

@@ -212,7 +212,9 @@ async def test_query_merges_inline_and_tool_rag_chunks_and_documents(
         )
 
         inline_chunk = RAGChunk(content="inline chunk content", source="byok")
-        inline_doc = ReferencedDocument(doc_title="Inline Doc", document_id="inline_doc_1")
+        inline_doc = ReferencedDocument(
+            doc_title="Inline Doc", document_id="inline_doc_1"
+        )
         inline_rag = RAGContext(
             context_text="",
             rag_chunks=[inline_chunk],

tests/unit/cache/test_postgres_cache.py

@@ -637,7 +637,11 @@ def test_insert_and_get_with_referenced_documents(
 
     # Create a CacheEntry with referenced documents
     docs = [
-        ReferencedDocument(doc_title="Test Doc", doc_url=AnyUrl("http://example.com/"), document_id="test_doc_postgres_1")
+        ReferencedDocument(
+            doc_title="Test Doc",
+            doc_url=AnyUrl("http://example.com/"),
+            document_id="test_doc_postgres_1",
+        )
     ]
     entry_with_docs = CacheEntry(
         query="user message",
@@ -664,7 +668,12 @@ def test_insert_and_get_with_referenced_documents(
     inserted_json_str = sql_params[-3]
 
     assert json.loads(inserted_json_str) == [
-        {"doc_url": "http://example.com/", "doc_title": "Test Doc", "source": None, "document_id": "test_doc_postgres_1"}
+        {
+            "doc_url": "http://example.com/",
+            "doc_title": "Test Doc",
+            "source": None,
+            "document_id": "test_doc_postgres_1",
+        }
     ]
 
     # Simulate the database returning that data
@@ -675,7 +684,13 @@ def test_insert_and_get_with_referenced_documents(
         "bar",
         "start_time",
         "end_time",
-        [{"doc_url": "http://example.com/", "doc_title": "Test Doc", "document_id": "test_doc_postgres_1"}],
+        [
+            {
+                "doc_url": "http://example.com/",
+                "doc_title": "Test Doc",
+                "document_id": "test_doc_postgres_1",
+            }
+        ],
         None,  # tool_calls
         None,  # tool_results
     )

tests/unit/cache/test_sqlite_cache.py

@@ -467,7 +467,11 @@ def test_insert_and_get_with_referenced_documents(tmpdir: Path) -> None:
 
     # Create a CacheEntry with referenced documents
     docs = [
-        ReferencedDocument(doc_title="Test Doc", doc_url=AnyUrl("http://example.com"), document_id="test_doc_cache_1")
+        ReferencedDocument(
+            doc_title="Test Doc",
+            doc_url=AnyUrl("http://example.com"),
+            document_id="test_doc_cache_1",
+        )
     ]
     entry_with_docs = CacheEntry(
         query="user message",
@@ -571,7 +575,11 @@ def test_insert_and_get_with_all_fields(tmpdir: Path) -> None:
 
     # Create all fields
     docs = [
-        ReferencedDocument(doc_title="Test Doc", doc_url=AnyUrl("http://example.com"), document_id="test_doc_cache_1")
+        ReferencedDocument(
+            doc_title="Test Doc",
+            doc_url=AnyUrl("http://example.com"),
+            document_id="test_doc_cache_1",
+        )
     ]
     tool_calls = [
         ToolCallSummary(

tests/unit/models/responses/test_successful_responses.py

@@ -321,7 +321,11 @@ def test_constructor_full(self) -> None:
             )
         ]
         referenced_docs = [
-            ReferencedDocument(doc_url=AnyUrl("https://example.com"), doc_title="Doc", document_id="test_doc_1")
+            ReferencedDocument(
+                doc_url=AnyUrl("https://example.com"),
+                doc_title="Doc",
+                document_id="test_doc_1",
+            )
         ]
 
         response = QueryResponse(  # type: ignore[call-arg]

tests/unit/utils/test_responses.py

@@ -2130,7 +2130,11 @@ def test_parse_referenced_documents_file_search_call(
         }
 
         mock_result2 = {
-            "attributes": {"url": "https://example.com/doc2", "title": "Document 2", "doc_id": "doc_2"},
+            "attributes": {
+                "url": "https://example.com/doc2",
+                "title": "Document 2",
+                "doc_id": "doc_2",
+            },
         }
 
         mock_output_item = mocker.Mock()