work/agent
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Рефакторинг

Browse Source
This commit is contained in:
zosimovaa
2026-03-12 23:33:51 +03:00
parent 9066c292de
commit 15586f9a8c
133 changed files with 1011 additions and 894 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
tests/pipeline_setup_v3/runtime/agent_runtime_adapter.py
View File

@@ -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