ruff

Author: KennyZhang1

backend/graphrag_app/api/data.py

@@ -97,7 +97,9 @@ async def upload_file_async(
 
                 # clean the output and upload to blob storage
                 cleaned_result = clean_output(result.text_content)
-                await converted_blob_client.upload_blob(cleaned_result, overwrite=overwrite)
+                await converted_blob_client.upload_blob(
+                    cleaned_result, overwrite=overwrite
+                )
 
                 # update the file cache
                 await update_cache(filename, file_stream, container_client)

backend/graphrag_app/utils/common.py

@@ -193,7 +193,7 @@ def desanitize_name(sanitized_container_name: str) -> str | None:
         raise HTTPException(
             status_code=500, detail="Error retrieving original container name."
         )
-    
+
 
 async def create_cache(container_client: ContainerClient) -> None:
     """
@@ -203,9 +203,7 @@ async def create_cache(container_client: ContainerClient) -> None:
         cache_blob_client = container_client.get_blob_client("uploaded_files_cache.csv")
         if not await cache_blob_client.exists():
             # create the empty file cache csv
-            headers = [
-                ['Filename', 'Hash']
-            ]
+            headers = [["Filename", "Hash"]]
             with open("uploaded_files_cache.csv", "w", newline="") as f:
                 writer = csv.writer(f, delimiter=",")
                 writer.writerows(headers)
@@ -247,7 +245,9 @@ async def check_cache(file_stream: BinaryIO, container_client: ContainerClient)
         )
 
 
-async def update_cache(filename: str, file_stream: BinaryIO, container_client: ContainerClient) -> None:
+async def update_cache(
+    filename: str, file_stream: BinaryIO, container_client: ContainerClient
+) -> None:
     """
     Update the file cache with the new file by appending a new row to the cache.
     """
@@ -274,9 +274,11 @@ async def update_cache(filename: str, file_stream: BinaryIO, container_client: C
 
         # Upload the updated cache to Azure Blob Storage
         updated_cache_content.seek(0)
-        await cache_blob_client.upload_blob(updated_cache_content.getvalue().encode("utf-8"), overwrite=True)
+        await cache_blob_client.upload_blob(
+            updated_cache_content.getvalue().encode("utf-8"), overwrite=True
+        )
     except Exception:
         raise HTTPException(
             status_code=500,
             detail="Error updating file cache in Azure Blob Storage.",
-        )
+        )

notebooks/2-Advanced_Getting_Started.ipynb

@@ -290,7 +290,10 @@
     "    return requests.post(\n",
     "        url,\n",
     "        files=prompts if len(prompts) > 0 else None,\n",
-    "        params={\"index_container_name\": index_name, \"storage_container_name\": storage_name},\n",
+    "        params={\n",
+    "            \"index_container_name\": index_name,\n",
+    "            \"storage_container_name\": storage_name,\n",
+    "        },\n",
     "        headers=headers,\n",
     "    )\n",
     "\n",
@@ -619,7 +622,9 @@
     "    community_summarization_prompt = prompts[\"community_summarization_prompt\"]\n",
     "    summarize_description_prompt = prompts[\"entity_summarization_prompt\"]\n",
     "else:\n",
-    "    entity_extraction_prompt = community_summarization_prompt = summarize_description_prompt = None\n",
+    "    entity_extraction_prompt = community_summarization_prompt = (\n",
+    "        summarize_description_prompt\n",
+    "    ) = None\n",
     "\n",
     "response = build_index(\n",
     "    storage_name=storage_name,\n",