AI-powered workflows from your own codebase

import { AIFunction, AIObject, AIFieldEnum, registerFunctionIntoAI } from "@sashimo/lib";

// Define a User object with multiple fields
const UserObject = new AIObject("User", "a user in the system")
  .field({
    name: "id",
    description: "unique user identifier",
    type: "number",
    required: true
  })
  .field({
    name: "email",
    description: "the email of the user",
    type: "string",
    required: true
  })
  .field({
    name: "status",
    description: "current user status",
    type: "string",
    required: false
  });

// Function to get user by ID
const GetUserByIdFunction = new AIFunction("get_user_by_id", "get a user by id")
  .args({
    name: "userId",
    description: "a user's id",
    type: "number",
    required: true
  })
  .returns(UserObject)
  .implement(async (userId: number) => {
    const user = await getUserById(userId);
    return user;
  });

// Function with enum parameter
const ChangeUserRoleFunction = new AIFunction("change_user_role", "change a user's role")
  .args(
    {
      name: "userId",
      description: "user's id",
      type: "number",
      required: true
    },
    new AIFieldEnum(
      "role",
      "the role to assign",
      ["ADMIN", "USER", "MODERATOR"],
      true
    )
  )
  .returns({
    name: "success",
    description: "operation result",
    type: "boolean"
  })
  .implement(async (userId: number, role: string) => {
    await updateUserRole(userId, role);
    return true;
  });

// Register functions
registerFunctionIntoAI("get_user_by_id", GetUserByIdFunction);
registerFunctionIntoAI("change_user_role", ChangeUserRoleFunction);