@squidcloud/client 1.0.106 → 1.0.108

package/dist/cjs/index.js

@@ -26616,6 +26616,8 @@ var IntegrationType;
     IntegrationType["redis"] = "redis";
     IntegrationType["xata"] = "xata";
     IntegrationType["azure_sql"] = "azure_sql";
+    IntegrationType["azure_postgresql"] = "azure_postgresql";
+    IntegrationType["azure_cosmosdb"] = "azure_cosmosdb";
 })(IntegrationType || (IntegrationType = {}));
 var IntegrationSchemaType;
 (function (IntegrationSchemaType) {
@@ -32196,6 +32198,7 @@ class AiAssistantClient {
         this.socketManager = socketManager;
         this.integrationId = integrationId;
         this.ongoingChatRequests = {};
+        this.ongoingChatSequences = {};
         this.socketManager
             .observeNotifications()
             .pipe((0,external_rxjs_namespaceObject.filter)((notification) => notification.type === 'aiAssistant'), map_map((n) => n))
@@ -32229,7 +32232,27 @@ class AiAssistantClient {
     chat(profileId, prompt) {
         const clientRequestId = generateId();
         const subject = new external_rxjs_namespaceObject.BehaviorSubject('');
+        const tokenSequence = new external_rxjs_namespaceObject.Subject();
         this.ongoingChatRequests[clientRequestId] = subject;
+        this.ongoingChatSequences[clientRequestId] = tokenSequence;
+        tokenSequence
+            .pipe((0,external_rxjs_namespaceObject.concatMap)(({ value, complete }) => {
+            if (complete) {
+                return (0,external_rxjs_namespaceObject.of)({ value, complete });
+            }
+            return (0,external_rxjs_namespaceObject.of)(value).pipe((0,external_rxjs_namespaceObject.delay)(5), map_map((char) => ({ value: char, complete: false })));
+        }), (0,external_rxjs_namespaceObject.takeWhile)(({ complete }) => !complete, true))
+            .subscribe({
+            next: ({ value }) => {
+                subject.next(subject.value + value);
+            },
+            error: (e) => {
+                console.error(e);
+            },
+            complete: () => {
+                subject.complete();
+            },
+        });
         const request = {
             profileId,
             prompt,
@@ -32242,18 +32265,22 @@ class AiAssistantClient {
         });
         return subject.pipe((0,external_rxjs_namespaceObject.finalize)(() => {
             delete this.ongoingChatRequests[clientRequestId];
+            delete this.ongoingChatSequences[clientRequestId];
         }), (0,external_rxjs_namespaceObject.share)());
     }
-    handleChatResponse(message) {
-        const subject = this.ongoingChatRequests[message.clientRequestId];
-        if (!subject) {
+    async handleChatResponse(message) {
+        const tokenSequence = this.ongoingChatSequences[message.clientRequestId];
+        if (!tokenSequence) {
             return;
         }
-        if (message.payload.complete) {
-            subject.complete();
+        const { token, complete } = message.payload;
+        if (complete && !token.length) {
+            tokenSequence.next({ value: '', complete: true });
         }
         else {
-            subject.next(subject.value + message.payload.token);
+            for (let i = 0; i < token.length; i++) {
+                tokenSequence.next({ value: token[i], complete: complete && i === token.length - 1 });
+            }
         }
     }
 }
@@ -32300,17 +32327,19 @@ class AiAssistantProfileReference {
      *
      * @param data An object containing options for creating the profile.
      * @param data.modelName - The name of the OpenAI model (`gpt-3.5 or `gpt-4`).
+     * @param data.isPublic - Whether the chat functionality of the profile can be accessed without security rules.
      * @returns A promise that resolves when the profile is successfully created.
      */
     insert(data) {
         // TODO: Investigate support strictContext.
-        const { modelName } = data;
+        const { modelName, isPublic = false } = data;
         const request = {
             type: 'insert',
             resource: 'profile',
             profileId: this.profileId,
             payload: {
                 modelName,
+                isPublic,
                 strictContext: false,
             },
             integrationId: this.integrationId,
@@ -32323,16 +32352,18 @@ class AiAssistantProfileReference {
      *
      * @param data An object containing options for updating the profile.
      * @param data.modelName - The name of the OpenAI model (`gpt-3.5 or `gpt-4`).
+     * @param data.isPublic - Whether the chat functionality of the profile can be accessed without security rules.
      * @returns A promise that resolves when the profile is successfully updated.
      */
     update(data) {
-        const { modelName } = data;
+        const { modelName, isPublic } = data;
         const request = {
             type: 'update',
             resource: 'profile',
             profileId: this.profileId,
             payload: {
                 modelName,
+                isPublic,
                 strictContext: false,
             },
             integrationId: this.integrationId,

package/dist/common/src/integrations/ai_assistant.types.d.ts

@@ -1,4 +1,5 @@
 import { BaseIntegrationConfig, IntegrationType } from './index';
+import { OpenAiModelName } from '../ai-assistant.types';
 export interface AiAssistantIntegrationConfig extends BaseIntegrationConfig {
     type: IntegrationType.ai_assistant;
     configuration: AiAssistantConfiguration;
@@ -7,7 +8,8 @@ export type AiAssistantConfiguration = {
     apiKey?: string;
 };
 export type AiAssistantProfileMetadata = {
-    modelName: string;
+    modelName: OpenAiModelName;
+    isPublic: boolean;
     strictContext: boolean;
     instructions: Record<string, string>;
 };

package/dist/common/src/integrations/index.d.ts

@@ -53,7 +53,9 @@ export declare enum IntegrationType {
     'oracledb' = "oracledb",
     'redis' = "redis",
     'xata' = "xata",
-    'azure_sql' = "azure_sql"
+    'azure_sql' = "azure_sql",
+    'azure_postgresql' = "azure_postgresql",
+    'azure_cosmosdb' = "azure_cosmosdb"
 }
 export declare enum IntegrationSchemaType {
     'data' = "data",

package/dist/typescript-client/src/ai-assistant-client.d.ts

@@ -7,6 +7,7 @@ export declare class AiAssistantClient {
     private readonly socketManager;
     private readonly integrationId;
     private readonly ongoingChatRequests;
+    private readonly ongoingChatSequences;
     constructor(rpcManager: RpcManager, socketManager: SocketManager, integrationId: IntegrationId);
     /**
      * Retrieves a profile reference for the provided id. A profile reference
@@ -64,10 +65,12 @@ declare class AiAssistantProfileReference {
      *
      * @param data An object containing options for creating the profile.
      * @param data.modelName - The name of the OpenAI model (`gpt-3.5 or `gpt-4`).
+     * @param data.isPublic - Whether the chat functionality of the profile can be accessed without security rules.
      * @returns A promise that resolves when the profile is successfully created.
      */
     insert(data: {
         modelName: OpenAiModelName;
+        isPublic: boolean;
     }): Promise<void>;
     /**
      * Updates an existing assistant profile. This will result in an error if a profile has not yet been created for the
@@ -75,10 +78,12 @@ declare class AiAssistantProfileReference {
      *
      * @param data An object containing options for updating the profile.
      * @param data.modelName - The name of the OpenAI model (`gpt-3.5 or `gpt-4`).
+     * @param data.isPublic - Whether the chat functionality of the profile can be accessed without security rules.
      * @returns A promise that resolves when the profile is successfully updated.
      */
     update(data: {
-        modelName: OpenAiModelName;
+        modelName?: OpenAiModelName;
+        isPublic?: boolean;
     }): Promise<void>;
     /**
      * Deletes an existing assistant profile. This will result in an error if a profile has not yet been created for the

package/package.json

@@ -1,9 +1,8 @@
 {
   "name": "@squidcloud/client",
-  "version": "1.0.106",
+  "version": "1.0.108",
   "description": "A typescript implementation of the Squid client",
   "main": "dist/cjs/index.js",
-  "module": "dist/esm/index.js",
   "types": "dist/typescript-client/src/index.d.ts",
   "files": [
     "dist/**/*"