Skip to content

Commit a5274f0

Browse files
author
Radovan Fuchs
committed
fix black
1 parent 1730b65 commit a5274f0

2 files changed

Lines changed: 7 additions & 5 deletions

File tree

src/utils/endpoints.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff 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")

tests/unit/utils/test_endpoints.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)