@kirosnn/mosaic 0.0.91 → 0.73.0

package/src/utils/questionBridge.ts

@@ -1,12 +1,15 @@
 export interface QuestionOption {
   label: string;
   value?: string | null;
+  group?: string;
 }
 
 export interface QuestionRequest {
   id: string;
   prompt: string;
   options: QuestionOption[];
+  timeout?: number;
+  validation?: { pattern: string; message?: string };
 }
 
 export interface QuestionAnswer {
@@ -23,6 +26,7 @@ let currentRequest: QuestionRequest | null = null;
 let listeners = new Set<QuestionListener>();
 let pendingResolve: ((answer: QuestionAnswer) => void) | null = null;
 let pendingReject: ((reason?: any) => void) | null = null;
+let pendingTimeoutId: ReturnType<typeof setTimeout> | null = null;
 
 function notify(): void {
   for (const listener of listeners) {
@@ -46,7 +50,12 @@ export function getCurrentQuestion(): QuestionRequest | null {
   return currentRequest;
 }
 
-export async function askQuestion(prompt: string, options: QuestionOption[]): Promise<QuestionAnswer> {
+export async function askQuestion(
+  prompt: string,
+  options: QuestionOption[],
+  timeout?: number,
+  validation?: { pattern: string; message?: string },
+): Promise<QuestionAnswer> {
   if (pendingResolve) {
     throw new Error('A question is already pending');
   }
@@ -63,6 +72,8 @@ export async function askQuestion(prompt: string, options: QuestionOption[]): Pr
     id: createId(),
     prompt,
     options,
+    ...(timeout !== undefined && { timeout }),
+    ...(validation !== undefined && { validation }),
   };
 
   currentRequest = request;
@@ -71,6 +82,20 @@ export async function askQuestion(prompt: string, options: QuestionOption[]): Pr
   const answer = await new Promise<QuestionAnswer>((resolve, reject) => {
     pendingResolve = resolve;
     pendingReject = reject;
+
+    if (timeout !== undefined && timeout > 0) {
+      pendingTimeoutId = setTimeout(() => {
+        pendingTimeoutId = null;
+        if (pendingReject) {
+          const rej = pendingReject;
+          pendingResolve = null;
+          pendingReject = null;
+          currentRequest = null;
+          notify();
+          rej(new Error(`Question timed out after ${timeout}s`));
+        }
+      }, timeout * 1000);
+    }
   });
 
   return answer;
@@ -101,6 +126,11 @@ export function answerQuestion(index: number, customText?: string): void {
 
   if (!answer) return;
 
+  if (pendingTimeoutId !== null) {
+    clearTimeout(pendingTimeoutId);
+    pendingTimeoutId = null;
+  }
+
   const resolve = pendingResolve;
   pendingResolve = null;
   pendingReject = null;
@@ -112,6 +142,11 @@ export function answerQuestion(index: number, customText?: string): void {
 export function cancelQuestion(): void {
   if (!currentRequest || !pendingReject) return;
 
+  if (pendingTimeoutId !== null) {
+    clearTimeout(pendingTimeoutId);
+    pendingTimeoutId = null;
+  }
+
   const reject = pendingReject;
   pendingResolve = null;
   pendingReject = null;

package/src/utils/tokenEstimator.ts

@@ -0,0 +1,32 @@
+const CODE_CHARS = /[{}()\[\]=<>:;]/g;
+const CODE_DENSITY_THRESHOLD = 0.04;
+
+export function estimateTokensFromText(text: string): number {
+  if (!text) return 0;
+  const matches = text.match(CODE_CHARS);
+  const density = matches ? matches.length / text.length : 0;
+  const ratio = density >= CODE_DENSITY_THRESHOLD ? 2.8 : 3.3;
+  return Math.ceil(text.length / ratio);
+}
+
+export function estimateTokensForContent(content: string, thinkingContent?: string): number {
+  const contentTokens = estimateTokensFromText(content);
+  const thinkingTokens = thinkingContent ? estimateTokensFromText(thinkingContent) : 0;
+  return contentTokens + thinkingTokens + 4;
+}
+
+const CONTEXT_BUDGETS: Record<string, number> = {
+  anthropic: 180000,
+  openai: 115000,
+  google: 900000,
+  mistral: 28000,
+  xai: 117000,
+  ollama: 7000,
+};
+
+const DEFAULT_BUDGET = 12000;
+
+export function getDefaultContextBudget(provider?: string): number {
+  if (!provider) return DEFAULT_BUDGET;
+  return CONTEXT_BUDGETS[provider] ?? DEFAULT_BUDGET;
+}