Роутер работает нормально в process v2
This commit is contained in:
@@ -19,10 +19,12 @@ class V3Runner:
|
||||
*,
|
||||
pipeline_mode: str = "full",
|
||||
router_llm_mode: str = "deterministic",
|
||||
workflow_llm_enabled: bool = True,
|
||||
) -> None:
|
||||
self._cases_dir = cases_dir
|
||||
self._pipeline_mode = pipeline_mode
|
||||
self._router_llm_mode = router_llm_mode
|
||||
self._workflow_llm_enabled = workflow_llm_enabled
|
||||
self._validator = CaseValidator()
|
||||
self._sessions = RagSessionProvider()
|
||||
self._agent_runtime = None
|
||||
@@ -50,9 +52,11 @@ class V3Runner:
|
||||
return results, self._writer.write_summary(results)
|
||||
|
||||
def _execute(self, case, rag_session_id):
|
||||
if case.runner != "agent_runtime":
|
||||
raise ValueError(f"Unsupported runner: {case.runner}")
|
||||
return self._agent_runtime_adapter().execute(case, rag_session_id)
|
||||
if case.runner == "agent_runtime":
|
||||
return self._agent_runtime_adapter().execute(case, rag_session_id)
|
||||
if case.runner == "process_v2":
|
||||
return self._v2_process_adapter().execute(case, rag_session_id)
|
||||
raise ValueError(f"Unsupported runner: {case.runner}")
|
||||
|
||||
def _agent_runtime_adapter(self):
|
||||
if self._agent_runtime is None:
|
||||
@@ -63,3 +67,10 @@ class V3Runner:
|
||||
router_llm_mode=self._router_llm_mode,
|
||||
)
|
||||
return self._agent_runtime
|
||||
|
||||
def _v2_process_adapter(self):
|
||||
if not hasattr(self, "_v2_process"):
|
||||
from tests.pipeline_setup_v3.runtime.v2_process_adapter import V2ProcessAdapter
|
||||
|
||||
self._v2_process = V2ProcessAdapter(workflow_llm_enabled=self._workflow_llm_enabled)
|
||||
return self._v2_process
|
||||
|
||||
Reference in New Issue
Block a user