diff --git a/src/llama_stack_provider_ragas/base.py b/src/llama_stack_provider_ragas/base.py index d1008b22..f1737eaa 100644 --- a/src/llama_stack_provider_ragas/base.py +++ b/src/llama_stack_provider_ragas/base.py @@ -121,3 +121,11 @@ async def register_benchmark(self, task_def: Benchmark) -> None: """Register a benchmark for evaluation.""" self.benchmarks[task_def.identifier] = task_def logger.info(f"Registered benchmark: {task_def.identifier}") + + async def unregister_benchmark(self, benchmark_id: str) -> None: + """Unregister a benchmark.""" + removed = self.benchmarks.pop(benchmark_id, None) + if removed is not None: + logger.info(f"Unregistered benchmark: {benchmark_id}") + else: + logger.info(f"Benchmark not found (nothing to unregister): {benchmark_id}")