diff --git a/caikit_nlp/toolkit/text_generation/tgis_utils.py b/caikit_nlp/toolkit/text_generation/tgis_utils.py index 0b870ba65..d4e1b1022 100644 --- a/caikit_nlp/toolkit/text_generation/tgis_utils.py +++ b/caikit_nlp/toolkit/text_generation/tgis_utils.py @@ -474,6 +474,13 @@ def unary_generate( batch_response = self.tgis_client.Generate( request, timeout=self.tgis_req_timeout ) + except grpc._channel._InactiveRpcError as err: + log.error("", err.details) + error_message = "The underlying TCP connection is closed" + caikit_status_code = GRPC_TO_CAIKIT_CORE_STATUS.get( + err.code(), CaikitCoreStatusCode.UNKNOWN + ) + raise CaikitCoreException(caikit_status_code, error_message) except grpc.RpcError as err: raise_caikit_core_exception(err) @@ -653,6 +660,13 @@ def stream_generate( input_tokens=input_token_list, details=details, ) + except grpc._channel._InactiveRpcError as err: + log.error("", err.details) + error_message = "The underlying TCP connection is closed" + caikit_status_code = GRPC_TO_CAIKIT_CORE_STATUS.get( + err.code(), CaikitCoreStatusCode.UNKNOWN + ) + raise CaikitCoreException(caikit_status_code, error_message) except grpc.RpcError as err: raise_caikit_core_exception(err)