wehub-resource-sync
36 lines
901 B
Python
36 lines
901 B
Python
"""Shared test utilities for Ragas tests.
|
|
|
|
This module provides reusable utilities for both pytest tests and Jupyter notebooks,
|
|
including LLM setup, embeddings configuration, and common test helpers.
|
|
"""
|
|
|
|
from .llm_setup import (
|
|
check_api_key,
|
|
create_legacy_embeddings,
|
|
create_legacy_llm,
|
|
create_modern_embeddings,
|
|
create_modern_llm,
|
|
)
|
|
from .metric_comparison import (
|
|
MetricDiffResult,
|
|
compare_metrics,
|
|
export_comparison_results,
|
|
run_metric_on_dataset,
|
|
run_metric_on_dataset_with_batching,
|
|
)
|
|
|
|
__all__ = [
|
|
# LLM and embeddings setup
|
|
"check_api_key",
|
|
"create_legacy_llm",
|
|
"create_modern_llm",
|
|
"create_legacy_embeddings",
|
|
"create_modern_embeddings",
|
|
# Metric comparison utilities
|
|
"MetricDiffResult",
|
|
"compare_metrics",
|
|
"export_comparison_results",
|
|
"run_metric_on_dataset",
|
|
"run_metric_on_dataset_with_batching",
|
|
]
|