teleemtry package base with OTELConfig export

AgentOps-AI · Jan 5, 2025 · dcf20c5 · dcf20c5
1 parent 9b36d7a
commit dcf20c5
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/agentops/telemetry/__init__.py b/agentops/telemetry/__init__.py
@@ -0,0 +1,3 @@
+from .config import OTELConfig
+
+__all__ = [OTELConfig]
diff --git a/agentops/telemetry/config.py b/agentops/telemetry/config.py
@@ -0,0 +1,20 @@
+from dataclasses import dataclass
+from typing import Dict, List, Optional
+
+from opentelemetry.sdk.trace.export import SpanExporter
+from opentelemetry.sdk.trace.sampling import Sampler
+
+
+@dataclass
+class OTELConfig:
+    """Configuration for OpenTelemetry integration"""
+
+    additional_exporters: Optional[List[SpanExporter]] = None
+    resource_attributes: Optional[Dict] = None
+    sampler: Optional[Sampler] = None
+    retry_config: Optional[Dict] = None
+    custom_formatters: Optional[List[callable]] = None
+    enable_metrics: bool = False
+    metric_readers: Optional[List] = None
+    enable_in_flight: bool = True
+    in_flight_interval: float = 1.0
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		from .config import OTELConfig
Check warning on line 1 in agentops/telemetry/__init__.py View check run for this annotation Codecov / codecov/patch agentops/telemetry/__init__.py#L1 `Added line #L1 was not covered by tests`

		__all__ = [OTELConfig]
Check warning on line 3 in agentops/telemetry/__init__.py View check run for this annotation Codecov / codecov/patch agentops/telemetry/__init__.py#L3 `Added line #L3 was not covered by tests`