src/model-config.ts

export interface ExamplePrompt {
  label: string;
  prompt: string;
}

export interface ModelConfig {
  modelId: string;
  label: string;
  description: string;
  examplePrompts: ExamplePrompt[];
}

export const MODEL_CONFIG: ModelConfig = {
  modelId: "onnx-community/NVIDIA-Nemotron-3-Nano-4B-BF16-ONNX",
  label: "Nemotron-3-Nano",
  description:
    "A 4B unified language model from NVIDIA for both direct answering and reasoning-heavy tasks.",
  examplePrompts: [
    {
      label: "Solve a logic puzzle",
      prompt:
        "Five people finished a race. Ava finished ahead of Ben but behind Chloe. Diego finished behind Ben but ahead of Emma. What was the finishing order?",
    },
    {
      label: "Summarize a meeting",
      prompt:
        "Turn these notes into a crisp follow-up email with action items:\n- launch moved to next Wednesday\n- Priya owns final QA\n- Marcus will update docs\n- follow-up check-in on Monday at 10 AM",
    },
    {
      label: "Write TypeScript code",
      prompt:
        "Write a clean TypeScript function that groups an array of objects by a given key and returns a Record. Include a short usage example.",
    },
    {
      label: "Explain a concept",
      prompt:
        "Explain the difference between caching and memoization in simple terms, then give one practical example of each.",
    },
  ],
};