mirror of
https://github.com/Azure-Samples/graphrag-accelerator.git
synced 2025-07-05 08:05:29 +00:00
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
|
# Copyright (c) Microsoft Corporation.
|
||
|
|
# Licensed under the MIT License.
|
||
|
|
|
||
|
|
import logging
|
||
|
|
from enum import Enum
|
||
|
|
from typing import Literal
|
||
|
|
|
||
|
|
from graphrag.index.config import (
|
||
|
|
PipelineReportingConfig,
|
||
|
|
reporting,
|
||
|
|
)
|
||
|
|
from pydantic import Field as pydantic_Field
|
||
|
|
|
||
|
|
|
||
|
|
class Reporters(Enum):
|
||
|
|
BLOB = (1, "blob")
|
||
|
|
CONSOLE = (2, "console")
|
||
|
|
FILE = (3, "file")
|
||
|
|
APP_INSIGHTS = (4, "application_insights")
|
||
|
|
|
||
|
|
|
||
|
|
class PipelineAppInsightsReportingConfig(
|
||
|
|
PipelineReportingConfig[Literal["application_insights"]]
|
||
|
|
):
|
||
|
|
"""Represents the ApplicationInsights reporting configuration for the pipeline."""
|
||
|
|
|
||
|
|
type: Literal["application_insights"] = Reporters.APP_INSIGHTS.name.lower()
|
||
|
|
"""The type of reporting."""
|
||
|
|
|
||
|
|
connection_string: str = pydantic_Field(
|
||
|
|
description="The connection string for the App Insights instance.",
|
||
|
|
default=None,
|
||
|
|
)
|
||
|
|
"""The connection string for the App Insights instance."""
|
||
|
|
|
||
|
|
logger_name: str = pydantic_Field(
|
||
|
|
description="The name for logger instance", default=None
|
||
|
|
)
|
||
|
|
"""The name for logger instance"""
|
||
|
|
|
||
|
|
logger_level: int = pydantic_Field(
|
||
|
|
description="The name of the logger. Defaults to None.", default=logging.INFO
|
||
|
|
)
|
||
|
|
"""The name of the logger. Defaults to None."""
|
||
|
|
|
||
|
|
|
||
|
|
# add the new type to the existing PipelineReportingConfigTypes
|
||
|
|
PipelineReportingConfigTypes = (
|
||
|
|
reporting.PipelineReportingConfigTypes | PipelineAppInsightsReportingConfig
|
||
|
|
)
|