fix format

Signed-off-by: Stephanie <yangcao@redhat.com>

Author: yangcao77

src/app/endpoints/query.py

@@ -363,7 +363,7 @@ async def query_endpoint_handler_base(  # pylint: disable=R0914
                 truncated=False,  # TODO(lucasagomes): implement truncation as part of quota work
                 attachments=query_request.attachments or [],
             )
-    
+
         logger.info("Persisting conversation details...")
         persist_user_conversation_details(
             user_id=user_id,

src/cache/postgres_cache.py

@@ -363,8 +363,7 @@ def insert_or_append(
                     tool_calls_json = json.dumps(tool_calls_as_dicts)
                 except (TypeError, ValueError) as e:
                     logger.warning(
-                        "Failed to serialize tool_calls for "
-                        "conversation %s: %s",
+                        "Failed to serialize tool_calls for " "conversation %s: %s",
                         conversation_id,
                         e,
                     )
@@ -378,8 +377,7 @@ def insert_or_append(
                     tool_results_json = json.dumps(tool_results_as_dicts)
                 except (TypeError, ValueError) as e:
                     logger.warning(
-                        "Failed to serialize tool_results for "
-                        "conversation %s: %s",
+                        "Failed to serialize tool_results for " "conversation %s: %s",
                         conversation_id,
                         e,
                     )

src/cache/sqlite_cache.py

@@ -247,8 +247,7 @@ def get(
                     ]
                 except (json.JSONDecodeError, ValueError) as e:
                     logger.warning(
-                        "Failed to deserialize tool_calls for "
-                        "conversation %s: %s",
+                        "Failed to deserialize tool_calls for " "conversation %s: %s",
                         conversation_id,
                         e,
                     )
@@ -264,8 +263,7 @@ def get(
                     ]
                 except (json.JSONDecodeError, ValueError) as e:
                     logger.warning(
-                        "Failed to deserialize tool_results for "
-                        "conversation %s: %s",
+                        "Failed to deserialize tool_results for " "conversation %s: %s",
                         conversation_id,
                         e,
                     )
@@ -334,8 +332,7 @@ def insert_or_append(
                 tool_calls_json = json.dumps(tool_calls_as_dicts)
             except (TypeError, ValueError) as e:
                 logger.warning(
-                    "Failed to serialize tool_calls for "
-                    "conversation %s: %s",
+                    "Failed to serialize tool_calls for " "conversation %s: %s",
                     conversation_id,
                     e,
                 )
@@ -349,8 +346,7 @@ def insert_or_append(
                 tool_results_json = json.dumps(tool_results_as_dicts)
             except (TypeError, ValueError) as e:
                 logger.warning(
-                    "Failed to serialize tool_results for "
-                    "conversation %s: %s",
+                    "Failed to serialize tool_results for " "conversation %s: %s",
                     conversation_id,
                     e,
                 )

tests/unit/cache/test_postgres_cache.py

@@ -737,11 +737,17 @@ def test_insert_and_get_with_tool_calls_and_results(
 
     # Create tool_calls and tool_results
     tool_calls = [
-        ToolCallSummary(id="call_1", name="test_tool", args={"param": "value"}, type="tool_call")
+        ToolCallSummary(
+            id="call_1", name="test_tool", args={"param": "value"}, type="tool_call"
+        )
     ]
     tool_results = [
         ToolResultSummary(
-            id="call_1", status="success", content="result data", type="tool_result", round=1
+            id="call_1",
+            status="success",
+            content="result data",
+            type="tool_result",
+            round=1,
         )
     ]
     entry_with_tools = CacheEntry(
@@ -770,13 +776,24 @@ def test_insert_and_get_with_tool_calls_and_results(
     # Verify tool_calls JSON
     tool_calls_json = sql_params[-2]
     assert json.loads(tool_calls_json) == [
-        {"id": "call_1", "name": "test_tool", "args": {"param": "value"}, "type": "tool_call"}
+        {
+            "id": "call_1",
+            "name": "test_tool",
+            "args": {"param": "value"},
+            "type": "tool_call",
+        }
     ]
 
     # Verify tool_results JSON
     tool_results_json = sql_params[-1]
     assert json.loads(tool_results_json) == [
-        {"id": "call_1", "status": "success", "content": "result data", "type": "tool_result", "round": 1}
+        {
+            "id": "call_1",
+            "status": "success",
+            "content": "result data",
+            "type": "tool_result",
+            "round": 1,
+        }
     ]
 
     # Simulate the database returning that data
@@ -788,8 +805,23 @@ def test_insert_and_get_with_tool_calls_and_results(
         "start_time",
         "end_time",
         None,  # referenced_documents
-        [{"id": "call_1", "name": "test_tool", "args": {"param": "value"}, "type": "tool_call"}],
-        [{"id": "call_1", "status": "success", "content": "result data", "type": "tool_result", "round": 1}],
+        [
+            {
+                "id": "call_1",
+                "name": "test_tool",
+                "args": {"param": "value"},
+                "type": "tool_call",
+            }
+        ],
+        [
+            {
+                "id": "call_1",
+                "status": "success",
+                "content": "result data",
+                "type": "tool_result",
+                "round": 1,
+            }
+        ],
     )
     mock_cursor.fetchall.return_value = [db_return_value]
 

tests/unit/cache/test_sqlite_cache.py

@@ -484,11 +484,17 @@ def test_insert_and_get_with_tool_calls_and_results(tmpdir: Path) -> None:
 
     # Create tool_calls and tool_results
     tool_calls = [
-        ToolCallSummary(id="call_1", name="test_tool", args={"param": "value"}, type="tool_call")
+        ToolCallSummary(
+            id="call_1", name="test_tool", args={"param": "value"}, type="tool_call"
+        )
     ]
     tool_results = [
         ToolResultSummary(
-            id="call_1", status="success", content="result data", type="tool_result", round=1
+            id="call_1",
+            status="success",
+            content="result data",
+            type="tool_result",
+            round=1,
         )
     ]
     entry_with_tools = CacheEntry(
@@ -531,7 +537,9 @@ def test_insert_and_get_with_all_fields(tmpdir: Path) -> None:
         ReferencedDocument(doc_title="Test Doc", doc_url=AnyUrl("http://example.com"))
     ]
     tool_calls = [
-        ToolCallSummary(id="call_1", name="test_tool", args={"key": "value"}, type="tool_call")
+        ToolCallSummary(
+            id="call_1", name="test_tool", args={"key": "value"}, type="tool_call"
+        )
     ]
     tool_results = [
         ToolResultSummary(