From 7e328c7d74216748d8b9be9f882d5ce522f5f9f1 Mon Sep 17 00:00:00 2001 From: Shonda-Adena-Witherspoon Date: Tue, 4 Jun 2024 14:32:20 -0500 Subject: [PATCH 1/4] added logging around tgis timout config setting Signed-off-by: Shonda-Adena-Witherspoon --- caikit_nlp/toolkit/text_generation/tgis_utils.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/caikit_nlp/toolkit/text_generation/tgis_utils.py b/caikit_nlp/toolkit/text_generation/tgis_utils.py index 8dd10d43..25026a65 100644 --- a/caikit_nlp/toolkit/text_generation/tgis_utils.py +++ b/caikit_nlp/toolkit/text_generation/tgis_utils.py @@ -329,6 +329,21 @@ def __init__( self.tgis_req_timeout = get_config().tgis_request_timeout + if ( + not self.tgis_req_timeout + or not isinstance(self.tgis_req_timeout, int) + or self.tgis_req_timeout <= 0 + ): + log.info("", "TGIS timeout not set") + self.tgis_req_timeout = None + + else: + log.info( + "", + "Setting TGIS timeout value to %d", + self.tgis_req_timeout, + ) + def unary_generate( self, text, From f4496b95b268d8326b9a3a6646704827ea67eb2a Mon Sep 17 00:00:00 2001 From: swith004 Date: Tue, 4 Jun 2024 18:03:19 -0400 Subject: [PATCH 2/4] Update caikit_nlp/toolkit/text_generation/tgis_utils.py change logging to debug Co-authored-by: Gaurav Kumbhat Signed-off-by: swith004 Signed-off-by: Shonda-Adena-Witherspoon --- caikit_nlp/toolkit/text_generation/tgis_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/caikit_nlp/toolkit/text_generation/tgis_utils.py b/caikit_nlp/toolkit/text_generation/tgis_utils.py index 25026a65..57d1b2e8 100644 --- a/caikit_nlp/toolkit/text_generation/tgis_utils.py +++ b/caikit_nlp/toolkit/text_generation/tgis_utils.py @@ -334,7 +334,7 @@ def __init__( or not isinstance(self.tgis_req_timeout, int) or self.tgis_req_timeout <= 0 ): - log.info("", "TGIS timeout not set") + log.debug("", "TGIS timeout not set") self.tgis_req_timeout = None else: From 7211497c771946dc90d0a46ed559b879013fe45d Mon Sep 17 00:00:00 2001 From: swith004 Date: Tue, 4 Jun 2024 18:03:42 -0400 Subject: [PATCH 3/4] Update caikit_nlp/toolkit/text_generation/tgis_utils.py change logging to debug Co-authored-by: Gaurav Kumbhat Signed-off-by: swith004 Signed-off-by: Shonda-Adena-Witherspoon --- caikit_nlp/toolkit/text_generation/tgis_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/caikit_nlp/toolkit/text_generation/tgis_utils.py b/caikit_nlp/toolkit/text_generation/tgis_utils.py index 57d1b2e8..f6185540 100644 --- a/caikit_nlp/toolkit/text_generation/tgis_utils.py +++ b/caikit_nlp/toolkit/text_generation/tgis_utils.py @@ -338,7 +338,7 @@ def __init__( self.tgis_req_timeout = None else: - log.info( + log.debug "", "Setting TGIS timeout value to %d", self.tgis_req_timeout, From b710556a587ca79f261879017764b74718b1e7bc Mon Sep 17 00:00:00 2001 From: Shonda-Adena-Witherspoon Date: Tue, 4 Jun 2024 17:12:37 -0500 Subject: [PATCH 4/4] fixed formatting Signed-off-by: Shonda-Adena-Witherspoon --- caikit_nlp/toolkit/text_generation/tgis_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/caikit_nlp/toolkit/text_generation/tgis_utils.py b/caikit_nlp/toolkit/text_generation/tgis_utils.py index f6185540..d10a0531 100644 --- a/caikit_nlp/toolkit/text_generation/tgis_utils.py +++ b/caikit_nlp/toolkit/text_generation/tgis_utils.py @@ -338,7 +338,7 @@ def __init__( self.tgis_req_timeout = None else: - log.debug + log.debug( "", "Setting TGIS timeout value to %d", self.tgis_req_timeout,