@superatomai/sdk-node 0.0.3 → 0.0.4-dsp

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.d.mts CHANGED
@@ -1,4 +1,78 @@
1
1
  import { z } from 'zod';
2
+ import Anthropic from '@anthropic-ai/sdk';
3
+
4
+ /**
5
+ * Unified UIBlock structure for database storage
6
+ * Used in both bookmarks and user-conversations tables
7
+ */
8
+ interface DBUIBlock {
9
+ id: string;
10
+ component: Record<string, any> | null;
11
+ analysis: string | null;
12
+ user_prompt: string;
13
+ }
14
+
15
+ /**
16
+ * Log levels in hierarchical order
17
+ * - errors: only error logs
18
+ * - warnings: warning + error logs
19
+ * - info: info + warning + error logs
20
+ * - verbose: all logs including debug
21
+ */
22
+ type LogLevel = 'errors' | 'warnings' | 'info' | 'verbose';
23
+ /**
24
+ * Logger class with environment-based log level support
25
+ */
26
+ declare class Logger {
27
+ private currentLevel;
28
+ private currentLevelPriority;
29
+ constructor();
30
+ /**
31
+ * Check if a string is a valid log level
32
+ */
33
+ private isValidLogLevel;
34
+ /**
35
+ * Check if a message should be logged based on current log level
36
+ */
37
+ private shouldLog;
38
+ /**
39
+ * Get current log level
40
+ */
41
+ getLogLevel(): LogLevel;
42
+ /**
43
+ * Set log level programmatically
44
+ */
45
+ setLogLevel(level: LogLevel): void;
46
+ /**
47
+ * Log info message (shown for info and verbose levels)
48
+ */
49
+ info(...args: any[]): void;
50
+ /**
51
+ * Log error message (shown for all levels)
52
+ */
53
+ error(...args: any[]): void;
54
+ /**
55
+ * Log warning message (shown for warnings, info, and verbose levels)
56
+ */
57
+ warn(...args: any[]): void;
58
+ /**
59
+ * Log debug message (only shown for verbose level)
60
+ */
61
+ debug(...args: any[]): void;
62
+ /**
63
+ * Write to log file
64
+ */
65
+ file(...args: any[]): void;
66
+ /**
67
+ * Clear the log file (call at start of new user request)
68
+ */
69
+ clearFile(): void;
70
+ /**
71
+ * Log LLM method prompts with clear labeling
72
+ */
73
+ logLLMPrompt(methodName: string, promptType: 'system' | 'user', content: string | object | any[]): void;
74
+ }
75
+ declare const logger: Logger;
2
76
 
