From 52bf829a84f89c89a37d6dcfd73e5395cc8c0f05 Mon Sep 17 00:00:00 2001 From: weiweiw <14335254+weiweiw22@user.noreply.gitee.com> Date: Mon, 6 Jan 2025 17:20:05 +0800 Subject: [PATCH] commit some debug log --- .../agent/tools_factory/tools_registry.py | 28 +++++++++++-------- 1 file changed, 16 insertions(+), 12 deletions(-) diff --git a/libs/chatchat-server/chatchat/server/agent/tools_factory/tools_registry.py b/libs/chatchat-server/chatchat/server/agent/tools_factory/tools_registry.py index 4eeda2b..20689a8 100644 --- a/libs/chatchat-server/chatchat/server/agent/tools_factory/tools_registry.py +++ b/libs/chatchat-server/chatchat/server/agent/tools_factory/tools_registry.py @@ -30,18 +30,22 @@ def _new_parse_input( tool_input: Union[str, Dict], ) -> Union[str, Dict[str, Any]]: """Convert tool input to pydantic model.""" - input_args = self.args_schema - if isinstance(tool_input, str): - print(f"tool_input:{tool_input}") - if input_args is not None: - key_ = next(iter(input_args.__fields__.keys())) - input_args.validate({key_: tool_input}) - return tool_input - else: - if input_args is not None: - result = input_args.parse_obj(tool_input) - logging.info(f"result:{result}") - return result.dict() + try: + input_args = self.args_schema + if isinstance(tool_input, str): + logging.info(f"tool_input:{tool_input}") + if input_args is not None: + key_ = next(iter(input_args.__fields__.keys())) + input_args.validate({key_: tool_input}) + return tool_input + else: + if input_args is not None: + result = input_args.parse_obj(tool_input) + logging.info(f"result:{result}") + return result.dict() + except Exception as e: + msg = f"Caught exception: {e}" + logging.error(f"{e.__class__.__name__}: {msg}") def _new_to_args_and_kwargs(self, tool_input: Union[str, Dict]) -> Tuple[Tuple, Dict]: