Skip to content

Commit 37a1bed

Browse files
committed
use Enum for span_format and enforce valid values
1 parent 240f66f commit 37a1bed

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

nemoguardrails/rails/llm/config.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -358,14 +358,19 @@ class LogAdapterConfig(BaseModel):
358358
model_config = ConfigDict(extra="allow")
359359

360360

361+
class SpanFormat(str, Enum):
362+
legacy = "legacy"
363+
opentelemetry = "opentelemetry"
364+
365+
361366
class TracingConfig(BaseModel):
362367
enabled: bool = False
363368
adapters: List[LogAdapterConfig] = Field(
364369
default_factory=lambda: [LogAdapterConfig()],
365370
description="The list of tracing adapters to use. If not specified, the default adapters are used.",
366371
)
367372
span_format: str = Field(
368-
default="opentelemetry",
373+
default=SpanFormat.opentelemetry,
369374
description="The span format to use. Options are 'legacy' (simple metrics) or 'opentelemetry' (OpenTelemetry semantic conventions).",
370375
)
371376
enable_content_capture: bool = Field(

0 commit comments

Comments
 (0)