Tools & Function Tools

from dynamiq.connections import OpenAI as OpenAIConnection
from dynamiq.nodes.agents import Agent
from dynamiq.nodes.llms import OpenAI
from dynamiq.nodes.tools.function_tool import function_tool

@function_tool
def multiply_numbers(a: int, b: int, **kwargs) -> int:
    """Multiply two numbers together."""
    return a * b

llm = OpenAI(connection=OpenAIConnection(), model="gpt-4o")

agent = Agent(
    name="math-agent",
    llm=llm,
    tools=[multiply_numbers()],  # call the decorated function to instantiate the tool
    max_loops=4,
)

result = agent.run(input_data={"input": "What is 6 times 7?"})
print(result.output["content"])

from pydantic import BaseModel

from dynamiq.nodes.tools.function_tool import FunctionTool

class AddNumbersInputSchema(BaseModel):
    a: int = -1
    b: int = -1

class AddNumbersTool(FunctionTool):
    name: str = "Add Numbers Tool"
    description: str = "A tool that adds two numbers together."

    def run_func(self, input_data: AddNumbersInputSchema, **kwargs) -> int:
        """Add two numbers together."""
        return input_data.a + input_data.b

from typing import Any, Literal

import sympy as sp
from pydantic import ConfigDict

from dynamiq.nodes import NodeGroup
from dynamiq.nodes.node import Node, ensure_config
from dynamiq.runnables import RunnableConfig

class CalculatorTool(Node):
    group: Literal[NodeGroup.TOOLS] = NodeGroup.TOOLS
    name: str = "Calculator"
    description: str = (
        "Tool to evaluate mathematical expressions. "
        "Provide a raw string of the math operation to parse and evaluate."
    )

    model_config = ConfigDict(arbitrary_types_allowed=True)

    def execute(
        self, input_data: dict[str, Any], config: RunnableConfig = None, **kwargs
    ) -> dict[str, Any]:
        config = ensure_config(config)
        self.run_on_node_execute_run(config.callbacks, **kwargs)

        expression = input_data.get("input", "")
        try:
            result = sp.sympify(expression)
        except Exception as e:
            result = str(e)
        return {"content": result}