3
77
  declare const DSLRendererPropsSchema$1: z.ZodObject<{
4
78
  dsl: z.ZodObject<{
@@ -27,7 +101,27 @@ declare const DSLRendererPropsSchema$1: z.ZodObject<{
27
101
  deps?: string[] | undefined;
28
102
  }>, "many">>;
29
103
  data: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
30
- render: z.ZodType<any, z.ZodTypeDef, any>;
104
+ render: z.ZodOptional<z.ZodType<any, z.ZodTypeDef, any>>;
105
+ pages: z.ZodOptional<z.ZodArray<z.ZodObject<{
106
+ id: z.ZodString;
107
+ name: z.ZodString;
108
+ order: z.ZodNumber;
109
+ icon: z.ZodOptional<z.ZodString>;
110
+ render: z.ZodType<any, z.ZodTypeDef, any>;
111
+ }, "strip", z.ZodTypeAny, {
112
+ id: string;
113
+ name: string;
114
+ order: number;
115
+ icon?: string | undefined;
116
+ render?: any;
117
+ }, {
118
+ id: string;
119
+ name: string;
120
+ order: number;
121
+ icon?: string | undefined;
122
+ render?: any;
123
+ }>, "many">>;
124
+ defaultPageId: z.ZodOptional<z.ZodString>;
31
125
  query: z.ZodOptional<z.ZodObject<{
32
126
  graphql: z.ZodOptional<z.ZodString>;
33
127
  sql: z.ZodOptional<z.ZodString>;
@@ -66,6 +160,7 @@ declare const DSLRendererPropsSchema$1: z.ZodObject<{
66
160
  dependencies?: string[] | undefined;
67
161
  } | undefined;
68
162
  props?: Record<string, any> | undefined;
163
+ render?: any;
69
164
  states?: Record<string, any> | undefined;
70
165
  methods?: Record<string, {
71
166
  fn: string;
@@ -76,7 +171,14 @@ declare const DSLRendererPropsSchema$1: z.ZodObject<{
76
171
  deps?: string[] | undefined;
77
172
  }[] | undefined;
78
173
  data?: Record<string, any> | undefined;
79
- render?: any;
174
+ pages?: {
175
+ id: string;
176
+ name: string;
177
+ order: number;
178
+ icon?: string | undefined;
179
+ render?: any;
180
+ }[] | undefined;
181
+ defaultPageId?: string | undefined;
80
182
  }, {
81
183
  id: string;
82
184
  name?: string | undefined;
@@ -90,6 +192,7 @@ declare const DSLRendererPropsSchema$1: z.ZodObject<{
90
192
  dependencies?: string[] | undefined;
91
193
  } | undefined;
92
194
  props?: Record<string, any> | undefined;
195
+ render?: any;
93
196
  states?: Record<string, any> | undefined;
94
197
  methods?: Record<string, {
95
198
  fn: string;
@@ -100,7 +203,14 @@ declare const DSLRendererPropsSchema$1: z.ZodObject<{
100
203
  deps?: string[] | undefined;
101
204
  }[] | undefined;
102
205
  data?: Record<string, any> | undefined;
103
- render?: any;
206
+ pages?: {
207
+ id: string;
208
+ name: string;
209
+ order: number;
210
+ icon?: string | undefined;
211
+ render?: any;
212
+ }[] | undefined;
213
+ defaultPageId?: string | undefined;
104
214
  }>;
105
215
  data: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
106
216
  context: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
@@ -118,6 +228,7 @@ declare const DSLRendererPropsSchema$1: z.ZodObject<{
118
228
  dependencies?: string[] | undefined;
119
229
  } | undefined;
120
230
  props?: Record<string, any> | undefined;
231
+ render?: any;
121
232
  states?: Record<string, any> | undefined;
122
233
  methods?: Record<string, {
123
234
  fn: string;
@@ -128,7 +239,14 @@ declare const DSLRendererPropsSchema$1: z.ZodObject<{
128
239
  deps?: string[] | undefined;
129
240
  }[] | undefined;
130
241
  data?: Record<string, any> | undefined;
131
- render?: any;
242
+ pages?: {
243
+ id: string;
244
+ name: string;
245
+ order: number;
246
+ icon?: string | undefined;
247
+ render?: any;
248
+ }[] | undefined;
249
+ defaultPageId?: string | undefined;
132
250
  };
133
251
  data?: Record<string, any> | undefined;
134
252
  context?: Record<string, any> | undefined;
@@ -146,6 +264,7 @@ declare const DSLRendererPropsSchema$1: z.ZodObject<{
146
264
  dependencies?: string[] | undefined;
147
265
  } | undefined;
148
266
  props?: Record<string, any> | undefined;
267
+ render?: any;
149
268
  states?: Record<string, any> | undefined;
150
269
  methods?: Record<string, {
151
270
  fn: string;
@@ -156,7 +275,14 @@ declare const DSLRendererPropsSchema$1: z.ZodObject<{
156
275
  deps?: string[] | undefined;
157
276
  }[] | undefined;
158
277
  data?: Record<string, any> | undefined;
159
- render?: any;
278
+ pages?: {
279
+ id: string;
280
+ name: string;
281
+ order: number;
282
+ icon?: string | undefined;
283
+ render?: any;
284
+ }[] | undefined;
285
+ defaultPageId?: string | undefined;
160
286
  };
161
287
  data?: Record<string, any> | undefined;
162
288
  context?: Record<string, any> | undefined;
@@ -229,6 +355,7 @@ declare const DSLRendererPropsSchema: z.ZodObject<{
229
355
  dependencies?: string[] | undefined;
230
356
  } | undefined;
231
357
  props?: Record<string, any> | undefined;
358
+ render?: any;
232
359
  states?: Record<string, any> | undefined;
233
360
  methods?: Record<string, {
234
361
  fn: string;
@@ -239,7 +366,6 @@ declare const DSLRendererPropsSchema: z.ZodObject<{
239
366
  deps?: string[] | undefined;
240
367
  }[] | undefined;
241
368
  data?: Record<string, any> | undefined;
242
- render?: any;
243
369
  }, {
244
370
  id: string;
245
371
  name?: string | undefined;
@@ -253,6 +379,7 @@ declare const DSLRendererPropsSchema: z.ZodObject<{
253
379
  dependencies?: string[] | undefined;
254
380
  } | undefined;
255
381
  props?: Record<string, any> | undefined;
382
+ render?: any;
256
383
  states?: Record<string, any> | undefined;
257
384
  methods?: Record<string, {
258
385
  fn: string;
@@ -263,7 +390,6 @@ declare const DSLRendererPropsSchema: z.ZodObject<{
263
390
  deps?: string[] | undefined;
264
391
  }[] | undefined;
265
392
  data?: Record<string, any> | undefined;
266
- render?: any;
267
393
  }>;
268
394
  data: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
269
395
  context: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
@@ -281,6 +407,7 @@ declare const DSLRendererPropsSchema: z.ZodObject<{
281
407
  dependencies?: string[] | undefined;
282
408
  } | undefined;
283
409
  props?: Record<string, any> | undefined;
410
+ render?: any;
284
411
  states?: Record<string, any> | undefined;
285
412
  methods?: Record<string, {
286
413
  fn: string;
@@ -291,7 +418,6 @@ declare const DSLRendererPropsSchema: z.ZodObject<{
291
418
  deps?: string[] | undefined;
292
419
  }[] | undefined;
293
420
  data?: Record<string, any> | undefined;
294
- render?: any;
295
421
  };
296
422
  data?: Record<string, any> | undefined;
297
423
  context?: Record<string, any> | undefined;
@@ -309,6 +435,7 @@ declare const DSLRendererPropsSchema: z.ZodObject<{
309
435
  dependencies?: string[] | undefined;
310
436
  } | undefined;
311
437
  props?: Record<string, any> | undefined;
438
+ render?: any;
312
439
  states?: Record<string, any> | undefined;
313
440
  methods?: Record<string, {
314
441
  fn: string;
@@ -319,13 +446,86 @@ declare const DSLRendererPropsSchema: z.ZodObject<{
319
446
  deps?: string[] | undefined;
320
447
  }[] | undefined;
321
448
  data?: Record<string, any> | undefined;
322
- render?: any;
323
449
  };
324
450
  data?: Record<string, any> | undefined;
325
451
  context?: Record<string, any> | undefined;
326
452
  }>;
327
453
  type DSLRendererProps = z.infer<typeof DSLRendererPropsSchema>;
328
454
 
455
+ declare const UserSchema: z.ZodObject<{
456
+ username: z.ZodString;
457
+ email: z.ZodOptional<z.ZodString>;
458
+ password: z.ZodString;
459
+ fullname: z.ZodOptional<z.ZodString>;
460
+ role: z.ZodOptional<z.ZodString>;
461
+ userInfo: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
462
+ wsIds: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
463
+ }, "strip", z.ZodTypeAny, {
464
+ username: string;
465
+ password: string;
466
+ email?: string | undefined;
467
+ fullname?: string | undefined;
468
+ role?: string | undefined;
469
+ userInfo?: Record<string, unknown> | undefined;
470
+ wsIds?: string[] | undefined;
471
+ }, {
472
+ username: string;
473
+ password: string;
474
+ email?: string | undefined;
475
+ fullname?: string | undefined;
476
+ role?: string | undefined;
477
+ userInfo?: Record<string, unknown> | undefined;
478
+ wsIds?: string[] | undefined;
479
+ }>;
480
+ type User = z.infer<typeof UserSchema>;
481
+ declare const UsersDataSchema: z.ZodObject<{
482
+ users: z.ZodArray<z.ZodObject<{
483
+ username: z.ZodString;
484
+ email: z.ZodOptional<z.ZodString>;
485
+ password: z.ZodString;
486
+ fullname: z.ZodOptional<z.ZodString>;
487
+ role: z.ZodOptional<z.ZodString>;
488
+ userInfo: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
489
+ wsIds: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
490
+ }, "strip", z.ZodTypeAny, {
491
+ username: string;
492
+ password: string;
493
+ email?: string | undefined;
494
+ fullname?: string | undefined;
495
+ role?: string | undefined;
496
+ userInfo?: Record<string, unknown> | undefined;
497
+ wsIds?: string[] | undefined;
498
+ }, {
499
+ username: string;
500
+ password: string;
501
+ email?: string | undefined;
502
+ fullname?: string | undefined;
503
+ role?: string | undefined;
504
+ userInfo?: Record<string, unknown> | undefined;
505
+ wsIds?: string[] | undefined;
506
+ }>, "many">;
507
+ }, "strip", z.ZodTypeAny, {
508
+ users: {
509
+ username: string;
510
+ password: string;
511
+ email?: string | undefined;
512
+ fullname?: string | undefined;
513
+ role?: string | undefined;
514
+ userInfo?: Record<string, unknown> | undefined;
515
+ wsIds?: string[] | undefined;
516
+ }[];
517
+ }, {
518
+ users: {
519
+ username: string;
520
+ password: string;
521
+ email?: string | undefined;
522
+ fullname?: string | undefined;
523
+ role?: string | undefined;
524
+ userInfo?: Record<string, unknown> | undefined;
525
+ wsIds?: string[] | undefined;
526
+ }[];
527
+ }>;
528
+ type UsersData = z.infer<typeof UsersDataSchema>;
329
529
  declare const MessageSchema: z.ZodObject<{
330
530
  id: z.ZodOptional<z.ZodString>;
331
531
  type: z.ZodString;
@@ -426,30 +626,441 @@ declare const IncomingMessageSchema: z.ZodObject<{
426
626
  payload?: unknown;
427
627
  }>;
428
628
  type IncomingMessage = z.infer<typeof IncomingMessageSchema>;
629
+ declare const ComponentSchema: z.ZodObject<{
630
+ id: z.ZodString;
631
+ name: z.ZodString;
632
+ displayName: z.ZodOptional<z.ZodString>;
633
+ isDisplayComp: z.ZodOptional<z.ZodBoolean>;
634
+ type: z.ZodString;
635
+ description: z.ZodString;
636
+ props: z.ZodObject<{
637
+ query: z.ZodOptional<z.ZodNullable<z.ZodUnion<[z.ZodString, z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>]>>>;
638
+ title: z.ZodOptional<z.ZodString>;
639
+ description: z.ZodOptional<z.ZodString>;
640
+ config: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
641
+ actions: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>;
642
+ }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
643
+ query: z.ZodOptional<z.ZodNullable<z.ZodUnion<[z.ZodString, z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>]>>>;
644
+ title: z.ZodOptional<z.ZodString>;
645
+ description: z.ZodOptional<z.ZodString>;
646
+ config: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
647
+ actions: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>;
648
+ }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
649
+ query: z.ZodOptional<z.ZodNullable<z.ZodUnion<[z.ZodString, z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>]>>>;
650
+ title: z.ZodOptional<z.ZodString>;
651
+ description: z.ZodOptional<z.ZodString>;
652
+ config: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
653
+ actions: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>;
654
+ }, z.ZodTypeAny, "passthrough">>;
655
+ category: z.ZodOptional<z.ZodString>;
656
+ keywords: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
657
+ }, "strip", z.ZodTypeAny, {
658
+ id: string;
659
+ type: string;
660
+ name: string;
661
+ description: string;
662
+ props: {
663
+ description?: string | undefined;
664
+ query?: string | {} | null | undefined;
665
+ title?: string | undefined;
666
+ config?: Record<string, unknown> | undefined;
667
+ actions?: any[] | undefined;
668
+ } & {
669
+ [k: string]: unknown;
670
+ };
671
+ displayName?: string | undefined;
672
+ isDisplayComp?: boolean | undefined;
673
+ category?: string | undefined;
674
+ keywords?: string[] | undefined;
675
+ }, {
676
+ id: string;
677
+ type: string;
678
+ name: string;
679
+ description: string;
680
+ props: {
681
+ description?: string | undefined;
682
+ query?: string | {} | null | undefined;
683
+ title?: string | undefined;
684
+ config?: Record<string, unknown> | undefined;
685
+ actions?: any[] | undefined;
686
+ } & {
687
+ [k: string]: unknown;
688
+ };
689
+ displayName?: string | undefined;
690
+ isDisplayComp?: boolean | undefined;
691
+ category?: string | undefined;
692
+ keywords?: string[] | undefined;
693
+ }>;
694
+ type Component = z.infer<typeof ComponentSchema>;
695
+ declare const OutputFieldSchema: z.ZodObject<{
696
+ name: z.ZodString;
697
+ type: z.ZodEnum<["string", "number", "boolean", "date"]>;
698
+ description: z.ZodString;
699
+ }, "strip", z.ZodTypeAny, {
700
+ type: "string" | "number" | "boolean" | "date";
701
+ name: string;
702
+ description: string;
703
+ }, {
704
+ type: "string" | "number" | "boolean" | "date";
705
+ name: string;
706
+ description: string;
707
+ }>;
708
+ type OutputField = z.infer<typeof OutputFieldSchema>;
709
+ declare const OutputSchema: z.ZodObject<{
710
+ description: z.ZodString;
711
+ fields: z.ZodArray<z.ZodObject<{
712
+ name: z.ZodString;
713
+ type: z.ZodEnum<["string", "number", "boolean", "date"]>;
714
+ description: z.ZodString;
715
+ }, "strip", z.ZodTypeAny, {
716
+ type: "string" | "number" | "boolean" | "date";
717
+ name: string;
718
+ description: string;
719
+ }, {
720
+ type: "string" | "number" | "boolean" | "date";
721
+ name: string;
722
+ description: string;
723
+ }>, "many">;
724
+ }, "strip", z.ZodTypeAny, {
725
+ description: string;
726
+ fields: {
727
+ type: "string" | "number" | "boolean" | "date";
728
+ name: string;
729
+ description: string;
730
+ }[];
731
+ }, {
732
+ description: string;
733
+ fields: {
734
+ type: "string" | "number" | "boolean" | "date";
735
+ name: string;
736
+ description: string;
737
+ }[];
738
+ }>;
739
+ type ToolOutputSchema = z.infer<typeof OutputSchema>;
740
+ declare const ToolSchema: z.ZodObject<{
741
+ id: z.ZodString;
742
+ name: z.ZodString;
743
+ description: z.ZodString;
744
+ /** Tool type: "source" = routed through SourceAgent, "direct" = called directly by MainAgent */
745
+ toolType: z.ZodOptional<z.ZodEnum<["source", "direct"]>>;
746
+ /** Full untruncated schema for source agent (all columns visible) */
747
+ fullSchema: z.ZodOptional<z.ZodString>;
748
+ params: z.ZodRecord<z.ZodString, z.ZodString>;
749
+ fn: z.ZodFunction<z.ZodTuple<[z.ZodAny], z.ZodUnknown>, z.ZodAny>;
750
+ outputSchema: z.ZodOptional<z.ZodObject<{
751
+ description: z.ZodString;
752
+ fields: z.ZodArray<z.ZodObject<{
753
+ name: z.ZodString;
754
+ type: z.ZodEnum<["string", "number", "boolean", "date"]>;
755
+ description: z.ZodString;
756
+ }, "strip", z.ZodTypeAny, {
757
+ type: "string" | "number" | "boolean" | "date";
758
+ name: string;
759
+ description: string;
760
+ }, {
761
+ type: "string" | "number" | "boolean" | "date";
762
+ name: string;
763
+ description: string;
764
+ }>, "many">;
765
+ }, "strip", z.ZodTypeAny, {
766
+ description: string;
767
+ fields: {
768
+ type: "string" | "number" | "boolean" | "date";
769
+ name: string;
770
+ description: string;
771
+ }[];
772
+ }, {
773
+ description: string;
774
+ fields: {
775
+ type: "string" | "number" | "boolean" | "date";
776
+ name: string;
777
+ description: string;
778
+ }[];
779
+ }>>;
780
+ /** Cache policy. `false` = never cache (live data, write ops). Mirrors HTTP `Cache-Control: no-store`. */
781
+ cache: z.ZodOptional<z.ZodUnion<[z.ZodLiteral<false>, z.ZodObject<{
782
+ ttlMs: z.ZodOptional<z.ZodNumber>;
783
+ }, "strip", z.ZodTypeAny, {
784
+ ttlMs?: number | undefined;
785
+ }, {
786
+ ttlMs?: number | undefined;
787
+ }>]>>;
788
+ }, "strip", z.ZodTypeAny, {
789
+ id: string;
790
+ params: Record<string, string>;
791
+ name: string;
792
+ description: string;
793
+ fn: (args_0: any, ...args: unknown[]) => any;
794
+ toolType?: "source" | "direct" | undefined;
795
+ fullSchema?: string | undefined;
796
+ outputSchema?: {
797
+ description: string;
798
+ fields: {
799
+ type: "string" | "number" | "boolean" | "date";
800
+ name: string;
801
+ description: string;
802
+ }[];
803
+ } | undefined;
804
+ cache?: false | {
805
+ ttlMs?: number | undefined;
806
+ } | undefined;
807
+ }, {
808
+ id: string;
809
+ params: Record<string, string>;
810
+ name: string;
811
+ description: string;
812
+ fn: (args_0: any, ...args: unknown[]) => any;
813
+ toolType?: "source" | "direct" | undefined;
814
+ fullSchema?: string | undefined;
815
+ outputSchema?: {
816
+ description: string;
817
+ fields: {
818
+ type: "string" | "number" | "boolean" | "date";
819
+ name: string;
820
+ description: string;
821
+ }[];
822
+ } | undefined;
823
+ cache?: false | {
824
+ ttlMs?: number | undefined;
825
+ } | undefined;
826
+ }>;
827
+ type Tool$1 = z.infer<typeof ToolSchema>;
429
828
  type CollectionOperation = 'getMany' | 'getOne' | 'query' | 'mutation' | 'updateOne' | 'deleteOne' | 'createOne';
430
829
  type CollectionHandler<TParams = any, TResult = any> = (params: TParams) => Promise<TResult> | TResult;
431
- type LLMProvider = 'anthropic' | 'groq';
830
+ type LLMProvider = 'anthropic' | 'groq' | 'gemini' | 'openai';
831
+
832
+ type DatabaseType = 'postgresql' | 'mssql' | 'snowflake' | 'mysql';
833
+ /**
834
+ * Model strategy for controlling which models are used for different tasks
835
+ * - 'best': Use the best model (e.g., Sonnet) for all tasks - highest quality, higher cost
836
+ * - 'fast': Use the fast model (e.g., Haiku) for all tasks - lower quality, lower cost
837
+ * - 'balanced': Use best model for complex tasks, fast model for simple tasks (default)
838
+ */
839
+ type ModelStrategy = 'best' | 'fast' | 'balanced';
840
+ /**
841
+ * Model configuration for DASH_COMP flow (dashboard component picking)
842
+ * Allows separate control of models used for component selection
843
+ */
844
+ interface DashCompModelConfig {
845
+ /**
846
+ * Primary model for DASH_COMP requests
847
+ * Format: "provider/model-name" (e.g., "anthropic/claude-sonnet-4-5-20250929")
848
+ */
849
+ model?: string;
850
+ /**
851
+ * Fast model for simpler DASH_COMP tasks (optional)
852
+ * Format: "provider/model-name" (e.g., "anthropic/claude-haiku-4-5-20251001")
853
+ */
854
+ fastModel?: string;
855
+ }
432
856
  interface SuperatomSDKConfig {
433
857
  url?: string;
434
- apiKey: string;
858
+ apiKey?: string;
435
859
  projectId: string;
436
- userId?: string;
437
860
  type?: string;
438
861
  bundleDir?: string;
439
862
  promptsDir?: string;
863
+ databaseType?: DatabaseType;
440
864
  ANTHROPIC_API_KEY?: string;
441
865
  GROQ_API_KEY?: string;
866
+ GEMINI_API_KEY?: string;
867
+ OPENAI_API_KEY?: string;
442
868
  LLM_PROVIDERS?: LLMProvider[];
869
+ logLevel?: LogLevel;
870
+ /**
871
+ * Model selection strategy for LLM API calls:
872
+ * - 'best': Use best model for all tasks (highest quality, higher cost)
873
+ * - 'fast': Use fast model for all tasks (lower quality, lower cost)
874
+ * - 'balanced': Use best model for complex tasks, fast model for simple tasks (default)
875
+ */
876
+ modelStrategy?: ModelStrategy;
877
+ /**
878
+ * Model for the main agent (routing + analysis).
879
+ * Format: "provider/model-name" (e.g., "anthropic/claude-haiku-4-5-20251001")
880
+ * If not set, uses the provider's default model.
881
+ */
882
+ mainAgentModel?: string;
883
+ /**
884
+ * Model for source agents (per-source query generation).
885
+ * Format: "provider/model-name" (e.g., "anthropic/claude-haiku-4-5-20251001")
886
+ * If not set, uses the provider's default model.
887
+ */
888
+ sourceAgentModel?: string;
889
+ /**
890
+ * Separate model configuration for DASH_COMP flow (dashboard component picking)
891
+ * If not provided, falls back to provider-based model selection
892
+ */
893
+ dashCompModels?: DashCompModelConfig;
894
+ /**
895
+ * Similarity threshold for conversation search (semantic matching)
896
+ * Value between 0 and 1 (e.g., 0.8 = 80% similarity required)
897
+ * Higher values require closer matches, lower values allow more distant matches
898
+ * Default: 0.8
899
+ */
900
+ conversationSimilarityThreshold?: number;
901
+ /**
902
+ * Query cache TTL (Time To Live) in minutes
903
+ * Cached query results expire after this duration
904
+ * Default: 5 minutes
905
+ */
906
+ queryCacheTTL?: number;
907
+ /**
908
+ * Dashboard conversation history TTL (Time To Live) in minutes
909
+ * Per-dashboard conversation histories expire after this duration
910
+ * Default: 30 minutes
911
+ */
912
+ dashboardHistoryTTL?: number;
443
913
  }
444
914
 
445
- interface User {
446
- username: string;
447
- password: string;
448
- wsIds: string[];
449
- }
450
- interface UsersData {
451
- users: User[];
915
+ declare const KbNodesQueryFiltersSchema: z.ZodObject<{
916
+ query: z.ZodOptional<z.ZodString>;
917
+ category: z.ZodOptional<z.ZodString>;
918
+ tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
919
+ type: z.ZodOptional<z.ZodEnum<["global", "user", "query"]>>;
920
+ createdBy: z.ZodOptional<z.ZodString>;
921
+ }, "strip", z.ZodTypeAny, {
922
+ type?: "query" | "user" | "global" | undefined;
923
+ query?: string | undefined;
924
+ category?: string | undefined;
925
+ createdBy?: string | undefined;
926
+ tags?: string[] | undefined;
927
+ }, {
928
+ type?: "query" | "user" | "global" | undefined;
929
+ query?: string | undefined;
930
+ category?: string | undefined;
931
+ createdBy?: string | undefined;
932
+ tags?: string[] | undefined;
933
+ }>;
934
+ type KbNodesQueryFilters = z.infer<typeof KbNodesQueryFiltersSchema>;
935
+ declare const KbNodesRequestPayloadSchema: z.ZodObject<{
936
+ operation: z.ZodEnum<["create", "update", "delete", "getAll", "getOne", "search", "getByCategory", "getByUser", "getCategories", "getTags"]>;
937
+ data: z.ZodOptional<z.ZodObject<{
938
+ id: z.ZodOptional<z.ZodNumber>;
939
+ title: z.ZodOptional<z.ZodString>;
940
+ content: z.ZodOptional<z.ZodString>;
941
+ category: z.ZodOptional<z.ZodString>;
942
+ tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
943
+ type: z.ZodOptional<z.ZodEnum<["global", "user", "query"]>>;
944
+ createdBy: z.ZodOptional<z.ZodString>;
945
+ updatedBy: z.ZodOptional<z.ZodString>;
946
+ userId: z.ZodOptional<z.ZodString>;
947
+ query: z.ZodOptional<z.ZodString>;
948
+ filters: z.ZodOptional<z.ZodObject<{
949
+ query: z.ZodOptional<z.ZodString>;
950
+ category: z.ZodOptional<z.ZodString>;
951
+ tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
952
+ type: z.ZodOptional<z.ZodEnum<["global", "user", "query"]>>;
953
+ createdBy: z.ZodOptional<z.ZodString>;
954
+ }, "strip", z.ZodTypeAny, {
955
+ type?: "query" | "user" | "global" | undefined;
956
+ query?: string | undefined;
957
+ category?: string | undefined;
958
+ createdBy?: string | undefined;
959
+ tags?: string[] | undefined;
960
+ }, {
961
+ type?: "query" | "user" | "global" | undefined;
962
+ query?: string | undefined;
963
+ category?: string | undefined;
964
+ createdBy?: string | undefined;
965
+ tags?: string[] | undefined;
966
+ }>>;
967
+ limit: z.ZodOptional<z.ZodNumber>;
968
+ offset: z.ZodOptional<z.ZodNumber>;
969
+ }, "strip", z.ZodTypeAny, {
970
+ id?: number | undefined;
971
+ type?: "query" | "user" | "global" | undefined;
972
+ query?: string | undefined;
973
+ title?: string | undefined;
974
+ category?: string | undefined;
975
+ userId?: string | undefined;
976
+ limit?: number | undefined;
977
+ filters?: {
978
+ type?: "query" | "user" | "global" | undefined;
979
+ query?: string | undefined;
980
+ category?: string | undefined;
981
+ createdBy?: string | undefined;
982
+ tags?: string[] | undefined;
983
+ } | undefined;
984
+ createdBy?: string | undefined;
985
+ updatedBy?: string | undefined;
986
+ offset?: number | undefined;
987
+ tags?: string[] | undefined;
988
+ content?: string | undefined;
989
+ }, {
990
+ id?: number | undefined;
991
+ type?: "query" | "user" | "global" | undefined;
992
+ query?: string | undefined;
993
+ title?: string | undefined;
994
+ category?: string | undefined;
995
+ userId?: string | undefined;
996
+ limit?: number | undefined;
997
+ filters?: {
998
+ type?: "query" | "user" | "global" | undefined;
999
+ query?: string | undefined;
1000
+ category?: string | undefined;
1001
+ createdBy?: string | undefined;
1002
+ tags?: string[] | undefined;
1003
+ } | undefined;
1004
+ createdBy?: string | undefined;
1005
+ updatedBy?: string | undefined;
1006
+ offset?: number | undefined;
1007
+ tags?: string[] | undefined;
1008
+ content?: string | undefined;
1009
+ }>>;
1010
+ }, "strip", z.ZodTypeAny, {
1011
+ operation: "create" | "getOne" | "update" | "delete" | "getAll" | "search" | "getByCategory" | "getByUser" | "getCategories" | "getTags";
1012
+ data?: {
1013
+ id?: number | undefined;
1014
+ type?: "query" | "user" | "global" | undefined;
1015
+ query?: string | undefined;
1016
+ title?: string | undefined;
1017
+ category?: string | undefined;
1018
+ userId?: string | undefined;
1019
+ limit?: number | undefined;
1020
+ filters?: {
1021
+ type?: "query" | "user" | "global" | undefined;
1022
+ query?: string | undefined;
1023
+ category?: string | undefined;
1024
+ createdBy?: string | undefined;
1025
+ tags?: string[] | undefined;
1026
+ } | undefined;
1027
+ createdBy?: string | undefined;
1028
+ updatedBy?: string | undefined;
1029
+ offset?: number | undefined;
1030
+ tags?: string[] | undefined;
1031
+ content?: string | undefined;
1032
+ } | undefined;
1033
+ }, {
1034
+ operation: "create" | "getOne" | "update" | "delete" | "getAll" | "search" | "getByCategory" | "getByUser" | "getCategories" | "getTags";
1035
+ data?: {
1036
+ id?: number | undefined;
1037
+ type?: "query" | "user" | "global" | undefined;
1038
+ query?: string | undefined;
1039
+ title?: string | undefined;
1040
+ category?: string | undefined;
1041
+ userId?: string | undefined;
1042
+ limit?: number | undefined;
1043
+ filters?: {
1044
+ type?: "query" | "user" | "global" | undefined;
1045
+ query?: string | undefined;
1046
+ category?: string | undefined;
1047
+ createdBy?: string | undefined;
1048
+ tags?: string[] | undefined;
1049
+ } | undefined;
1050
+ createdBy?: string | undefined;
1051
+ updatedBy?: string | undefined;
1052
+ offset?: number | undefined;
1053
+ tags?: string[] | undefined;
1054
+ content?: string | undefined;
1055
+ } | undefined;
1056
+ }>;
1057
+ type KbNodesRequestPayload = z.infer<typeof KbNodesRequestPayloadSchema>;
1058
+ interface T_RESPONSE {
1059
+ success: boolean;
1060
+ data?: any;
1061
+ errors: string[];
452
1062
  }
1063
+
453
1064
  /**
454
1065
  * UserManager class to handle CRUD operations on users with file persistence
455
1066
  * and in-memory caching. Changes are synced to file periodically.
@@ -503,6 +1114,18 @@ declare class UserManager {
503
1114
  * @returns The user if found, undefined otherwise
504
1115
  */
505
1116
  getUser(username: string): User | undefined;
1117
+ /**
1118
+ * Read a user by email
1119
+ * @param email - Email to retrieve
1120
+ * @returns The user if found, undefined otherwise
1121
+ */
1122
+ getUserByEmail(email: string): User | undefined;
1123
+ /**
1124
+ * Find user by username or email
1125
+ * @param identifier - Username or email to search for
1126
+ * @returns The user if found, undefined otherwise
1127
+ */
1128
+ getUserByUsernameOrEmail(identifier: string): User | undefined;
506
1129
  /**
507
1130
  * Read all users
508
1131
  * @returns Array of all users
@@ -697,115 +1320,888 @@ declare class ReportManager {
697
1320
  getReportCount(): number;
698
1321
  }
699
1322
 
700
- interface LLMMessages {
701
- sys: string;
702
- user: string;
703
- }
704
- interface LLMOptions {
705
- model?: string;
706
- maxTokens?: number;
707
- temperature?: number;
708
- topP?: number;
709
- apiKey?: string;
710
- partial?: (chunk: string) => void;
711
- }
712
- declare class LLM {
713
- static text(messages: LLMMessages, options?: LLMOptions): Promise<string>;
714
- static stream<T = string>(messages: LLMMessages, options?: LLMOptions, json?: boolean): Promise<T extends string ? string : any>;
715
- /**
716
- * Parse model string to extract provider and model name
717
- * @param modelString - Format: "provider/model-name" or just "model-name"
718
- * @returns [provider, modelName]
719
- *
720
- * @example
721
- * "anthropic/claude-sonnet-4-5" → ["anthropic", "claude-sonnet-4-5"]
722
- * "groq/openai/gpt-oss-120b" → ["groq", "openai/gpt-oss-120b"]
723
- * "claude-sonnet-4-5" → ["anthropic", "claude-sonnet-4-5"] (default)
724
- */
725
- private static _parseModel;
726
- private static _anthropicText;
727
- private static _anthropicStream;
728
- private static _groqText;
729
- private static _groqStream;
730
- /**
731
- * Parse JSON string, handling markdown code blocks and surrounding text
732
- * Enhanced version from anthropic.ts implementation
733
- * @param text - Text that may contain JSON wrapped in ```json...``` or with surrounding text
734
- * @returns Parsed JSON object
735
- */
736
- private static _parseJSON;
737
- }
738
-
739
- interface CapturedLog {
740
- timestamp: number;
741
- level: 'info' | 'error' | 'warn' | 'debug';
742
- message: string;
743
- type?: 'explanation' | 'query' | 'general';
744
- data?: Record<string, any>;
745
- }
746
1323
  /**
747
- * UILogCollector captures logs during user prompt processing
748
- * and sends them to runtime via ui_logs message with uiBlockId as the message id
749
- * Logs are sent in real-time for streaming effect in the UI
1324
+ * StreamBuffer - Buffered streaming utility for smoother text delivery
1325
+ * Batches small chunks together and flushes at regular intervals
750
1326
  */
751
- declare class UILogCollector {
752
- private logs;
753
- private uiBlockId;
754
- private clientId;
755
- private sendMessage;
756
- constructor(clientId: string, sendMessage: (message: Message) => void, uiBlockId?: string);
757
- /**
758
- * Check if logging is enabled (uiBlockId is provided)
759
- */
760
- isEnabled(): boolean;
761
- /**
762
- * Add a log entry with timestamp and immediately send to runtime
763
- */
764
- private addLog;
765
- /**
766
- * Send a single log to runtime immediately
767
- */
768
- private sendLogImmediately;
769
- /**
770
- * Log info message
771
- */
772
- info(message: string, type?: 'explanation' | 'query' | 'general', data?: Record<string, any>): void;
773
- /**
774
- * Log error message
775
- */
776
- error(message: string, type?: 'explanation' | 'query' | 'general', data?: Record<string, any>): void;
777
- /**
778
- * Log warning message
779
- */
780
- warn(message: string, type?: 'explanation' | 'query' | 'general', data?: Record<string, any>): void;
1327
+ type StreamCallback = (chunk: string) => void;
1328
+ /**
1329
+ * StreamBuffer class for managing buffered streaming output
1330
+ * Provides smooth text delivery by batching small chunks
1331
+ */
1332
+ declare class StreamBuffer {
1333
+ private buffer;
1334
+ private flushTimer;
1335
+ private callback;
1336
+ private fullText;
1337
+ constructor(callback?: StreamCallback);
781
1338
  /**
782
- * Log debug message
1339
+ * Check if the buffer has a callback configured
783
1340
  */
784
- debug(message: string, type?: 'explanation' | 'query' | 'general', data?: Record<string, any>): void;
1341
+ hasCallback(): boolean;
785
1342
  /**
786
- * Log LLM explanation with typed metadata
1343
+ * Get all text that has been written (including already flushed)
787
1344
  */
788
- logExplanation(message: string, explanation: string, data?: Record<string, any>): void;
1345
+ getFullText(): string;
789
1346
  /**
790
- * Log generated query with typed metadata
1347
+ * Write a chunk to the buffer
1348
+ * Large chunks or chunks with newlines are flushed immediately
1349
+ * Small chunks are batched and flushed after a short interval
1350
+ *
1351
+ * @param chunk - Text chunk to write
791
1352
  */
792
- logQuery(message: string, query: string, data?: Record<string, any>): void;
1353
+ write(chunk: string): void;
793
1354
  /**
794
- * Send all collected logs at once (optional, for final summary)
1355
+ * Flush the buffer immediately
1356
+ * Call this before tool execution or other operations that need clean output
795
1357
  */
796
- sendAllLogs(): void;
1358
+ flush(): void;
797
1359
  /**
798
- * Get all collected logs
1360
+ * Internal flush implementation
799
1361
  */
800
- getLogs(): CapturedLog[];
1362
+ private flushNow;
801
1363
  /**
802
- * Clear all logs
1364
+ * Clean up resources
1365
+ * Call this when done with the buffer
803
1366
  */
804
- clearLogs(): void;
805
- /**
806
- * Set uiBlockId (in case it's provided later)
1367
+ dispose(): void;
1368
+ }
1369
+
1370
+ /**
1371
+ * ToolExecutorService - Handles execution of SQL queries and external tools
1372
+ * Extracted from BaseLLM.generateTextResponse for better separation of concerns
1373
+ */
1374
+
1375
+ /**
1376
+ * External tool definition
1377
+ */
1378
+ interface ExternalTool {
1379
+ id: string;
1380
+ name: string;
1381
+ description?: string;
1382
+ /** Tool type: "source" = routed through SourceAgent, "direct" = called directly by MainAgent */
1383
+ toolType?: 'source' | 'direct';
1384
+ /** Full untruncated schema for source agent (all columns visible) */
1385
+ fullSchema?: string;
1386
+ /** Schema size tier: small (≤50 tables), medium (51-200), large (201-500), very_large (500+) */
1387
+ schemaTier?: string;
1388
+ /** Schema search function for very_large tier — keyword search over entities */
1389
+ schemaSearchFn?: (keywords: string[]) => string;
1390
+ fn: (input: any) => Promise<any>;
1391
+ limit?: number;
1392
+ outputSchema?: any;
1393
+ executionType?: 'immediate' | 'deferred';
1394
+ userProvidedData?: any;
1395
+ params?: Record<string, any>;
1396
+ }
1397
+ /**
1398
+ * Executed tool tracking info
1399
+ */
1400
+ interface ExecutedToolInfo {
1401
+ id: string;
1402
+ name: string;
1403
+ params: any;
1404
+ result: {
1405
+ _totalRecords: number;
1406
+ _recordsShown: number;
1407
+ _metadata?: any;
1408
+ _sampleData: any[];
1409
+ /** Bounded summary over the FULL fetched result (complete structure). */
1410
+ _summary?: any;
1411
+ /** Up to MAIN_AGENT_COMPLETE_ROWS rows — the complete result when small. */
1412
+ _mainAgentRows?: any[];
1413
+ };
1414
+ outputSchema?: any;
1415
+ sourceSchema?: string;
1416
+ sourceType?: string;
1417
+ }
1418
+
1419
+ /**
1420
+ * Multi-Agent Architecture Types
1421
+ *
1422
+ * Defines interfaces for the hierarchical agent system:
1423
+ * - Main Agent: ONE LLM.streamWithTools() call with source agent tools
1424
+ * - Source Agents: independent agents that query individual data sources
1425
+ *
1426
+ * The main agent sees only source summaries. When it calls a source tool,
1427
+ * the SourceAgent runs independently (own LLM, own retries) and returns clean data.
1428
+ */
1429
+
1430
+ /**
1431
+ * Per-entity detail: name, row count, and column names.
1432
+ * Gives the main agent enough context to route to the right source.
1433
+ */
1434
+ interface EntityDetail {
1435
+ /** Entity name (table, sheet, endpoint) */
1436
+ name: string;
1437
+ /** Approximate row count */
1438
+ rowCount?: number;
1439
+ /** Column/field names */
1440
+ columns: string[];
1441
+ }
1442
+ /**
1443
+ * Representation of a data source for the main agent.
1444
+ * Contains entity names WITH column names so the LLM can route accurately.
1445
+ */
1446
+ interface SourceSummary {
1447
+ /** Source ID (matches tool ID prefix) */
1448
+ id: string;
1449
+ /** Human-readable source name */
1450
+ name: string;
1451
+ /** Source type: postgres, excel, rest_api, etc. */
1452
+ type: string;
1453
+ /** Brief description of what data this source contains */
1454
+ description: string;
1455
+ /** Detailed entity info with column names for routing */
1456
+ entityDetails: EntityDetail[];
1457
+ /** The tool ID associated with this source */
1458
+ toolId: string;
1459
+ }
1460
+ /**
1461
+ * What a source agent returns after querying its data source.
1462
+ * The main agent uses this to analyze and compose the final response.
1463
+ */
1464
+ interface SourceAgentResult {
1465
+ /** Source ID */
1466
+ sourceId: string;
1467
+ /** Source name */
1468
+ sourceName: string;
1469
+ /** Whether the query succeeded */
1470
+ success: boolean;
1471
+ /** Result data rows */
1472
+ data: any[];
1473
+ /** Metadata about the query execution */
1474
+ metadata: SourceAgentMetadata;
1475
+ /** Tool execution info for the last successful query (backward compat) */
1476
+ executedTool: ExecutedToolInfo;
1477
+ /** All successful tool executions (primary + follow-up queries) */
1478
+ allExecutedTools?: ExecutedToolInfo[];
1479
+ /** Error message if failed */
1480
+ error?: string;
1481
+ }
1482
+ interface SourceAgentMetadata {
1483
+ /** Total rows that matched the query (before limit) */
1484
+ totalRowsMatched: number;
1485
+ /** Rows actually returned (after limit) */
1486
+ rowsReturned: number;
1487
+ /** Whether the result was truncated by the row limit */
1488
+ isLimited: boolean;
1489
+ /** The query/params that were executed */
1490
+ queryExecuted?: string;
1491
+ /** Execution time in milliseconds */
1492
+ executionTimeMs: number;
1493
+ }
1494
+ /**
1495
+ * A pre-built, multi-step UI flow registered with the SDK.
1496
+ *
1497
+ * When the main agent decides a user's question matches a workflow's whenToUse
1498
+ * trigger, it picks the workflow instead of running source agents / generating
1499
+ * dashboard components. The LLM extracts the workflow's required props from the
1500
+ * prompt (using `propsSchema` as the tool input_schema) and the SDK returns the
1501
+ * workflow component directly — no analysis text, no chart generation. The
1502
+ * frontend renders the registered workflow component with the LLM-extracted
1503
+ * props.
1504
+ */
1505
+ interface WorkflowDescriptor {
1506
+ /** Unique workflow id (used as the LLM tool name) */
1507
+ id: string;
1508
+ /** Component name on the frontend (matches the registered React component) */
1509
+ name: string;
1510
+ /** Short human-readable description of what this workflow does */
1511
+ description: string;
1512
+ /**
1513
+ * 1–2 sentence trigger condition. The LLM uses this to decide if the
1514
+ * user's prompt matches this workflow. Be specific — e.g.
1515
+ * "User wants to *initiate* an inventory transfer (review + submit POs),
1516
+ * not just see analysis or charts."
807
1517
  */
808
- setUIBlockId(uiBlockId: string): void;
1518
+ whenToUse: string;
1519
+ /**
1520
+ * JSON-schema-style description of the props the workflow needs. Becomes
1521
+ * the LLM tool's input_schema, so the model fills these from the prompt.
1522
+ * Use the same shape as `params` on direct tools — string descriptors with
1523
+ * an optional "(optional)" suffix.
1524
+ *
1525
+ * Example:
1526
+ * ```
1527
+ * {
1528
+ * selectedStore: 'object — { id, name } of the source branch',
1529
+ * minROI: 'number (optional) — only show transfers with ROI ≥ this',
1530
+ * }
1531
+ * ```
1532
+ */
1533
+ propsSchema: Record<string, string>;
1534
+ /**
1535
+ * Optional: static prop defaults merged with LLM-extracted props before
1536
+ * the component is returned. Useful for things like the embedded
1537
+ * `externalTool` config that the workflow uses to fetch its own data.
1538
+ */
1539
+ defaultProps?: Record<string, any>;
1540
+ }
1541
+ /**
1542
+ * The workflow selection captured during a routing call.
1543
+ * Set on AgentResponse when the LLM picks a workflow tool.
1544
+ */
1545
+ interface SelectedWorkflow {
1546
+ /** Component name (matches WorkflowDescriptor.name) */
1547
+ name: string;
1548
+ /** Props extracted from the prompt + merged with workflow.defaultProps */
1549
+ props: Record<string, any>;
1550
+ }
1551
+ /**
1552
+ * The complete response from the multi-agent system.
1553
+ * Contains everything needed for text display + component generation.
1554
+ */
1555
+ interface AgentResponse {
1556
+ /** Generated text response (analysis of the data) */
1557
+ text: string;
1558
+ /** All executed tools across all source agents (for component generation) */
1559
+ executedTools: ExecutedToolInfo[];
1560
+ /** Individual results from each source agent */
1561
+ sourceResults: SourceAgentResult[];
1562
+ /**
1563
+ * Populated when MainAgent wrote AND successfully executed a script during its turn.
1564
+ * Caller (agent-user-response.ts) persists it via ScriptStore.save().
1565
+ * Absent when MainAgent didn't write one (trivial question / all attempts failed).
1566
+ */
1567
+ savedScript?: AgentWrittenScript;
1568
+ /**
1569
+ * Set when the LLM routed the question to a registered workflow component.
1570
+ * When present, the upstream caller should skip component generation and
1571
+ * return this workflow as the response.
1572
+ */
1573
+ workflow?: SelectedWorkflow;
1574
+ }
1575
+ /**
1576
+ * A script MainAgent authored + verified during its turn. Shape aligns with
1577
+ * what ScriptStore.save() needs — minus store-assigned fields (id, timestamps, counts).
1578
+ */
1579
+ interface AgentWrittenScript {
1580
+ /**
1581
+ * `ScriptRecipe.id` of the draft that was authored + verified during this turn.
1582
+ * The caller passes this to `ScriptStore.promoteToVerified(recipeId, …)` to
1583
+ * flip the draft to verified status and (when possible) drop the turn-suffix
1584
+ * from its filename.
1585
+ */
1586
+ recipeId: string;
1587
+ name: string;
1588
+ intentDescription: string;
1589
+ tags: string[];
1590
+ parameters: Array<{
1591
+ name: string;
1592
+ type: 'string' | 'number' | 'date' | 'date_range' | 'enum' | 'boolean';
1593
+ required: boolean;
1594
+ default?: any;
1595
+ enumValues?: Record<string, string>;
1596
+ description: string;
1597
+ }>;
1598
+ scriptBody: string;
1599
+ /** Source IDs referenced by the script (extracted from ctx.query calls) */
1600
+ sourceIds: string[];
1601
+ /** Tables referenced in the script's SQL (regex-extracted) */
1602
+ tables: string[];
1603
+ /** Executed queries from the verified run — fed to component generation */
1604
+ executedQueries: Array<{
1605
+ sourceId: string;
1606
+ sourceName: string;
1607
+ sql: string;
1608
+ data: any[];
1609
+ count: number;
1610
+ totalCount?: number;
1611
+ executionTimeMs: number;
1612
+ /**
1613
+ * True for synthetic entries (ctx.emit datasets, the computed:_final
1614
+ * post-JS data). The component generator routes virtual sources through
1615
+ * the script_dataset sentinel toolId so the frontend resolves them via
1616
+ * queryCache instead of attempting to re-execute SQL.
1617
+ */
1618
+ virtual?: boolean;
1619
+ }>;
1620
+ }
1621
+ /**
1622
+ * Configuration for the multi-agent system.
1623
+ * Controls limits, models, and behavior.
1624
+ */
1625
+ interface AgentConfig {
1626
+ /** Max rows shown to the UI preview / inlined per source (default: 10) */
1627
+ maxRowsPerSource: number;
1628
+ /**
1629
+ * Max rows a source query may FETCH from the DB server-side (default: 2000).
1630
+ * Decoupled from what the main agent is shown: the full result is fetched and
1631
+ * summarized (bounded), but only a small/complete slice enters LLM context.
1632
+ * This lets small lookups (benchmark maps) arrive COMPLETE without letting
1633
+ * large results blow up context.
1634
+ */
1635
+ maxRowsFetched: number;
1636
+ /** Model for the main agent (routing + analysis in one LLM call) */
1637
+ mainAgentModel: string;
1638
+ /** Model for source agent query generation */
1639
+ sourceAgentModel: string;
1640
+ /** API key for LLM calls */
1641
+ apiKey?: string;
1642
+ /** Max retry attempts per source agent */
1643
+ maxRetries: number;
1644
+ /** Max tool calling iterations for the main agent loop */
1645
+ maxIterations: number;
1646
+ /** Global knowledge base context (static, same for all users/questions — cached in system prompt) */
1647
+ globalKnowledgeBase?: string;
1648
+ /** Per-request knowledge base context (user-specific + query-matched — dynamic, not cached) */
1649
+ knowledgeBaseContext?: string;
1650
+ /** Collections registry (ChromaDB search hooks) for embedding-based schema + source search */
1651
+ collections?: any;
1652
+ /** Optional project ID for scoping embedding searches */
1653
+ projectId?: string;
1654
+ }
1655
+ /**
1656
+ * Default agent configuration
1657
+ */
1658
+ declare const DEFAULT_AGENT_CONFIG: AgentConfig;
1659
+
1660
+ /**
1661
+ * Script Flow Types
1662
+ *
1663
+ * Defines interfaces for the script-based query architecture:
1664
+ * - ScriptRecipe: metadata for matching, validation, and quality tracking
1665
+ * - ScriptResult: output from executing a script
1666
+ * - ScriptMatch: result from the LLM-based script matcher
1667
+ */
1668
+ /**
1669
+ * Recipe metadata stored alongside each script.
1670
+ * Used for matching, validation, and quality tracking.
1671
+ */
1672
+ interface ScriptRecipe {
1673
+ /** Unique script identifier */
1674
+ id: string;
1675
+ /** Version number (incremented on regeneration) */
1676
+ version: number;
1677
+ /** Human-readable name (e.g., "Revenue by Dimension") */
1678
+ name: string;
1679
+ /** Natural language description of what this script does */
1680
+ intentDescription: string;
1681
+ /** Keyword tags for quick filtering */
1682
+ tags: string[];
1683
+ /** Source tool IDs this script queries (e.g., ["mssql-abc123_query"]) */
1684
+ sourceIds: string[];
1685
+ /** Table names used (for future schema drift detection) */
1686
+ tables: string[];
1687
+ /** Parameter definitions — what can vary */
1688
+ parameters: ScriptParameter[];
1689
+ /** The script function body as a string. Loaded from disk (scripts-store/<fileBase>.ts). */
1690
+ scriptBody: string;
1691
+ /**
1692
+ * On-disk filename stem for the body: scripts-store/<fileBase>.ts.
1693
+ * Editable in the IDE. Decided at authoring time (slug of `name`, with a
1694
+ * short id suffix on collision) and stable across promotion.
1695
+ */
1696
+ fileBase?: string;
1697
+ /** sha256 of the on-disk body — lets the runtime detect manual edits. */
1698
+ bodyHash?: string;
1699
+ /** Project scope (single-VM deployments may leave this undefined). */
1700
+ projectId?: string;
1701
+ /** Times this script was used successfully */
1702
+ successCount: number;
1703
+ /** Times this script failed */
1704
+ failureCount: number;
1705
+ /** ISO timestamp of last usage */
1706
+ lastUsed: string;
1707
+ /** Original user question that created this script */
1708
+ createdFrom: string;
1709
+ /** ISO timestamp */
1710
+ createdAt: string;
1711
+ /** ISO timestamp */
1712
+ updatedAt: string;
1713
+ /**
1714
+ * `recipe.id` of the parent this script was forked from.
1715
+ * Undefined for root scripts (those written from scratch by MainAgent).
1716
+ * See backend/docs/SCRIPT-FLOW-FORK-ADAPT.md.
1717
+ */
1718
+ parentId?: string;
1719
+ /** 0 for root scripts; `parent.forkDepth + 1` for forks. Capped at 3. */
1720
+ forkDepth?: number;
1721
+ /**
1722
+ * Brief description of what this fork changed vs its parent
1723
+ * (sourced from the matcher's `modificationHint`).
1724
+ */
1725
+ forkReason?: string;
1726
+ /**
1727
+ * Validated component specs captured at authoring time. On a tier-high
1728
+ * replay these are rebound to fresh queryIds deterministically — no
1729
+ * component-generation LLM call, and the rendered columns can't drift from
1730
+ * what was validated when the script was authored. Absent on recipes
1731
+ * authored before this landed; those fall back to LLM component generation.
1732
+ * See backend/docs/SCRIPT-COMPONENT-CONSISTENCY.md.
1733
+ */
1734
+ components?: ScriptComponentSpec[];
1735
+ /**
1736
+ * Lifecycle stage of this recipe on disk.
1737
+ * - 'draft': written by MainAgent's write_script during a turn; filtered out
1738
+ * of FTS results (status='verified' only) so the matcher never picks it.
1739
+ * Filename is suffixed with `turnId` to keep concurrent turns
1740
+ * from clobbering each other's drafts.
1741
+ * - 'verified': promoted after `execute_script` succeeded; the matcher sees it.
1742
+ * Filename drops the turn suffix unless a verified file with
1743
+ * the same slug already exists (collision case keeps the suffix).
1744
+ *
1745
+ * Recipes loaded from disk without this field default to 'verified' so
1746
+ * existing scripts keep working unchanged.
1747
+ */
1748
+ status?: 'draft' | 'verified';
1749
+ /**
1750
+ * Per-turn unique suffix used for draft filenames (e.g. `1714745623-x9k2`).
1751
+ * Set when the draft is saved; carried until the recipe is promoted.
1752
+ */
1753
+ turnId?: string;
1754
+ /**
1755
+ * Last execution error captured by `recordDraftError` while the recipe was
1756
+ * still a draft. Lets users open the draft .json file and see why it failed
1757
+ * without grepping logs. Cleared on promotion to 'verified'.
1758
+ */
1759
+ lastError?: {
1760
+ phase: 'compile' | 'runtime' | 'timeout' | 'ipc';
1761
+ message: string;
1762
+ at: string;
1763
+ attempt: number;
1764
+ };
1765
+ }
1766
+ interface ScriptParameter {
1767
+ /** Parameter name (used in script body as params.name) */
1768
+ name: string;
1769
+ /** Parameter type */
1770
+ type: 'string' | 'number' | 'date' | 'date_range' | 'enum' | 'boolean';
1771
+ /** Whether this parameter is required */
1772
+ required: boolean;
1773
+ /** Default value if not provided */
1774
+ default?: any;
1775
+ /** For enum type — maps user-facing values to internal values */
1776
+ enumValues?: Record<string, string>;
1777
+ /** Human-readable description (used in the matcher LLM prompt) */
1778
+ description: string;
1779
+ }
1780
+ /**
1781
+ * A reusable component binding captured when a script is authored. Stored on
1782
+ * the recipe so tier-high replays rebuild components deterministically (rebind
1783
+ * to fresh queryIds) instead of re-running the component-picker LLM.
1784
+ */
1785
+ interface ScriptComponentSpec {
1786
+ /** Registered component name (e.g. "DynamicBarChart") — matched against the available component library. */
1787
+ componentType: string;
1788
+ /** `executedQuery.sourceId` to bind to (e.g. a tool id or 'computed:_final'), 'federation' for a cross-source component, or 'markdown' for a content-only narrative block (no data source). */
1789
+ sourceRef: string;
1790
+ /** Present only when sourceRef === 'federation' — the DuckDB SQL to re-execute on replay. */
1791
+ federationSql?: string;
1792
+ /** Present only when sourceRef === 'markdown' — the narrative text to render on replay (markdown has no data source, so its content must be persisted). */
1793
+ content?: string;
1794
+ title?: string;
1795
+ description?: string;
1796
+ /** Validated axis/value keys + aggregation — all referencing real columns of the bound source. */
1797
+ config: Record<string, any>;
1798
+ }
1799
+ /**
1800
+ * Result from executing a script via ScriptRunner.
1801
+ */
1802
+ interface ScriptResult {
1803
+ /** Whether the script executed successfully */
1804
+ success: boolean;
1805
+ /** Combined data from all queries */
1806
+ data: any[];
1807
+ /** Individual query results tracked during execution */
1808
+ executedQueries: ScriptQueryResult[];
1809
+ /** Error message if failed */
1810
+ error?: string;
1811
+ /**
1812
+ * Where in the lifecycle the error occurred. Lets MainAgent's fix-loop
1813
+ * decide between "rewrite the whole draft" (compile) and "patch the
1814
+ * specific line" (runtime).
1815
+ */
1816
+ errorPhase?: 'compile' | 'runtime' | 'timeout' | 'ipc';
1817
+ /** Total execution time in milliseconds */
1818
+ executionTimeMs: number;
1819
+ }
1820
+ /**
1821
+ * A single query executed during script runtime.
1822
+ * Tracked by ScriptContext for component generation and debugging.
1823
+ */
1824
+ interface ScriptQueryResult {
1825
+ /** Source tool ID */
1826
+ sourceId: string;
1827
+ /** Human-readable source name */
1828
+ sourceName: string;
1829
+ /** The SQL that was executed */
1830
+ sql: string;
1831
+ /** Result data rows */
1832
+ data: any[];
1833
+ /** Number of rows returned */
1834
+ count: number;
1835
+ /** Total rows that matched before limit (if available) */
1836
+ totalCount?: number;
1837
+ /** Query execution time in milliseconds */
1838
+ executionTimeMs: number;
1839
+ /**
1840
+ * True for rows that did NOT come from a real SQL execution — either a
1841
+ * ctx.emit() dataset or the synthesized "computed:_final" entry that
1842
+ * carries the script's post-JS returned data. The component generator
1843
+ * uses this to route the resulting component through the script_dataset
1844
+ * sentinel toolId so the frontend resolves it via the queryCache short-circuit.
1845
+ */
1846
+ virtual?: boolean;
1847
+ }
1848
+ /**
1849
+ * Match tier returned by the LLM script matcher.
1850
+ *
1851
+ * - 'high': the script answers the question directly; only parameter values
1852
+ * may differ. The runtime replays it with extracted params (cheapest path).
1853
+ * - 'near': the script answers a STRUCTURALLY similar question but needs
1854
+ * body modification (different metric, dimension, table, filter shape).
1855
+ * The runtime forks the parent and adapts the body via MainAgent's normal
1856
+ * write_script + execute_script loop — no SourceAgent dispatch needed.
1857
+ * See backend/docs/SCRIPT-FLOW-FORK-ADAPT.md for the full design.
1858
+ * - 'none': no script is relevant; full agent flow runs.
1859
+ */
1860
+ type MatchTier = 'high' | 'near' | 'none';
1861
+ /**
1862
+ * Result from the LLM-based script matcher.
1863
+ *
1864
+ * For `tier: 'high'`, `extractedParams` carries the values to pass to the
1865
+ * existing script. For `tier: 'near'`, `gaps` and `modificationHint` describe
1866
+ * what the fork-author needs to change in the parent body.
1867
+ */
1868
+ interface ScriptMatch {
1869
+ /** The matched script recipe */
1870
+ recipe: ScriptRecipe;
1871
+ /** Match tier — see MatchTier docs */
1872
+ tier: MatchTier;
1873
+ /** Similarity score (0-1, derived from LLM tier) */
1874
+ similarity: number;
1875
+ /**
1876
+ * Legacy confidence level. Mirrors `tier === 'high'`/`'near'` for now;
1877
+ * kept so existing callers compile while we migrate to tier-based logic.
1878
+ */
1879
+ confidence: 'high' | 'medium';
1880
+ /** Parameters extracted from the user question by the LLM (tier='high') */
1881
+ extractedParams?: Record<string, any>;
1882
+ /** What the user question needs that the parent doesn't cover (tier='near') */
1883
+ gaps?: string[];
1884
+ /** One-sentence description of the change the fork-author should make (tier='near') */
1885
+ modificationHint?: string;
1886
+ /** Why the matcher made this choice (for logs and telemetry) */
1887
+ reasoning?: string;
1888
+ }
1889
+
1890
+ /**
1891
+ * ScriptRecipeStore — injected metadata backend for the script flow.
1892
+ *
1893
+ * The SDK is standalone (no DB dependency). The backend implements this
1894
+ * interface over Postgres (full-text search + atomic counters) and injects it
1895
+ * via `collections['script-recipes']`, exactly like `collections['source-embeddings']`.
1896
+ * `ScriptStore` consumes it for all METADATA operations while keeping the
1897
+ * executable body on disk as scripts-store/<fileBase>.ts.
1898
+ *
1899
+ * All metadata rows are plain JSON (no scriptBody — that lives on disk).
1900
+ * See backend/docs/SCRIPT-FLOW-SCALING-ISSUES.md (#1, #3, #7).
1901
+ */
1902
+
1903
+ /** One recipe's metadata as stored in Postgres (mirrors the script_recipes table). */
1904
+ interface ScriptRecipeMetaRow {
1905
+ id: string;
1906
+ projectId?: string | null;
1907
+ version: number;
1908
+ name: string;
1909
+ intentDescription: string;
1910
+ tags: string[] | null;
1911
+ createdFrom: string | null;
1912
+ sourceIds: string[] | null;
1913
+ tables: string[] | null;
1914
+ parameters: ScriptParameter[] | null;
1915
+ components?: ScriptComponentSpec[] | null;
1916
+ fileBase: string;
1917
+ bodyHash?: string | null;
1918
+ successCount: number;
1919
+ failureCount: number;
1920
+ lastUsed: string | null;
1921
+ parentId?: string | null;
1922
+ forkDepth?: number | null;
1923
+ forkReason?: string | null;
1924
+ status: 'draft' | 'verified' | string;
1925
+ turnId?: string | null;
1926
+ lastError?: {
1927
+ phase: 'compile' | 'runtime' | 'timeout' | 'ipc';
1928
+ message: string;
1929
+ at: string;
1930
+ attempt: number;
1931
+ } | null;
1932
+ createdAt?: string | null;
1933
+ updatedAt?: string | null;
1934
+ }
1935
+ interface ScriptRecipeStore {
1936
+ /** FTS shortlist of healthy verified recipes for the matcher (metadata only). */
1937
+ search(params: {
1938
+ prompt: string;
1939
+ projectId?: string;
1940
+ limit?: number;
1941
+ }): Promise<ScriptRecipeMetaRow[]>;
1942
+ /** Fetch one recipe by id (any status). */
1943
+ getById(id: string): Promise<ScriptRecipeMetaRow | null>;
1944
+ /** Count healthy verified recipes (drives the "any scripts?" gate). */
1945
+ count(params?: {
1946
+ projectId?: string;
1947
+ }): Promise<number>;
1948
+ /** Insert or update a recipe row (keyed by id). */
1949
+ upsert(row: ScriptRecipeMetaRow): Promise<void>;
1950
+ /** Atomically bump counters / last-used. */
1951
+ updateStats(id: string, patch: {
1952
+ successDelta?: number;
1953
+ failureDelta?: number;
1954
+ lastUsed?: string;
1955
+ }): Promise<void>;
1956
+ /** Flip a draft to verified, applying provenance + optional fork lineage. */
1957
+ promote(id: string, patch: {
1958
+ sourceIds: string[];
1959
+ tables: string[];
1960
+ fileBase?: string;
1961
+ parentId?: string;
1962
+ forkDepth?: number;
1963
+ forkReason?: string;
1964
+ components?: ScriptComponentSpec[];
1965
+ }): Promise<ScriptRecipeMetaRow | null>;
1966
+ /** Stamp a draft's last execution error. */
1967
+ recordDraftError(id: string, err: {
1968
+ phase: string;
1969
+ message: string;
1970
+ attempt: number;
1971
+ at: string;
1972
+ }): Promise<void>;
1973
+ /** Delete a recipe row (body file removed separately). */
1974
+ remove(id: string): Promise<void>;
1975
+ /** True if `fileBase` is taken by a different recipe in this project. */
1976
+ fileBaseTaken(fileBase: string, excludeId: string, projectId?: string): Promise<boolean>;
1977
+ }
1978
+ /** Pull the injected store off the collections bag (or null if not wired). */
1979
+ declare function resolveScriptRecipeStore(collections: any): ScriptRecipeStore | null;
1980
+
1981
+ /**
1982
+ * ScriptStore — Postgres metadata + on-disk body for script recipes.
1983
+ *
1984
+ * Split of responsibilities:
1985
+ * - METADATA → injected `ScriptRecipeStore` (Postgres FTS + atomic counters),
1986
+ * resolved from `collections['script-recipes']`.
1987
+ * - BODY → scripts-store/<fileBase>.ts, editable in your IDE. Written
1988
+ * atomically (temp + rename); `bodyHash` (sha256) detects edits.
1989
+ *
1990
+ * The old "read every file every turn + send the whole catalog to the LLM"
1991
+ * matcher is gone — matching is `store.search(prompt)` (FTS shortlist). The
1992
+ * draft/verified filename dance is gone too: `status` is a DB column and the
1993
+ * file keeps a stable `<fileBase>.ts` name across promotion.
1994
+ *
1995
+ * When no metadata store is injected, the store degrades to a safe no-op
1996
+ * (count 0 → script flow disabled) instead of crashing.
1997
+ *
1998
+ * See backend/docs/SCRIPT-FLOW-SCALING-ISSUES.md.
1999
+ */
2000
+
2001
+ interface SaveDraftInput {
2002
+ /** Reuse an existing draft (retry); omit to mint a new one. */
2003
+ recipeId?: string;
2004
+ /** Per-turn unique suffix, stable across retries within the turn. */
2005
+ turnId: string;
2006
+ name: string;
2007
+ intentDescription: string;
2008
+ tags: string[];
2009
+ parameters: ScriptParameter[];
2010
+ scriptBody: string;
2011
+ createdFrom: string;
2012
+ }
2013
+ interface PromoteToVerifiedInput {
2014
+ sourceIds: string[];
2015
+ tables: string[];
2016
+ parentId?: string;
2017
+ forkDepth?: number;
2018
+ forkReason?: string;
2019
+ components?: ScriptComponentSpec[];
2020
+ }
2021
+ interface ScriptStoreOptions {
2022
+ /** Explicit metadata store, or resolved from `collections['script-recipes']`. */
2023
+ store?: ScriptRecipeStore | null;
2024
+ collections?: any;
2025
+ /** Body directory (defaults to <cwd>/scripts-store). */
2026
+ baseDir?: string;
2027
+ /** Project scope stamped on every row. */
2028
+ projectId?: string;
2029
+ }
2030
+ /**
2031
+ * Normalize a scriptBody into the on-disk form (strip a leading comment block,
2032
+ * ensure `export async function getData`). Exported for MainAgent.
2033
+ */
2034
+ declare function normalizeScriptBody(scriptBody: string): string;
2035
+ declare class ScriptStore {
2036
+ private store;
2037
+ private storeDir;
2038
+ private projectId?;
2039
+ constructor(opts?: ScriptStoreOptions);
2040
+ /** Whether a metadata store is wired (matcher / authoring are gated on this). */
2041
+ hasStore(): boolean;
2042
+ /** Number of healthy verified recipes (gates the script-matching path). */
2043
+ count(): Promise<number>;
2044
+ /**
2045
+ * FTS shortlist for the matcher (metadata only — bodies are loaded lazily by
2046
+ * `get()` once the LLM picks one). Returns verified, healthy recipes ranked
2047
+ * by relevance.
2048
+ */
2049
+ search(prompt: string, limit?: number): Promise<ScriptRecipe[]>;
2050
+ /** Fetch one recipe by id with its body loaded from disk. */
2051
+ get(id: string): Promise<ScriptRecipe | null>;
2052
+ /** Create or update a recipe (metadata upsert + body write when changed). */
2053
+ save(recipe: ScriptRecipe): Promise<void>;
2054
+ /**
2055
+ * Persist (or update) a draft. Within a turn, retries that pass the same
2056
+ * `recipeId` overwrite the same row + file; a fresh `recipeId` mints a new
2057
+ * draft. The body is visible at scripts-store/<fileBase>.ts immediately.
2058
+ */
2059
+ saveDraft(input: SaveDraftInput): Promise<ScriptRecipe>;
2060
+ /** Stamp a draft's last execution error (metadata only). */
2061
+ recordDraftError(recipeId: string, err: {
2062
+ phase: 'compile' | 'runtime' | 'timeout' | 'ipc';
2063
+ message: string;
2064
+ attempt: number;
2065
+ }): Promise<void>;
2066
+ /**
2067
+ * Promote a successfully-executed draft into a verified script.
2068
+ * The on-disk body already exists at <fileBase>.ts (written at write_script
2069
+ * time) and keeps its name — only the DB row flips status + provenance.
2070
+ */
2071
+ promoteToVerified(recipeId: string, input: PromoteToVerifiedInput): Promise<ScriptRecipe | null>;
2072
+ /**
2073
+ * Drop a draft (row + body file). MainAgent calls this at end-of-turn when a
2074
+ * draft was authored but never verified — failed drafts are never matched, so
2075
+ * deleting them immediately avoids unbounded accumulation (#5). No-op if the
2076
+ * recipe isn't a draft (so a promoted/verified script is never removed here).
2077
+ */
2078
+ discardDraft(recipeId: string): Promise<void>;
2079
+ /** Delete a recipe (row + body file). */
2080
+ delete(id: string): Promise<void>;
2081
+ /** Record a successful execution (atomic counter bump). */
2082
+ recordSuccess(id: string): Promise<void>;
2083
+ /** Record a failed execution (atomic counter bump). */
2084
+ recordFailure(id: string): Promise<void>;
2085
+ /** Absolute path to the .ts body for a recipe (used by the runner/MainAgent). */
2086
+ getScriptPath(recipe: ScriptRecipe): string;
2087
+ private removeById;
2088
+ private rowToRecipe;
2089
+ private recipeToRow;
2090
+ /** slug of name, with a short id suffix when the bare slug is already taken. */
2091
+ private computeFileBase;
2092
+ private toSlug;
2093
+ private hash;
2094
+ private bodyPath;
2095
+ private readBody;
2096
+ /** Atomic body write (temp + rename) so concurrent reads never see a partial file. */
2097
+ private writeBody;
2098
+ private unlinkBody;
2099
+ }
2100
+
2101
+ /**
2102
+ * Main Agent (Orchestrator)
2103
+ *
2104
+ * A single LLM.streamWithTools() call that handles everything:
2105
+ * - Routing: decides which source(s) to query based on summaries
2106
+ * - Querying: calls source tools (each wraps an independent SourceAgent)
2107
+ * - Direct tools: calls pre-built function tools directly with LLM-provided params
2108
+ * - Re-querying: if data is wrong/incomplete, calls tools again with modified intent
2109
+ * - Analysis: generates final text response from the data
2110
+ *
2111
+ * Two tool types:
2112
+ * - "source" tools: main agent sees summaries, SourceAgent handles SQL generation independently
2113
+ * - "direct" tools: main agent calls fn() directly with structured params (no SourceAgent)
2114
+ */
2115
+
2116
+ declare class MainAgent {
2117
+ private externalTools;
2118
+ private workflows;
2119
+ private config;
2120
+ private streamBuffer;
2121
+ /**
2122
+ * Optional: when provided, MainAgent exposes the `write_script` /
2123
+ * `execute_script` tools to the LLM and persists drafts to disk via the
2124
+ * store. Headless callers (alert analyzer, metric resolver) omit these to
2125
+ * suppress script authoring entirely — drafts would otherwise leak onto
2126
+ * disk with no caller to promote or clean them up.
2127
+ */
2128
+ private scriptStore;
2129
+ private turnId;
2130
+ private createdFromPrompt;
2131
+ private scriptState;
2132
+ constructor(externalTools: ExternalTool[], config: AgentConfig, scriptStore?: ScriptStore, turnId?: string, streamBuffer?: StreamBuffer, workflows?: WorkflowDescriptor[]);
2133
+ private get scriptingEnabled();
2134
+ /**
2135
+ * Handle a user question using the multi-agent system.
2136
+ *
2137
+ * This is ONE LLM.streamWithTools() call. The LLM:
2138
+ * 1. Sees source summaries + direct tool descriptions in system prompt
2139
+ * 2. Decides which tool(s) to call (routing)
2140
+ * 3. Source tools → SourceAgent runs independently → returns data
2141
+ * 4. Direct tools → fn() called directly with LLM params → returns data
2142
+ * 5. Generates final analysis text
2143
+ */
2144
+ handleQuestion(userPrompt: string, apiKey?: string, conversationHistory?: string, streamCallback?: (chunk: string) => void): Promise<AgentResponse>;
2145
+ private handleWriteScript;
2146
+ private handleExecuteScript;
2147
+ /**
2148
+ * Build the AgentWrittenScript payload the caller will hand to
2149
+ * `ScriptStore.promoteToVerified()`. Only returned when a verified
2150
+ * successful execution is on record.
2151
+ */
2152
+ private buildSavedScript;
2153
+ private normalizeParameterList;
2154
+ /**
2155
+ * Use the schema embedding collection to pre-select relevant tables for
2156
+ * this source + intent. Returns a formatted schema block if confidence is
2157
+ * high (top match ≥ 0.55 and ≥3 candidates), otherwise null.
2158
+ *
2159
+ * When this returns a block, we can skip the SourceAgent's `search_schema`
2160
+ * loop and reduce iteration budget. When it returns null, the SourceAgent
2161
+ * falls back to the existing LLM-driven keyword search (same as today).
2162
+ */
2163
+ private preResolveSchema;
2164
+ /**
2165
+ * Execute a direct tool — call fn() with LLM-provided params, no SourceAgent.
2166
+ */
2167
+ private handleDirectTool;
2168
+ /**
2169
+ * Build the main agent's system prompt with source summaries, direct tool descriptions,
2170
+ * and workflow component descriptions.
2171
+ */
2172
+ private buildSystemPrompt;
2173
+ /**
2174
+ * Build tool definitions for source tools — summary-only descriptions.
2175
+ * The full schema is inside the SourceAgent which runs independently.
2176
+ */
2177
+ private buildSourceToolDefinitions;
2178
+ /**
2179
+ * Build tool definitions for direct tools — expose their actual params.
2180
+ * These are called directly by the main agent LLM, no SourceAgent.
2181
+ */
2182
+ private buildDirectToolDefinitions;
2183
+ /**
2184
+ * Capture a workflow selection. We do NOT execute anything — the LLM has
2185
+ * already extracted the props it wants the workflow rendered with. We
2186
+ * record the selection (via the capture callback) and return a short
2187
+ * acknowledgement so the LLM ends its turn cleanly without writing
2188
+ * analysis text or calling more tools.
2189
+ */
2190
+ private handleWorkflow;
2191
+ /**
2192
+ * Build LLM tool definitions for workflow components. The workflow's
2193
+ * propsSchema becomes the tool's input_schema so the LLM extracts props
2194
+ * directly from the prompt — same mechanic as direct tools.
2195
+ */
2196
+ private buildWorkflowToolDefinitions;
2197
+ /**
2198
+ * Format a source agent's result as a clean string for the main agent LLM.
2199
+ */
2200
+ private formatResultForMainAgent;
2201
+ /**
2202
+ * Get source summaries (for external inspection/debugging).
2203
+ */
2204
+ getSourceSummaries(): SourceSummary[];
809
2205
  }
810
2206
 
811
2207
  /**
@@ -818,345 +2214,1276 @@ interface Action {
818
2214
  [key: string]: any;
819
2215
  }
820
2216
 
2217
+ type SystemPrompt = string | Anthropic.Messages.TextBlockParam[];
2218
+ interface LLMMessages {
2219
+ sys: SystemPrompt;
2220
+ user: string;
2221
+ prefill?: string;
2222
+ }
2223
+ interface LLMOptions {
2224
+ model?: string;
2225
+ maxTokens?: number;
2226
+ temperature?: number;
2227
+ topP?: number;
2228
+ apiKey?: string;
2229
+ partial?: (chunk: string) => void;
2230
+ }
2231
+ interface Tool {
2232
+ name: string;
2233
+ description: string;
2234
+ input_schema: {
2235
+ type: string;
2236
+ properties: Record<string, any>;
2237
+ required?: string[];
2238
+ };
2239
+ }
2240
+ declare class LLM {
2241
+ static text(messages: LLMMessages, options?: LLMOptions): Promise<string>;
2242
+ static stream<T = string>(messages: LLMMessages, options?: LLMOptions, json?: boolean): Promise<T extends string ? string : any>;
2243
+ static streamWithTools(messages: LLMMessages, tools: Tool[], toolHandler: (toolName: string, toolInput: any) => Promise<any>, options?: LLMOptions, maxIterations?: number): Promise<string>;
2244
+ /**
2245
+ * Normalize system prompt to Anthropic format
2246
+ * Converts string to array format if needed
2247
+ * @param sys - System prompt (string or array of blocks)
2248
+ * @returns Normalized system prompt for Anthropic API
2249
+ */
2250
+ private static _normalizeSystemPrompt;
2251
+ /**
2252
+ * Strip unpaired UTF-16 surrogates from every text field of a message set.
2253
+ *
2254
+ * A lone surrogate (from mid-pair string slicing or corrupt source data)
2255
+ * serializes to a bare `\udXXX` escape that strict JSON parsers — including
2256
+ * the one on Anthropic's API — reject with "no low surrogate in string",
2257
+ * failing the whole request. Sanitizing here, at the single boundary every
2258
+ * provider call flows through, guarantees no request can carry one.
2259
+ */
2260
+ private static _sanitizeMessages;
2261
+ /**
2262
+ * Log cache usage metrics from Anthropic API response
2263
+ * Shows cache hits, costs, and savings
2264
+ */
2265
+ private static _logCacheUsage;
2266
+ /**
2267
+ * Parse model string to extract provider and model name
2268
+ * @param modelString - Format: "provider/model-name" or just "model-name"
2269
+ * @returns [provider, modelName]
2270
+ *
2271
+ * @example
2272
+ * "anthropic/claude-sonnet-4-5" → ["anthropic", "claude-sonnet-4-5"]
2273
+ * "groq/openai/gpt-oss-120b" → ["groq", "openai/gpt-oss-120b"]
2274
+ * "claude-sonnet-4-5" → ["anthropic", "claude-sonnet-4-5"] (default)
2275
+ */
2276
+ private static _parseModel;
2277
+ private static _anthropicText;
2278
+ private static _anthropicStream;
2279
+ private static _anthropicStreamWithTools;
2280
+ private static _groqText;
2281
+ private static _groqStream;
2282
+ private static _geminiText;
2283
+ private static _geminiStream;
2284
+ /**
2285
+ * Recursively strip unsupported JSON Schema properties for Gemini
2286
+ * Gemini doesn't support: additionalProperties, $schema, etc.
2287
+ */
2288
+ private static _cleanSchemaForGemini;
2289
+ private static _geminiStreamWithTools;
2290
+ private static _openaiText;
2291
+ private static _openaiStream;
2292
+ private static _openaiStreamWithTools;
2293
+ /**
2294
+ * Parse JSON string, handling markdown code blocks and surrounding text
2295
+ * Enhanced version with jsonrepair to handle malformed JSON from LLMs
2296
+ * @param text - Text that may contain JSON wrapped in ```json...``` or with surrounding text
2297
+ * @returns Parsed JSON object or array
2298
+ */
2299
+ private static _parseJSON;
2300
+ }
2301
+
2302
+ interface CapturedLog {
2303
+ timestamp: number;
2304
+ level: 'info' | 'error' | 'warn' | 'debug';
2305
+ message: string;
2306
+ type?: 'explanation' | 'query' | 'general';
2307
+ data?: Record<string, any>;
2308
+ }
821
2309
  /**
822
- * UIBlock represents a single user and assistant message block in a thread
823
- * Contains user question, component metadata, component data, and available actions
2310
+ * UILogCollector captures logs during user prompt processing
2311
+ * and sends them to runtime via ui_logs message with uiBlockId as the message id
2312
+ * Logs are sent in real-time for streaming effect in the UI
2313
+ * Respects the global log level configuration
824
2314
  */
825
- declare class UIBlock {
826
- private id;
827
- private userQuestion;
828
- private generatedComponentMetadata;
829
- private componentData;
830
- private actions;
831
- private createdAt;
2315
+ declare class UILogCollector {
2316
+ private logs;
2317
+ private uiBlockId;
2318
+ private clientId;
2319
+ private sendMessage;
2320
+ private currentLogLevel;
2321
+ constructor(clientId: string, sendMessage: (message: Message) => void, uiBlockId?: string);
832
2322
  /**
833
- * Creates a new UIBlock instance
834
- * @param userQuestion - The user's question or input
835
- * @param componentData - The component data object
836
- * @param generatedComponentMetadata - Optional metadata about the generated component
837
- * @param actions - Optional array of available actions
838
- * @param id - Optional custom ID, generates UUID if not provided
2323
+ * Check if logging is enabled (uiBlockId is provided)
2324
+ */
2325
+ isEnabled(): boolean;
2326
+ /**
2327
+ * Check if a message should be logged based on current log level
839
2328
  */
840
- constructor(userQuestion: string, componentData?: Record<string, any>, generatedComponentMetadata?: Record<string, any>, actions?: Action[], id?: string);
2329
+ private shouldLog;
841
2330
  /**
842
- * Get the UIBlock ID
2331
+ * Add a log entry with timestamp and immediately send to runtime
2332
+ * Only logs that pass the log level filter are captured and sent
843
2333
  */
844
- getId(): string;
2334
+ private addLog;
845
2335
  /**
846
- * Get the user question
2336
+ * Send a single log to runtime immediately
847
2337
  */
848
- getUserQuestion(): string;
2338
+ private sendLogImmediately;
849
2339
  /**
850
- * Set or update the user question
2340
+ * Log info message
851
2341
  */
852
- setUserQuestion(question: string): void;
2342
+ info(message: string, type?: 'explanation' | 'query' | 'general', data?: Record<string, any>): void;
853
2343
  /**
854
- * Get component metadata
2344
+ * Log error message
855
2345
  */
856
- getComponentMetadata(): Record<string, any>;
2346
+ error(message: string, type?: 'explanation' | 'query' | 'general', data?: Record<string, any>): void;
857
2347
  /**
858
- * Set or update component metadata
2348
+ * Log warning message
859
2349
  */
860
- setComponentMetadata(metadata: Record<string, any>): void;
2350
+ warn(message: string, type?: 'explanation' | 'query' | 'general', data?: Record<string, any>): void;
2351
+ /**
2352
+ * Log debug message
2353
+ */
2354
+ debug(message: string, type?: 'explanation' | 'query' | 'general', data?: Record<string, any>): void;
2355
+ /**
2356
+ * Log LLM explanation with typed metadata
2357
+ */
2358
+ logExplanation(message: string, explanation: string, data?: Record<string, any>): void;
2359
+ /**
2360
+ * Log generated query with typed metadata
2361
+ */
2362
+ logQuery(message: string, query: string, data?: Record<string, any>): void;
2363
+ /**
2364
+ * Send all collected logs at once (optional, for final summary)
2365
+ */
2366
+ sendAllLogs(): void;
2367
+ /**
2368
+ * Get all collected logs
2369
+ */
2370
+ getLogs(): CapturedLog[];
2371
+ /**
2372
+ * Clear all logs
2373
+ */
2374
+ clearLogs(): void;
2375
+ /**
2376
+ * Set uiBlockId (in case it's provided later)
2377
+ */
2378
+ setUIBlockId(uiBlockId: string): void;
2379
+ }
2380
+
2381
+ /**
2382
+ * UIBlock represents a single user and assistant message block in a thread
2383
+ * Contains user question, component metadata, component data, text response, and available actions
2384
+ */
2385
+ declare class UIBlock {
2386
+ private id;
2387
+ private userQuestion;
2388
+ private generatedComponentMetadata;
2389
+ private componentData;
2390
+ private textResponse;
2391
+ private actions;
2392
+ private createdAt;
2393
+ /**
2394
+ * Creates a new UIBlock instance
2395
+ * @param userQuestion - The user's question or input
2396
+ * @param componentData - The component data object
2397
+ * @param generatedComponentMetadata - Optional metadata about the generated component
2398
+ * @param actions - Optional array of available actions
2399
+ * @param id - Optional custom ID, generates UUID if not provided
2400
+ * @param textResponse - Optional text response from LLM
2401
+ */
2402
+ constructor(userQuestion: string, componentData?: Record<string, any>, generatedComponentMetadata?: Record<string, any>, actions?: Action[], id?: string, textResponse?: string | null);
2403
+ /**
2404
+ * Get the UIBlock ID
2405
+ */
2406
+ getId(): string;
2407
+ /**
2408
+ * Get the user question
2409
+ */
2410
+ getUserQuestion(): string;
2411
+ /**
2412
+ * Set or update the user question
2413
+ */
2414
+ setUserQuestion(question: string): void;
2415
+ /**
2416
+ * Get component metadata
2417
+ */
2418
+ getComponentMetadata(): Record<string, any>;
2419
+ getTextResponse(): string;
2420
+ /**
2421
+ * Set or update component metadata
2422
+ */
2423
+ setComponentMetadata(metadata: Record<string, any>): void;
2424
+ /**
2425
+ * Get component data
2426
+ */
2427
+ getComponentData(): Record<string, any>;
2428
+ /**
2429
+ * Calculate size of data in bytes
2430
+ */
2431
+ private getDataSizeInBytes;
2432
+ /**
2433
+ * Limit array data to maximum rows
2434
+ */
2435
+ private limitArrayData;
2436
+ /**
2437
+ * Check if data exceeds size limit
2438
+ */
2439
+ private exceedsSizeLimit;
2440
+ /**
2441
+ * Process and limit data before storing
2442
+ */
2443
+ private processDataForStorage;
2444
+ /**
2445
+ * Set or update component data with size and row limits
2446
+ */
2447
+ setComponentData(data: Record<string, any>): void;
2448
+ /**
2449
+ * Set or update text response
2450
+ */
2451
+ setTextResponse(textResponse: string | null): void;
2452
+ /**
2453
+ * Get all actions (only if they are resolved, not if fetching)
2454
+ */
2455
+ getActions(): Action[] | null | Promise<Action[]>;
2456
+ /**
2457
+ * Get or fetch actions
2458
+ * If actions don't exist or are a Promise, calls the generateFn and stores the promise
2459
+ * If actions already exist, returns them
2460
+ * @param generateFn - Async function to generate actions
2461
+ * @returns Promise resolving to Action[]
2462
+ */
2463
+ getOrFetchActions(generateFn: () => Promise<Action[]>): Promise<Action[]>;
2464
+ /**
2465
+ * Set or replace all actions
2466
+ */
2467
+ setActions(actions: Action[]): void;
2468
+ /**
2469
+ * Add a single action (only if actions are resolved)
2470
+ */
2471
+ addAction(action: Action): void;
2472
+ /**
2473
+ * Add multiple actions (only if actions are resolved)
2474
+ */
2475
+ addActions(actions: Action[]): void;
2476
+ /**
2477
+ * Remove an action by ID (only if actions are resolved)
2478
+ */
2479
+ removeAction(actionId: string): boolean;
2480
+ /**
2481
+ * Clear all actions
2482
+ */
2483
+ clearActions(): void;
2484
+ /**
2485
+ * Get creation timestamp
2486
+ */
2487
+ getCreatedAt(): Date;
2488
+ /**
2489
+ * Convert UIBlock to JSON-serializable object
2490
+ */
2491
+ toJSON(): Record<string, any>;
2492
+ }
2493
+
2494
+ /**
2495
+ * Thread represents a conversation thread containing multiple UIBlocks
2496
+ * Each UIBlock in a thread represents a user question and assistant response pair
2497
+ */
2498
+ declare class Thread {
2499
+ private id;
2500
+ private uiblocks;
2501
+ private createdAt;
2502
+ /**
2503
+ * Creates a new Thread instance
2504
+ * @param id - Optional custom ID, generates UUID if not provided
2505
+ */
2506
+ constructor(id?: string);
2507
+ /**
2508
+ * Get the thread ID
2509
+ */
2510
+ getId(): string;
2511
+ /**
2512
+ * Add a UIBlock to the thread
2513
+ */
2514
+ addUIBlock(uiblock: UIBlock): void;
2515
+ /**
2516
+ * Get a UIBlock by ID
2517
+ */
2518
+ getUIBlock(id: string): UIBlock | undefined;
2519
+ /**
2520
+ * Get all UIBlocks in the thread
2521
+ */
2522
+ getUIBlocks(): UIBlock[];
2523
+ /**
2524
+ * Get UIBlocks as a Map
2525
+ */
2526
+ getUIBlocksMap(): Map<string, UIBlock>;
2527
+ /**
2528
+ * Remove a UIBlock by ID
2529
+ */
2530
+ removeUIBlock(id: string): boolean;
2531
+ /**
2532
+ * Check if UIBlock exists
2533
+ */
2534
+ hasUIBlock(id: string): boolean;
2535
+ /**
2536
+ * Get number of UIBlocks in the thread
2537
+ */
2538
+ getUIBlockCount(): number;
2539
+ /**
2540
+ * Clear all UIBlocks from the thread
2541
+ */
2542
+ clear(): void;
2543
+ /**
2544
+ * Get creation timestamp
2545
+ */
2546
+ getCreatedAt(): Date;
2547
+ /**
2548
+ * Get conversation context from recent UIBlocks (excluding current one)
2549
+ * Returns formatted string with previous questions and component summaries
2550
+ * @param limit - Maximum number of previous UIBlocks to include (default: 5)
2551
+ * @param currentUIBlockId - ID of current UIBlock to exclude from context (optional)
2552
+ * @returns Formatted conversation history string
2553
+ */
2554
+ getConversationContext(limit?: number, currentUIBlockId?: string): string;
2555
+ /**
2556
+ * Convert Thread to JSON-serializable object
2557
+ */
2558
+ toJSON(): Record<string, any>;
2559
+ }
2560
+
2561
+ /**
2562
+ * ThreadManager manages all threads globally
2563
+ * Provides methods to create, retrieve, and delete threads.
2564
+ * Includes automatic cleanup to prevent unbounded memory growth.
2565
+ */
2566
+ declare class ThreadManager {
2567
+ private static instance;
2568
+ private threads;
2569
+ private cleanupInterval;
2570
+ private readonly threadTtlMs;
2571
+ private constructor();
2572
+ /**
2573
+ * Periodically remove threads older than 7 days.
2574
+ * Runs every hour to avoid frequent iteration over the map.
2575
+ */
2576
+ private startCleanup;
2577
+ /**
2578
+ * Get singleton instance of ThreadManager
2579
+ */
2580
+ static getInstance(): ThreadManager;
2581
+ /**
2582
+ * Create a new thread
2583
+ * @param id - Optional custom ID, generates UUID if not provided
2584
+ * @returns The created Thread instance
2585
+ */
2586
+ createThread(id?: string): Thread;
2587
+ /**
2588
+ * Get a thread by ID
2589
+ */
2590
+ getThread(id: string): Thread | undefined;
2591
+ /**
2592
+ * Get all threads
2593
+ */
2594
+ getAllThreads(): Thread[];
2595
+ /**
2596
+ * Get threads as a Map
2597
+ */
2598
+ getThreadsMap(): Map<string, Thread>;
2599
+ /**
2600
+ * Delete a thread by ID
2601
+ */
2602
+ deleteThread(id: string): boolean;
2603
+ /**
2604
+ * Check if thread exists
2605
+ */
2606
+ hasThread(id: string): boolean;
2607
+ /**
2608
+ * Get number of threads
2609
+ */
2610
+ getThreadCount(): number;
2611
+ /**
2612
+ * Clear all threads
2613
+ */
2614
+ clearAll(): void;
2615
+ /**
2616
+ * Find a UIBlock by ID across all threads
2617
+ * @param uiBlockId - The UIBlock ID to search for
2618
+ * @returns Object with thread and uiBlock if found, undefined otherwise
2619
+ */
2620
+ findUIBlockById(uiBlockId: string): {
2621
+ thread: Thread;
2622
+ uiBlock: UIBlock;
2623
+ } | undefined;
2624
+ /**
2625
+ * Convert all threads to JSON-serializable object
2626
+ */
2627
+ toJSON(): Record<string, any>;
2628
+ }
2629
+
2630
+ /**
2631
+ * CleanupService handles cleanup of old threads and UIBlocks
2632
+ * to prevent memory bloat and maintain optimal performance
2633
+ */
2634
+ declare class CleanupService {
2635
+ private static instance;
2636
+ private cleanupInterval;
2637
+ private constructor();
2638
+ /**
2639
+ * Get singleton instance of CleanupService
2640
+ */
2641
+ static getInstance(): CleanupService;
2642
+ /**
2643
+ * Clean up old threads based on retention period
2644
+ * @param retentionDays - Number of days to keep threads (defaults to config)
2645
+ * @returns Number of threads deleted
2646
+ */
2647
+ cleanupOldThreads(retentionDays?: number): number;
2648
+ /**
2649
+ * Clean up old UIBlocks within threads based on retention period
2650
+ * @param retentionDays - Number of days to keep UIBlocks (defaults to config)
2651
+ * @returns Object with number of UIBlocks deleted per thread
2652
+ */
2653
+ cleanupOldUIBlocks(retentionDays?: number): {
2654
+ [threadId: string]: number;
2655
+ };
2656
+ /**
2657
+ * Clear all component data from UIBlocks to free memory
2658
+ * Keeps metadata but removes the actual data
2659
+ * @param retentionDays - Number of days to keep full data (defaults to config)
2660
+ * @returns Number of UIBlocks whose data was cleared
2661
+ */
2662
+ clearOldUIBlockData(retentionDays?: number): number;
2663
+ /**
2664
+ * Run full cleanup (threads, UIBlocks, and data)
2665
+ * @returns Cleanup statistics
2666
+ */
2667
+ runFullCleanup(): {
2668
+ threadsDeleted: number;
2669
+ uiblocksDeleted: {
2670
+ [threadId: string]: number;
2671
+ };
2672
+ dataCleared: number;
2673
+ };
2674
+ /**
2675
+ * Start automatic cleanup at regular intervals
2676
+ * @param intervalHours - Hours between cleanup runs (default: 24)
2677
+ */
2678
+ startAutoCleanup(intervalHours?: number): void;
2679
+ /**
2680
+ * Stop automatic cleanup
2681
+ */
2682
+ stopAutoCleanup(): void;
2683
+ /**
2684
+ * Check if auto cleanup is running
2685
+ */
2686
+ isAutoCleanupRunning(): boolean;
2687
+ /**
2688
+ * Get current memory usage statistics
2689
+ */
2690
+ getMemoryStats(): {
2691
+ threadCount: number;
2692
+ totalUIBlocks: number;
2693
+ avgUIBlocksPerThread: number;
2694
+ };
2695
+ }
2696
+
2697
+ /**
2698
+ * Configuration for data storage limits in UIBlocks
2699
+ */
2700
+ declare const STORAGE_CONFIG: {
2701
+ /**
2702
+ * Maximum number of rows to store in UIBlock data
2703
+ */
2704
+ MAX_ROWS_PER_BLOCK: number;
2705
+ /**
2706
+ * Maximum size in bytes per UIBlock (500KB - reduced to save memory)
2707
+ */
2708
+ MAX_SIZE_PER_BLOCK_BYTES: number;
2709
+ /**
2710
+ * Number of days to keep threads before cleanup
2711
+ * Note: This is for in-memory storage. Conversations are also persisted to database.
2712
+ */
2713
+ THREAD_RETENTION_DAYS: number;
2714
+ /**
2715
+ * Number of days to keep UIBlocks before cleanup
2716
+ * Note: This is for in-memory storage. Data is also persisted to database.
2717
+ */
2718
+ UIBLOCK_RETENTION_DAYS: number;
2719
+ };
2720
+
2721
+ /**
2722
+ * Configuration for conversation context and history management
2723
+ */
2724
+ declare const CONTEXT_CONFIG: {
2725
+ /**
2726
+ * Maximum number of previous UIBlocks to include as conversation context
2727
+ * Set to 0 to disable conversation history
2728
+ * Higher values provide more context but may increase token usage
2729
+ */
2730
+ MAX_CONVERSATION_CONTEXT_BLOCKS: number;
2731
+ };
2732
+
2733
+ /**
2734
+ * LLM Usage Logger - Tracks token usage, costs, and timing for all LLM API calls
2735
+ */
2736
+ interface LLMUsageEntry {
2737
+ timestamp: string;
2738
+ requestId: string;
2739
+ provider: string;
2740
+ model: string;
2741
+ method: string;
2742
+ inputTokens: number;
2743
+ outputTokens: number;
2744
+ cacheReadTokens?: number;
2745
+ cacheWriteTokens?: number;
2746
+ totalTokens: number;
2747
+ costUSD: number;
2748
+ durationMs: number;
2749
+ toolCalls?: number;
2750
+ success: boolean;
2751
+ error?: string;
2752
+ }
2753
+ declare class LLMUsageLogger {
2754
+ private logStream;
2755
+ private logPath;
2756
+ private enabled;
2757
+ private sessionStats;
2758
+ constructor();
2759
+ private initLogStream;
2760
+ private writeHeader;
2761
+ /**
2762
+ * Calculate cost based on token usage and model
2763
+ */
2764
+ calculateCost(model: string, inputTokens: number, outputTokens: number, cacheReadTokens?: number, cacheWriteTokens?: number): number;
2765
+ /**
2766
+ * Log an LLM API call
2767
+ */
2768
+ log(entry: LLMUsageEntry): void;
2769
+ /**
2770
+ * Log session summary (call at end of request)
2771
+ */
2772
+ logSessionSummary(requestContext?: string): void;
2773
+ /**
2774
+ * Reset session stats (call at start of new user request)
2775
+ */
2776
+ resetSession(): void;
2777
+ /**
2778
+ * Reset the log file for a new request (clears previous logs)
2779
+ * Call this at the start of each USER_PROMPT_REQ
2780
+ */
2781
+ resetLogFile(requestContext?: string): void;
2782
+ /**
2783
+ * Get current session stats
2784
+ */
2785
+ getSessionStats(): {
2786
+ totalCalls: number;
2787
+ totalInputTokens: number;
2788
+ totalOutputTokens: number;
2789
+ totalCacheReadTokens: number;
2790
+ totalCacheWriteTokens: number;
2791
+ totalCostUSD: number;
2792
+ totalDurationMs: number;
2793
+ };
2794
+ /**
2795
+ * Generate a unique request ID
2796
+ */
2797
+ generateRequestId(): string;
2798
+ }
2799
+ declare const llmUsageLogger: LLMUsageLogger;
2800
+
2801
+ /**
2802
+ * User Prompt Error Logger - Captures detailed errors for USER_PROMPT_REQ
2803
+ * Logs full error details including raw strings for parse failures
2804
+ */
2805
+ declare class UserPromptErrorLogger {
2806
+ private logStream;
2807
+ private logPath;
2808
+ private enabled;
2809
+ private hasErrors;
2810
+ constructor();
2811
+ /**
2812
+ * Reset the error log file for a new request
2813
+ */
2814
+ resetLogFile(requestContext?: string): void;
2815
+ /**
2816
+ * Log a JSON parse error with the raw string that failed
2817
+ */
2818
+ logJsonParseError(context: string, rawString: string, error: Error): void;
2819
+ /**
2820
+ * Log a general error with full details
2821
+ */
2822
+ logError(context: string, error: Error | string, additionalData?: Record<string, any>): void;
2823
+ /**
2824
+ * Log a SQL query error with the full query
2825
+ */
2826
+ logSqlError(query: string, error: Error | string, params?: any[]): void;
2827
+ /**
2828
+ * Log an LLM API error
2829
+ */
2830
+ logLlmError(provider: string, model: string, method: string, error: Error | string, requestData?: any): void;
2831
+ /**
2832
+ * Log tool execution error
2833
+ */
2834
+ logToolError(toolName: string, toolInput: any, error: Error | string): void;
2835
+ /**
2836
+ * Write final summary if there were errors
2837
+ */
2838
+ writeSummary(): void;
2839
+ /**
2840
+ * Check if any errors were logged
2841
+ */
2842
+ hadErrors(): boolean;
2843
+ private write;
2844
+ }
2845
+ declare const userPromptErrorLogger: UserPromptErrorLogger;
2846
+
2847
+ /**
2848
+ * BM25L Reranker for hybrid semantic search
2849
+ *
2850
+ * BM25L is an improved variant of BM25 that provides better handling of
2851
+ * long documents and term frequency saturation. This implementation is
2852
+ * designed to rerank semantic search results from ChromaDB.
2853
+ *
2854
+ * The hybrid approach combines:
2855
+ * 1. Semantic similarity from ChromaDB embeddings (dense vectors)
2856
+ * 2. Lexical matching from BM25L (sparse, keyword-based)
2857
+ *
2858
+ * This addresses the weakness of pure semantic search which may miss
2859
+ * exact keyword matches that are important for user intent.
2860
+ */
2861
+ interface BM25LOptions {
2862
+ /** Term frequency saturation parameter (default: 1.5) */
2863
+ k1?: number;
2864
+ /** Length normalization parameter (default: 0.75) */
2865
+ b?: number;
2866
+ /** Lower-bound adjustment from BM25L paper (default: 0.5) */
2867
+ delta?: number;
2868
+ }
2869
+ interface RerankedResult<T> {
2870
+ item: T;
2871
+ originalIndex: number;
2872
+ semanticScore: number;
2873
+ bm25Score: number;
2874
+ hybridScore: number;
2875
+ }
2876
+ interface HybridSearchOptions extends BM25LOptions {
2877
+ /** Weight for semantic score (0-1, default: 0.7) */
2878
+ semanticWeight?: number;
2879
+ /** Weight for BM25 score (0-1, default: 0.3) */
2880
+ bm25Weight?: number;
2881
+ /** Minimum hybrid score threshold (0-1, default: 0) */
2882
+ minScore?: number;
2883
+ }
2884
+ /**
2885
+ * BM25L implementation for lexical scoring
2886
+ */
2887
+ declare class BM25L {
2888
+ private k1;
2889
+ private b;
2890
+ private delta;
2891
+ private documents;
2892
+ private docLengths;
2893
+ private avgDocLength;
2894
+ private termDocFreq;
2895
+ /**
2896
+ * @param documents - Array of raw documents (strings)
2897
+ * @param opts - Optional BM25L parameters
2898
+ */
2899
+ constructor(documents?: string[], opts?: BM25LOptions);
2900
+ /**
2901
+ * Tokenize text into lowercase alphanumeric tokens
2902
+ */
2903
+ tokenize(text: string): string[];
2904
+ /**
2905
+ * Compute IDF (Inverse Document Frequency) with smoothing
2906
+ */
2907
+ private idf;
2908
+ /**
2909
+ * Compute BM25L score for a single document
2910
+ */
2911
+ score(query: string, docIndex: number): number;
2912
+ /**
2913
+ * Search and rank all documents
2914
+ */
2915
+ search(query: string): Array<{
2916
+ index: number;
2917
+ score: number;
2918
+ }>;
2919
+ }
2920
+ /**
2921
+ * Hybrid reranker that combines semantic and BM25L scores
2922
+ *
2923
+ * @param query - The search query
2924
+ * @param items - Array of items to rerank
2925
+ * @param getDocument - Function to extract document text from an item
2926
+ * @param getSemanticScore - Function to extract semantic similarity score from an item
2927
+ * @param options - Hybrid search options
2928
+ * @returns Reranked items with hybrid scores
2929
+ */
2930
+ declare function hybridRerank<T>(query: string, items: T[], getDocument: (item: T) => string, getSemanticScore: (item: T) => number, options?: HybridSearchOptions): RerankedResult<T>[];
2931
+ /**
2932
+ * Simple reranking function for ChromaDB results
2933
+ *
2934
+ * This is a convenience wrapper for reranking ChromaDB query results
2935
+ * that follow the standard { ids, documents, metadatas, distances } format.
2936
+ *
2937
+ * @param query - The search query
2938
+ * @param chromaResults - ChromaDB query results
2939
+ * @param options - Hybrid search options
2940
+ * @returns Reranked results with hybrid scores
2941
+ */
2942
+ declare function rerankChromaResults(query: string, chromaResults: {
2943
+ ids: string[][];
2944
+ documents: (string | null)[][];
2945
+ metadatas: Record<string, any>[][];
2946
+ distances: number[][];
2947
+ }, options?: HybridSearchOptions): Array<{
2948
+ id: string;
2949
+ document: string | null;
2950
+ metadata: Record<string, any>;
2951
+ distance: number;
2952
+ semanticScore: number;
2953
+ bm25Score: number;
2954
+ hybridScore: number;
2955
+ }>;
2956
+ /**
2957
+ * Rerank conversation search results specifically
2958
+ *
2959
+ * This function is designed to work with the conversation-history.search collection
2960
+ * where we need to fetch more results initially and then rerank them.
2961
+ *
2962
+ * @param query - The user's search query
2963
+ * @param results - Array of conversation search results from ChromaDB
2964
+ * @param options - Hybrid search options
2965
+ * @returns Reranked results sorted by hybrid score
2966
+ */
2967
+ declare function rerankConversationResults<T extends {
2968
+ userPrompt?: string;
2969
+ similarity?: number;
2970
+ }>(query: string, results: T[], options?: HybridSearchOptions): Array<T & {
2971
+ hybridScore: number;
2972
+ bm25Score: number;
2973
+ }>;
2974
+
2975
+ /**
2976
+ * QueryExecutionService - Handles all query execution, validation, and retry logic
2977
+ * Extracted from BaseLLM for better separation of concerns
2978
+ */
2979
+
2980
+ /**
2981
+ * Context for component when requesting query fix
2982
+ */
2983
+ interface ComponentContext {
2984
+ name: string;
2985
+ type: string;
2986
+ title?: string;
2987
+ }
2988
+ /**
2989
+ * Result of query validation
2990
+ */
2991
+ interface QueryValidationResult {
2992
+ component: Component | null;
2993
+ queryKey: string;
2994
+ result: any;
2995
+ validated: boolean;
2996
+ }
2997
+ /**
2998
+ * Result of batch query validation
2999
+ */
3000
+ interface BatchValidationResult {
3001
+ components: Component[];
3002
+ queryResults: Map<string, any>;
3003
+ }
3004
+ /**
3005
+ * Configuration for QueryExecutionService
3006
+ */
3007
+ interface QueryExecutionServiceConfig {
3008
+ defaultLimit: number;
3009
+ getModelForTask: (taskType: 'simple' | 'complex') => string;
3010
+ getApiKey: (apiKey?: string) => string | undefined;
3011
+ providerName: string;
3012
+ }
3013
+ /**
3014
+ * QueryExecutionService handles all query-related operations
3015
+ */
3016
+ declare class QueryExecutionService {
3017
+ private config;
3018
+ constructor(config: QueryExecutionServiceConfig);
861
3019
  /**
862
- * Get component data
3020
+ * Get the cache key for a query
3021
+ * This ensures the cache key matches what the frontend will send
863
3022
  */
864
- getComponentData(): Record<string, any>;
3023
+ getQueryCacheKey(query: any): string;
865
3024
  /**
866
- * Calculate size of data in bytes
3025
+ * Execute a query against the database
3026
+ * @param query - The SQL query to execute (string or object with sql/values)
3027
+ * @param collections - Collections object containing database execute function
3028
+ * @returns Object with result data and cache key
867
3029
  */
868
- private getDataSizeInBytes;
3030
+ executeQuery(query: any, collections: any): Promise<{
3031
+ result: any;
3032
+ cacheKey: string;
3033
+ }>;
869
3034
  /**
870
- * Limit array data to maximum rows
3035
+ * Request the LLM to fix a failed SQL query
3036
+ * @param failedQuery - The query that failed execution
3037
+ * @param errorMessage - The error message from the failed execution
3038
+ * @param componentContext - Context about the component
3039
+ * @param apiKey - Optional API key
3040
+ * @returns Fixed query string
871
3041
  */
872
- private limitArrayData;
3042
+ requestQueryFix(failedQuery: string, errorMessage: string, componentContext: ComponentContext, apiKey?: string): Promise<string>;
873
3043
  /**
874
- * Check if data exceeds size limit
3044
+ * Validate a single component's query with retry logic
3045
+ * @param component - The component to validate
3046
+ * @param collections - Collections object containing database execute function
3047
+ * @param apiKey - Optional API key for LLM calls
3048
+ * @returns Validation result with component, query key, and result
875
3049
  */
876
- private exceedsSizeLimit;
3050
+ validateSingleQuery(component: Component, collections: any, apiKey?: string): Promise<QueryValidationResult>;
877
3051
  /**
878
- * Process and limit data before storing
3052
+ * Validate multiple component queries in parallel
3053
+ * @param components - Array of components with potential queries
3054
+ * @param collections - Collections object containing database execute function
3055
+ * @param apiKey - Optional API key for LLM calls
3056
+ * @returns Object with validated components and query results map
879
3057
  */
880
- private processDataForStorage;
3058
+ validateComponentQueries(components: Component[], collections: any, apiKey?: string): Promise<BatchValidationResult>;
3059
+ }
3060
+
3061
+ /**
3062
+ * Task types for model selection
3063
+ * - 'complex': Text generation, component matching, parameter adaptation (uses best model in balanced mode)
3064
+ * - 'simple': Classification, action generation (uses fast model in balanced mode)
3065
+ */
3066
+ type TaskType = 'complex' | 'simple';
3067
+ interface BaseLLMConfig {
3068
+ model?: string;
3069
+ fastModel?: string;
3070
+ defaultLimit?: number;
3071
+ apiKey?: string;
881
3072
  /**
882
- * Set or update component data with size and row limits
3073
+ * Model selection strategy:
3074
+ * - 'best': Use best model for all tasks (highest quality, higher cost)
3075
+ * - 'fast': Use fast model for all tasks (lower quality, lower cost)
3076
+ * - 'balanced': Use best model for complex tasks, fast model for simple tasks (default)
883
3077
  */
884
- setComponentData(data: Record<string, any>): void;
3078
+ modelStrategy?: ModelStrategy;
3079
+ conversationSimilarityThreshold?: number;
3080
+ }
3081
+ /**
3082
+ * BaseLLM abstract class for AI-powered component generation and matching
3083
+ * Provides common functionality for all LLM providers
3084
+ */
3085
+ declare abstract class BaseLLM {
3086
+ protected model: string;
3087
+ protected fastModel: string;
3088
+ protected defaultLimit: number;
3089
+ protected apiKey?: string;
3090
+ protected modelStrategy: ModelStrategy;
3091
+ protected conversationSimilarityThreshold: number;
3092
+ protected queryService: QueryExecutionService;
3093
+ constructor(config?: BaseLLMConfig);
885
3094
  /**
886
- * Get all actions (only if they are resolved, not if fetching)
3095
+ * Get the appropriate model based on task type and model strategy
3096
+ * @param taskType - 'complex' for text generation/matching, 'simple' for classification/actions
3097
+ * @returns The model string to use for this task
887
3098
  */
888
- getActions(): Action[] | null | Promise<Action[]>;
3099
+ protected getModelForTask(taskType: TaskType): string;
889
3100
  /**
890
- * Get or fetch actions
891
- * If actions don't exist or are a Promise, calls the generateFn and stores the promise
892
- * If actions already exist, returns them
893
- * @param generateFn - Async function to generate actions
894
- * @returns Promise resolving to Action[]
3101
+ * Set the model strategy at runtime
3102
+ * @param strategy - 'best', 'fast', or 'balanced'
895
3103
  */
896
- getOrFetchActions(generateFn: () => Promise<Action[]>): Promise<Action[]>;
3104
+ setModelStrategy(strategy: ModelStrategy): void;
897
3105
  /**
898
- * Add a single action (only if actions are resolved)
3106
+ * Get the current model strategy
3107
+ * @returns The current model strategy
899
3108
  */
900
- addAction(action: Action): void;
3109
+ getModelStrategy(): ModelStrategy;
901
3110
  /**
902
- * Add multiple actions (only if actions are resolved)
3111
+ * Set the conversation similarity threshold at runtime
3112
+ * @param threshold - Value between 0 and 1 (e.g., 0.8 = 80% similarity required)
903
3113
  */
904
- addActions(actions: Action[]): void;
3114
+ setConversationSimilarityThreshold(threshold: number): void;
905
3115
  /**
906
- * Remove an action by ID (only if actions are resolved)
3116
+ * Get the current conversation similarity threshold
3117
+ * @returns The current threshold value
907
3118
  */
908
- removeAction(actionId: string): boolean;
3119
+ getConversationSimilarityThreshold(): number;
909
3120
  /**
910
- * Clear all actions
3121
+ * Get the default model for this provider (used for complex tasks like text generation)
911
3122
  */
912
- clearActions(): void;
3123
+ protected abstract getDefaultModel(): string;
913
3124
  /**
914
- * Get creation timestamp
3125
+ * Get the default fast model for this provider (used for simple tasks: classification, matching, actions)
3126
+ * Should return a cheaper/faster model like Haiku for Anthropic
915
3127
  */
916
- getCreatedAt(): Date;
3128
+ protected abstract getDefaultFastModel(): string;
917
3129
  /**
918
- * Convert UIBlock to JSON-serializable object
3130
+ * Get the default API key from environment
919
3131
  */
920
- toJSON(): Record<string, any>;
921
- }
922
-
923
- /**
924
- * Thread represents a conversation thread containing multiple UIBlocks
925
- * Each UIBlock in a thread represents a user question and assistant response pair
926
- */
927
- declare class Thread {
928
- private id;
929
- private uiblocks;
930
- private createdAt;
3132
+ protected abstract getDefaultApiKey(): string | undefined;
931
3133
  /**
932
- * Creates a new Thread instance
933
- * @param id - Optional custom ID, generates UUID if not provided
3134
+ * Get the provider name (for logging)
934
3135
  */
935
- constructor(id?: string);
3136
+ protected abstract getProviderName(): string;
936
3137
  /**
937
- * Get the thread ID
3138
+ * Get the API key (from instance, parameter, or environment)
938
3139
  */
939
- getId(): string;
3140
+ protected getApiKey(apiKey?: string): string | undefined;
940
3141
  /**
941
- * Add a UIBlock to the thread
3142
+ * Check if a component contains a Form (data_modification component)
3143
+ * Forms have hardcoded defaultValues that become stale when cached
3144
+ * This checks both single Form components and Forms inside MultiComponentContainer
942
3145
  */
943
- addUIBlock(uiblock: UIBlock): void;
3146
+ protected containsFormComponent(component: any): boolean;
944
3147
  /**
945
- * Get a UIBlock by ID
3148
+ * Match components from text response suggestions and generate follow-up questions
3149
+ * Takes a text response with component suggestions (c1:type format) and matches with available components
3150
+ * Also generates title, description, and intelligent follow-up questions (actions) based on the analysis
3151
+ * All components are placed in a default MultiComponentContainer layout
3152
+ * @param analysisContent - The text response containing component suggestions
3153
+ * @param components - List of available components
3154
+ * @param apiKey - Optional API key
3155
+ * @param componentStreamCallback - Optional callback to stream primary KPI component as soon as it's identified
3156
+ * @returns Object containing matched components, layout title/description, and follow-up actions
946
3157
  */
947
- getUIBlock(id: string): UIBlock | undefined;
3158
+ matchComponentsFromAnalysis(analysisContent: string, components: Component[], userPrompt: string, apiKey?: string, componentStreamCallback?: (component: Component) => void, deferredTools?: any[], executedTools?: any[], collections?: any, userId?: string): Promise<{
3159
+ components: Component[];
3160
+ layoutTitle: string;
3161
+ layoutDescription: string;
3162
+ actions: Action[];
3163
+ }>;
948
3164
  /**
949
- * Get all UIBlocks in the thread
950
- */
951
- getUIBlocks(): UIBlock[];
3165
+ * Classify user question into category and detect external tools needed
3166
+ * Determines if question is for data analysis, requires external tools, or needs text response
3167
+ */
3168
+ classifyQuestionCategory(userPrompt: string, apiKey?: string, conversationHistory?: string, externalTools?: any[]): Promise<{
3169
+ category: 'data_analysis' | 'data_modification' | 'general';
3170
+ externalTools: Array<{
3171
+ type: string;
3172
+ name: string;
3173
+ description: string;
3174
+ parameters: Record<string, any>;
3175
+ }>;
3176
+ dataAnalysisType?: 'visualization' | 'calculation' | 'comparison' | 'trend';
3177
+ reasoning: string;
3178
+ confidence: number;
3179
+ }>;
952
3180
  /**
953
- * Get UIBlocks as a Map
954
- */
955
- getUIBlocksMap(): Map<string, UIBlock>;
3181
+ * Adapt UI block parameters based on current user question
3182
+ * Takes a matched UI block from semantic search and modifies its props to answer the new question
3183
+ * Also adapts the cached text response to match the new question
3184
+ */
3185
+ adaptUIBlockParameters(currentUserPrompt: string, originalUserPrompt: string, matchedUIBlock: any, apiKey?: string, cachedTextResponse?: string): Promise<{
3186
+ success: boolean;
3187
+ adaptedComponent?: Component;
3188
+ adaptedTextResponse?: string;
3189
+ parametersChanged?: Array<{
3190
+ field: string;
3191
+ reason: string;
3192
+ }>;
3193
+ explanation: string;
3194
+ }>;
956
3195
  /**
957
- * Remove a UIBlock by ID
3196
+ * Generate text-based response for user question
3197
+ * This provides conversational text responses instead of component generation
3198
+ * Supports tool calling for query execution with automatic retry on errors (max 3 attempts)
3199
+ * After generating text response, if components are provided, matches suggested components
958
3200
  */
959
- removeUIBlock(id: string): boolean;
3201
+ generateTextResponse(userPrompt: string, apiKey?: string, conversationHistory?: string, streamCallback?: (chunk: string) => void, collections?: any, components?: Component[], externalTools?: any[], category?: 'data_analysis' | 'data_modification' | 'general', userId?: string): Promise<T_RESPONSE>;
960
3202
  /**
961
- * Check if UIBlock exists
3203
+ * Main orchestration function with semantic search and multi-step classification
3204
+ * NEW FLOW (Recommended):
3205
+ * 1. Semantic search: Check previous conversations (>60% match)
3206
+ * - If match found → Adapt UI block parameters and return
3207
+ * 2. Category classification: Determine if data_analysis, requires_external_tools, or text_response
3208
+ * 3. Route appropriately based on category and response mode
962
3209
  */
963
- hasUIBlock(id: string): boolean;
3210
+ handleUserRequest(userPrompt: string, components: Component[], apiKey?: string, conversationHistory?: string, responseMode?: 'component' | 'text', streamCallback?: (chunk: string) => void, collections?: any, externalTools?: any[], userId?: string): Promise<T_RESPONSE>;
964
3211
  /**
965
- * Get number of UIBlocks in the thread
3212
+ * Generate next questions that the user might ask based on the original prompt and generated component
3213
+ * This helps provide intelligent suggestions for follow-up queries
3214
+ * For general/conversational questions without components, pass textResponse instead
966
3215
  */
967
- getUIBlockCount(): number;
3216
+ generateNextQuestions(originalUserPrompt: string, component?: Component | null, componentData?: Record<string, unknown>, apiKey?: string, conversationHistory?: string, textResponse?: string): Promise<string[]>;
3217
+ }
3218
+
3219
+ interface AnthropicLLMConfig extends BaseLLMConfig {
3220
+ }
3221
+ /**
3222
+ * AnthropicLLM class for handling AI-powered component generation and matching using Anthropic Claude
3223
+ */
3224
+ declare class AnthropicLLM extends BaseLLM {
3225
+ constructor(config?: AnthropicLLMConfig);
3226
+ protected getDefaultModel(): string;
3227
+ protected getDefaultFastModel(): string;
3228
+ protected getDefaultApiKey(): string | undefined;
3229
+ protected getProviderName(): string;
3230
+ }
3231
+ declare const anthropicLLM: AnthropicLLM;
3232
+
3233
+ interface GroqLLMConfig extends BaseLLMConfig {
3234
+ }
3235
+ /**
3236
+ * GroqLLM class for handling AI-powered component generation and matching using Groq
3237
+ */
3238
+ declare class GroqLLM extends BaseLLM {
3239
+ constructor(config?: GroqLLMConfig);
3240
+ protected getDefaultModel(): string;
3241
+ protected getDefaultFastModel(): string;
3242
+ protected getDefaultApiKey(): string | undefined;
3243
+ protected getProviderName(): string;
3244
+ }
3245
+ declare const groqLLM: GroqLLM;
3246
+
3247
+ interface GeminiLLMConfig extends BaseLLMConfig {
3248
+ }
3249
+ /**
3250
+ * GeminiLLM class for handling AI-powered component generation and matching using Google Gemini
3251
+ */
3252
+ declare class GeminiLLM extends BaseLLM {
3253
+ constructor(config?: GeminiLLMConfig);
3254
+ protected getDefaultModel(): string;
3255
+ protected getDefaultFastModel(): string;
3256
+ protected getDefaultApiKey(): string | undefined;
3257
+ protected getProviderName(): string;
3258
+ }
3259
+ declare const geminiLLM: GeminiLLM;
3260
+
3261
+ interface OpenAILLMConfig extends BaseLLMConfig {
3262
+ }
3263
+ /**
3264
+ * OpenAILLM class for handling AI-powered component generation and matching using OpenAI GPT models
3265
+ */
3266
+ declare class OpenAILLM extends BaseLLM {
3267
+ constructor(config?: OpenAILLMConfig);
3268
+ protected getDefaultModel(): string;
3269
+ protected getDefaultFastModel(): string;
3270
+ protected getDefaultApiKey(): string | undefined;
3271
+ protected getProviderName(): string;
3272
+ }
3273
+ declare const openaiLLM: OpenAILLM;
3274
+
3275
+ /**
3276
+ * Query Cache — Two mechanisms:
3277
+ *
3278
+ * 1. `cache` (query string → result data) — TTL-based with max size, for avoiding re-execution
3279
+ * of recently validated queries. True LRU eviction: reads bubble entries to the back via
3280
+ * delete+re-set so the oldest *unused* entry is evicted, not the oldest *inserted*.
3281
+ *
3282
+ * 2. Encrypted queryId tokens — SQL is encrypted into the queryId itself (self-contained).
3283
+ * No server-side storage needed for SQL mappings. The token is decrypted on each request.
3284
+ * This eliminates the unbounded queryIdCache that previously grew forever and caused
3285
+ * memory bloat (hundreds of MBs after thousands of queries).
3286
+ *
3287
+ * Result data can still be cached temporarily via the data cache (mechanism 1).
3288
+ */
3289
+ declare class QueryCache {
3290
+ private cache;
3291
+ private ttlMs;
3292
+ private maxCacheSize;
3293
+ private cleanupInterval;
3294
+ private readonly algorithm;
3295
+ private encryptionKey;
3296
+ constructor();
968
3297
  /**
969
- * Clear all UIBlocks from the thread
3298
+ * Set the cache TTL (Time To Live)
3299
+ * @param minutes - TTL in minutes (default: 10)
970
3300
  */
971
- clear(): void;
3301
+ setTTL(minutes: number): void;
972
3302
  /**
973
- * Get creation timestamp
3303
+ * Get the current TTL in minutes
974
3304
  */
975
- getCreatedAt(): Date;
3305
+ getTTL(): number;
976
3306
  /**
977
- * Get conversation context from recent UIBlocks (excluding current one)
978
- * Returns formatted string with previous questions and component summaries
979
- * @param limit - Maximum number of previous UIBlocks to include (default: 5)
980
- * @param currentUIBlockId - ID of current UIBlock to exclude from context (optional)
981
- * @returns Formatted conversation history string
3307
+ * Store query result in data cache.
3308
+ * If the key already exists, it's removed first so the re-insert places it
3309
+ * at the back of the iteration order (LRU). Eviction only fires when adding
3310
+ * a genuinely new key past the size limit.
982
3311
  */
983
- getConversationContext(limit?: number, currentUIBlockId?: string): string;
3312
+ set(query: string, data: any): void;
984
3313
  /**
985
- * Convert Thread to JSON-serializable object
3314
+ * Get cached result if exists and not expired.
3315
+ * On hit, re-inserts the entry so it moves to the back of the Map's
3316
+ * iteration order — turning FIFO eviction into true LRU.
986
3317
  */
987
- toJSON(): Record<string, any>;
988
- }
989
-
990
- /**
991
- * ThreadManager manages all threads globally
992
- * Provides methods to create, retrieve, and delete threads
993
- */
994
- declare class ThreadManager {
995
- private static instance;
996
- private threads;
997
- private constructor();
3318
+ get(query: string): any | null;
998
3319
  /**
999
- * Get singleton instance of ThreadManager
3320
+ * Check if query exists in cache (not expired)
1000
3321
  */
1001
- static getInstance(): ThreadManager;
3322
+ has(query: string): boolean;
1002
3323
  /**
1003
- * Create a new thread
1004
- * @param id - Optional custom ID, generates UUID if not provided
1005
- * @returns The created Thread instance
3324
+ * Remove a specific query from cache
1006
3325
  */
1007
- createThread(id?: string): Thread;
3326
+ delete(query: string): void;
1008
3327
  /**
1009
- * Get a thread by ID
3328
+ * Clear all cached entries
1010
3329
  */
1011
- getThread(id: string): Thread | undefined;
3330
+ clear(): void;
1012
3331
  /**
1013
- * Get all threads
3332
+ * Get cache statistics
1014
3333
  */
1015
- getAllThreads(): Thread[];
3334
+ getStats(): {
3335
+ size: number;
3336
+ queryIdCount: number;
3337
+ oldestEntryAge: number | null;
3338
+ };
1016
3339
  /**
1017
- * Get threads as a Map
3340
+ * Start periodic cleanup of expired data cache entries.
1018
3341
  */
1019
- getThreadsMap(): Map<string, Thread>;
3342
+ private startCleanup;
1020
3343
  /**
1021
- * Delete a thread by ID
3344
+ * Encrypt a payload into a self-contained token.
1022
3345
  */
1023
- deleteThread(id: string): boolean;
3346
+ private encrypt;
1024
3347
  /**
1025
- * Check if thread exists
3348
+ * Decrypt a token back to the original payload.
1026
3349
  */
1027
- hasThread(id: string): boolean;
3350
+ private decrypt;
1028
3351
  /**
1029
- * Get number of threads
3352
+ * Store a query by generating an encrypted token as queryId.
3353
+ * The SQL is encrypted INTO the token — nothing stored in memory.
3354
+ * If data is provided, it's cached temporarily in the data cache.
1030
3355
  */
1031
- getThreadCount(): number;
3356
+ storeQuery(query: any, data?: any): string;
1032
3357
  /**
1033
- * Clear all threads
3358
+ * Get a stored query by decrypting its token.
3359
+ * Returns the SQL + any cached result data.
1034
3360
  */
1035
- clearAll(): void;
3361
+ getQuery(queryId: string): {
3362
+ query: any;
3363
+ data: any;
3364
+ } | null;
1036
3365
  /**
1037
- * Find a UIBlock by ID across all threads
1038
- * @param uiBlockId - The UIBlock ID to search for
1039
- * @returns Object with thread and uiBlock if found, undefined otherwise
3366
+ * Update cached data for a queryId token
1040
3367
  */
1041
- findUIBlockById(uiBlockId: string): {
1042
- thread: Thread;
1043
- uiBlock: UIBlock;
1044
- } | undefined;
3368
+ setQueryData(queryId: string, data: any): void;
1045
3369
  /**
1046
- * Convert all threads to JSON-serializable object
3370
+ * Stop cleanup interval (for graceful shutdown)
1047
3371
  */
1048
- toJSON(): Record<string, any>;
3372
+ destroy(): void;
1049
3373
  }
3374
+ declare const queryCache: QueryCache;
1050
3375
 
1051
3376
  /**
1052
- * CleanupService handles cleanup of old threads and UIBlocks
1053
- * to prevent memory bloat and maintain optimal performance
3377
+ * Manages conversation history scoped per user + dashboard.
3378
+ * Each user-dashboard pair has its own isolated history that expires after a configurable TTL.
1054
3379
  */
1055
- declare class CleanupService {
1056
- private static instance;
3380
+ declare class DashboardConversationHistory {
3381
+ private histories;
3382
+ private ttlMs;
3383
+ private maxEntries;
1057
3384
  private cleanupInterval;
1058
- private constructor();
1059
- /**
1060
- * Get singleton instance of CleanupService
1061
- */
1062
- static getInstance(): CleanupService;
3385
+ constructor();
1063
3386
  /**
1064
- * Clean up old threads based on retention period
1065
- * @param retentionDays - Number of days to keep threads (defaults to config)
1066
- * @returns Number of threads deleted
3387
+ * Set the TTL for dashboard histories
3388
+ * @param minutes - TTL in minutes
1067
3389
  */
1068
- cleanupOldThreads(retentionDays?: number): number;
3390
+ setTTL(minutes: number): void;
1069
3391
  /**
1070
- * Clean up old UIBlocks within threads based on retention period
1071
- * @param retentionDays - Number of days to keep UIBlocks (defaults to config)
1072
- * @returns Object with number of UIBlocks deleted per thread
3392
+ * Set max entries per dashboard
1073
3393
  */
1074
- cleanupOldUIBlocks(retentionDays?: number): {
1075
- [threadId: string]: number;
1076
- };
3394
+ setMaxEntries(max: number): void;
1077
3395
  /**
1078
- * Clear all component data from UIBlocks to free memory
1079
- * Keeps metadata but removes the actual data
1080
- * @param retentionDays - Number of days to keep full data (defaults to config)
1081
- * @returns Number of UIBlocks whose data was cleared
3396
+ * Add a conversation entry for a user's dashboard
1082
3397
  */
1083
- clearOldUIBlockData(retentionDays?: number): number;
3398
+ addEntry(dashboardId: string, userPrompt: string, componentSummary: string, userId?: string): void;
1084
3399
  /**
1085
- * Run full cleanup (threads, UIBlocks, and data)
1086
- * @returns Cleanup statistics
3400
+ * Get formatted conversation history for a user's dashboard
1087
3401
  */
1088
- runFullCleanup(): {
1089
- threadsDeleted: number;
1090
- uiblocksDeleted: {
1091
- [threadId: string]: number;
1092
- };
1093
- dataCleared: number;
1094
- };
3402
+ getHistory(dashboardId: string, userId?: string): string;
1095
3403
  /**
1096
- * Start automatic cleanup at regular intervals
1097
- * @param intervalHours - Hours between cleanup runs (default: 24)
3404
+ * Clear history for a specific user's dashboard
1098
3405
  */
1099
- startAutoCleanup(intervalHours?: number): void;
3406
+ clearDashboard(dashboardId: string, userId?: string): void;
1100
3407
  /**
1101
- * Stop automatic cleanup
3408
+ * Clear all dashboard histories
1102
3409
  */
1103
- stopAutoCleanup(): void;
3410
+ clearAll(): void;
1104
3411
  /**
1105
- * Check if auto cleanup is running
3412
+ * Start periodic cleanup of expired histories
1106
3413
  */
1107
- isAutoCleanupRunning(): boolean;
3414
+ private startCleanup;
1108
3415
  /**
1109
- * Get current memory usage statistics
3416
+ * Stop cleanup interval (for graceful shutdown)
1110
3417
  */
1111
- getMemoryStats(): {
1112
- threadCount: number;
1113
- totalUIBlocks: number;
1114
- avgUIBlocksPerThread: number;
1115
- };
3418
+ destroy(): void;
1116
3419
  }
3420
+ declare const dashboardConversationHistory: DashboardConversationHistory;
1117
3421
 
1118
3422
  /**
1119
- * Configuration for data storage limits in UIBlocks
3423
+ * ScriptMatcher LLM-Based Script Matching + Parameter Extraction
3424
+ *
3425
+ * Uses ONE LLM call to:
3426
+ * 1. Pick the best matching script from the library (or "none")
3427
+ * 2. Extract parameter values from the user question
3428
+ *
3429
+ * Why LLM over embeddings:
3430
+ * - Embeddings capture topic similarity ("overstock" ≈ "inventory" ≈ "revenue")
3431
+ * but can't distinguish structurally different questions about the same domain
3432
+ * - LLM understands that "overstock by warehouse" needs a different script than
3433
+ * "revenue by warehouse" even though they're semantically close
3434
+ * - One call does both matching AND parameter extraction
3435
+ *
3436
+ * When script library grows past ~50, add an embedding pre-filter
3437
+ * (ChromaDB narrows to top 10 → LLM picks from those 10).
1120
3438
  */
1121
- declare const STORAGE_CONFIG: {
1122
- /**
1123
- * Maximum number of rows to store in UIBlock data
1124
- */
1125
- MAX_ROWS_PER_BLOCK: number;
1126
- /**
1127
- * Maximum size in bytes per UIBlock (1MB)
1128
- */
1129
- MAX_SIZE_PER_BLOCK_BYTES: number;
3439
+
3440
+ declare class ScriptMatcher {
3441
+ private store;
3442
+ constructor(store: ScriptStore);
1130
3443
  /**
1131
- * Number of days to keep threads before cleanup
3444
+ * Find the best matching script for a user question.
3445
+ * Uses ONE LLM call that picks the script AND extracts parameters.
3446
+ * Returns null if no script matches.
1132
3447
  */
1133
- THREAD_RETENTION_DAYS: number;
3448
+ match(userPrompt: string, apiKey?: string, model?: string): Promise<ScriptMatch | null>;
1134
3449
  /**
1135
- * Number of days to keep UIBlocks before cleanup
3450
+ * Build the script catalog string for the LLM prompt.
3451
+ * Each script gets: index, ID, name, description, and parameter definitions.
1136
3452
  */
1137
- UIBLOCK_RETENTION_DAYS: number;
1138
- };
3453
+ private buildScriptCatalog;
3454
+ }
1139
3455
 
1140
3456
  /**
1141
- * Configuration for conversation context and history management
3457
+ * ScriptRunner Execute scripts in an isolated tsx subprocess.
3458
+ *
3459
+ * The subprocess approach replaces the earlier `new Function()` eval and gives us:
3460
+ * - Real sandbox (separate process, SIGKILL on timeout).
3461
+ * - Real TypeScript (tsx transpiles on the fly).
3462
+ * - npm imports available to scripts (clustering, stats, geo, etc.).
3463
+ *
3464
+ * Protocol: NDJSON over the child's stdin/stdout. See script-ipc.ts + backend/docs/SCRIPT-FLOW-IMPLEMENTATION.md.
1142
3465
  */
1143
- declare const CONTEXT_CONFIG: {
1144
- /**
1145
- * Maximum number of previous UIBlocks to include as conversation context
1146
- * Set to 0 to disable conversation history
1147
- * Higher values provide more context but may increase token usage
1148
- */
1149
- MAX_CONVERSATION_CONTEXT_BLOCKS: number;
1150
- };
1151
3466
 
1152
- declare const SDK_VERSION = "0.0.8";
3467
+ interface RunScriptOptions {
3468
+ /** Data sources the script is allowed to query via ctx.query */
3469
+ externalTools: ExternalTool[];
3470
+ /** Optional — for propagating per-query UI progress to the user */
3471
+ streamBuffer?: StreamBuffer;
3472
+ /** Override the wall-clock timeout (default `SCRIPT_TIMEOUT_MS`, 60s). */
3473
+ timeoutMs?: number;
3474
+ }
3475
+ /**
3476
+ * Execute a recipe by spawning a tsx child on the script's .ts file.
3477
+ * `scriptPath` is the absolute path to the saved `.ts` body.
3478
+ */
3479
+ declare function runScript(recipe: ScriptRecipe, scriptPath: string, params: Record<string, any>, options: RunScriptOptions): Promise<ScriptResult>;
3480
+
1153
3481
  type MessageTypeHandler = (message: IncomingMessage) => void | Promise<void>;
1154
3482
  declare class SuperatomSDK {
1155
3483
  private ws;
1156
3484
  private url;
1157
- private apiKey;
3485
+ private apiKey?;
1158
3486
  private projectId;
1159
- private userId;
1160
3487
  private type;
1161
3488
  private bundleDir;
1162
3489
  private messageHandlers;
@@ -1166,15 +3493,30 @@ declare class SuperatomSDK {
1166
3493
  private maxReconnectAttempts;
1167
3494
  private collections;
1168
3495
  private components;
3496
+ private tools;
3497
+ private workflows;
1169
3498
  private anthropicApiKey;
1170
3499
  private groqApiKey;
3500
+ private geminiApiKey;
3501
+ private openaiApiKey;
1171
3502
  private llmProviders;
3503
+ private databaseType;
3504
+ private modelStrategy;
3505
+ private mainAgentModel;
3506
+ private sourceAgentModel;
3507
+ private dashCompModels?;
3508
+ private conversationSimilarityThreshold;
1172
3509
  private userManager;
1173
3510
  private dashboardManager;
1174
3511
  private reportManager;
3512
+ private pingInterval;
3513
+ private lastPong;
3514
+ private readonly PING_INTERVAL_MS;
3515
+ private readonly PONG_TIMEOUT_MS;
1175
3516
  constructor(config: SuperatomSDKConfig);
1176
3517
  /**
1177
3518
  * Initialize PromptLoader and load prompts into memory
3519
+ * Tries to load from file system first, then falls back to hardcoded prompts
1178
3520
  */
1179
3521
  private initializePromptLoader;
1180
3522
  /**
@@ -1210,9 +3552,11 @@ declare class SuperatomSDK {
1210
3552
  */
1211
3553
  private handleMessage;
1212
3554
  /**
1213
- * Send a message to the Superatom service
3555
+ * Send a message to the Superatom service.
3556
+ * Returns true if the message was sent, false if the WebSocket is not connected.
3557
+ * Does NOT throw on closed connections — callers can check the return value if needed.
1214
3558
  */
1215
- send(message: Message): void;
3559
+ send(message: Message): boolean;
1216
3560
  /**
1217
3561
  * Register a message handler to receive all messages
1218
3562
  */
@@ -1238,7 +3582,69 @@ declare class SuperatomSDK {
1238
3582
  */
1239
3583
  addCollection<TParams = any, TResult = any>(collectionName: string, operation: CollectionOperation | string, handler: CollectionHandler<TParams, TResult>): void;
1240
3584
  private handleReconnect;
3585
+ /**
3586
+ * Start heartbeat to keep WebSocket connection alive
3587
+ * Sends PING every 3 minutes to prevent idle timeout from cloud infrastructure
3588
+ */
3589
+ private startHeartbeat;
3590
+ /**
3591
+ * Stop the heartbeat interval
3592
+ */
3593
+ private stopHeartbeat;
3594
+ /**
3595
+ * Handle PONG response from server
3596
+ */
3597
+ private handlePong;
1241
3598
  private storeComponents;
3599
+ /**
3600
+ * Set tools for the SDK instance
3601
+ */
3602
+ setTools(tools: Tool$1[]): void;
3603
+ /**
3604
+ * Get the stored tools
3605
+ */
3606
+ getTools(): Tool$1[];
3607
+ /**
3608
+ * Register workflow components for the SDK instance.
3609
+ *
3610
+ * Workflows are pre-built multi-step UI flows the main agent can pick when
3611
+ * the user's prompt matches a workflow's `whenToUse` trigger. Picking a
3612
+ * workflow short-circuits analysis text + dashboard component generation —
3613
+ * the workflow component is returned directly, with the LLM-extracted props.
3614
+ */
3615
+ setWorkflows(workflows: WorkflowDescriptor[]): void;
3616
+ /**
3617
+ * Get the registered workflow components.
3618
+ */
3619
+ getWorkflows(): WorkflowDescriptor[];
3620
+ /**
3621
+ * Apply model strategy to all LLM provider singletons
3622
+ * @param strategy - 'best', 'fast', or 'balanced'
3623
+ */
3624
+ private applyModelStrategy;
3625
+ /**
3626
+ * Set model strategy at runtime
3627
+ * @param strategy - 'best', 'fast', or 'balanced'
3628
+ */
3629
+ setModelStrategy(strategy: ModelStrategy): void;
3630
+ /**
3631
+ * Get current model strategy
3632
+ */
3633
+ getModelStrategy(): ModelStrategy;
3634
+ /**
3635
+ * Apply conversation similarity threshold to all LLM provider singletons
3636
+ * @param threshold - Value between 0 and 1 (e.g., 0.8 = 80% similarity required)
3637
+ */
3638
+ private applyConversationSimilarityThreshold;
3639
+ /**
3640
+ * Set conversation similarity threshold at runtime
3641
+ * @param threshold - Value between 0 and 1 (e.g., 0.8 = 80% similarity required)
3642
+ */
3643
+ setConversationSimilarityThreshold(threshold: number): void;
3644
+ /**
3645
+ * Get current conversation similarity threshold
3646
+ */
3647
+ getConversationSimilarityThreshold(): number;
1242
3648
  }
1243
3649
 
1244
- export { type Action, CONTEXT_CONFIG, type CapturedLog, CleanupService, type CollectionHandler, type CollectionOperation, type IncomingMessage, LLM, type Message, SDK_VERSION, STORAGE_CONFIG, SuperatomSDK, type SuperatomSDKConfig, Thread, ThreadManager, UIBlock, UILogCollector, type User, UserManager, type UsersData };
3650
+ export { type Action, type AgentConfig, type AgentResponse, BM25L, type BM25LOptions, type BaseLLMConfig, CONTEXT_CONFIG, type CapturedLog, CleanupService, type CollectionHandler, type CollectionOperation, type DBUIBlock, DEFAULT_AGENT_CONFIG, type DatabaseType, type HybridSearchOptions, type IncomingMessage, type KbNodesQueryFilters, type KbNodesRequestPayload, LLM, type LLMUsageEntry, type LogLevel, MainAgent, type Message, type ModelStrategy, type OutputField, type RerankedResult, STORAGE_CONFIG, type ScriptComponentSpec, ScriptMatcher, type ScriptParameter, type ScriptRecipe, type ScriptRecipeMetaRow, type ScriptRecipeStore, type ScriptResult, ScriptStore, type ScriptStoreOptions, type SelectedWorkflow, SuperatomSDK, type SuperatomSDKConfig, type TaskType, Thread, ThreadManager, type Tool$1 as Tool, type ToolOutputSchema, UIBlock, UILogCollector, type User, UserManager, type UsersData, type WorkflowDescriptor, anthropicLLM, dashboardConversationHistory, geminiLLM, groqLLM, hybridRerank, llmUsageLogger, logger, normalizeScriptBody, openaiLLM, queryCache, rerankChromaResults, rerankConversationResults, resolveScriptRecipeStore, runScript, userPromptErrorLogger };