File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -680,9 +680,7 @@ async def cleanup_after_streaming(
680680 query_request .query , client , llama_stack_model_id
681681 )
682682 else :
683- logger .debug (
684- "Topic summary generation disabled by request parameter"
685- )
683+ logger .debug ("Topic summary generation disabled by request parameter" )
686684 topic_summary = None
687685
688686 completed_at = datetime .now (UTC ).strftime ("%Y-%m-%dT%H:%M:%SZ" )
Original file line number Diff line number Diff line change @@ -1056,7 +1056,9 @@ async def test_cleanup_after_streaming_generate_topic_summary_default_true(
10561056 mock_session .__exit__ = mocker .Mock (return_value = None )
10571057 mocker .patch ("utils.endpoints.get_session" , return_value = mock_session )
10581058
1059- mocker .patch ("utils.endpoints.create_referenced_documents_with_metadata" , return_value = [])
1059+ mocker .patch (
1060+ "utils.endpoints.create_referenced_documents_with_metadata" , return_value = []
1061+ )
10601062 mocker .patch ("utils.endpoints.store_conversation_into_cache" )
10611063
10621064 query_request = QueryRequest (query = "test query" )
@@ -1106,7 +1108,9 @@ async def test_cleanup_after_streaming_generate_topic_summary_explicit_false(
11061108 mock_session .__exit__ = mocker .Mock (return_value = None )
11071109 mocker .patch ("utils.endpoints.get_session" , return_value = mock_session )
11081110
1109- mocker .patch ("utils.endpoints.create_referenced_documents_with_metadata" , return_value = [])
1111+ mocker .patch (
1112+ "utils.endpoints.create_referenced_documents_with_metadata" , return_value = []
1113+ )
11101114 mocker .patch ("utils.endpoints.store_conversation_into_cache" )
11111115
11121116 query_request = QueryRequest (query = "test query" , generate_topic_summary = False )
You can’t perform that action at this time.
0 commit comments