Custom Spans

Create custom spans with specific names and attributes for advanced observability needs.

Custom Span Names

Use the span_name parameter to set a semantic span name:

app.py
from rhesis.sdk import observe

# Custom span name following ai.<domain>.<action> pattern
@observe(span_name="ai.llm.invoke")
def my_custom_llm_call(prompt: str) -> str:
    return llm.complete(prompt)

Custom Attributes

Pass additional attributes directly to the decorator:

app.py
from rhesis.sdk import observe
from rhesis.sdk.telemetry import AIAttributes

@observe(
    span_name="ai.llm.invoke",
    **{
        AIAttributes.MODEL_PROVIDER: "custom-provider",
        AIAttributes.MODEL_NAME: "custom-model",
        AIAttributes.LLM_TEMPERATURE: 0.7,
    }
)
def custom_llm(prompt: str) -> str:
    return custom_model.generate(prompt)

Attribute Constants

Import attribute constants from rhesis.sdk.telemetry:

app.py
from rhesis.sdk.telemetry import AIAttributes, AIEvents

See Semantic Conventions for the full list of AIAttributes keys and AIEvents names.

Helper Functions

Use helper functions to create attribute dictionaries:

app.py
from rhesis.sdk.telemetry import create_llm_attributes, create_tool_attributes

# Create LLM attributes
attrs = create_llm_attributes(
    provider="openai",
    model_name="gpt-4",
    tokens_input=150,
    tokens_output=200,
)

# Create tool attributes
attrs = create_tool_attributes(
    tool_name="weather_api",
    tool_type="http",
)

Building Custom Decorators

Create your own convenience decorators by wrapping @observe:

decorators.py
from rhesis.sdk import observe
from rhesis.sdk.telemetry import AIAttributes
from rhesis.telemetry.schemas import AIOperationType

def my_custom_llm(provider: str, model: str, **extra):
    """Custom decorator for your specific LLM setup."""
    return observe(
        span_name=AIOperationType.LLM_INVOKE,
        **{
            AIAttributes.MODEL_PROVIDER: provider,
            AIAttributes.MODEL_NAME: model,
            "custom.attribute": "my-value",
            **extra,
        }
    )

# Usage
@my_custom_llm(provider="my-provider", model="my-model")
def generate(prompt: str) -> str:
    return my_llm.complete(prompt)

Example: Full Custom Implementation

app.py
from rhesis.sdk import RhesisClient, observe
from rhesis.sdk.telemetry import AIAttributes, AIEvents
from rhesis.telemetry.schemas import AIOperationType

client = RhesisClient(
    api_key="your-api-key",
    project_id="your-project-id",
    environment="development",
)

# Custom LLM decorator with all attributes
@observe(
    span_name=AIOperationType.LLM_INVOKE,
    **{
        AIAttributes.MODEL_PROVIDER: "my-provider",
        AIAttributes.MODEL_NAME: "my-model-v2",
        AIAttributes.LLM_TEMPERATURE: 0.8,
        AIAttributes.LLM_MAX_TOKENS: 1000,
        "custom.deployment": "us-east-1",
        "custom.version": "2.0",
    }
)
def generate_with_custom_model(prompt: str) -> str:
    response = my_model.complete(prompt)
    return response.text

Next: Learn about auto-instrumentation for zero-config tracing.