diff --git a/kag/builder/prompt/utils.py b/kag/builder/prompt/utils.py index 79984001..ef07045f 100644 --- a/kag/builder/prompt/utils.py +++ b/kag/builder/prompt/utils.py @@ -9,15 +9,17 @@ # Unless required by applicable law or agreed to in writing, software distributed under the License # is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express # or implied. - +import logging from kag.interface import PromptABC +logger = logging.getLogger(__name__) + def init_prompt_with_fallback(prompt_name, biz_scene): try: return PromptABC.from_config({"type": f"{biz_scene}_{prompt_name}"}) except Exception as e: - print( + logger.debug( f"fail to initialize prompts with biz scene {biz_scene}, fallback to default biz scene, info: {e}" ) diff --git a/kag/common/registry/registrable.py b/kag/common/registry/registrable.py index 7e0bbacd..212226af 100644 --- a/kag/common/registry/registrable.py +++ b/kag/common/registry/registrable.py @@ -809,7 +809,7 @@ class Registrable: params.clear() setattr(instant, "__from_config_kwargs__", remaining_kwargs) except Exception as e: - logger.warn(f"Failed to initialize class {cls}, info: {e}") + logger.debug(f"Failed to initialize class {cls}, info: {e}") raise e if len(params) > 0: logger.warn(f"These params are not used for constructing {cls}:\n{params}")