ййй
This commit is contained in:
@@ -11,8 +11,18 @@ from tests.pipeline_setup_v3.core.validators import CaseValidator
|
||||
|
||||
|
||||
class V3Runner:
|
||||
def __init__(self, cases_dir: Path, results_dir: Path, run_name: str) -> None:
|
||||
def __init__(
|
||||
self,
|
||||
cases_dir: Path,
|
||||
results_dir: Path,
|
||||
run_name: str,
|
||||
*,
|
||||
pipeline_mode: str = "full",
|
||||
router_llm_mode: str = "deterministic",
|
||||
) -> None:
|
||||
self._cases_dir = cases_dir
|
||||
self._pipeline_mode = pipeline_mode
|
||||
self._router_llm_mode = router_llm_mode
|
||||
self._validator = CaseValidator()
|
||||
self._sessions = RagSessionProvider()
|
||||
self._agent_runtime = None
|
||||
@@ -48,5 +58,8 @@ class V3Runner:
|
||||
if self._agent_runtime is None:
|
||||
from tests.pipeline_setup_v3.runtime.agent_runtime_adapter import AgentRuntimeAdapter
|
||||
|
||||
self._agent_runtime = AgentRuntimeAdapter()
|
||||
self._agent_runtime = AgentRuntimeAdapter(
|
||||
pipeline_mode=self._pipeline_mode,
|
||||
router_llm_mode=self._router_llm_mode,
|
||||
)
|
||||
return self._agent_runtime
|
||||
|
||||
Reference in New Issue
Block a user