Skip to content

Commit 849f950

Browse files
author
Danny Zheng
committed
tested telemetry
1 parent cd844df commit 849f950

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

graphrag/config/load_config.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414

1515
from graphrag.config.create_graphrag_config import create_graphrag_config
1616
from graphrag.config.models.graph_rag_config import GraphRagConfig
17+
from graphrag.telemetry.decorators import add_trace
1718

1819
_default_config_files = ["settings.yaml", "settings.yml", "settings.json"]
1920

@@ -142,7 +143,9 @@ def _parse(file_extension: str, contents: str) -> dict[str, Any]:
142143
)
143144
raise ValueError(msg)
144145

145-
146+
@add_trace(
147+
operation_name="graphrag.config.load_config",
148+
attributes={"component": "config"})
146149
def load_config(
147150
root_dir: Path,
148151
config_filepath: Path | None = None,

graphrag/telemetry/setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def _setup_tracing(config: TelemetryConfig, resource: Resource) -> None:
109109
exporter = ZipkinExporter(endpoint=config.obs_endpoint)
110110
logger.info(f"Tracing configured with Zipkin endpoint: {config.obs_endpoint}")
111111
else:
112-
logger.warning("Zipkin exporter not available. Install with: pip install opentelemetry-exporter-zipkin-json")
112+
logger.warning("Zipkin exporter not available. Please install with: pip install opentelemetry-exporter-zipkin-json")
113113
logger.info("Falling back to OTLP exporter")
114114
credentials, isInSecure = _get_credentials()
115115
exporter = OTLPSpanExporter(

0 commit comments

Comments
 (0)