From adebc73a36faa41f1073c155a11b68a53203a78c Mon Sep 17 00:00:00 2001 From: Luv Bansal Date: Mon, 12 May 2025 16:41:37 +0530 Subject: [PATCH] Added req_id in logger in debug mode --- clarifai/runners/models/model_runner.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/clarifai/runners/models/model_runner.py b/clarifai/runners/models/model_runner.py index 72f44c58..2e299bc8 100644 --- a/clarifai/runners/models/model_runner.py +++ b/clarifai/runners/models/model_runner.py @@ -5,6 +5,8 @@ from clarifai_protocol import BaseRunner from clarifai_protocol.utils.health import HealthProbeRequestHandler +from clarifai.utils.logging import logger + from ..utils.url_fetcher import ensure_urls_downloaded from .model_class import ModelClass @@ -84,6 +86,9 @@ def runner_item_predict( ensure_urls_downloaded(request) resp = self.model.predict_wrapper(request) + logger.debug( + f'Request_id: {request.status.req_id}, ', + ) if resp.status.code != status_code_pb2.SUCCESS: return service_pb2.RunnerItemOutput(multi_output_response=resp) successes = [o.status.code == status_code_pb2.SUCCESS for o in resp.outputs] @@ -117,6 +122,9 @@ def runner_item_generate( ensure_urls_downloaded(request) for resp in self.model.generate_wrapper(request): + logger.debug( + f'Request_id: {request.status.req_id}, ', + ) if resp.status.code != status_code_pb2.SUCCESS: yield service_pb2.RunnerItemOutput(multi_output_response=resp) continue @@ -151,6 +159,9 @@ def runner_item_stream( ) -> Iterator[service_pb2.RunnerItemOutput]: # Call the generate() method the underlying model implements. for resp in self.model.stream_wrapper(pmo_iterator(runner_item_iterator)): + logger.debug( + f'Request_id: {resp.status.req_id}, ', + ) if resp.status.code != status_code_pb2.SUCCESS: yield service_pb2.RunnerItemOutput(multi_output_response=resp) continue