ran linter

This commit is contained in:
Gabriel Nieves 2025-04-11 01:56:17 +00:00
parent 946b95c0f4
commit 2d805053ad
7 changed files with 5 additions and 13 deletions

View File

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

View File

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

View File

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

View File

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

View File

@ -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.cosmos import (
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 @@ class AzureClientManager:
else:
self.storage_account_name = self.storage_account_hostname.split(".")[0]
def get_blob_service_client(self) -> BlobServiceClient:
"""
Returns the blob service client.

View File

@ -9,20 +9,18 @@ import hashlib
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).

View File

@ -12,7 +12,6 @@ from graphrag.callbacks.workflow_callbacks import WorkflowCallbacks
# 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