DF_Final_Assignment_Template

Paused

DF_Final_Assignment_Template / tools /add_integers.py

Create add_integers.py

d276273 verified over 1 year ago

496 Bytes

from langchain_core.tools import tool
from loguru import logger

@tool("add_integers_tool", parse_docstring=True)
def add_integers(a: int, b: int) -> int:
    """
    Adds two integers together.
    Args:
        a (int): The first integer.
        b (int): The second integer.
    Returns:
        int: The sum of the two integers.
    """
    logger.info(f"add_integers called with a={a}, b={b}")
    return a + b


if __name__ == "__main__":
    print(add_integers.invoke({"a": 2, "b": 3}))

	from langchain_core.tools import tool
	from loguru import logger

	@tool("add_integers_tool", parse_docstring=True)
	def add_integers(a: int, b: int) -> int:
	"""
	Adds two integers together.

	Args:
	a (int): The first integer.
	b (int): The second integer.

	Returns:
	int: The sum of the two integers.
	"""
	logger.info(f"add_integers called with a={a}, b={b}")
	return a + b


	if __name__ == "__main__":
	print(add_integers.invoke({"a": 2, "b": 3}))