Рефакторинг

tests/pipeline_setup_v3/runtime/agent_runtime_adapter.py

@@ -2,14 +2,15 @@ from __future__ import annotations
 
 import math
 
-from app.modules.agent.code_qa_runtime import CodeQaRuntimeExecutor
-from app.modules.agent.code_qa_runtime.repo_context import CodeQaRepoContextFactory
-from app.modules.agent.code_qa_runtime.retrieval_adapter import CodeQaRetrievalAdapter
+from app.modules.agent.runtime import (
+    AgentRuntimeExecutor,
+    RuntimeRepoContextFactory,
+    RuntimeRetrievalAdapter,
+)
 from app.modules.agent.llm import AgentLlmService
-from app.modules.agent.prompt_loader import PromptLoader
-from app.modules.rag.code_qa_pipeline.retrieval_request_builder import build_retrieval_request
-from app.modules.rag.code_qa_pipeline.retrieval_result_builder import build_retrieval_result
-from app.modules.rag.intent_router_v2 import ConversationState, IntentRouterV2
+from app.modules.agent.llm.prompt_loader import PromptLoader
+from app.modules.agent.runtime.steps.context import build_retrieval_request, build_retrieval_result
+from app.modules.agent.intent_router_v2 import ConversationState, IntentRouterV2
 from app.modules.shared.gigachat.client import GigaChatClient
 from app.modules.shared.gigachat.settings import GigaChatSettings
 from app.modules.shared.gigachat.token_provider import GigaChatTokenProvider
@@ -19,9 +20,9 @@ from tests.pipeline_setup_v3.core.models import ExecutionPayload, V3Case
 class AgentRuntimeAdapter:
     def __init__(self) -> None:
         self._router = IntentRouterV2()
-        self._repo_context_factory = CodeQaRepoContextFactory()
-        self._retrieval = CodeQaRetrievalAdapter()
-        self._executor: CodeQaRuntimeExecutor | None = None
+        self._repo_context_factory = RuntimeRepoContextFactory()
+        self._retrieval = RuntimeRetrievalAdapter()
+        self._executor: AgentRuntimeExecutor | None = None
 
     def execute(self, case: V3Case, rag_session_id: str | None) -> ExecutionPayload:
         if case.mode == "router_only":
@@ -156,9 +157,9 @@ class AgentRuntimeAdapter:
             "layers": tuple(layers or []),
         }
 
-    def _executor_instance(self) -> CodeQaRuntimeExecutor:
+    def _executor_instance(self) -> AgentRuntimeExecutor:
         if self._executor is None:
-            self._executor = CodeQaRuntimeExecutor(_build_llm())
+            self._executor = AgentRuntimeExecutor(_build_llm())
         return self._executor