@trapar-waves/llm-template/src/index.ts

A template for LLM development integrating AI tools, TypeScript, Zod validation, and development utilities like Vitest and Rslib.

import { generateObject } from "ai";
import { z } from "zod";
import { iOpenAi } from "./model";
import { testPrompt } from "./prompt";

interface RunOptions {
  apiKey: string;
  baseURL?: string;
}

export async function run(message: string, options: RunOptions) {
  const openai = iOpenAi(options.apiKey, options.baseURL);
  return await generateObject({
    model: openai("qwen-2.5-max"),
    schema: z.object({
      markdown: z.string(),
    }),
    prompt: testPrompt(message),
  });
}