Удаление легаси
This commit is contained in:
@@ -1,6 +1,8 @@
|
||||
from app.modules.agent.module import AgentModule
|
||||
from app.modules.agent.repository import AgentRepository
|
||||
from app.modules.agent.story_context_repository import StoryContextRepository, StoryContextSchemaRepository
|
||||
from app.modules.agent.code_qa_runtime import CodeQaRuntimeExecutor
|
||||
from app.modules.agent.code_qa_runtime.retrieval_adapter import CodeQaRetrievalAdapter
|
||||
from app.modules.agent.code_qa_runner_adapter import CodeQaRunnerAdapter
|
||||
from app.modules.agent.llm import AgentLlmService
|
||||
from app.modules.agent.prompt_loader import PromptLoader
|
||||
from app.modules.chat.direct_service import CodeExplainChatService
|
||||
from app.modules.chat.dialog_store import DialogSessionStore
|
||||
from app.modules.chat.repository import ChatRepository
|
||||
@@ -8,6 +10,7 @@ from app.modules.chat.module import ChatModule
|
||||
from app.modules.chat.session_resolver import ChatSessionResolver
|
||||
from app.modules.chat.task_store import TaskStore
|
||||
from app.modules.rag.persistence.repository import RagRepository
|
||||
from app.modules.rag.persistence.story_context_repository import StoryContextRepository, StoryContextSchemaRepository
|
||||
from app.modules.rag.explain import CodeExplainRetrieverV2, CodeGraphRepository, LayeredRetrievalGateway
|
||||
from app.modules.rag.module import RagModule, RagRepoModule
|
||||
from app.modules.shared.bootstrap import bootstrap_database
|
||||
@@ -21,7 +24,6 @@ class ModularApplication:
|
||||
self.retry = RetryExecutor()
|
||||
self.rag_repository = RagRepository()
|
||||
self.chat_repository = ChatRepository()
|
||||
self.agent_repository = AgentRepository()
|
||||
self.story_context_schema_repository = StoryContextSchemaRepository()
|
||||
self.story_context_repository = StoryContextRepository()
|
||||
self.chat_tasks = TaskStore()
|
||||
@@ -35,15 +37,20 @@ class ModularApplication:
|
||||
gateway=LayeredRetrievalGateway(self.rag_repository, self.rag.embedder),
|
||||
graph_repository=CodeGraphRepository(),
|
||||
)
|
||||
self.agent = AgentModule(
|
||||
rag_retriever=self.rag.rag,
|
||||
agent_repository=self.agent_repository,
|
||||
story_context_repository=self.story_context_repository,
|
||||
code_explain_retriever=self.code_explain_retriever,
|
||||
)
|
||||
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
|
||||
|
||||
_giga_settings = GigaChatSettings.from_env()
|
||||
_giga_client = GigaChatClient(_giga_settings, GigaChatTokenProvider(_giga_settings))
|
||||
_prompt_loader = PromptLoader()
|
||||
self._agent_llm = AgentLlmService(client=_giga_client, prompts=_prompt_loader)
|
||||
_retrieval = CodeQaRetrievalAdapter(self.rag_repository)
|
||||
_executor = CodeQaRuntimeExecutor(llm=self._agent_llm, retrieval=_retrieval)
|
||||
self._agent_runner = CodeQaRunnerAdapter(_executor)
|
||||
self.direct_chat = CodeExplainChatService(
|
||||
retriever=self.code_explain_retriever,
|
||||
llm=self.agent.llm,
|
||||
llm=self._agent_llm,
|
||||
session_resolver=ChatSessionResolver(
|
||||
dialogs=DialogSessionStore(self.chat_repository),
|
||||
rag_session_exists=lambda rag_session_id: self.rag.sessions.get(rag_session_id) is not None,
|
||||
@@ -52,7 +59,7 @@ class ModularApplication:
|
||||
message_sink=self.chat_repository.add_message,
|
||||
)
|
||||
self.chat = ChatModule(
|
||||
agent_runner=self.agent.runtime,
|
||||
agent_runner=self._agent_runner,
|
||||
event_bus=self.events,
|
||||
retry=self.retry,
|
||||
rag_sessions=self.rag.sessions,
|
||||
@@ -65,6 +72,5 @@ class ModularApplication:
|
||||
bootstrap_database(
|
||||
self.rag_repository,
|
||||
self.chat_repository,
|
||||
self.agent_repository,
|
||||
self.story_context_schema_repository,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user