@dremio/js-sdk 0.42.0 → 0.43.0

package/dist/enterprise/ai/AIResource.js

@@ -13,18 +13,19 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { catchError, concatMap, endWith, map, Observable, of, scan, share, startWith } from "rxjs";
 import { ChatSession } from "./chat/ChatSession.js";
-import { AgentChatResponse, } from "./chat/eventWrappers/AgentChatResponse.js";
 import { ModelProvider } from "./modelProvider/ModelProvider.js";
 import * as z from "zod/mini";
 import { modelProviderCreateCodec, modelProviderRetrieveCodec, } from "./modelProvider/modelProviderCodec.js";
 import { safeParseResultToResult } from "../../common/safeParseResultToResult.js";
 import { listModelsPropertiesSchema } from "./modelProvider/modelProviderConfigSchemas.js";
-import { AgentChatExchange } from "./chat/eventWrappers/AgentChatExchange.js";
-import { AgentEndTurn } from "../interfaces.js";
-import { ChatSessionClient } from "./chat/ChatSessionClient.js";
 import { createSendChatMessage } from "./chat/methods/sendChatMessage.js";
+import { listConversations } from "./conversations/methods/listConversations.js";
+import { createConversation, createConversationCodec, } from "./conversations/methods/createConversation.js";
+import { retrieveConversation } from "./conversations/methods/retrieveConversation.js";
+import { AgentConversation } from "../interfaces.js";
+import { retrieveConversationHistory } from "./conversations/methods/retrieveConversationHistory.js";
+import { createConversationMachine } from "./conversations/createConversationMachine.js";
 export function listAvailableModels(config) {
     return function listAvailableModels(properties, { signal } = {}) {
         return config
@@ -46,6 +47,12 @@ export class AIResource {
     constructor(config) {
         this.#config = config;
     }
+    get conversationMachine() {
+        return createConversationMachine(this.#config);
+    }
+    createConversation(body) {
+        return createConversation(this.#config)(body);
+    }
     createModelProvider(properties) {
         const body = z.encode(modelProviderCreateCodec, properties);
         return this.#config
@@ -61,6 +68,15 @@ export class AIResource {
             .map((res) => res.json())
             .map((entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)));
     }
+    listConversations() {
+        const getPage = listConversations(this.#config);
+        return {
+            async *data({ signal } = {}) {
+                yield* (await getPage({ maxResults: 100 }, { signal }).map((response) => response.data).promise).unwrap();
+            },
+            getPage,
+        };
+    }
     listModelProviders() {
         const getPage = ({ signal } = {}) => {
             return this.#config
@@ -85,6 +101,24 @@ export class AIResource {
     listAvailableModels(properties) {
         return listAvailableModels(this.#config)(properties);
     }
+    retrieveConversation(id, { signal } = {}) {
+        return retrieveConversation(this.#config)(id, { signal }).map((properties) => new AgentConversation(this.#config, properties));
+    }
+    retrieveConversationHistory(id) {
+        // eslint-disable-next-line @typescript-eslint/no-empty-object-type
+        const getPage = (params, { signal } = {}) => retrieveConversationHistory(this.#config)(id, { signal });
+        return {
+            async *data() {
+                const history = await getPage({}).then((result) => {
+                    if (result.isErr()) {
+                        throw result.error;
+                    }
+                    return result.value;
+                });
+                yield* history.data;
+            },
+        };
+    }
     retrieveModelProvider(id, { signal } = {}) {
         return this.#config
             .v4Request(`model-provider/config/${id}`, {
@@ -96,23 +130,11 @@ export class AIResource {
             .map((res) => res.json())
             .map((entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)));
     }
-    /**
-     * Creates an Observable that emits each time the `AgentChatExchange` is updated
-     * with events from the Agent.
-     */
-    createAgentChatExchange$(chatSession, userChatMessage, options = {}) {
-        const initialAgentChatExchange = new AgentChatExchange(userChatMessage, []);
-        return this.sendChatMessage(chatSession, userChatMessage).pipe(map((eventResult) => eventResult.unwrap()), concatMap((agentChatEvent) => options.mapChatEvents?.(agentChatEvent) || of(agentChatEvent)), endWith(new AgentEndTurn({ type: "reply_finished" })), catchError((error) => of(new AgentEndTurn({ error, type: "error" }))), scan((agentChatExchange, agentChatEvent) => agentChatExchange._addChatEvent(agentChatEvent), initialAgentChatExchange), startWith(initialAgentChatExchange), share());
-    }
-    createChatSessionClient(chatSession) {
-        return new ChatSessionClient({
-            chatSession,
-            createAgentChatExchange$: this.createAgentChatExchange$.bind(this),
-        });
-    }
     /**
      * Sends a chat message to the Dremio AI Agent and returns an Observable containing
      * Agent replies.
+     *
+     * @deprecated
      */
     sendChatMessage(chatSession, userChatMessage) {
         return createSendChatMessage(this.#config)(chatSession, userChatMessage);

package/dist/enterprise/ai/AIResource.js.map

@@ -1 +1 @@
-{"version":3,"file":"AIResource.js","sourceRoot":"","sources":["../../../src/enterprise/ai/AIResource.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,MAAM,CAAC;AACnG,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAEpD,OAAO,EACL,iBAAiB,GAElB,MAAM,2CAA2C,CAAC;AAEnD,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EACL,wBAAwB,EACxB,0BAA0B,GAC3B,MAAM,uCAAuC,CAAC;AAE/C,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAClF,OAAO,EAAE,0BAA0B,EAAE,MAAM,+CAA+C,CAAC;AAC3F,OAAO,EAAE,iBAAiB,EAAE,MAAM,2CAA2C,CAAC;AAC9E,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAChE,OAAO,EAAE,qBAAqB,EAAE,MAAM,mCAAmC,CAAC;AAE1E,MAAM,UAAU,mBAAmB,CAAC,MAAgB;IAClD,OAAO,SAAS,mBAAmB,CACjC,UAAsD,EACtD,EAAE,MAAM,KAAkB,EAAE;QAE5B,OAAO,MAAM;aACV,SAAS,CAAC,2BAA2B,EAAE;YACtC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,0BAA0B,EAAE,UAAU,CAAC,CAAC;YACrE,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;aACnC;YACD,MAAM,EAAE,MAAM;YACd,MAAM;SACP,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAmC,CAAC;aACzD,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,UAAU;IACrB,OAAO,CAAoC;IAE3C,YAAY,MAAyC;QACnD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED,mBAAmB,CAAC,UAAqD;QACvE,MAAM,IAAI,GAAG,CAAC,CAAC,MAAM,CAAC,wBAAwB,EAAE,UAAU,CAAC,CAAC;QAC5D,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,uBAAuB,EAAE;YAClC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAC1B,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;aACnC;YACD,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAyD,CAAC;aAC/E,GAAG,CACF,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC,CAC1F,CAAC;IACN,CAAC;IAED,kBAAkB;QAChB,MAAM,OAAO,GAAG,CAAC,EAAE,MAAM,KAAkB,EAAE,EAAE,EAAE;YAC/C,OAAO,IAAI,CAAC,OAAO;iBAChB,SAAS,CAAC,uBAAuB,EAAE;gBAClC,OAAO,EAAE;oBACP,MAAM,EAAE,kBAAkB;iBAC3B;gBACD,MAAM;aACP,CAAC;iBACD,GAAG,CACF,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAwE,CAC1F;iBACA,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBAClB,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACpC,uBAAuB,CAAC,CAAC,CAAC,UAAU,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,CAC3E,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAC5D,CACF;aACF,CAAC,CAAC,CAAC;QACR,CAAC,CAAC;QACF,OAAO;YACL,KAAK,CAAC,CAAC,IAAI,CAAC,EAAE,MAAM,KAAkB,EAAE;gBACtC,KAAK,CAAC,CAAC,CAAC,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC;YACvF,CAAC;YACD,OAAO;SACR,CAAC;IACJ,CAAC;IAED,mBAAmB,CAAC,UAAsD;QACxE,OAAO,mBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,UAAU,CAAC,CAAC;IACvD,CAAC;IAED,qBAAqB,CAAC,EAAU,EAAE,EAAE,MAAM,KAAkB,EAAE;QAC5D,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,yBAAyB,EAAE,EAAE,EAAE;YACxC,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;aAC3B;YACD,MAAM;SACP,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAyD,CAAC;aAC/E,GAAG,CACF,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC,CAC1F,CAAC;IACN,CAAC;IAED;;;OAGG;IACH,wBAAwB,CACtB,WAAwB,EACxB,eAAgC,EAChC,UAII,EAAE;QAEN,MAAM,wBAAwB,GAAG,IAAI,iBAAiB,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;QAE5E,OAAO,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC,IAAI,CAC5D,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,EAC1C,SAAS,CAAC,CAAC,cAAc,EAAE,EAAE,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,cAAc,CAAC,CAAC,EAC5F,OAAO,CAAC,IAAI,YAAY,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC,EACrD,UAAU,CAAC,CAAC,KAAc,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,YAAY,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAC9E,IAAI,CACF,CAAC,iBAAiB,EAAE,cAAc,EAAE,EAAE,CAAC,iBAAiB,CAAC,aAAa,CAAC,cAAc,CAAC,EACtF,wBAAwB,CACzB,EACD,SAAS,CAAC,wBAAwB,CAAC,EACnC,KAAK,EAAE,CACR,CAAC;IACJ,CAAC;IAED,uBAAuB,CAAC,WAAwB;QAC9C,OAAO,IAAI,iBAAiB,CAAC;YAC3B,WAAW;YACX,wBAAwB,EAAE,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,IAAI,CAAC;SACnE,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACH,eAAe,CAAC,WAAwB,EAAE,eAAgC;QACxE,OAAO,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC;IAC3E,CAAC;CACF","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { catchError, concatMap, endWith, map, Observable, of, scan, share, startWith } from \"rxjs\";\nimport { ChatSession } from \"./chat/ChatSession.ts\";\nimport type { UserChatMessage } from \"./chat/UserChatMessage.ts\";\nimport {\n  AgentChatResponse,\n  type AgentResponseContent,\n} from \"./chat/eventWrappers/AgentChatResponse.ts\";\nimport type { Logger, SonarV4Config, V4Config } from \"../../common/Config.ts\";\nimport { ModelProvider } from \"./modelProvider/ModelProvider.ts\";\nimport * as z from \"zod/mini\";\nimport {\n  modelProviderCreateCodec,\n  modelProviderRetrieveCodec,\n} from \"./modelProvider/modelProviderCodec.ts\";\nimport type { SignalParam } from \"../../common/Params.ts\";\nimport { safeParseResultToResult } from \"../../common/safeParseResultToResult.ts\";\nimport { listModelsPropertiesSchema } from \"./modelProvider/modelProviderConfigSchemas.ts\";\nimport { AgentChatExchange } from \"./chat/eventWrappers/AgentChatExchange.ts\";\nimport { AgentEndTurn } from \"../interfaces.ts\";\nimport { ChatSessionClient } from \"./chat/ChatSessionClient.ts\";\nimport { createSendChatMessage } from \"./chat/methods/sendChatMessage.ts\";\n\nexport function listAvailableModels(config: V4Config) {\n  return function listAvailableModels(\n    properties: z.infer<typeof listModelsPropertiesSchema>,\n    { signal }: SignalParam = {},\n  ) {\n    return config\n      .v4Request(\"model-provider:listModels\", {\n        body: JSON.stringify(z.parse(listModelsPropertiesSchema, properties)),\n        headers: {\n          Accept: \"application/json\",\n          \"Content-Type\": \"application/json\",\n        },\n        method: \"POST\",\n        signal,\n      })\n      .map((res) => res.json() as Promise<{ models: string[] }>)\n      .map((result) => new Set(result.models));\n  };\n}\n\nexport class AIResource {\n  #config: Logger & V4Config & SonarV4Config;\n\n  constructor(config: Logger & V4Config & SonarV4Config) {\n    this.#config = config;\n  }\n\n  createModelProvider(properties: z.output<typeof modelProviderCreateCodec>) {\n    const body = z.encode(modelProviderCreateCodec, properties);\n    return this.#config\n      .v4Request(`model-provider/config`, {\n        body: JSON.stringify(body),\n        headers: {\n          Accept: \"application/json\",\n          \"Content-Type\": \"application/json\",\n        },\n        keepalive: true,\n        method: \"POST\",\n      })\n      .map((res) => res.json() as Promise<z.input<typeof modelProviderRetrieveCodec>>)\n      .map(\n        (entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)),\n      );\n  }\n\n  listModelProviders() {\n    const getPage = ({ signal }: SignalParam = {}) => {\n      return this.#config\n        .v4Request(\"model-provider/config\", {\n          headers: {\n            Accept: \"application/json\",\n          },\n          signal,\n        })\n        .map(\n          (res) => res.json() as Promise<{ configs: z.input<typeof modelProviderRetrieveCodec>[] }>,\n        )\n        .map((response) => ({\n          data: response.configs.map((entity) =>\n            safeParseResultToResult(z.safeDecode(modelProviderRetrieveCodec, entity)).map(\n              (properties) => new ModelProvider(this.#config, properties),\n            ),\n          ),\n        }));\n    };\n    return {\n      async *data({ signal }: SignalParam = {}) {\n        yield* (await getPage({ signal }).promise).map((response) => response.data).unwrap();\n      },\n      getPage,\n    };\n  }\n\n  listAvailableModels(properties: z.infer<typeof listModelsPropertiesSchema>) {\n    return listAvailableModels(this.#config)(properties);\n  }\n\n  retrieveModelProvider(id: string, { signal }: SignalParam = {}) {\n    return this.#config\n      .v4Request(`model-provider/config/${id}`, {\n        headers: {\n          Accept: \"application/json\",\n        },\n        signal,\n      })\n      .map((res) => res.json() as Promise<z.input<typeof modelProviderRetrieveCodec>>)\n      .map(\n        (entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)),\n      );\n  }\n\n  /**\n   * Creates an Observable that emits each time the `AgentChatExchange` is updated\n   * with events from the Agent.\n   */\n  createAgentChatExchange$<T extends AgentResponseContent>(\n    chatSession: ChatSession,\n    userChatMessage: UserChatMessage,\n    options: {\n      mapChatEvents?: (\n        v: AgentChatResponse<AgentResponseContent>,\n      ) => Observable<AgentChatResponse<T>>;\n    } = {},\n  ) {\n    const initialAgentChatExchange = new AgentChatExchange(userChatMessage, []);\n\n    return this.sendChatMessage(chatSession, userChatMessage).pipe(\n      map((eventResult) => eventResult.unwrap()),\n      concatMap((agentChatEvent) => options.mapChatEvents?.(agentChatEvent) || of(agentChatEvent)),\n      endWith(new AgentEndTurn({ type: \"reply_finished\" })),\n      catchError((error: unknown) => of(new AgentEndTurn({ error, type: \"error\" }))),\n      scan(\n        (agentChatExchange, agentChatEvent) => agentChatExchange._addChatEvent(agentChatEvent),\n        initialAgentChatExchange,\n      ),\n      startWith(initialAgentChatExchange),\n      share(),\n    );\n  }\n\n  createChatSessionClient(chatSession: ChatSession) {\n    return new ChatSessionClient({\n      chatSession,\n      createAgentChatExchange$: this.createAgentChatExchange$.bind(this),\n    });\n  }\n\n  /**\n   * Sends a chat message to the Dremio AI Agent and returns an Observable containing\n   * Agent replies.\n   */\n  sendChatMessage(chatSession: ChatSession, userChatMessage: UserChatMessage) {\n    return createSendChatMessage(this.#config)(chatSession, userChatMessage);\n  }\n}\n"]}
+{"version":3,"file":"AIResource.js","sourceRoot":"","sources":["../../../src/enterprise/ai/AIResource.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAGpD,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EACL,wBAAwB,EACxB,0BAA0B,GAC3B,MAAM,uCAAuC,CAAC;AAE/C,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAClF,OAAO,EAAE,0BAA0B,EAAE,MAAM,+CAA+C,CAAC;AAC3F,OAAO,EAAE,qBAAqB,EAAE,MAAM,mCAAmC,CAAC;AAC1E,OAAO,EAAE,iBAAiB,EAAE,MAAM,8CAA8C,CAAC;AACjF,OAAO,EACL,kBAAkB,EAClB,uBAAuB,GACxB,MAAM,+CAA+C,CAAC;AACvD,OAAO,EAAE,oBAAoB,EAAE,MAAM,iDAAiD,CAAC;AACvF,OAAO,EAAE,iBAAiB,EAAkB,MAAM,kBAAkB,CAAC;AACrE,OAAO,EAAE,2BAA2B,EAAE,MAAM,wDAAwD,CAAC;AAErG,OAAO,EAAE,yBAAyB,EAAE,MAAM,8CAA8C,CAAC;AAEzF,MAAM,UAAU,mBAAmB,CAAC,MAAgB;IAClD,OAAO,SAAS,mBAAmB,CACjC,UAAsD,EACtD,EAAE,MAAM,KAAkB,EAAE;QAE5B,OAAO,MAAM;aACV,SAAS,CAAC,2BAA2B,EAAE;YACtC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,0BAA0B,EAAE,UAAU,CAAC,CAAC;YACrE,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;aACnC;YACD,MAAM,EAAE,MAAM;YACd,MAAM;SACP,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAqE,CAAC;aAC3F,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,UAAU;IACrB,OAAO,CAAoC;IAE3C,YAAY,MAAyC;QACnD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED,IAAI,mBAAmB;QACrB,OAAO,yBAAyB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACjD,CAAC;IAED,kBAAkB,CAAC,IAA6C;QAC9D,OAAO,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC;IAChD,CAAC;IAED,mBAAmB,CAAC,UAAqD;QACvE,MAAM,IAAI,GAAG,CAAC,CAAC,MAAM,CAAC,wBAAwB,EAAE,UAAU,CAAC,CAAC;QAC5D,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,uBAAuB,EAAE;YAClC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAC1B,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;aACnC;YACD,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAyD,CAAC;aAC/E,GAAG,CACF,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC,CAC1F,CAAC;IACN,CAAC;IAED,iBAAiB;QACf,MAAM,OAAO,GAAG,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChD,OAAO;YACL,KAAK,CAAC,CAAC,IAAI,CAAC,EAAE,MAAM,KAAkB,EAAE;gBACtC,KAAK,CAAC,CAAC,CACL,MAAM,OAAO,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,CACxF,CAAC,MAAM,EAAE,CAAC;YACb,CAAC;YACD,OAAO;SACR,CAAC;IACJ,CAAC;IAED,kBAAkB;QAChB,MAAM,OAAO,GAAG,CAAC,EAAE,MAAM,KAAkB,EAAE,EAAE,EAAE;YAC/C,OAAO,IAAI,CAAC,OAAO;iBAChB,SAAS,CAAC,uBAAuB,EAAE;gBAClC,OAAO,EAAE;oBACP,MAAM,EAAE,kBAAkB;iBAC3B;gBACD,MAAM;aACP,CAAC;iBACD,GAAG,CACF,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAwE,CAC1F;iBACA,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBAClB,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACpC,uBAAuB,CAAC,CAAC,CAAC,UAAU,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,CAC3E,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAC5D,CACF;aACF,CAAC,CAAC,CAAC;QACR,CAAC,CAAC;QACF,OAAO;YACL,KAAK,CAAC,CAAC,IAAI,CAAC,EAAE,MAAM,KAAkB,EAAE;gBACtC,KAAK,CAAC,CAAC,CAAC,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC;YACvF,CAAC;YACD,OAAO;SACR,CAAC;IACJ,CAAC;IAED,mBAAmB,CAAC,UAAsD;QACxE,OAAO,mBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,UAAU,CAAC,CAAC;IACvD,CAAC;IAED,oBAAoB,CAAC,EAAU,EAAE,EAAE,MAAM,KAAkB,EAAE;QAC3D,OAAO,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,GAAG,CAC3D,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAChE,CAAC;IACJ,CAAC;IAED,2BAA2B,CAAC,EAAU;QACpC,mEAAmE;QACnE,MAAM,OAAO,GAAG,CAAC,MAAU,EAAE,EAAE,MAAM,KAAkB,EAAE,EAAE,EAAE,CAC3D,2BAA2B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QAC5D,OAAO;YACL,KAAK,CAAC,CAAC,IAAI;gBACT,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;oBAChD,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;wBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;oBACrB,CAAC;oBACD,OAAO,MAAM,CAAC,KAAK,CAAC;gBACtB,CAAC,CAAC,CAAC;gBACH,KAAK,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;YACtB,CAAC;SACF,CAAC;IACJ,CAAC;IAED,qBAAqB,CAAC,EAAU,EAAE,EAAE,MAAM,KAAkB,EAAE;QAC5D,OAAO,IAAI,CAAC,OAAO;aAChB,SAAS,CAAC,yBAAyB,EAAE,EAAE,EAAE;YACxC,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;aAC3B;YACD,MAAM;SACP,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAyD,CAAC;aAC/E,GAAG,CACF,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC,CAC1F,CAAC;IACN,CAAC;IAED;;;;;OAKG;IACH,eAAe,CAAC,WAAwB,EAAE,eAAgC;QACxE,OAAO,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC;IAC3E,CAAC;CACF","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { ChatSession } from \"./chat/ChatSession.ts\";\nimport type { UserChatMessage } from \"./chat/UserChatMessage.ts\";\nimport type { Logger, SonarV4Config, V4Config } from \"../../common/Config.ts\";\nimport { ModelProvider } from \"./modelProvider/ModelProvider.ts\";\nimport * as z from \"zod/mini\";\nimport {\n  modelProviderCreateCodec,\n  modelProviderRetrieveCodec,\n} from \"./modelProvider/modelProviderCodec.ts\";\nimport type { SignalParam } from \"../../common/Params.ts\";\nimport { safeParseResultToResult } from \"../../common/safeParseResultToResult.ts\";\nimport { listModelsPropertiesSchema } from \"./modelProvider/modelProviderConfigSchemas.ts\";\nimport { createSendChatMessage } from \"./chat/methods/sendChatMessage.ts\";\nimport { listConversations } from \"./conversations/methods/listConversations.ts\";\nimport {\n  createConversation,\n  createConversationCodec,\n} from \"./conversations/methods/createConversation.ts\";\nimport { retrieveConversation } from \"./conversations/methods/retrieveConversation.ts\";\nimport { AgentConversation, type ChatEvent } from \"../interfaces.ts\";\nimport { retrieveConversationHistory } from \"./conversations/methods/retrieveConversationHistory.ts\";\nimport type { HttpError } from \"../../common/HttpError.ts\";\nimport { createConversationMachine } from \"./conversations/createConversationMachine.ts\";\n\nexport function listAvailableModels(config: V4Config) {\n  return function listAvailableModels(\n    properties: z.infer<typeof listModelsPropertiesSchema>,\n    { signal }: SignalParam = {},\n  ) {\n    return config\n      .v4Request(\"model-provider:listModels\", {\n        body: JSON.stringify(z.parse(listModelsPropertiesSchema, properties)),\n        headers: {\n          Accept: \"application/json\",\n          \"Content-Type\": \"application/json\",\n        },\n        method: \"POST\",\n        signal,\n      })\n      .map((res) => res.json() as Promise<{ models: { name: string; isRecommended: boolean }[] }>)\n      .map((result) => new Set(result.models));\n  };\n}\n\nexport class AIResource {\n  #config: Logger & V4Config & SonarV4Config;\n\n  constructor(config: Logger & V4Config & SonarV4Config) {\n    this.#config = config;\n  }\n\n  get conversationMachine() {\n    return createConversationMachine(this.#config);\n  }\n\n  createConversation(body: z.input<typeof createConversationCodec>) {\n    return createConversation(this.#config)(body);\n  }\n\n  createModelProvider(properties: z.output<typeof modelProviderCreateCodec>) {\n    const body = z.encode(modelProviderCreateCodec, properties);\n    return this.#config\n      .v4Request(`model-provider/config`, {\n        body: JSON.stringify(body),\n        headers: {\n          Accept: \"application/json\",\n          \"Content-Type\": \"application/json\",\n        },\n        keepalive: true,\n        method: \"POST\",\n      })\n      .map((res) => res.json() as Promise<z.input<typeof modelProviderRetrieveCodec>>)\n      .map(\n        (entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)),\n      );\n  }\n\n  listConversations() {\n    const getPage = listConversations(this.#config);\n    return {\n      async *data({ signal }: SignalParam = {}) {\n        yield* (\n          await getPage({ maxResults: 100 }, { signal }).map((response) => response.data).promise\n        ).unwrap();\n      },\n      getPage,\n    };\n  }\n\n  listModelProviders() {\n    const getPage = ({ signal }: SignalParam = {}) => {\n      return this.#config\n        .v4Request(\"model-provider/config\", {\n          headers: {\n            Accept: \"application/json\",\n          },\n          signal,\n        })\n        .map(\n          (res) => res.json() as Promise<{ configs: z.input<typeof modelProviderRetrieveCodec>[] }>,\n        )\n        .map((response) => ({\n          data: response.configs.map((entity) =>\n            safeParseResultToResult(z.safeDecode(modelProviderRetrieveCodec, entity)).map(\n              (properties) => new ModelProvider(this.#config, properties),\n            ),\n          ),\n        }));\n    };\n    return {\n      async *data({ signal }: SignalParam = {}) {\n        yield* (await getPage({ signal }).promise).map((response) => response.data).unwrap();\n      },\n      getPage,\n    };\n  }\n\n  listAvailableModels(properties: z.infer<typeof listModelsPropertiesSchema>) {\n    return listAvailableModels(this.#config)(properties);\n  }\n\n  retrieveConversation(id: string, { signal }: SignalParam = {}) {\n    return retrieveConversation(this.#config)(id, { signal }).map(\n      (properties) => new AgentConversation(this.#config, properties),\n    );\n  }\n\n  retrieveConversationHistory(id: string) {\n    // eslint-disable-next-line @typescript-eslint/no-empty-object-type\n    const getPage = (params: {}, { signal }: SignalParam = {}) =>\n      retrieveConversationHistory(this.#config)(id, { signal });\n    return {\n      async *data(): AsyncGenerator<ChatEvent, void, HttpError> {\n        const history = await getPage({}).then((result) => {\n          if (result.isErr()) {\n            throw result.error;\n          }\n          return result.value;\n        });\n        yield* history.data;\n      },\n    };\n  }\n\n  retrieveModelProvider(id: string, { signal }: SignalParam = {}) {\n    return this.#config\n      .v4Request(`model-provider/config/${id}`, {\n        headers: {\n          Accept: \"application/json\",\n        },\n        signal,\n      })\n      .map((res) => res.json() as Promise<z.input<typeof modelProviderRetrieveCodec>>)\n      .map(\n        (entity) => new ModelProvider(this.#config, z.decode(modelProviderRetrieveCodec, entity)),\n      );\n  }\n\n  /**\n   * Sends a chat message to the Dremio AI Agent and returns an Observable containing\n   * Agent replies.\n   *\n   * @deprecated\n   */\n  sendChatMessage(chatSession: ChatSession, userChatMessage: UserChatMessage) {\n    return createSendChatMessage(this.#config)(chatSession, userChatMessage);\n  }\n}\n"]}

package/dist/enterprise/ai/chat/ChatSession.d.ts

@@ -1,40 +1,22 @@
 import * as z from "zod/mini";
-import { Temporal } from "temporal-polyfill";
 declare const chatSessionInputSchema: z.ZodMiniObject<{
-    createdAt: z.ZodMiniString<string>;
     id: z.ZodMiniString<string>;
     modelName: z.ZodMiniString<string>;
     modelProviderId: z.ZodMiniString<string>;
-    title: z.ZodMiniOptional<z.ZodMiniString<string>>;
-    updatedAt: z.ZodMiniString<string>;
 }, z.core.$strip>;
 declare const newChatSessionSchema: z.ZodMiniObject<{
     modelName: z.ZodMiniOptional<z.ZodMiniString<string>>;
     modelProviderId: z.ZodMiniOptional<z.ZodMiniString<string>>;
 }, z.core.$strip>;
-/**
- * Tracks an Agent chat session. Each field is optional and intentionally mutable
- * so that if any of these fields become available during agent reply (such as which
- * default model provider and name it picked, as well as the server-generated sessionId),
- * they're mutated on the instance of the class.
- */
 export declare class ChatSession {
-    #private;
-    get createdAt(): Temporal.Instant;
-    get id(): string | undefined;
-    get modelName(): string | undefined;
-    get modelProviderId(): string | undefined;
-    get title(): string | undefined;
-    get updatedAt(): Temporal.Instant;
-    private constructor();
-    [Symbol.observable](): import("rxjs").Observable<ChatSession>;
+    id?: string;
+    modelName?: string;
+    modelProviderId?: string;
+    constructor(properties?: Partial<z.infer<typeof chatSessionInputSchema>>);
     toJSON(): {
-        createdAt: string;
         id: string;
         modelName: string;
         modelProviderId: string;
-        updatedAt: string;
-        title?: string | undefined;
     };
     assign(properties: Partial<z.infer<typeof chatSessionInputSchema>>): void;
     equals(other: unknown): boolean;
@@ -45,12 +27,9 @@ export declare class ChatSession {
      */
     static new(options?: z.infer<typeof newChatSessionSchema>): ChatSession;
     static codec: z.ZodMiniCodec<z.ZodMiniObject<{
-        createdAt: z.ZodMiniString<string>;
         id: z.ZodMiniString<string>;
         modelName: z.ZodMiniString<string>;
         modelProviderId: z.ZodMiniString<string>;
-        title: z.ZodMiniOptional<z.ZodMiniString<string>>;
-        updatedAt: z.ZodMiniString<string>;
     }, z.core.$strip>, z.ZodMiniCustom<ChatSession, ChatSession>>;
 }
 export {};

package/dist/enterprise/ai/chat/ChatSession.js

@@ -14,81 +14,35 @@
  * limitations under the License.
  */
 import * as z from "zod/mini";
-import { Temporal } from "temporal-polyfill";
-import { BehaviorSubject } from "rxjs";
 const chatSessionInputSchema = z.object({
-    createdAt: z.string().check(z.iso.datetime()),
     id: z.string(),
     modelName: z.string(),
     modelProviderId: z.string(),
-    title: z.optional(z.string()),
-    updatedAt: z.string().check(z.iso.datetime()),
 });
 const newChatSessionSchema = z.partial(z.pick(chatSessionInputSchema, {
     modelName: true,
     modelProviderId: true,
 }));
-/**
- * Tracks an Agent chat session. Each field is optional and intentionally mutable
- * so that if any of these fields become available during agent reply (such as which
- * default model provider and name it picked, as well as the server-generated sessionId),
- * they're mutated on the instance of the class.
- */
 export class ChatSession {
-    #createdAt = Temporal.Now.instant();
-    #id;
-    #modelName;
-    #modelProviderId;
-    #title;
-    #updatedAt = Temporal.Now.instant();
-    #bus = new BehaviorSubject(this);
-    get createdAt() {
-        return this.#createdAt;
-    }
-    get id() {
-        return this.#id;
-    }
-    get modelName() {
-        return this.#modelName;
-    }
-    get modelProviderId() {
-        return this.#modelProviderId;
-    }
-    get title() {
-        return this.#title;
-    }
-    get updatedAt() {
-        return this.#updatedAt;
-    }
+    id;
+    modelName;
+    modelProviderId;
     constructor(properties = {}) {
         this.assign(properties);
     }
-    [Symbol.observable]() {
-        return this.#bus.asObservable();
-    }
     toJSON() {
         return z.encode(ChatSession.codec, this);
     }
     assign(properties) {
-        if (properties.createdAt) {
-            this.#createdAt = Temporal.Instant.from(properties.createdAt);
-        }
         if (properties.id) {
-            this.#id = properties.id;
+            this.id = properties.id;
         }
         if (properties.modelName) {
-            this.#modelName = properties.modelName;
+            this.modelName = properties.modelName;
         }
         if (properties.modelProviderId) {
-            this.#modelProviderId = properties.modelProviderId;
-        }
-        if (properties.title) {
-            this.#title = properties.title;
-        }
-        if (properties.updatedAt) {
-            this.#updatedAt = Temporal.Instant.from(properties.updatedAt);
+            this.modelProviderId = properties.modelProviderId;
         }
-        this.#bus.next(this);
     }
     equals(other) {
         return (other instanceof ChatSession &&
@@ -113,12 +67,9 @@ export class ChatSession {
                 throw new Error("An uninitialized ChatSession cannot be encoded.");
             }
             return {
-                createdAt: v.createdAt.toString(),
                 id: v.id,
                 modelName: v.modelName,
                 modelProviderId: v.modelProviderId,
-                title: v.title,
-                updatedAt: v.updatedAt.toString(),
             };
         },
     });

package/dist/enterprise/ai/chat/ChatSession.js.map

@@ -1 +1 @@
-{"version":3,"file":"ChatSession.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/ChatSession.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAE,MAAM,MAAM,CAAC;AAEvC,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;IAC7C,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE;IAC3B,KAAK,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC7B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;CAC9C,CAAC,CAAC;AAEH,MAAM,oBAAoB,GAAG,CAAC,CAAC,OAAO,CACpC,CAAC,CAAC,IAAI,CAAC,sBAAsB,EAAE;IAC7B,SAAS,EAAE,IAAI;IACf,eAAe,EAAE,IAAI;CACtB,CAAC,CACH,CAAC;AAEF;;;;;GAKG;AACH,MAAM,OAAO,WAAW;IACtB,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACpC,GAAG,CAAU;IACb,UAAU,CAAU;IACpB,gBAAgB,CAAU;IAC1B,MAAM,CAAU;IAChB,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IAEpC,IAAI,GAAG,IAAI,eAAe,CAAc,IAAI,CAAC,CAAC;IAE9C,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,IAAI,EAAE;QACJ,OAAO,IAAI,CAAC,GAAG,CAAC;IAClB,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,IAAI,eAAe;QACjB,OAAO,IAAI,CAAC,gBAAgB,CAAC;IAC/B,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,YAAoB,aAA8D,EAAE;QAClF,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAC1B,CAAC;IAED,CAAC,MAAM,CAAC,UAAU,CAAC;QACjB,OAAO,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;IAClC,CAAC;IAED,MAAM;QACJ,OAAO,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,CAAC,UAA2D;QAChE,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YACzB,IAAI,CAAC,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAChE,CAAC;QACD,IAAI,UAAU,CAAC,EAAE,EAAE,CAAC;YAClB,IAAI,CAAC,GAAG,GAAG,UAAU,CAAC,EAAE,CAAC;QAC3B,CAAC;QACD,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YACzB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,SAAS,CAAC;QACzC,CAAC;QACD,IAAI,UAAU,CAAC,eAAe,EAAE,CAAC;YAC/B,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAC,eAAe,CAAC;QACrD,CAAC;QACD,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;YACrB,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC;QACjC,CAAC;QACD,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YACzB,IAAI,CAAC,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAChE,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;IAED,MAAM,CAAC,KAAc;QACnB,OAAO,CACL,KAAK,YAAY,WAAW;YAC5B,CAAC,IAAI,KAAK,KAAK,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CACxE,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,IAAuC;QACrD,OAAO,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,GAAG,CAAC,UAAgD,EAAE;QAC3D,OAAO,IAAI,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,oBAAoB,EAAE,OAAO,CAAC,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CACpB,sBAAsB,EACtB,CAAC,CAAC,MAAM,CAAc,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,WAAW,CAAC,EACtD;QACE,MAAM,CAAC,CAAC;YACN,OAAO,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC;QACD,MAAM,CAAC,CAAC;YACN,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;gBACV,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;YACrE,CAAC;YACD,OAAO;gBACL,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE;gBACjC,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,SAAS,EAAE,CAAC,CAAC,SAAU;gBACvB,eAAe,EAAE,CAAC,CAAC,eAAgB;gBACnC,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE;aAClC,CAAC;QACJ,CAAC;KACF,CACF,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport * as z from \"zod/mini\";\nimport { Temporal } from \"temporal-polyfill\";\nimport { BehaviorSubject } from \"rxjs\";\n\nconst chatSessionInputSchema = z.object({\n  createdAt: z.string().check(z.iso.datetime()),\n  id: z.string(),\n  modelName: z.string(),\n  modelProviderId: z.string(),\n  title: z.optional(z.string()),\n  updatedAt: z.string().check(z.iso.datetime()),\n});\n\nconst newChatSessionSchema = z.partial(\n  z.pick(chatSessionInputSchema, {\n    modelName: true,\n    modelProviderId: true,\n  }),\n);\n\n/**\n * Tracks an Agent chat session. Each field is optional and intentionally mutable\n * so that if any of these fields become available during agent reply (such as which\n * default model provider and name it picked, as well as the server-generated sessionId),\n * they're mutated on the instance of the class.\n */\nexport class ChatSession {\n  #createdAt = Temporal.Now.instant();\n  #id?: string;\n  #modelName?: string;\n  #modelProviderId?: string;\n  #title?: string;\n  #updatedAt = Temporal.Now.instant();\n\n  #bus = new BehaviorSubject<ChatSession>(this);\n\n  get createdAt(): Temporal.Instant {\n    return this.#createdAt;\n  }\n\n  get id(): string | undefined {\n    return this.#id;\n  }\n\n  get modelName(): string | undefined {\n    return this.#modelName;\n  }\n\n  get modelProviderId(): string | undefined {\n    return this.#modelProviderId;\n  }\n\n  get title(): string | undefined {\n    return this.#title;\n  }\n\n  get updatedAt(): Temporal.Instant {\n    return this.#updatedAt;\n  }\n\n  private constructor(properties: Partial<z.infer<typeof chatSessionInputSchema>> = {}) {\n    this.assign(properties);\n  }\n\n  [Symbol.observable]() {\n    return this.#bus.asObservable();\n  }\n\n  toJSON() {\n    return z.encode(ChatSession.codec, this);\n  }\n\n  assign(properties: Partial<z.infer<typeof chatSessionInputSchema>>) {\n    if (properties.createdAt) {\n      this.#createdAt = Temporal.Instant.from(properties.createdAt);\n    }\n    if (properties.id) {\n      this.#id = properties.id;\n    }\n    if (properties.modelName) {\n      this.#modelName = properties.modelName;\n    }\n    if (properties.modelProviderId) {\n      this.#modelProviderId = properties.modelProviderId;\n    }\n    if (properties.title) {\n      this.#title = properties.title;\n    }\n    if (properties.updatedAt) {\n      this.#updatedAt = Temporal.Instant.from(properties.updatedAt);\n    }\n    this.#bus.next(this);\n  }\n\n  equals(other: unknown) {\n    return (\n      other instanceof ChatSession &&\n      (this === other || (this.id === other.id && (!!this.id || !!other.id)))\n    );\n  }\n\n  static fromJSON(data: z.input<typeof ChatSession.codec>) {\n    return z.decode(ChatSession.codec, data);\n  }\n\n  /**\n   * Creates a new ChatSession with fields that will be populated automatically\n   * once a reply is received from the Dremio AI Agent.\n   */\n  static new(options: z.infer<typeof newChatSessionSchema> = {}) {\n    return new ChatSession(z.parse(newChatSessionSchema, options));\n  }\n\n  static codec = z.codec(\n    chatSessionInputSchema,\n    z.custom<ChatSession>((v) => v instanceof ChatSession),\n    {\n      decode(v) {\n        return new ChatSession(v);\n      },\n      encode(v) {\n        if (!v.id) {\n          throw new Error(\"An uninitialized ChatSession cannot be encoded.\");\n        }\n        return {\n          createdAt: v.createdAt.toString(),\n          id: v.id,\n          modelName: v.modelName!,\n          modelProviderId: v.modelProviderId!,\n          title: v.title,\n          updatedAt: v.updatedAt.toString(),\n        };\n      },\n    },\n  );\n}\n"]}
+{"version":3,"file":"ChatSession.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/ChatSession.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE;CAC5B,CAAC,CAAC;AAEH,MAAM,oBAAoB,GAAG,CAAC,CAAC,OAAO,CACpC,CAAC,CAAC,IAAI,CAAC,sBAAsB,EAAE;IAC7B,SAAS,EAAE,IAAI;IACf,eAAe,EAAE,IAAI;CACtB,CAAC,CACH,CAAC;AAEF,MAAM,OAAO,WAAW;IACtB,EAAE,CAAU;IACZ,SAAS,CAAU;IACnB,eAAe,CAAU;IAEzB,YAAY,aAA8D,EAAE;QAC1E,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAC1B,CAAC;IAED,MAAM;QACJ,OAAO,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,CAAC,UAA2D;QAChE,IAAI,UAAU,CAAC,EAAE,EAAE,CAAC;YAClB,IAAI,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;QAC1B,CAAC;QACD,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YACzB,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACxC,CAAC;QACD,IAAI,UAAU,CAAC,eAAe,EAAE,CAAC;YAC/B,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC,eAAe,CAAC;QACpD,CAAC;IACH,CAAC;IAED,MAAM,CAAC,KAAc;QACnB,OAAO,CACL,KAAK,YAAY,WAAW;YAC5B,CAAC,IAAI,KAAK,KAAK,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CACxE,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,IAAuC;QACrD,OAAO,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,GAAG,CAAC,UAAgD,EAAE;QAC3D,OAAO,IAAI,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,oBAAoB,EAAE,OAAO,CAAC,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CACpB,sBAAsB,EACtB,CAAC,CAAC,MAAM,CAAc,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,WAAW,CAAC,EACtD;QACE,MAAM,CAAC,CAAC;YACN,OAAO,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC;QACD,MAAM,CAAC,CAAC;YACN,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;gBACV,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;YACrE,CAAC;YACD,OAAO;gBACL,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,SAAS,EAAE,CAAC,CAAC,SAAU;gBACvB,eAAe,EAAE,CAAC,CAAC,eAAgB;aACpC,CAAC;QACJ,CAAC;KACF,CACF,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport * as z from \"zod/mini\";\n\nconst chatSessionInputSchema = z.object({\n  id: z.string(),\n  modelName: z.string(),\n  modelProviderId: z.string(),\n});\n\nconst newChatSessionSchema = z.partial(\n  z.pick(chatSessionInputSchema, {\n    modelName: true,\n    modelProviderId: true,\n  }),\n);\n\nexport class ChatSession {\n  id?: string;\n  modelName?: string;\n  modelProviderId?: string;\n\n  constructor(properties: Partial<z.infer<typeof chatSessionInputSchema>> = {}) {\n    this.assign(properties);\n  }\n\n  toJSON() {\n    return z.encode(ChatSession.codec, this);\n  }\n\n  assign(properties: Partial<z.infer<typeof chatSessionInputSchema>>) {\n    if (properties.id) {\n      this.id = properties.id;\n    }\n    if (properties.modelName) {\n      this.modelName = properties.modelName;\n    }\n    if (properties.modelProviderId) {\n      this.modelProviderId = properties.modelProviderId;\n    }\n  }\n\n  equals(other: unknown) {\n    return (\n      other instanceof ChatSession &&\n      (this === other || (this.id === other.id && (!!this.id || !!other.id)))\n    );\n  }\n\n  static fromJSON(data: z.input<typeof ChatSession.codec>) {\n    return z.decode(ChatSession.codec, data);\n  }\n\n  /**\n   * Creates a new ChatSession with fields that will be populated automatically\n   * once a reply is received from the Dremio AI Agent.\n   */\n  static new(options: z.infer<typeof newChatSessionSchema> = {}) {\n    return new ChatSession(z.parse(newChatSessionSchema, options));\n  }\n\n  static codec = z.codec(\n    chatSessionInputSchema,\n    z.custom<ChatSession>((v) => v instanceof ChatSession),\n    {\n      decode(v) {\n        return new ChatSession(v);\n      },\n      encode(v) {\n        if (!v.id) {\n          throw new Error(\"An uninitialized ChatSession cannot be encoded.\");\n        }\n        return {\n          id: v.id,\n          modelName: v.modelName!,\n          modelProviderId: v.modelProviderId!,\n        };\n      },\n    },\n  );\n}\n"]}

package/dist/enterprise/ai/chat/UserChatMessage.d.ts

@@ -1,30 +1,46 @@
-import * as z from "zod/mini";
 import { Temporal } from "temporal-polyfill";
-import { type UserChatMessageContent } from "./UserChatMessageContent.ts";
+import * as z from "zod/mini";
 export declare class UserChatMessage {
-    readonly createdAt: Temporal.Instant;
-    readonly content: UserChatMessageContent;
+    readonly createdAt: Temporal.ZonedDateTime;
+    readonly content: string;
     readonly id: string;
-    readonly context?: Record<string, z.core.util.JSONType>;
-    constructor(content: UserChatMessage["content"], context?: UserChatMessage["context"], id?: string, createdAt?: Temporal.Instant);
-    withContext(context: UserChatMessage["context"]): UserChatMessage;
-    mergeContext(context: NonNullable<UserChatMessage["context"]>): UserChatMessage;
-    toString(): string;
+    readonly prompt?: Omit<z.infer<typeof createConversationPromptSchema>, "text">;
+    constructor(id: UserChatMessage["id"], createdAt: UserChatMessage["createdAt"], content: UserChatMessage["content"], prompt?: UserChatMessage["prompt"]);
+    withPrompt(prompt: UserChatMessage["prompt"]): UserChatMessage;
     toJSON(): {
-        content: {
-            content: string;
-            type: "string";
-        };
+        content: string;
         createdAt: string;
         id: string;
+        prompt: Omit<{
+            text: string;
+            approvals?: {
+                allToolsAllowed: boolean;
+                approvalNonce: string;
+                toolDecisions: {
+                    approved: boolean;
+                    arguments: Record<string, unknown>;
+                    executionId: string;
+                    name: string;
+                }[];
+            } | undefined;
+            context?: string | undefined;
+            skillIds?: string[] | undefined;
+        }, "text"> | undefined;
     };
-    static fromJSON(data: z.input<typeof UserChatMessage.codec>): UserChatMessage;
-    static codec: z.ZodMiniCodec<z.ZodMiniObject<{
-        content: z.ZodMiniObject<{
-            content: z.ZodMiniString<string>;
-            type: z.ZodMiniLiteral<"string">;
-        }, z.core.$strip>;
-        createdAt: z.ZodMiniString<string>;
-        id: z.ZodMiniString<string>;
-    }, z.core.$strip>, z.ZodMiniCustom<UserChatMessage, UserChatMessage>>;
+    static new(content: UserChatMessage["content"], prompt?: UserChatMessage["prompt"]): UserChatMessage;
 }
+export declare const createConversationPromptSchema: z.ZodMiniObject<{
+    approvals: z.ZodMiniOptional<z.ZodMiniObject<{
+        allToolsAllowed: z.ZodMiniBoolean<boolean>;
+        approvalNonce: z.ZodMiniString<string>;
+        toolDecisions: z.ZodMiniArray<z.ZodMiniObject<{
+            approved: z.ZodMiniBoolean<boolean>;
+            arguments: z.ZodMiniRecord<z.ZodMiniString<string>, z.ZodMiniUnknown>;
+            executionId: z.ZodMiniString<string>;
+            name: z.ZodMiniString<string>;
+        }, z.core.$strip>>;
+    }, z.core.$strip>>;
+    context: z.ZodMiniOptional<z.ZodMiniString<string>>;
+    skillIds: z.ZodMiniOptional<z.ZodMiniArray<z.ZodMiniString<string>>>;
+    text: z.ZodMiniString<string>;
+}, z.core.$strict>;

package/dist/enterprise/ai/chat/UserChatMessage.js

@@ -14,55 +14,47 @@
  * limitations under the License.
  */
 import { nanoid } from "nanoid/non-secure";
-import * as z from "zod/mini";
 import { Temporal } from "temporal-polyfill";
-import { userChatMessageContentCodec, } from "./UserChatMessageContent.js";
+import * as z from "zod/mini";
 export class UserChatMessage {
-    createdAt = Temporal.Now.instant();
+    createdAt;
     content;
-    id = nanoid();
-    context;
-    constructor(content, context, id, createdAt) {
+    id;
+    prompt;
+    constructor(id, createdAt, content, prompt) {
+        this.id = id;
+        this.createdAt = createdAt;
         this.content = content;
-        this.context = context;
-        if (id) {
-            this.id = id;
-        }
-        if (createdAt) {
-            this.createdAt = createdAt;
-        }
-    }
-    withContext(context) {
-        return new UserChatMessage(this.content, context, this.id, this.createdAt);
-    }
-    mergeContext(context) {
-        return new UserChatMessage(this.content, { ...this.context, ...context }, this.id);
+        this.prompt = prompt;
     }
-    toString() {
-        return (this.content.content +
-            (this.context ? `\n\n<system-context>${JSON.stringify(this.context)}</system-context>` : ""));
+    withPrompt(prompt) {
+        return new UserChatMessage(this.id, this.createdAt, this.content, prompt);
     }
     toJSON() {
-        return z.encode(UserChatMessage.codec, this);
+        return {
+            content: this.content,
+            createdAt: this.createdAt.toString(),
+            id: this.id,
+            prompt: this.prompt,
+        };
     }
-    static fromJSON(data) {
-        return z.decode(UserChatMessage.codec, data);
+    static new(content, prompt) {
+        return new UserChatMessage(nanoid(), Temporal.Now.zonedDateTimeISO(), content, prompt);
     }
-    static codec = z.codec(z.object({
-        content: userChatMessageContentCodec.def.in,
-        createdAt: z.string().check(z.iso.datetime()),
-        id: z.string(),
-    }), z.instanceof(UserChatMessage), {
-        decode(v) {
-            return new UserChatMessage(z.decode(userChatMessageContentCodec, v.content), undefined, v.id);
-        },
-        encode(v) {
-            return {
-                ...v,
-                content: z.encode(userChatMessageContentCodec, v.content),
-                createdAt: v.createdAt.toString(),
-            };
-        },
-    });
 }
+export const createConversationPromptSchema = z.strictObject({
+    approvals: z.optional(z.object({
+        allToolsAllowed: z.boolean(),
+        approvalNonce: z.string(),
+        toolDecisions: z.array(z.object({
+            approved: z.boolean(),
+            arguments: z.record(z.string(), z.unknown()),
+            executionId: z.string(),
+            name: z.string(),
+        })),
+    })),
+    context: z.optional(z.string()),
+    skillIds: z.optional(z.array(z.string())),
+    text: z.string(),
+});
 //# sourceMappingURL=UserChatMessage.js.map

package/dist/enterprise/ai/chat/UserChatMessage.js.map

@@ -1 +1 @@
-{"version":3,"file":"UserChatMessage.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/UserChatMessage.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EACL,2BAA2B,GAE5B,MAAM,6BAA6B,CAAC;AAErC,MAAM,OAAO,eAAe;IACjB,SAAS,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnC,OAAO,CAAyB;IAChC,EAAE,GAAG,MAAM,EAAE,CAAC;IACd,OAAO,CAAwC;IAExD,YACE,OAAmC,EACnC,OAAoC,EACpC,EAAW,EACX,SAA4B;QAE5B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,IAAI,EAAE,EAAE,CAAC;YACP,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACf,CAAC;QAED,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC7B,CAAC;IACH,CAAC;IAED,WAAW,CAAC,OAAmC;QAC7C,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAC7E,CAAC;IAED,YAAY,CAAC,OAAgD;QAC3D,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,OAAO,EAAE,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IACrF,CAAC;IAED,QAAQ;QACN,OAAO,CACL,IAAI,CAAC,OAAO,CAAC,OAAO;YACpB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,uBAAuB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,CAAC,EAAE,CAAC,CAC7F,CAAC;IACJ,CAAC;IAED,MAAM;QACJ,OAAO,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,IAA2C;QACzD,OAAO,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CACpB,CAAC,CAAC,MAAM,CAAC;QACP,OAAO,EAAE,2BAA2B,CAAC,GAAG,CAAC,EAAE;QAC3C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC7C,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;KACf,CAAC,EACF,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,EAC7B;QACE,MAAM,CAAC,CAAC;YACN,OAAO,IAAI,eAAe,CACxB,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE,CAAC,CAAC,OAAO,CAAC,EAChD,SAAS,EACT,CAAC,CAAC,EAAE,CACL,CAAC;QACJ,CAAC;QACD,MAAM,CAAC,CAAC;YACN,OAAO;gBACL,GAAG,CAAC;gBACJ,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,2BAA2B,EAAE,CAAC,CAAC,OAAO,CAAC;gBACzD,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE;aACzB,CAAC;QACb,CAAC;KACF,CACF,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { nanoid } from \"nanoid/non-secure\";\nimport * as z from \"zod/mini\";\nimport { Temporal } from \"temporal-polyfill\";\nimport {\n  userChatMessageContentCodec,\n  type UserChatMessageContent,\n} from \"./UserChatMessageContent.ts\";\n\nexport class UserChatMessage {\n  readonly createdAt = Temporal.Now.instant();\n  readonly content: UserChatMessageContent;\n  readonly id = nanoid();\n  readonly context?: Record<string, z.core.util.JSONType>;\n\n  constructor(\n    content: UserChatMessage[\"content\"],\n    context?: UserChatMessage[\"context\"],\n    id?: string,\n    createdAt?: Temporal.Instant,\n  ) {\n    this.content = content;\n    this.context = context;\n\n    if (id) {\n      this.id = id;\n    }\n\n    if (createdAt) {\n      this.createdAt = createdAt;\n    }\n  }\n\n  withContext(context: UserChatMessage[\"context\"]) {\n    return new UserChatMessage(this.content, context, this.id, this.createdAt);\n  }\n\n  mergeContext(context: NonNullable<UserChatMessage[\"context\"]>) {\n    return new UserChatMessage(this.content, { ...this.context, ...context }, this.id);\n  }\n\n  toString() {\n    return (\n      this.content.content +\n      (this.context ? `\\n\\n<system-context>${JSON.stringify(this.context)}</system-context>` : \"\")\n    );\n  }\n\n  toJSON() {\n    return z.encode(UserChatMessage.codec, this);\n  }\n\n  static fromJSON(data: z.input<typeof UserChatMessage.codec>) {\n    return z.decode(UserChatMessage.codec, data);\n  }\n\n  static codec = z.codec(\n    z.object({\n      content: userChatMessageContentCodec.def.in,\n      createdAt: z.string().check(z.iso.datetime()),\n      id: z.string(),\n    }),\n    z.instanceof(UserChatMessage),\n    {\n      decode(v) {\n        return new UserChatMessage(\n          z.decode(userChatMessageContentCodec, v.content),\n          undefined,\n          v.id,\n        );\n      },\n      encode(v) {\n        return {\n          ...v,\n          content: z.encode(userChatMessageContentCodec, v.content),\n          createdAt: v.createdAt.toString(),\n        } as const;\n      },\n    },\n  );\n}\n"]}
+{"version":3,"file":"UserChatMessage.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/chat/UserChatMessage.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAE9B,MAAM,OAAO,eAAe;IACjB,SAAS,CAAyB;IAClC,OAAO,CAAS;IAChB,EAAE,CAAS;IACX,MAAM,CAAgE;IAE/E,YACE,EAAyB,EACzB,SAAuC,EACvC,OAAmC,EACnC,MAAkC;QAElC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,UAAU,CAAC,MAAiC;QAC1C,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;IAED,MAAM;QACJ,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE;YACpC,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,GAAG,CAAC,OAAmC,EAAE,MAAkC;QAChF,OAAO,IAAI,eAAe,CAAC,MAAM,EAAE,EAAE,QAAQ,CAAC,GAAG,CAAC,gBAAgB,EAAE,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACzF,CAAC;CACF;AAED,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAAC,CAAC,YAAY,CAAC;IAC3D,SAAS,EAAE,CAAC,CAAC,QAAQ,CACnB,CAAC,CAAC,MAAM,CAAC;QACP,eAAe,EAAE,CAAC,CAAC,OAAO,EAAE;QAC5B,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;QACzB,aAAa,EAAE,CAAC,CAAC,KAAK,CACpB,CAAC,CAAC,MAAM,CAAC;YACP,QAAQ,EAAE,CAAC,CAAC,OAAO,EAAE;YACrB,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;YAC5C,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;YACvB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;SACjB,CAAC,CACH;KACF,CAAC,CACH;IACD,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC/B,QAAQ,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IACzC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;CACjB,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { nanoid } from \"nanoid/non-secure\";\nimport { Temporal } from \"temporal-polyfill\";\nimport * as z from \"zod/mini\";\n\nexport class UserChatMessage {\n  readonly createdAt: Temporal.ZonedDateTime;\n  readonly content: string;\n  readonly id: string;\n  readonly prompt?: Omit<z.infer<typeof createConversationPromptSchema>, \"text\">;\n\n  constructor(\n    id: UserChatMessage[\"id\"],\n    createdAt: UserChatMessage[\"createdAt\"],\n    content: UserChatMessage[\"content\"],\n    prompt?: UserChatMessage[\"prompt\"],\n  ) {\n    this.id = id;\n    this.createdAt = createdAt;\n    this.content = content;\n    this.prompt = prompt;\n  }\n\n  withPrompt(prompt: UserChatMessage[\"prompt\"]) {\n    return new UserChatMessage(this.id, this.createdAt, this.content, prompt);\n  }\n\n  toJSON() {\n    return {\n      content: this.content,\n      createdAt: this.createdAt.toString(),\n      id: this.id,\n      prompt: this.prompt,\n    };\n  }\n\n  static new(content: UserChatMessage[\"content\"], prompt?: UserChatMessage[\"prompt\"]) {\n    return new UserChatMessage(nanoid(), Temporal.Now.zonedDateTimeISO(), content, prompt);\n  }\n}\n\nexport const createConversationPromptSchema = z.strictObject({\n  approvals: z.optional(\n    z.object({\n      allToolsAllowed: z.boolean(),\n      approvalNonce: z.string(),\n      toolDecisions: z.array(\n        z.object({\n          approved: z.boolean(),\n          arguments: z.record(z.string(), z.unknown()),\n          executionId: z.string(),\n          name: z.string(),\n        }),\n      ),\n    }),\n  ),\n  context: z.optional(z.string()),\n  skillIds: z.optional(z.array(z.string())),\n  text: z.string(),\n});\n"]}