ййй

src/app/modules/chat/module.py

@@ -1,13 +1,11 @@
 from __future__ import annotations
 
-import os
 from typing import TYPE_CHECKING
 
 from fastapi import APIRouter, Header
 from fastapi.responses import StreamingResponse
 
 from app.core.exceptions import AppError
-from app.modules.chat.direct_service import CodeExplainChatService
 from app.modules.chat.dialog_store import DialogSessionStore
 from app.modules.chat.service import ChatOrchestrator
 from app.modules.chat.task_store import TaskStore
@@ -37,16 +35,13 @@ class ChatModule:
         retry: RetryExecutor,
         rag_sessions: RagSessionStore,
         repository: ChatRepository,
-        direct_chat: CodeExplainChatService | None = None,
         task_store: TaskStore | None = None,
     ) -> None:
         self._rag_sessions = rag_sessions
-        self._simple_code_explain_only = os.getenv("SIMPLE_CODE_EXPLAIN_ONLY", "true").lower() in {"1", "true", "yes"}
         self.tasks = task_store or TaskStore()
         self.dialogs = DialogSessionStore(repository)
         self.idempotency = IdempotencyStore()
         self.events = event_bus
-        self.direct_chat = direct_chat
         self.chat = ChatOrchestrator(
             task_store=self.tasks,
             dialogs=self.dialogs,
@@ -76,8 +71,6 @@ class ChatModule:
             request: ChatMessageRequest,
             idempotency_key: str | None = Header(default=None, alias="Idempotency-Key"),
         ) -> TaskQueuedResponse | TaskResultResponse:
-            if self._simple_code_explain_only and self.direct_chat is not None:
-                return await self.direct_chat.handle_message(request)
             task = await self.chat.enqueue_message(request, idempotency_key)
             return TaskQueuedResponse(task_id=task.task_id, status=task.status.value)
 
@@ -91,6 +84,7 @@ class ChatModule:
                 status=task.status,
                 result_type=task.result_type,
                 answer=task.answer,
+                artifacts=task.artifacts,
                 changeset=task.changeset,
                 error=task.error,
             )