1111)
1212from src .core .config .app_config import AppConfig
1313from src .core .di .container import ServiceCollection
14+ from src .core .domain .chat import ChatRequest
15+ from src .core .domain .processed_result import ProcessedResult
16+ from src .core .domain .request_context import RequestContext
17+ from src .core .domain .responses import ResponseEnvelope , StreamingResponseEnvelope
1418from src .core .interfaces .application_state_interface import IApplicationState
1519from src .core .interfaces .backend_service_interface import IBackendService
1620from src .core .interfaces .command_service_interface import ICommandService
2226from src .core .interfaces .session_resolver_interface import ISessionResolver
2327from src .core .interfaces .session_service_interface import ISessionService
2428from src .core .interfaces .wire_capture_interface import IWireCapture
29+ from src .core .repositories .in_memory_session_repository import InMemorySessionRepository
2530from src .core .services .application_state_service import ApplicationStateService
2631from src .core .services .response_manager_service import AgentResponseFormatter
2732from src .core .services .session_resolver_service import DefaultSessionResolver
2833from src .core .services .session_service_impl import SessionService
29- from src .core .domain .processed_result import ProcessedResult
30- from src .core .domain .responses import ResponseEnvelope , StreamingResponseEnvelope
31- from src .core .domain .chat import ChatRequest
32- from src .core .domain .request_context import RequestContext
33- from src .core .repositories .in_memory_session_repository import InMemorySessionRepository
3434
3535
3636class _StubCommandService (ICommandService ):
@@ -53,6 +53,7 @@ async def call_completion(
5353 context : RequestContext | None = None ,
5454 ) -> ResponseEnvelope | StreamingResponseEnvelope :
5555 if stream :
56+
5657 async def _stream () -> AsyncIterator [StreamingResponseEnvelope ]:
5758 yield StreamingResponseEnvelope (content = {}, headers = {}, status_code = 200 )
5859
@@ -84,9 +85,7 @@ async def _generator() -> AsyncIterator[ProcessedResponse]:
8485
8586 return _generator ()
8687
87- async def register_middleware (
88- self , middleware : Any , priority : int = 0
89- ) -> None :
88+ async def register_middleware (self , middleware : Any , priority : int = 0 ) -> None :
9089 return None
9190
9291
0 commit comments