mirror of
https://github.com/Azure-Samples/graphrag-accelerator.git
synced 2026-01-06 12:22:57 +00:00
ran linter
This commit is contained in:
parent
946b95c0f4
commit
2d805053ad
@ -6,7 +6,6 @@ import traceback
|
||||
from pathlib import Path
|
||||
|
||||
import graphrag.api as api
|
||||
import yaml
|
||||
from fastapi import (
|
||||
APIRouter,
|
||||
Depends,
|
||||
|
||||
@ -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,
|
||||
|
||||
@ -4,7 +4,6 @@
|
||||
import json
|
||||
import os
|
||||
import traceback
|
||||
from typing import Dict, List
|
||||
|
||||
from fastapi import (
|
||||
APIRouter,
|
||||
|
||||
@ -4,7 +4,6 @@
|
||||
from io import StringIO
|
||||
from typing import (
|
||||
Any,
|
||||
Dict,
|
||||
List,
|
||||
Optional,
|
||||
)
|
||||
|
||||
@ -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.
|
||||
|
||||
@ -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).
|
||||
|
||||
@ -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
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user