Store and search data

import { LuaTool, Data } from 'lua-cli';
import { z } from 'zod';

export default class AddKnownIssueTool implements LuaTool {
  name = 'add_known_issue';
  description = 'Record a known product issue and its workaround so support can find it later.';
  inputSchema = z.object({
    product: z.string().describe('Product code, for example router-x1'),
    title: z.string().describe('One-line summary of the issue'),
    workaround: z.string().describe('What the customer can do until it is fixed'),
  });

  async execute(input: z.infer<typeof this.inputSchema>) {
    // Everything a support question might mention goes into the indexed text.
    const searchText = `${input.product} ${input.title} ${input.workaround}`;
    const entry = await Data.create('known-issues', { ...input, status: 'open' }, searchText);
    return { id: entry.id, product: entry.product, status: entry.status };
  }
}

import { LuaTool, Data } from 'lua-cli';
import { z } from 'zod';

export default class FindKnownIssueTool implements LuaTool {
  name = 'find_known_issue';
  description = "Find known issues that match a customer's description of a problem.";
  inputSchema = z.object({
    problem: z.string().describe("The problem in the customer's own words"),
  });

  async execute(input: z.infer<typeof this.inputSchema>) {
    const matches = await Data.search('known-issues', input.problem, 5, 0.7);
    return matches.map((match) => ({
      id: match.id,
      product: match.product,
      title: match.title,
      workaround: match.workaround,
      score: match.score,
    }));
  }
}

import { LuaTool, Data } from 'lua-cli';
import { z } from 'zod';

export default class ListOpenIssuesTool implements LuaTool {
  name = 'list_open_issues';
  description = 'List the unresolved known issues for one product.';
  inputSchema = z.object({
    product: z.string().describe('Product code, for example router-x1'),
  });

  async execute(input: z.infer<typeof this.inputSchema>) {
    const page = await Data.get(
      'known-issues',
      { product: input.product, status: { $ne: 'resolved' } },
      1,
      20
    );
    return {
      total: page.pagination.totalCount,
      issues: page.data.map((entry) => ({ id: entry.id, ...entry.data })),
    };
  }
}

import { LuaTool, Data } from 'lua-cli';
import { z } from 'zod';

export default class ResolveKnownIssueTool implements LuaTool {
  name = 'resolve_known_issue';
  description = 'Mark a known issue as resolved, or remove it when it was recorded by mistake.';
  inputSchema = z.object({
    id: z.string().describe('The entry id returned by add_known_issue'),
    remove: z.boolean().default(false).describe('True to delete the entry instead of resolving it'),
  });

  async execute(input: z.infer<typeof this.inputSchema>) {
    if (input.remove) {
      await Data.delete('known-issues', input.id);
      return { id: input.id, removed: true };
    }
    await Data.update('known-issues', input.id, {
      status: 'resolved',
      resolvedAt: new Date().toISOString(),
    });
    return { id: input.id, status: 'resolved' };
  }
}

import { LuaSkill } from 'lua-cli';
import AddKnownIssueTool from './tools/AddKnownIssueTool';
import FindKnownIssueTool from './tools/FindKnownIssueTool';
import ListOpenIssuesTool from './tools/ListOpenIssuesTool';
import ResolveKnownIssueTool from './tools/ResolveKnownIssueTool';

export default new LuaSkill({
  name: 'known-issues',
  description: 'Known product issues and their workarounds',
  context:
    'When a customer describes a problem, call find_known_issue with their words before answering ' +
    'and offer the workaround. Only support staff add or resolve issues; refer customers to support.',
  tools: [new AddKnownIssueTool(), new FindKnownIssueTool(), new ListOpenIssuesTool(), new ResolveKnownIssueTool()],
});

…
Tool returned: Array[1] of Object — fields: id, product, title, workaround, score
Output:
[
  {
    id: '7a112ce1-5c55-4973-b4f9-ec482d21b60c',
    product: 'router-x1',
    title: 'Wi-Fi drops every hour',
    workaround: 'Disable band steering in the admin panel',
    score: 0.734224796295166
  }
]