tool_wrapper

[maoxiaomeng2021]

i want to create a tool_wrapper_fun to do sth extensional ,eg: just print sth or save log ,this is the code i :
`
def tool_log_recorder(*, func):
sfunc = StructuredTool.from_function(func=func)

@tool(sfunc.name, description=sfunc.description, args_schema=sfunc.args_schema)
def tool_wrapper(**kwargs):
    logger.info(
        f"==========tool_log_recorder 函数名:{sfunc.get_name()} 函数描述:{repr(func)}  args: {sfunc.input_schema}")
    logger.info(f"==========tool_log_recorder {sfunc.get_name()} start")
    logger.info(f"tool_call_id:{tool_call_id}")
    try:
        res = func(**kwargs)
        return res
    except Exception as e:
        logger.error(f"==========tool_log_recorder {sfunc.get_name()} error: {e}")
    finally:
        logger.error(f"==========tool_log_recorder {sfunc.get_name()} end")

return tool_wrapper

`

and i use it by:
`
def add(a,b):
"""plus a and b"""
return a+b

agent = create_react_agent(
model=model,
tools=[
tool_log_recorder(func=add),
],
)
`

and it works;

but i want to get tool_call_id or other sth such as state, it it failed

i try it such as:

def tool_log_recorder(*, func, tool_call_id: Annotated[str, InjectedToolCallId] = None):
sfunc = StructuredTool.from_function(func=func)

@tool(sfunc.name, description=sfunc.description, args_schema=sfunc.args_schema)
def tool_wrapper(**kwargs):
    logger.info(
        f"==========tool_log_recorder 函数名:{sfunc.get_name()} 函数描述:{repr(func)}  args: {sfunc.input_schema}")
    logger.info(f"==========tool_log_recorder {sfunc.get_name()} start")
    logger.info(f"tool_call_id:{tool_call_id}")
    try:
        res = func(**kwargs)
        return res
    except Exception as e:
        logger.error(f"==========tool_log_recorder {sfunc.get_name()} error: {e}")
    finally:
        logger.error(f"==========tool_log_recorder {sfunc.get_name()} end")

return tool_wrapper

it seems failed too ,
plz help me

[hinthornw]

https://docs.python.org/3/library/functools.html#functools.wraps

?