@garrix82/reactgenie-dsl 1.0.0 → 1.0.2

package/lib/nl/__test__/context-aware-prompt.test.ts

@@ -1,372 +0,0 @@
-import { ContextAwareDescriptorPromptGen, ExampleParse } from "../prompt-gen";
-
-function fakeDescriptor(
-  description: string,
-  genieObjectType: "DataClass" | "HelperClass" = "DataClass"
-) {
-  const match = description.match(/class\s+([A-Za-z_][A-Za-z0-9_]*)/);
-  return {
-    className: match?.[1] || "",
-    description: () => description,
-    classConstructor: {
-      __genieObjectType: genieObjectType,
-    },
-  } as any;
-}
-
-describe("ContextAwareDescriptorPromptGen", () => {
-  test("zero-shot prompt keeps filtered top-N classes and top-M examples", () => {
-    const prompt = new ContextAwareDescriptorPromptGen(
-      [
-        fakeDescriptor("class Restaurant { static Restaurant Current(); string name; }"),
-        fakeDescriptor("class Counter { static Counter Current(); void increment(); int count; }"),
-        fakeDescriptor("class Probe { static Probe Current(); string status; }"),
-      ],
-      [
-        new ExampleParse("increment counter", "Counter.Current().increment()"),
-        new ExampleParse("what is counter count", "Counter.Current().count"),
-        new ExampleParse("show restaurant", "Restaurant.Current().name"),
-        new ExampleParse("show probe status", "Probe.Current().status"),
-      ],
-      undefined,
-      {
-        topNClassDescriptions: 1,
-        topMExamples: 2,
-      }
-    );
-
-    const built = prompt.zero_shot_prompt("counter action");
-
-    expect(built).not.toContain("class Restaurant");
-    expect(built).toContain("class Counter");
-    expect(built).not.toContain("class Probe");
-
-    expect(built).toContain("increment counter");
-    expect(built).toContain("what is counter count");
-    expect(built).not.toContain("show restaurant");
-    expect(built).not.toContain("show probe status");
-  });
-
-  test("constrained selection does not fall back to all examples when no root examples match", () => {
-    const prompt = new ContextAwareDescriptorPromptGen(
-      [
-        fakeDescriptor("class Probe { static Probe Current(); string status; }"),
-        fakeDescriptor("class Counter { static Counter Current(); void increment(); }"),
-      ],
-      [new ExampleParse("increment counter", "Counter.Current().increment()")],
-      undefined,
-      {
-        topNClassDescriptions: 1,
-        topMExamples: 1,
-      }
-    );
-
-    prompt.setRuntimeContext({
-      tapClassName: "Probe",
-      visibleClassNames: ["Probe"],
-      isDeicticUtterance: true,
-    });
-
-    const built = prompt.prompt("show this status");
-    expect(built).not.toContain("increment counter");
-  });
-
-  test("selection does not auto-include helper classes beyond ranked top-N", () => {
-    const prompt = new ContextAwareDescriptorPromptGen(
-      [
-        fakeDescriptor(
-          "class ProbeDraftIntent { static ProbeDraftIntent Current(); ProbeDateTime sampledAt; }",
-          "DataClass"
-        ),
-        fakeDescriptor(
-          "class ProbeDateTime { static ProbeDateTime now(); int minute; }",
-          "HelperClass"
-        ),
-        fakeDescriptor(
-          "class DateTime { static DateTime now(); int minute; }",
-          "HelperClass"
-        ),
-      ],
-      [
-        new ExampleParse(
-          "set sampled at now",
-          "ProbeDraftIntent.Current().setSampledAt(value: ProbeDateTime.now())"
-        ),
-      ],
-      undefined,
-      {
-        topNClassDescriptions: 1,
-        topMExamples: 1,
-      }
-    );
-
-    const built = prompt.zero_shot_prompt("draft intent");
-    expect(built).toContain("class ProbeDraftIntent");
-    expect(built).not.toContain("class ProbeDateTime");
-    expect(built).not.toContain("class DateTime");
-  });
-
-  test("deictic utterance with no tap and no visible classes rescues self-resolving matches", () => {
-    const prompt = new ContextAwareDescriptorPromptGen(
-      [
-        fakeDescriptor(
-          "class Probe { static Probe[] getAllProbes(); static Probe Current(); string status; ProbeAnalysis openAnalysis(); }"
-        ),
-        fakeDescriptor(
-          "class ProbeAnalysis { static ProbeAnalysis openForProbe(string probeId); string probeId; }"
-        ),
-        fakeDescriptor("class Counter { static Counter Current(); void increment(); }"),
-      ],
-      [
-        new ExampleParse(
-          "show me the latest probe and start analysis for it",
-          "Probe.getAllProbes().sort(field: .receivedAt, ascending: false).index(index: 0).openAnalysis()"
-        ),
-        new ExampleParse("show probe status", "Probe.Current().status"),
-        new ExampleParse("increment counter", "Counter.Current().increment()"),
-      ]
-    );
-
-    prompt.setRuntimeContext({
-      tapClassName: undefined,
-      visibleClassNames: [],
-      isDeicticUtterance: true,
-    });
-
-    const built = prompt.prompt("show me the latest probe and start analysis for it");
-    expect(built).toContain("class Probe");
-    expect(built).not.toContain("class Counter");
-    expect(built).toContain("show me the latest probe and start analysis for it");
-    expect(built).not.toContain("show probe status");
-    expect(built).not.toContain("increment counter");
-  });
-
-  test("selects top-N class descriptions and top-M examples", () => {
-    const prompt = new ContextAwareDescriptorPromptGen(
-      [
-        fakeDescriptor("class Restaurant { static Restaurant[] all(); string name; int price; }"),
-        fakeDescriptor("class Counter { static Counter Current(); void increment(); }"),
-      ],
-      [
-        new ExampleParse("show me cheap restaurants", "Restaurant.all().sort(field: .price, ascending: true)"),
-        new ExampleParse("increment counter", "Counter.Current().increment()"),
-      ],
-      undefined,
-      {
-        topNClassDescriptions: 1,
-        topMExamples: 1,
-      }
-    );
-
-    const built = prompt.prompt("find restaurant by name");
-
-    expect(built).toContain("class Restaurant");
-    expect(built).not.toContain("class Counter");
-    expect(built).toContain("show me cheap restaurants");
-    expect(built).not.toContain("increment counter");
-  });
-
-  test("injects short structured history", () => {
-    const prompt = new ContextAwareDescriptorPromptGen(
-      [fakeDescriptor("class Order { static Order current(); void submit(); }")],
-      [new ExampleParse("submit order", "Order.current().submit()")],
-      undefined,
-      {
-        recentTurnsWindow: 2,
-      }
-    );
-
-    prompt.recordInteraction({
-      utterance: "open order",
-      parsed: "Order.current()",
-      success: true,
-      resultSummary: "execute_ok",
-    });
-    prompt.recordInteraction({
-      utterance: "submit it",
-      parsed: "Order.current().submit()",
-      success: true,
-      resultSummary: "execute_ok",
-    });
-
-    const built = prompt.prompt("submit again");
-    expect(built).toContain("// Conversation context");
-    expect(built).toContain("recent_turns:");
-    expect(built).toContain("user=open order");
-    expect(built).toContain("user=submit it");
-  });
-
-  test("response prompt is lightweight and excludes DSL context", () => {
-    const prompt = new ContextAwareDescriptorPromptGen(
-      [fakeDescriptor("class Restaurant { static Restaurant[] all(); }")],
-      [new ExampleParse("show restaurants", "Restaurant.all()")],
-      "extra-instruction"
-    );
-
-    const built = prompt.response_prompt(
-      "show cheapest restaurant",
-      "Restaurant.all().sort(field: .price, ascending: true)[0]",
-      '{"summary":"success"}'
-    );
-
-    expect(built).toContain("user: show cheapest restaurant");
-    expect(built).toContain("parsed: Restaurant.all().sort(field: .price, ascending: true)[0]");
-    expect(built).toContain("response_context:");
-    expect(built).not.toContain("// Here are all the functions");
-    expect(built).not.toContain("class Restaurant");
-    expect(built).not.toContain("show restaurants");
-    expect(built).not.toContain("// Conversation context");
-  });
-
-  test("parse prompt forbids quoted field arguments for array query operators", () => {
-    const prompt = new ContextAwareDescriptorPromptGen(
-      [fakeDescriptor("class Probe { static Probe[] getAllProbes(); string customerName; }")],
-      [new ExampleParse("show probes for customer Acme", 'Probe.getAllProbes().contains(field: .customerName, value: "Acme")')]
-    );
-
-    const built = prompt.prompt("show probes for customer Freshmarket");
-
-    expect(built).toContain("For matching, contains, equals, between, and sort, the field parameter MUST be a dotted accessor");
-    expect(built).toContain('✅ CORRECT:   Probe.getAllProbes().contains(field: .customerName, value: "Freshmarket")');
-    expect(built).toContain('❌ WRONG:     Probe.getAllProbes().contains(field: "customerName", value: "Freshmarket")');
-  });
-
-  test("injects runtime UI context and deictic guidance into parse prompt", () => {
-    const prompt = new ContextAwareDescriptorPromptGen(
-      [
-        fakeDescriptor("class Probe { static Probe Current(); string testPlanName; }"),
-        fakeDescriptor("class ProbeDraftIntent { static ProbeDraftIntent Current(); string testPlan; }"),
-      ],
-      [
-        new ExampleParse("what test plan is used", "Probe.getCurrent().testPlanName"),
-        new ExampleParse("what is the test plan", "ProbeDraftIntent.Current().testPlan"),
-      ]
-    );
-
-    prompt.setRuntimeContext({
-      tapClassName: "Probe",
-      visibleClassNames: ["Probe", "Probe[]"],
-      isDeicticUtterance: true,
-    });
-
-    const built = prompt.prompt("what is the test plan for this");
-    expect(built).toContain("// Runtime UI context");
-    expect(built).toContain("current_tap_class=Probe");
-    expect(built).toContain("visible_classes=Probe, Probe[]");
-    expect(built).toContain("deictic_utterance=true");
-    expect(built).toContain("deictic_rules:");
-  });
-
-  test("updates class/example selection when deictic runtime context changes", () => {
-    const prompt = new ContextAwareDescriptorPromptGen(
-      [
-        fakeDescriptor("class Restaurant { static Restaurant Current(); string name; }"),
-        fakeDescriptor("class Counter { static Counter Current(); int count; void increment(); }"),
-      ],
-      [
-        new ExampleParse("show this restaurant", "Restaurant.Current().name"),
-        new ExampleParse("show this counter", "Counter.Current().count"),
-      ],
-      undefined,
-      {
-        topNClassDescriptions: 1,
-        topMExamples: 1,
-      }
-    );
-
-    prompt.setRuntimeContext({
-      tapClassName: "Restaurant",
-      visibleClassNames: ["Restaurant"],
-      isDeicticUtterance: true,
-    });
-    const restaurantPrompt = prompt.prompt("show this");
-    expect(restaurantPrompt).toContain("class Restaurant");
-    expect(restaurantPrompt).toContain("show this restaurant");
-
-    prompt.setRuntimeContext({
-      tapClassName: "Counter",
-      visibleClassNames: ["Counter"],
-      isDeicticUtterance: true,
-    });
-    const counterPrompt = prompt.prompt("show this");
-    expect(counterPrompt).toContain("class Counter");
-    expect(counterPrompt).toContain("show this counter");
-  });
-
-  test("biases non-deictic selection toward a single visible root class", () => {
-    const prompt = new ContextAwareDescriptorPromptGen(
-      [
-        fakeDescriptor("class Probe { static Probe getCurrent(); Probe updateProbe(string batch); }"),
-        fakeDescriptor("class ProbeDraftIntent { static ProbeDraftIntent Current(); ProbeDraftIntent setBatch(string value); }"),
-      ],
-      [
-        new ExampleParse("update probe batch to B-245", 'Probe.getCurrent().updateProbe(batch: "B-245")'),
-        new ExampleParse("update the batch to B-245", 'ProbeDraftIntent.Current().setBatch(value: "B-245")'),
-      ],
-      undefined,
-      {
-        topNClassDescriptions: 1,
-        topMExamples: 1,
-      }
-    );
-
-    prompt.setRuntimeContext({
-      tapClassName: undefined,
-      visibleClassNames: ["Probe"],
-      isDeicticUtterance: false,
-    });
-
-    const built = prompt.prompt("update the batch to batch-2026-010");
-
-    expect(built).toContain("class Probe");
-    expect(built).not.toContain("class ProbeDraftIntent");
-    expect(built).toContain("update probe batch to B-245");
-    expect(built).not.toContain("update the batch to B-245");
-  });
-
-  test("does not auto-pin helper class descriptions or examples", () => {
-    const prompt = new ContextAwareDescriptorPromptGen(
-      [
-        fakeDescriptor(
-          "class ProbeDraftIntent { static ProbeDraftIntent Current(); ProbeDraftIntent setSampledAt(value: DateTime); DateTime sampledAt; }",
-        ),
-        fakeDescriptor(
-          "class DateTime { static DateTime now(); DateTime addDateOffset(minute: int? = `0`); }",
-        ),
-        fakeDescriptor("class Counter { static Counter Current(); void increment(); }"),
-      ],
-      [
-        new ExampleParse(
-          "set sampled at to now",
-          "ProbeDraftIntent.Current().setSampledAt(value: DateTime.now())",
-        ),
-        new ExampleParse(
-          "10 minutes from now",
-          "DateTime.now().addDateOffset(minute: 10)",
-        ),
-        new ExampleParse("increment counter", "Counter.Current().increment()"),
-      ],
-      undefined,
-      {
-        topNClassDescriptions: 1,
-        topMExamples: 1,
-      },
-    );
-
-    prompt.setRuntimeContext({
-      tapClassName: "ProbeDraftIntent",
-      visibleClassNames: ["ProbeDraftIntent"],
-      isDeicticUtterance: true,
-    });
-
-    const built = prompt.prompt("set sampled at to 10 minutes from now");
-
-    expect(built).toContain("class ProbeDraftIntent");
-    expect(built).not.toContain("class DateTime");
-    expect(built).toContain("set sampled at to now");
-    expect(built).toContain("10 minutes from now");
-    expect(built).not.toContain("DateTime.now().addDateOffset");
-    expect(built).not.toContain("class Counter");
-    expect(built).not.toContain("increment counter");
-  });
-});

