Platform to build and scale agents inside your product

import { workflow, llm, datasets, memory } from '@calljmp/agent';
import { z } from 'zod';

export function run(input: { message: string }) {
  return workflow.phase(
    'Chat',
    () => lm.generate({
      input: [
        {
          role: 'system',
          content: 'You are a copilot for a dashboard.',
        },
        {
          role: 'user',
          content: input.message,
        },
      ],
      memory: memory.short.context(),
      tools: [
        llm.tool({
          name: 'queryKnowledgeBase',
          description: 'Search product documentation and knowledge',
          parameters: z.object({
            query: z.string(),
          }),
          execute: async ({ query }) => {
            const { segments } = await datasets.query(query);
            return segments.map(segment => segment.content).join('\n---\n\n');
          },
        }),
      ],
    })
  );
}