ran linter

Author: Gabriel Nieves

backend/graphrag_app/api/prompt_tuning.py

@@ -6,7 +6,6 @@
 from pathlib import Path
 
 import graphrag.api as api
-import yaml
 from fastapi import (
     APIRouter,
     Depends,

backend/graphrag_app/api/query.py

@@ -3,7 +3,6 @@
 
 import os
 import traceback
-from pathlib import Path
 
 from fastapi import (
     APIRouter,
@@ -17,7 +16,6 @@
 
 from graphrag_app.logger.load_logger import load_pipeline_logger
 from graphrag_app.typing.models import (
-    GraphDriftRequest,
     GraphGlobalRequest,
     GraphRequest,
     GraphResponse,

backend/graphrag_app/api/query_streaming.py

@@ -4,7 +4,6 @@
 import json
 import os
 import traceback
-from typing import Dict, List
 
 from fastapi import (
     APIRouter,

backend/graphrag_app/typing/models.py

@@ -4,7 +4,6 @@
 from io import StringIO
 from typing import (
     Any,
-    Dict,
     List,
     Optional,
 )

backend/graphrag_app/utils/azure_clients.py

@@ -2,6 +2,7 @@
 # Licensed under the MIT License.
 
 import os
+from pathlib import PurePosixPath
 
 from azure.cosmos import (
     ContainerProxy,
@@ -11,7 +12,6 @@
 from azure.identity import DefaultAzureCredential
 from azure.storage.blob import BlobServiceClient
 from azure.storage.blob.aio import BlobServiceClient as BlobServiceClientAsync
-from pathlib import PurePosixPath
 
 ENDPOINT_ERROR_MSG = "Could not find connection string in environment variables"
 
@@ -133,7 +133,6 @@ def __init__(self) -> None:
         else:
             self.storage_account_name = self.storage_account_hostname.split(".")[0]
 
-
     def get_blob_service_client(self) -> BlobServiceClient:
         """
         Returns the blob service client.

backend/graphrag_app/utils/common.py

@@ -9,20 +9,18 @@
 import os
 import sys
 import traceback
-from pathlib import Path
-from typing import Annotated, Dict, List
 from io import StringIO
-from typing import Annotated, Tuple
+from pathlib import Path
+from typing import Annotated, Dict, Tuple
 
 import pandas as pd
 from azure.core.exceptions import ResourceNotFoundError
 from azure.cosmos import ContainerProxy, exceptions
 from azure.identity import DefaultAzureCredential
 from azure.storage.blob.aio import ContainerClient
-from fastapi import Header, HTTPException
+from fastapi import Header, HTTPException, status
 from graphrag.config.load_config import load_config
 from graphrag.config.models.graph_rag_config import GraphRagConfig
-from fastapi import Header, HTTPException, status
 
 from graphrag_app.logger.load_logger import load_pipeline_logger
 from graphrag_app.typing.models import QueryData
@@ -393,6 +391,7 @@ def update_multi_index_context_data(
         updated_context_data[key] = updated_entry
     return updated_context_data
 
+
 async def create_cache(container_client: ContainerClient) -> None:
     """
     Create a file cache (csv).

backend/scripts/indexer.py

@@ -12,7 +12,6 @@
 # from graphrag.index.create_pipeline_config import create_pipeline_config
 from graphrag.config.enums import IndexingMethod
 from graphrag.config.load_config import load_config
-from graphrag.config.models.community_reports_config import CommunityReportsConfig
 from graphrag.config.models.graph_rag_config import GraphRagConfig
 from graphrag.index.typing.pipeline_run_result import PipelineRunResult
 from graphrag.index.workflows.factory import PipelineFactory