package/lib/nl/__test__/context-selector.test.ts

@@ -1,27 +0,0 @@
-import { rankByQuery, scoreTextOverlap } from "../context-selector";
-
-describe("context selector", () => {
-  test("splits camelCase tokens for overlap scoring", () => {
-    const score = scoreTextOverlap("sampled at", "setSampledAt value");
-    expect(score).toBeGreaterThan(0);
-  });
-
-  test("splits snake_case tokens for overlap scoring", () => {
-    const score = scoreTextOverlap("sampled at", "sampled_at value");
-    expect(score).toBeGreaterThan(0);
-  });
-
-  test("rankByQuery can disable zero-score backfill", () => {
-    const ranked = rankByQuery(
-      "probe",
-      ["counter", "probe"],
-      (item) => item,
-      2,
-      { allowZeroScoreBackfill: false }
-    );
-
-    expect(ranked).toHaveLength(1);
-    expect(ranked[0].index).toBe(1);
-    expect(ranked[0].score).toBeGreaterThan(0);
-  });
-});

package/lib/nl/__test__/nl-parser-groq-transport.test.ts

@@ -1,139 +0,0 @@
-import { NlParserGroq } from "../nl-parser-groq";
-import { PromptGen } from "../prompt-gen";
-
-function toHeaderRecord(headers: any): Record<string, string> {
-  if (!headers) return {};
-  if (typeof headers.entries === "function") {
-    return Object.fromEntries(
-      Array.from(headers.entries()).map(([key, value]) => [
-        String(key).toLowerCase(),
-        String(value),
-      ])
-    );
-  }
-  return Object.fromEntries(
-    Object.entries(headers).map(([key, value]) => [
-      String(key).toLowerCase(),
-      String(value),
-    ])
-  );
-}
-
-const promptStub: PromptGen = {
-  prompt: () => "legacy prompt",
-  zero_shot_prompt: () => "parse prompt",
-  response_prompt: () => "respond prompt",
-};
-
-describe("NlParserGroq transport", () => {
-  const originalFetch = global.fetch;
-
-  afterEach(() => {
-    global.fetch = originalFetch;
-    jest.restoreAllMocks();
-  });
-
-  test("uses proxy chat completions endpoint without sdk stainless headers", async () => {
-    const fetchMock = jest.fn().mockResolvedValue(
-      new Response(
-        JSON.stringify({
-          choices: [{ message: { content: JSON.stringify({ command: "Light.on()" }) } }],
-          usage: { prompt_tokens: 10, completion_tokens: 4, total_tokens: 14 },
-        }),
-        {
-          status: 200,
-          headers: { "content-type": "application/json" },
-        }
-      )
-    );
-    global.fetch = fetchMock as any;
-
-    const parser = new NlParserGroq(
-      promptStub,
-      "proxy-token",
-      "llama-3.3-70b-versatile",
-      undefined,
-      "https://proxy.example.com/openai/v1"
-    );
-
-    const parsed = await parser.parse("turn the lights on");
-
-    expect(parsed).toBe("Light.on()");
-    expect(fetchMock).toHaveBeenCalledTimes(1);
-
-    const [url, init] = fetchMock.mock.calls[0] as [string, any];
-    expect(String(url)).toBe("https://proxy.example.com/openai/v1/chat/completions");
-    expect(init?.method).toBe("POST");
-
-    const headers = toHeaderRecord(init?.headers);
-    expect(headers.authorization).toBe("Bearer proxy-token");
-    expect(headers["x-genie-provider"]).toBe("groq");
-    expect(
-      Object.keys(headers).some((key) => key.startsWith("x-stainless-"))
-    ).toBe(false);
-
-    const body = JSON.parse(String(init?.body || "{}"));
-    expect(body.response_format?.type).toBe("json_schema");
-    expect(body.response_format?.json_schema?.name).toBe("dsl_command");
-  });
-
-  test("repairs one invalid command after semantic validation fails", async () => {
-    const validateCommand = jest
-      .fn<Promise<void>, [string]>()
-      .mockImplementation(async (command: string) => {
-        if (command === 'Light.on(field: "name")') {
-          throw new Error(
-            "Invalid DSL: matching(field: ...) requires a dotted accessor such as .name, but received string."
-          );
-        }
-      });
-
-    const fetchMock = jest
-      .fn()
-      .mockResolvedValueOnce(
-        new Response(
-          JSON.stringify({
-            choices: [{ message: { content: JSON.stringify({ command: 'Light.on(field: "name")' }) } }],
-            usage: {},
-          }),
-          {
-            status: 200,
-            headers: { "content-type": "application/json" },
-          }
-        )
-      )
-      .mockResolvedValueOnce(
-        new Response(
-          JSON.stringify({
-            choices: [{ message: { content: JSON.stringify({ command: "Light.on(field: .name)" }) } }],
-            usage: {},
-          }),
-          {
-            status: 200,
-            headers: { "content-type": "application/json" },
-          }
-        )
-      );
-    global.fetch = fetchMock as any;
-
-    const parser = new NlParserGroq(
-      promptStub,
-      "proxy-token",
-      "llama-3.3-70b-versatile",
-      undefined,
-      "https://proxy.example.com/openai/v1",
-      undefined,
-      undefined,
-      undefined,
-      validateCommand
-    );
-
-    await expect(parser.parse("turn the lights on")).resolves.toBe("Light.on(field: .name)");
-    expect(validateCommand).toHaveBeenCalledTimes(2);
-
-    const repairBody = JSON.parse(String(fetchMock.mock.calls[1][1]?.body || "{}"));
-    expect(repairBody.messages[1].content).toBe('Light.on(field: "name")');
-    expect(repairBody.messages[2].content).toContain("The previous DSL command is invalid.");
-    expect(repairBody.messages[2].content).toContain("requires a dotted accessor");
-  });
-});