refactor(queries): Split schema.ts for maintainability

bin/build-schema-json.mjs

@@ -9,7 +9,7 @@ import tsj from 'ts-json-schema-generator'
 /** @type {import('ts-json-schema-generator/dist/src/Config').Config} */
 const config = {
     ...tsj.DEFAULT_CONFIG,
-    path: 'frontend/src/queries/schema.ts',
+    path: 'frontend/src/queries/schema/index.ts',
     tsconfig: 'tsconfig.json',
     discriminatorType: 'open-api',
     skipTypeCheck: true,

frontend/src/queries/schema/index.ts

@@ -0,0 +1,5 @@
+/* eslint-disable simple-import-sort/exports */
+export * from './schema-general'
+export * from './schema-assistant-queries'
+export * from './schema-assistant-messages'
+/* eslint-enable simple-import-sort/exports */

frontend/src/queries/schema/schema-assistant-messages.ts

@@ -0,0 +1,70 @@
+import { AssistantFunnelsQuery, AssistantTrendsQuery } from './schema-assistant-queries'
+
+export enum AssistantMessageType {
+    Human = 'human',
+    Assistant = 'ai',
+    Reasoning = 'ai/reasoning',
+    Visualization = 'ai/viz',
+    Failure = 'ai/failure',
+    Router = 'ai/router',
+}
+
+export interface BaseAssistantMessage {
+    id?: string
+}
+
+export interface HumanMessage extends BaseAssistantMessage {
+    type: AssistantMessageType.Human
+    content: string
+}
+
+export interface AssistantMessage extends BaseAssistantMessage {
+    type: AssistantMessageType.Assistant
+    content: string
+}
+
+export interface ReasoningMessage extends BaseAssistantMessage {
+    type: AssistantMessageType.Reasoning
+    content: string
+    substeps?: string[]
+}
+
+export interface VisualizationMessage extends BaseAssistantMessage {
+    type: AssistantMessageType.Visualization
+    plan?: string
+    answer?: AssistantTrendsQuery | AssistantFunnelsQuery
+    initiator?: string
+}
+
+export interface FailureMessage extends BaseAssistantMessage {
+    type: AssistantMessageType.Failure
+    content?: string
+}
+
+export interface RouterMessage extends BaseAssistantMessage {
+    type: AssistantMessageType.Router
+    content: string
+}
+
+export type RootAssistantMessage =
+    | VisualizationMessage
+    | ReasoningMessage
+    | AssistantMessage
+    | HumanMessage
+    | FailureMessage
+    | RouterMessage
+
+export enum AssistantEventType {
+    Status = 'status',
+    Message = 'message',
+    Conversation = 'conversation',
+}
+
+export enum AssistantGenerationStatusType {
+    Acknowledged = 'ack',
+    GenerationError = 'generation_error',
+}
+
+export interface AssistantGenerationStatusEvent {
+    type: AssistantGenerationStatusType
+}