@garrix82/reactgenie-dsl 1.0.1 → 1.0.2

package/lib/nl/prompt-gen.ts

@@ -1,607 +0,0 @@
-import {
-  agent_response_prompt,
-  array_definition,
-  accessor_parameter_rules,
-  class_prequel,
-  class_separator,
-  example_parses,
-  example_prequel,
-  post_section_separator,
-  pre_section_separator,
-  user_interaction_prequel,
-  user_interaction_prompt,
-  parse_issues, parseCodePrompt
-} from "./prompt-res";
-import { ClassDescriptor, GenieObject } from "../dsl-descriptor";
-import { BoundedCache, normalizeCacheKey, rankByQuery } from "./context-selector";
-
-export interface PromptGen {
-  prompt(user_utterance: string): string;
-  zero_shot_prompt(user_utterance: string): string;
-
-  response_prompt(
-    user_utterance: string,
-    parsed: string,
-    result: string
-  ): string;
-}
-
-export type ResponseSeverity = "info" | "error" | "success" | "warning";
-
-export interface AgentResponsePayload {
-  minimalText: string;
-  fullText: string;
-  type: ResponseSeverity;
-  shouldSpeak: boolean;
-}
-
-export interface PromptInteractionRecord {
-  utterance: string;
-  parsed?: string | null;
-  success: boolean;
-  resultSummary?: string;
-}
-
-export interface PromptInteractionRecorder {
-  recordInteraction(record: PromptInteractionRecord): void;
-}
-
-export interface PromptSelectionConfig {
-  enabled: boolean;
-  topNClassDescriptions: number;
-  topMExamples: number;
-  recentTurnsWindow: number;
-  olderSummaryMaxChars: number;
-  cacheSize: number;
-}
-
-export interface RuntimeUiContext {
-  tapClassName?: string;
-  visibleClassNames: string[];
-  isDeicticUtterance: boolean;
-}
-
-export interface PromptRuntimeContextSetter {
-  setRuntimeContext(context: RuntimeUiContext | null): void;
-}
-
-interface PromptSelection {
-  classDescriptions: string[];
-  examples: ExampleParse[];
-}
-
-interface RankedEntry {
-  index: number;
-}
-
-const DEFAULT_PROMPT_SELECTION_CONFIG: PromptSelectionConfig = {
-  enabled: true,
-  topNClassDescriptions: 12,
-  topMExamples: 4,
-  recentTurnsWindow: 4,
-  olderSummaryMaxChars: 600,
-  cacheSize: 64,
-};
-
-export class ExampleParse {
-  constructor(readonly user_utterance, readonly example_parsed) {}
-}
-
-export class BasicPromptGen {
-  constructor(
-    private class_descriptions: string[],
-    private examples: ExampleParse[],
-    private extraPrompt?: string
-  ) {}
-
-  prompt_basic(): string {
-    let prompt =
-      `${class_prequel}${pre_section_separator}` +
-      `${array_definition}${class_separator}` +
-      `${this.class_descriptions.join(
-        class_separator
-      )}${post_section_separator}` +
-      `${example_prequel}${pre_section_separator}` +
-      `${example_parses(this.examples)}${post_section_separator}` +
-      `${accessor_parameter_rules}${post_section_separator}` +
-      // `${parse_issues}${post_section_separator}` +
-      `${user_interaction_prequel}${pre_section_separator}`;
-    if (this.extraPrompt !== undefined) {
-      prompt += `${this.extraPrompt}${pre_section_separator}`;
-    }
-    return prompt;
-  }
-
-  zero_shot_prompt_basic(): string {
-    return parseCodePrompt(this.prompt_basic())
-  }
-
-  prompt(user_utterance: string): string {
-    let prompt = this.prompt_basic();
-    prompt += `${user_interaction_prompt(user_utterance)}`;
-    return prompt;
-  }
-
-  zero_shot_prompt(user_utterance: string): string {
-    let prompt = this.zero_shot_prompt_basic();
-    prompt += `${user_interaction_prompt(user_utterance)}`;
-    return prompt;
-  }
-
-  response_prompt(
-    user_utterance: string,
-    parsed: string,
-    result: string
-  ): string {
-    return agent_response_prompt(user_utterance, parsed, result);
-  }
-}
-
-export class DescriptorPromptGen extends BasicPromptGen {
-  constructor(
-    private class_descriptors: ClassDescriptor<GenieObject>[],
-    examples: ExampleParse[],
-    extraPrompt?: string
-  ) {
-    super(
-      class_descriptors.map((c) => c.description()),
-      examples,
-      extraPrompt
-    );
-  }
-}
-
-export class ContextAwareDescriptorPromptGen
-  implements PromptGen, PromptInteractionRecorder, PromptRuntimeContextSetter
-{
-  private readonly allClassNames: string[];
-  private readonly allClassDescriptions: string[];
-  private readonly allExamples: ExampleParse[];
-  private readonly allExampleRootClasses: string[];
-  private readonly config: PromptSelectionConfig;
-  private readonly selectionCache: BoundedCache<PromptSelection>;
-  private recentTurns: PromptInteractionRecord[] = [];
-  private olderTurnsSummary = "";
-  private runtimeUiContext: RuntimeUiContext | null = null;
-
-  constructor(
-    class_descriptors: ClassDescriptor<GenieObject>[],
-    examples: ExampleParse[],
-    private extraPrompt?: string,
-    config?: Partial<PromptSelectionConfig>
-  ) {
-    this.allClassNames = class_descriptors.map((c) => c.className);
-    this.allClassDescriptions = class_descriptors.map((c) => c.description());
-    this.allExamples = examples;
-    this.allExampleRootClasses = examples.map((example) =>
-      ContextAwareDescriptorPromptGen.extractRootClass(example.example_parsed)
-    );
-    this.config = {
-      ...DEFAULT_PROMPT_SELECTION_CONFIG,
-      ...config,
-    };
-    this.selectionCache = new BoundedCache<PromptSelection>(this.config.cacheSize);
-  }
-
-  setRuntimeContext(context: RuntimeUiContext | null): void {
-    if (!context) {
-      this.runtimeUiContext = null;
-      return;
-    }
-
-    this.runtimeUiContext = {
-      tapClassName: context.tapClassName,
-      visibleClassNames: Array.from(
-        new Set((context.visibleClassNames || []).filter((entry) => !!entry))
-      ),
-      isDeicticUtterance: !!context.isDeicticUtterance,
-    };
-  }
-
-  private static normalizeClassName(className: string): string {
-    return className.replace(/\[\]$/g, "");
-  }
-
-  private static extractRootClass(parsedExpression: string): string {
-    const match = parsedExpression.match(/^\s*([A-Za-z_][A-Za-z0-9_]*)\s*\./);
-    return match?.[1] || "";
-  }
-
-  private static isScreenIndependentExample(example: ExampleParse): boolean {
-    return !/\b(?:AllCurrent|Current|getCurrent|current)\s*\(/.test(
-      example.example_parsed
-    );
-  }
-
-  private getAllowedRootClassesForDeicticContext(): Set<string> {
-    const context = this.runtimeUiContext;
-    const allowed = new Set<string>();
-    if (!context || !context.isDeicticUtterance) {
-      return allowed;
-    }
-
-    if (context.tapClassName) {
-      allowed.add(ContextAwareDescriptorPromptGen.normalizeClassName(context.tapClassName));
-      return allowed;
-    }
-
-    context.visibleClassNames
-      .map((className) => ContextAwareDescriptorPromptGen.normalizeClassName(className))
-      .forEach((className) => {
-        if (this.allClassNames.includes(className)) {
-          allowed.add(className);
-        }
-      });
-    return allowed;
-  }
-
-  private getPreferredRootClassesForVisibleScreenContext(): Set<string> {
-    const context = this.runtimeUiContext;
-    const preferred = new Set<string>();
-    if (!context || context.isDeicticUtterance) {
-      return preferred;
-    }
-
-    if (context.tapClassName) {
-      const normalizedTapClass = ContextAwareDescriptorPromptGen.normalizeClassName(
-        context.tapClassName
-      );
-      if (this.allClassNames.includes(normalizedTapClass)) {
-        preferred.add(normalizedTapClass);
-      }
-      return preferred;
-    }
-
-    const visibleRootClasses = Array.from(
-      new Set(
-        context.visibleClassNames
-          .map((className) =>
-            ContextAwareDescriptorPromptGen.normalizeClassName(className)
-          )
-          .filter((className) => this.allClassNames.includes(className))
-      )
-    );
-
-    if (visibleRootClasses.length === 1) {
-      preferred.add(visibleRootClasses[0]);
-    }
-    return preferred;
-  }
-
-  private buildRuntimeUiContextBlock(): string {
-    const context = this.runtimeUiContext;
-    if (!context) {
-      return "";
-    }
-    const hasExplicitContext =
-      !!context.tapClassName || context.visibleClassNames.length > 0;
-
-    const lines = ["// Runtime UI context"];
-    lines.push(
-      `ui_context: current_tap_class=${context.tapClassName || "none"}; visible_classes=${
-        context.visibleClassNames.length > 0
-          ? context.visibleClassNames.join(", ")
-          : "none"
-      }; deictic_utterance=${context.isDeicticUtterance ? "true" : "false"}`
-    );
-    lines.push("deictic_rules:");
-    lines.push(
-      "1. If deictic_utterance=true and current_tap_class exists, root the DSL chain on current_tap_class."
-    );
-    lines.push(
-      "2. If deictic_utterance=true and current_tap_class=none, prefer root classes that are in visible_classes."
-    );
-    lines.push(
-      "3. For deictic utterances, avoid root classes outside visible_classes unless the user explicitly names another class."
-    );
-    if (context.isDeicticUtterance && !hasExplicitContext) {
-      lines.push(
-        "4. No visible object context is available. Do not use Current()/AllCurrent()/getCurrent(); resolve the command from the utterance itself."
-      );
-    }
-    lines.push("deictic_examples:");
-    lines.push('- user: "show this item" parsed: Item.Current()');
-    lines.push('- user: "update this status to done" parsed: Task.Current().setStatus(value: "done")');
-    return lines.join("\n");
-  }
-
-  private buildPromptBasic(
-    classDescriptions: string[],
-    examples: ExampleParse[],
-    historyContext: string,
-    uiContext: string
-  ): string {
-    let prompt =
-      `${class_prequel}${pre_section_separator}` +
-      `${array_definition}${class_separator}` +
-      `${classDescriptions.join(class_separator)}${post_section_separator}` +
-      `${example_prequel}${pre_section_separator}` +
-      `${example_parses(examples)}${post_section_separator}` +
-      `${accessor_parameter_rules}${post_section_separator}` +
-      `${user_interaction_prequel}${pre_section_separator}`;
-
-    if (historyContext) {
-      prompt += `${historyContext}${post_section_separator}`;
-    }
-
-    if (uiContext) {
-      prompt += `${uiContext}${post_section_separator}`;
-    }
-
-    if (this.extraPrompt !== undefined) {
-      prompt += `${this.extraPrompt}${pre_section_separator}`;
-    }
-
-    return prompt;
-  }
-
-  private buildHistoryContext(): string {
-    if (this.recentTurns.length === 0 && !this.olderTurnsSummary) {
-      return "";
-    }
-
-    const sections: string[] = ["// Conversation context"]; 
-    if (this.olderTurnsSummary) {
-      sections.push(`summary: ${this.olderTurnsSummary}`);
-    }
-
-    if (this.recentTurns.length > 0) {
-      sections.push("recent_turns:");
-      this.recentTurns.forEach((turn, index) => {
-        const parsed = turn.parsed ? ` parsed=${turn.parsed}` : "";
-        const result = turn.resultSummary ? ` result=${turn.resultSummary}` : "";
-        sections.push(
-          `- t${index + 1}: user=${turn.utterance} success=${turn.success}${parsed}${result}`
-        );
-      });
-    }
-
-    return sections.join("\n");
-  }
-
-  private summarizeRecord(record: PromptInteractionRecord): string {
-    const parsed = record.parsed ? `parsed=${record.parsed}` : "parsed=<none>";
-    const result = record.resultSummary
-      ? `result=${record.resultSummary}`
-      : "result=<none>";
-    return `user=${record.utterance} success=${record.success} ${parsed} ${result}`;
-  }
-
-  private appendOlderSummary(summaryLine: string): void {
-    const next = this.olderTurnsSummary
-      ? `${this.olderTurnsSummary} | ${summaryLine}`
-      : summaryLine;
-
-    if (next.length <= this.config.olderSummaryMaxChars) {
-      this.olderTurnsSummary = next;
-      return;
-    }
-
-    this.olderTurnsSummary = next.slice(next.length - this.config.olderSummaryMaxChars);
-  }
-
-  private selectionFromCache(cacheKey: string): PromptSelection | undefined {
-    const key = normalizeCacheKey(cacheKey);
-    if (!key) {
-      return undefined;
-    }
-    return this.selectionCache.get(key);
-  }
-
-  private cacheSelection(cacheKey: string, selection: PromptSelection): void {
-    const key = normalizeCacheKey(cacheKey);
-    if (!key) {
-      return;
-    }
-    this.selectionCache.set(key, selection);
-  }
-
-  private selectionCacheKeyForUtterance(user_utterance: string): string {
-    const context = this.runtimeUiContext;
-    if (!context) {
-      return user_utterance;
-    }
-
-    const preferredVisibleRoots = this.getPreferredRootClassesForVisibleScreenContext();
-    if (!context.isDeicticUtterance && preferredVisibleRoots.size === 0) {
-      return user_utterance;
-    }
-
-    const visibleKey = [...context.visibleClassNames].sort().join(",");
-    return `${user_utterance}::tap=${context.tapClassName || ""}::visible=${visibleKey}::deictic=${
-      context.isDeicticUtterance ? 1 : 0
-    }`;
-  }
-
-  private buildSelectionFromExampleIndexes(exampleIndexes: number[]): PromptSelection {
-    if (exampleIndexes.length === 0) {
-      return {
-        classDescriptions: [],
-        examples: [],
-      };
-    }
-
-    const uniqueExampleIndexes = Array.from(new Set(exampleIndexes)).sort((a, b) => a - b);
-    const examples = uniqueExampleIndexes.map((index) => this.allExamples[index]);
-    const exampleRootClasses = new Set(
-      uniqueExampleIndexes
-        .map((index) => this.allExampleRootClasses[index])
-        .filter((className) => className.length > 0)
-    );
-    const classDescriptions = this.allClassNames
-      .map((className, index) => ({ className, index }))
-      .filter((entry) => exampleRootClasses.has(entry.className))
-      .map((entry) => this.allClassDescriptions[entry.index]);
-
-    return {
-      classDescriptions,
-      examples,
-    };
-  }
-
-  private rescueSelectionWithoutVisibleContext(
-    user_utterance: string,
-    rankedExamples: RankedEntry[]
-  ): PromptSelection {
-    const rescuedExampleIndexes = rankedExamples
-      .filter((entry) =>
-        ContextAwareDescriptorPromptGen.isScreenIndependentExample(
-          this.allExamples[entry.index]
-        )
-      )
-      .slice(0, this.config.topMExamples)
-      .map((entry) => entry.index);
-
-    return this.buildSelectionFromExampleIndexes(rescuedExampleIndexes);
-  }
-
-  private computeSelection(user_utterance: string): PromptSelection {
-    if (!this.config.enabled) {
-      return {
-        classDescriptions: this.allClassDescriptions,
-        examples: this.allExamples,
-      };
-    }
-
-    const cacheKey = this.selectionCacheKeyForUtterance(user_utterance);
-    const cached = this.selectionFromCache(cacheKey);
-    if (cached) {
-      return cached;
-    }
-
-    const allowedDeicticRootClasses = this.getAllowedRootClassesForDeicticContext();
-    const preferredVisibleRootClasses =
-      this.getPreferredRootClassesForVisibleScreenContext();
-    const contextRootClasses =
-      allowedDeicticRootClasses.size > 0
-        ? allowedDeicticRootClasses
-        : preferredVisibleRootClasses;
-    const isDeicticUtterance = !!this.runtimeUiContext?.isDeicticUtterance;
-    const isContextConstrained = contextRootClasses.size > 0;
-    const rankingQuery = isContextConstrained
-      ? `${user_utterance} ${Array.from(contextRootClasses).join(" ")}`
-      : user_utterance;
-
-    const rankedClassDescriptions = rankByQuery(
-      rankingQuery,
-      this.allClassDescriptions,
-      (description) => description,
-      isContextConstrained
-        ? this.allClassDescriptions.length
-        : this.config.topNClassDescriptions,
-      {
-        allowZeroScoreBackfill: !isContextConstrained,
-      }
-    );
-    const candidateClassEntries = isContextConstrained
-      ? rankedClassDescriptions.filter((entry) =>
-          contextRootClasses.has(this.allClassNames[entry.index])
-        )
-      : rankedClassDescriptions;
-    const classIndexes = candidateClassEntries
-      .slice(0, this.config.topNClassDescriptions)
-      .map((entry) => entry.index)
-      .sort((a, b) => a - b);
-    const selectedClassDescriptions = classIndexes.map(
-      (index) => this.allClassDescriptions[index]
-    );
-
-    const rankedExamples = rankByQuery(
-      rankingQuery,
-      this.allExamples,
-      (example) => `${example.user_utterance}\n${example.example_parsed}`,
-      isContextConstrained ? this.allExamples.length : this.config.topMExamples,
-      {
-        allowZeroScoreBackfill: !isContextConstrained,
-      }
-    );
-    const candidateExampleEntries = isContextConstrained
-      ? rankedExamples.filter((entry) =>
-          contextRootClasses.has(this.allExampleRootClasses[entry.index])
-        )
-      : rankedExamples;
-
-    if (isDeicticUtterance && contextRootClasses.size === 0) {
-      const rescuedSelection = this.rescueSelectionWithoutVisibleContext(
-        user_utterance,
-        rankedExamples
-      );
-      this.cacheSelection(cacheKey, rescuedSelection);
-      return rescuedSelection;
-    }
-
-    const exampleIndexes = candidateExampleEntries
-      .slice(0, this.config.topMExamples)
-      .map((entry) => entry.index)
-      .sort((a, b) => a - b);
-    const selectedExamples = exampleIndexes.map((index) => this.allExamples[index]);
-
-    const selection = {
-      classDescriptions:
-        selectedClassDescriptions.length > 0
-          ? selectedClassDescriptions
-          : isDeicticUtterance
-            ? []
-            : this.allClassDescriptions,
-      examples:
-        selectedExamples.length > 0
-          ? selectedExamples
-          : isContextConstrained
-            ? []
-            : this.allExamples,
-    };
-
-    this.cacheSelection(cacheKey, selection);
-    return selection;
-  }
-
-  private zeroShotPromptForSelection(
-    user_utterance: string,
-    selection: PromptSelection
-  ): string {
-    const historyContext = this.buildHistoryContext();
-    const uiContext = this.buildRuntimeUiContextBlock();
-    const basic = this.buildPromptBasic(
-      selection.classDescriptions,
-      selection.examples,
-      historyContext,
-      uiContext
-    );
-    return `${parseCodePrompt(basic)}${user_interaction_prompt(user_utterance)}`;
-  }
-
-  prompt(user_utterance: string): string {
-    const selection = this.computeSelection(user_utterance);
-    const historyContext = this.buildHistoryContext();
-    const uiContext = this.buildRuntimeUiContextBlock();
-    const basic = this.buildPromptBasic(
-      selection.classDescriptions,
-      selection.examples,
-      historyContext,
-      uiContext
-    );
-    return `${basic}${user_interaction_prompt(user_utterance)}`;
-  }
-
-  zero_shot_prompt(user_utterance: string): string {
-    const selection = this.computeSelection(user_utterance);
-    return this.zeroShotPromptForSelection(user_utterance, selection);
-  }
-
-  response_prompt(user_utterance: string, parsed: string, result: string): string {
-    return agent_response_prompt(user_utterance, parsed, result);
-  }
-
-  recordInteraction(record: PromptInteractionRecord): void {
-    this.recentTurns.push(record);
-
-    while (this.recentTurns.length > this.config.recentTurnsWindow) {
-      const shifted = this.recentTurns.shift();
-      if (shifted) {
-        this.appendOlderSummary(this.summarizeRecord(shifted));
-      }
-    }
-  }
-}
-
-// TODO: automatically generate prompt from DSL Descriptors