mirror of
https://github.com/Azure-Samples/graphrag-accelerator.git
synced 2025-07-04 07:33:58 +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.reporting import (
|
|
PipelineReportingConfig,
|
|
PipelineReportingConfigTypes,
|
|
)
|
|
from pydantic import Field as pydantic_Field
|
|
|
|
|
|
class Logger(Enum):
|
|
BLOB = (1, "blob")
|
|
CONSOLE = (2, "console")
|
|
FILE = (3, "file")
|
|
APP_INSIGHTS = (4, "app_insights")
|
|
|
|
|
|
class PipelineAppInsightsReportingConfig(
|
|
PipelineReportingConfig[Literal["app_insights"]]
|
|
):
|
|
"""Represents the ApplicationInsights reporting configuration for the pipeline."""
|
|
|
|
type: Literal["app_insights"] = Logger.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 = (
|
|
PipelineReportingConfigTypes | PipelineAppInsightsReportingConfig
|
|
)
|