diff --git a/surfsense_backend/app/services/connector_service.py b/surfsense_backend/app/services/connector_service.py index 251241e96..3bd9a4421 100644 --- a/surfsense_backend/app/services/connector_service.py +++ b/surfsense_backend/app/services/connector_service.py @@ -1303,10 +1303,9 @@ class ConnectorService: sources_list = self._build_chunk_sources_from_documents( github_docs, - description_fn=lambda chunk, _doc_info, metadata: metadata.get( - "description" - ) - or chunk.get("content", ""), + description_fn=lambda chunk, _doc_info, metadata: ( + metadata.get("description") or chunk.get("content", "") + ), url_fn=lambda _doc_info, metadata: metadata.get("url", "") or "", ) diff --git a/surfsense_backend/tests/e2e/test_page_limits.py b/surfsense_backend/tests/e2e/test_page_limits.py index bed59d952..6d54bcb09 100644 --- a/surfsense_backend/tests/e2e/test_page_limits.py +++ b/surfsense_backend/tests/e2e/test_page_limits.py @@ -175,7 +175,10 @@ class TestPageLimitNotification: ) latest = notifications[0] - assert "page limit" in latest["title"].lower() or "page limit" in latest["message"].lower(), ( + assert ( + "page limit" in latest["title"].lower() + or "page limit" in latest["message"].lower() + ), ( f"Notification should mention page limit: title={latest['title']!r}, " f"message={latest['message']!r}" ) @@ -217,7 +220,11 @@ class TestDocumentProcessingNotification: type_filter="document_processing", search_space_id=search_space_id, ) - completed = [n for n in notifications if n.get("metadata", {}).get("processing_stage") == "completed"] + completed = [ + n + for n in notifications + if n.get("metadata", {}).get("processing_stage") == "completed" + ] assert len(completed) >= 1, ( "Expected at least one document_processing notification with 'completed' stage" ) @@ -257,9 +264,7 @@ class TestPagesUnchangedOnProcessingFailure: assert statuses[did]["status"]["state"] == "failed" used, _ = await page_limits.get() - assert used == 10, ( - f"pages_used should remain 10 after ETL failure, got {used}" - ) + assert used == 10, f"pages_used should remain 10 after ETL failure, got {used}" # --------------------------------------------------------------------------- diff --git a/surfsense_backend/tests/e2e/test_upload_limits.py b/surfsense_backend/tests/e2e/test_upload_limits.py index c37bd4c6a..b81e701e6 100644 --- a/surfsense_backend/tests/e2e/test_upload_limits.py +++ b/surfsense_backend/tests/e2e/test_upload_limits.py @@ -130,7 +130,10 @@ class TestTotalSizeLimit: ): chunk_size = 45 * 1024 * 1024 # 45 MB each files = [ - ("files", (f"chunk_{i}.txt", io.BytesIO(b"\x00" * chunk_size), "text/plain")) + ( + "files", + (f"chunk_{i}.txt", io.BytesIO(b"\x00" * chunk_size), "text/plain"), + ) for i in range(5) # 5 x 45 MB = 225 MB > 200 MB ] resp = await client.post(