Function Tools

import { tool } from "@llamaindex/core/tools";
import { z } from "zod";

const weatherTool = tool({
  name: "get_weather",
  description: "Get current weather for a location",
  parameters: z.object({
    location: z.string().describe("City name"),
    units: z.enum(["celsius", "fahrenheit"]).optional()
  }),
  execute: async ({ location, units = "celsius" }) => {
    // Fetch weather data
    const weather = await fetchWeather(location);
    return `Weather in ${location}: ${weather.temp}°${units === "celsius" ? "C" : "F"}`;
  }
});

function tool<T>(config: {
  name: string;
  description: string;
  parameters: ZodSchema<T>;
  execute: (input: T) => JSONValue | Promise<JSONValue>;
}): FunctionTool<T>

import { OpenAI } from "@llamaindex/openai";

const llm = new OpenAI({ model: "gpt-4" });

const response = await llm.chat({
  messages: [{ role: "user", content: "What's the weather in Paris?" }],
  tools: [weatherTool]
});

console.log(response.message.content);

import { Workflow, StartEvent, StopEvent } from "@llamaindex/workflow";
import { OpenAI } from "@llamaindex/openai";

class AgentWorkflow extends Workflow {
  private llm = new OpenAI({ model: "gpt-4" });
  private tools = [weatherTool, calculatorTool];
  
  async run(ctx: Context, ev: StartEvent) {
    const response = await this.llm.chat({
      messages: [{ role: "user", content: ev.message }],
      tools: this.tools
    });
    
    return new StopEvent({ result: response.message.content });
  }
}

const calculatorTool = tool({
  name: "calculator",
  description: "Perform mathematical calculations",
  parameters: z.object({
    expression: z.string().describe("Math expression to evaluate")
  }),
  execute: async ({ expression }) => {
    try {
      const result = eval(expression);
      return String(result);
    } catch (error) {
      return `Error: ${error.message}`;
    }
  }
});

const searchTool = tool({
  name: "web_search",
  description: "Search the web for information",
  parameters: z.object({
    query: z.string(),
    max_results: z.number().min(1).max(10).default(5)
  }),
  execute: async ({ query, max_results }) => {
    const results = await searchAPI(query, max_results);
    return JSON.stringify(results);
  }
});

const dbQueryTool = tool({
  name: "query_database",
  description: "Query the database",
  parameters: z.object({
    query: z.string().describe("SQL query to execute"),
    limit: z.number().optional()
  }),
  execute: async ({ query, limit }) => {
    const results = await db.query(query, limit);
    return JSON.stringify(results);
  }
});

const fileReaderTool = tool({
  name: "read_file",
  description: "Read contents of a file",
  parameters: z.object({
    filepath: z.string().describe("Path to the file")
  }),
  execute: async ({ filepath }) => {
    const fs = await import("fs/promises");
    const content = await fs.readFile(filepath, "utf-8");
    return content;
  }
});

const tool1 = tool({
  name: "simple_tool",
  description: "Tool with simple parameters",
  parameters: z.object({
    text: z.string(),
    count: z.number(),
    enabled: z.boolean()
  }),
  execute: async ({ text, count, enabled }) => {
    // ...
  }
});

const tool2 = tool({
  name: "optional_params",
  description: "Tool with optional parameters",
  parameters: z.object({
    required: z.string(),
    optional: z.string().optional(),
    withDefault: z.number().default(10)
  }),
  execute: async ({ required, optional, withDefault }) => {
    // ...
  }
});

const tool3 = tool({
  name: "enum_params",
  description: "Tool with enum parameters",
  parameters: z.object({
    action: z.enum(["create", "update", "delete"]),
    format: z.enum(["json", "xml", "yaml"])
  }),
  execute: async ({ action, format }) => {
    // ...
  }
});

const tool4 = tool({
  name: "array_params",
  description: "Tool with array parameters",
  parameters: z.object({
    items: z.array(z.string()),
    numbers: z.array(z.number()).min(1).max(10)
  }),
  execute: async ({ items, numbers }) => {
    // ...
  }
});

const tool5 = tool({
  name: "nested_params",
  description: "Tool with nested parameters",
  parameters: z.object({
    user: z.object({
      name: z.string(),
      email: z.string().email(),
      metadata: z.object({
        age: z.number(),
        country: z.string()
      })
    })
  }),
  execute: async ({ user }) => {
    // ...
  }
});

const robustTool = tool({
  name: "robust_tool",
  description: "Tool with error handling",
  parameters: z.object({
    input: z.string()
  }),
  execute: async ({ input }) => {
    try {
      const result = await riskyOperation(input);
      return JSON.stringify({ success: true, result });
    } catch (error) {
      return JSON.stringify({
        success: false,
        error: error.message
      });
    }
  }
});

const myTool = tool({
  name: "my_tool",
  description: "My custom tool",
  parameters: z.object({ input: z.string() }),
  execute: async ({ input }) => input
});

console.log(myTool.metadata.name);        // "my_tool"
console.log(myTool.metadata.description); // "My custom tool"
console.log(myTool.metadata.parameters);  // JSON Schema object

import { FunctionTool } from "@llamaindex/core/tools";

const customTool = new FunctionTool(
  async (input: { query: string }) => {
    return `Result for: ${input.query}`;
  },
  {
    name: "custom_tool",
    description: "A custom tool",
    parameters: {
      type: "object",
      properties: {
        query: { type: "string" }
      },
      required: ["query"]
    }
  }
);

import { VectorStoreIndex } from "llamaindex";

const index = await VectorStoreIndex.fromDocuments(documents);

const queryTool = tool({
  name: "query_documents",
  description: "Search the document index for relevant information",
  parameters: z.object({
    query: z.string().describe("Search query")
  }),
  execute: async ({ query }) => {
    const queryEngine = index.asQueryEngine();
    const response = await queryEngine.query({ query });
    return response.response;
  }
});

const llm = new OpenAI({ model: "gpt-4" });

const response = await llm.chat({
  messages: [{ role: "user", content: "Find information about pricing" }],
  tools: [queryTool]
});