headroom/benchmarks/scenarios/__init__.py
chopratejas e4a41faa33 Fix all ruff lint and format errors for CI
- Fix E402: Move module-level imports to top of file
- Fix F401: Add noqa for availability check imports
- Fix F402: Rename loop variables shadowing imports
- Fix E722: Replace bare except with except Exception
- Fix B904: Add exception chaining (from e)
- Fix F811: Remove duplicate imports
- Fix B027: Add noqa for empty close() method
- Fix E741: Rename ambiguous variable l -> label
- Fix I001: Import sorting issues
- Apply ruff format to all 106 files

All 902 tests pass.
2026-01-10 15:33:44 -08:00

29 lines
767 B
Python

"""Benchmark scenario generators for Headroom SDK.
This package provides realistic data generators for benchmarking Headroom
transforms and relevance scorers.
Modules:
tool_outputs: Generators for tool output data (search, logs, API responses)
conversations: Generators for conversation history (agentic, RAG)
"""
from .conversations import (
generate_agentic_conversation,
generate_rag_conversation,
)
from .tool_outputs import (
generate_api_responses,
generate_database_rows,
generate_log_entries,
generate_search_results,
)
__all__ = [
"generate_search_results",
"generate_log_entries",
"generate_api_responses",
"generate_database_rows",
"generate_agentic_conversation",
"generate_rag_conversation",
]