From 1eeba2cc8acbc509a5f61a78ac7033d5780ffe05 Mon Sep 17 00:00:00 2001 From: ilya-kolchinsky Date: Fri, 7 Mar 2025 16:29:04 +0100 Subject: [PATCH] Changes after merge. --- llama_stack/distribution/routers/routers.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/llama_stack/distribution/routers/routers.py b/llama_stack/distribution/routers/routers.py index 8ea2ccb74..dc932e867 100644 --- a/llama_stack/distribution/routers/routers.py +++ b/llama_stack/distribution/routers/routers.py @@ -702,12 +702,15 @@ class PreprocessingRouter(Preprocessing): self, routing_table: RoutingTable, ) -> None: + logcat.debug("core", "Initializing PreprocessingRouter") self.routing_table = routing_table async def initialize(self) -> None: + logcat.debug("core", "PreprocessingRouter.initialize") pass async def shutdown(self) -> None: + logcat.debug("core", "PreprocessingRouter.shutdown") pass async def preprocess( @@ -716,6 +719,10 @@ class PreprocessingRouter(Preprocessing): preprocessor_inputs: List[PreprocessingDataElement], options: Optional[PreprocessorOptions] = None, ) -> PreprocessorResponse: + logcat.debug( + "core", + f"PreprocessingRouter.preprocess: {preprocessor_id}, {len(preprocessor_inputs)} inputs, options={options}", + ) return await self.routing_table.get_provider_impl(preprocessor_id).preprocess( preprocessor_id=preprocessor_id, preprocessor_inputs=preprocessor_inputs, @@ -727,5 +734,9 @@ class PreprocessingRouter(Preprocessing): preprocessors: PreprocessorChain, preprocessor_inputs: List[PreprocessingDataElement], ) -> PreprocessorResponse: + logcat.debug( + "core", + f"PreprocessingRouter.chain_preprocess: preprocessors {[p.preprocessor_id for p in preprocessors]}, {len(preprocessor_inputs)} inputs", + ) preprocessor_impls = [self.routing_table.get_provider_impl(p.preprocessor_id) for p in preprocessors] return await execute_preprocessor_chain(preprocessors, preprocessor_impls, preprocessor_inputs)