commit some debug log

This commit is contained in:
weiweiw 2025-01-06 17:20:05 +08:00
parent b225390080
commit 52bf829a84
1 changed files with 16 additions and 12 deletions

View File

@ -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]: