QueryCondition

from fiddler_evals import (
    OperatorType,
    QueryCondition,
    QueryConditionType,
    QueryRule,
)

# LLM spans that either used many tokens or ran on a GPT-4 model
span_filter = QueryCondition(
    rules=[
        QueryRule(
            field="Span::span_type",
            operator=OperatorType.EQUAL,
            value="llm",
        ),
        QueryCondition(
            condition=QueryConditionType.OR,
            rules=[
                QueryRule(
                    field="SpanAttribute::gen_ai.usage.total_tokens",
                    operator=OperatorType.GREATER,
                    value=4000,
                ),
                QueryRule(
                    field="SpanAttribute::gen_ai.request.model",
                    operator=OperatorType.BEGINS_WITH,
                    value="gpt-4",
                ),
            ],
        ),
    ]
)