rag-v2-improved

import { createConfigSchematics } from "@lmstudio/sdk";

export const configSchematics = createConfigSchematics()
  .field(
    "retrievalLimit",
    "numeric",
    {
      int: true,
      min: 1,
      displayName: "Retrieval Limit",
      subtitle: "Maximum number of retrieved chunks.",
      slider: { min: 1, max: 12, step: 1 },
    },
    5,
  )
  .field(
    "retrievalAffinityThreshold",
    "numeric",
    {
      min: 0,
      max: 1,
      displayName: "Retrieval Threshold",
      subtitle: "Minimum similarity score required.",
      slider: { min: 0, max: 1, step: 0.05 },
    },
    0.55,
  )
  .build();