@squidcloud/client 1.0.124 → 1.0.126

package/dist/cjs/index.js

@@ -26388,7 +26388,7 @@ const AiAssistantMutateRequestSchema = {
                     properties: {
                         modelName: {
                             type: 'string',
-                            enum: ['gpt-3.5-turbo', 'gpt-4'],
+                            enum: ['gpt-3.5-turbo', 'gpt-4', 'claude-2'],
                         },
                         strictContext: {
                             type: 'boolean',
@@ -26442,7 +26442,7 @@ const AiAssistantMutateRequestSchema = {
                     properties: {
                         modelName: {
                             type: 'string',
-                            enum: ['gpt-3.5-turbo', 'gpt-4'],
+                            enum: ['gpt-3.5-turbo', 'gpt-4', 'claude-2'],
                         },
                         strictContext: {
                             type: 'boolean',
@@ -26531,13 +26531,16 @@ const AiAssistantMutateRequestSchema = {
 
 ;// CONCATENATED MODULE: ../common/src/ai-assistant.types.ts
 /** @internal */
-const OpenAiModelData = {
+const AiModelData = {
     'gpt-4': {
         tokens: 8192,
     },
     'gpt-3.5-turbo': {
         tokens: 4096,
     },
+    'claude-2': {
+        tokens: 100000,
+    },
 };
 
 ;// CONCATENATED MODULE: ../common/src/integrations/observability.types.ts
@@ -26613,6 +26616,7 @@ var IntegrationType;
     IntegrationType["db2"] = "db2";
     IntegrationType["mariadb"] = "mariadb";
     IntegrationType["oracledb"] = "oracledb";
+    IntegrationType["pinot"] = "pinot";
     IntegrationType["redis"] = "redis";
     IntegrationType["xata"] = "xata";
     IntegrationType["azure_sql"] = "azure_sql";
@@ -26634,6 +26638,7 @@ const DatabaseIntegrationTypes = [
     IntegrationType.cockroach,
     IntegrationType.snowflake,
     IntegrationType.oracledb,
+    IntegrationType.pinot,
 ];
 const ApiIntegrationTypes = [IntegrationType.api, IntegrationType.graphql];
 const ObservabilityIntegrationTypes = [IntegrationType.datadog, IntegrationType.newrelic];
@@ -26713,6 +26718,25 @@ const OracleConnectionOptionsSchema = {
         connectionLimit: { type: 'number', nullable: false },
     },
 };
+const PinotConnectionOptionsSchema = {
+    type: 'object',
+    nullable: false,
+    required: ['brokerUrl', 'controllerUrl'],
+    properties: {
+        brokerUrl: { type: 'string', nullable: false },
+        controllerUrl: { type: 'string', nullable: false },
+        user: { type: 'string', nullable: false },
+        secrets: {
+            type: 'object',
+            required: ['password'],
+            nullable: false,
+            properties: {
+                password: { type: 'string', nullable: false, isSecret: {} },
+            },
+        },
+        connectionLimit: { type: 'number', nullable: false },
+    },
+};
 const MssqlConnectionOptionsSchema = {
     type: 'object',
     nullable: false,
@@ -26940,6 +26964,27 @@ const UpsertIntegrationRequestSchema = {
                 },
             },
         },
+        {
+            type: 'object',
+            properties: {
+                type: { const: IntegrationType.pinot },
+                config: {
+                    type: 'object',
+                    required: ['configuration'],
+                    properties: {
+                        supportsExternalChanges: { type: 'boolean', nullable: true },
+                        configuration: {
+                            type: 'object',
+                            nullable: false,
+                            required: ['connectionOptions'],
+                            properties: {
+                                connectionOptions: PinotConnectionOptionsSchema,
+                            },
+                        },
+                    },
+                },
+            },
+        },
         {
             type: 'object',
             properties: {
@@ -27327,6 +27372,20 @@ const TestDataConnectionRequestSchema = {
                 },
             },
         },
+        {
+            type: 'object',
+            required: ['configuration'],
+            properties: {
+                type: { const: IntegrationType.pinot },
+                configuration: {
+                    type: 'object',
+                    required: ['connectionOptions'],
+                    properties: {
+                        connectionOptions: PinotConnectionOptionsSchema,
+                    },
+                },
+            },
+        },
         {
             type: 'object',
             required: ['configuration'],
@@ -27434,6 +27493,14 @@ const DiscoverDataConnectionSchemaRequestSchema = {
                 connectionOptions: OracleConnectionOptionsSchema,
             },
         },
+        {
+            type: 'object',
+            required: ['connectionOptions'],
+            properties: {
+                integrationType: { const: IntegrationType.pinot },
+                connectionOptions: PinotConnectionOptionsSchema,
+            },
+        },
         {
             type: 'object',
             required: ['connectionOptions'],
@@ -32287,7 +32354,7 @@ class AiAssistantClient {
             .observeNotifications()
             .pipe((0,external_rxjs_namespaceObject.filter)((notification) => notification.type === 'aiAssistant'), map_map((n) => n))
             .subscribe((notification) => {
-            this.handleChatResponse(notification);
+            this.handleChatResponse(notification).then();
         });
     }
     /**
@@ -32310,10 +32377,11 @@ class AiAssistantClient {
      *
      * @param profileId - The profile id.
      * @param prompt - The prompt.
+     * @param options - The options to send to the chat model.
      * @returns An observable that emits when a new response token is received. The emitted value is the entire response
      * that has been received so far.
      */
-    chat(profileId, prompt) {
+    chat(profileId, prompt, options) {
         const clientRequestId = generateId();
         let accumulatedValue = '';
         const subject = new external_rxjs_namespaceObject.Subject();
@@ -32341,6 +32409,7 @@ class AiAssistantClient {
         const request = {
             profileId,
             prompt,
+            options,
             integrationId: this.integrationId,
             clientRequestId,
         };
@@ -32378,11 +32447,12 @@ class AiAssistantProfileReference {
      * Sends a prompt to the current profile.
      *
      * @param prompt - The prompt.
+     * @param options - The chat options.
      * @returns An observable that emits when a new response token is received. The emitted value is the entire response
      * that has been received so far.
      */
-    chat(prompt) {
-        return this.client.chat(this.profileId, prompt);
+    chat(prompt, options) {
+        return this.client.chat(this.profileId, prompt, options);
     }
     /**
      * Retrieves a context reference for the current profile. A context reference can be used to add a new context entry
@@ -32410,7 +32480,7 @@ class AiAssistantProfileReference {
      * Adds a new profile to the assistant. This will result in an error if a profile already exists with the same id.
      *
      * @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.modelName - The name of the OpenAI model (`gpt-3.5, `gpt-4` or `claude-2`).
      * @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.
      */
@@ -32435,7 +32505,7 @@ class AiAssistantProfileReference {
      * current profile id.
      *
      * @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.modelName - The name of the OpenAI model (`gpt-3.5, `gpt-4` or `claude-2`).
      * @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.
      */
@@ -32485,6 +32555,7 @@ class AiAssistantContextReference {
      * @param data An object containing options for creating the entry.
      * @param data.title - The title of the entry.
      * @param data.context - The context data.
+     * @param file - The file to insert.
      * @returns A promise that resolves when the context is successfully created.
      */
     insert(data, file) {

package/dist/common/src/ai-assistant.types.d.ts

@@ -1,5 +1,5 @@
-/** The supported Open AI model names. */
-export type OpenAiModelName = 'gpt-3.5-turbo' | 'gpt-4';
+/** The supported AI model names. */
+export type AiModelName = 'gpt-3.5-turbo' | 'gpt-4' | 'claude-2';
 export type AiAssistantContextType = 'text' | 'url' | 'file';
 export interface AiAssistantTextContext {
     type: 'text';
@@ -14,3 +14,8 @@ export interface AiAssistantFileContext {
     data: string;
 }
 export type AiAssistantContext = AiAssistantTextContext | AiAssistantUrlContext | AiAssistantFileContext;
+export interface AiAssistantChatOptions {
+    maxTokens?: number;
+    chatId?: string;
+    disableHistory?: boolean;
+}

package/dist/common/src/communication.types.d.ts

@@ -9,4 +9,5 @@ export type EnvironmentId = 'prod' | 'dev';
 export declare const allEnvironmentIds: Array<EnvironmentId>;
 /** A type alias for an integration id. */
 export type IntegrationId = string;
+export type ChatId = string;
 export type SquidDeveloperId = string;

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

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

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

@@ -26,6 +26,9 @@ export interface MySqlConnectionSecretOptions {
 export interface OracleConnectionSecretOptions {
     password: string;
 }
+export interface PinotConnectionSecretOptions {
+    password?: string;
+}
 export interface MssqlConnectionSecretOptions {
     password: string;
 }
@@ -48,6 +51,13 @@ export interface OracleConnectionOptions {
     database: string;
     connectionLimit?: number;
 }
+export interface PinotConnectionOptions {
+    secrets: PinotConnectionSecretOptions;
+    brokerUrl: string;
+    controllerUrl: string;
+    user?: string;
+    connectionLimit?: number;
+}
 export interface MssqlConnectionOptions {
     secrets: MssqlConnectionSecretOptions;
     host: string;
@@ -92,6 +102,9 @@ export interface MySqlIntegrationConfiguration {
 export interface OracleIntegrationConfiguration {
     connectionOptions: OracleConnectionOptions;
 }
+export interface PinotIntegrationConfiguration {
+    connectionOptions: PinotConnectionOptions;
+}
 export interface MssqlIntegrationConfiguration {
     connectionOptions: MssqlConnectionOptions;
 }
@@ -112,6 +125,10 @@ export interface OracleIntegrationConfig extends BaseDatabaseIntegrationConfig {
     type: IntegrationType.oracledb;
     configuration: OracleIntegrationConfiguration;
 }
+export interface PinotIntegrationConfig extends BaseDatabaseIntegrationConfig {
+    type: IntegrationType.pinot;
+    configuration: PinotIntegrationConfiguration;
+}
 export interface MssqlIntegrationConfig extends BaseDatabaseIntegrationConfig {
     type: IntegrationType.mssql;
     configuration: MssqlIntegrationConfiguration;

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

@@ -7,7 +7,7 @@ import { IntegrationId } from '../communication.types';
 import { AiAssistantIntegrationConfig } from './ai_assistant.types';
 import { GraphQLIntegrationConfig, HttpApiIntegrationConfig, IntegrationApiSchema, IntegrationGraphQLSchema, TestGraphQLDataConnectionRequest } from './api.types';
 import { Auth0IntegrationConfig, CognitoIntegrationConfig, JwtHmacIntegrationConfig, JwtRsaIntegrationConfig, OktaIntegrationConfig } from './auth.types';
-import { BaseDatabaseIntegrationConfig, CockroachIntegrationConfig, IntegrationDataSchema, InternalIntegrationConfig, MongoIntegrationConfig, MssqlIntegrationConfig, MySqlIntegrationConfig, OracleIntegrationConfig, PostgresIntegrationConfig, SnowflakeIntegrationConfig, TestDataConnectionRequest } from './database.types';
+import { BaseDatabaseIntegrationConfig, CockroachIntegrationConfig, IntegrationDataSchema, InternalIntegrationConfig, MongoIntegrationConfig, MssqlIntegrationConfig, MySqlIntegrationConfig, OracleIntegrationConfig, PinotIntegrationConfig, PostgresIntegrationConfig, SnowflakeIntegrationConfig, TestDataConnectionRequest } from './database.types';
 import { DatadogIntegrationConfig, NewRelicIntegrationConfig } from './observability.types';
 export declare enum IntegrationCategory {
     'database' = "database",
@@ -51,6 +51,7 @@ export declare enum IntegrationType {
     'db2' = "db2",
     'mariadb' = "mariadb",
     'oracledb' = "oracledb",
+    'pinot' = "pinot",
     'redis' = "redis",
     'xata' = "xata",
     'azure_sql' = "azure_sql",
@@ -67,6 +68,7 @@ export interface IntegrationConfigTypes {
     [IntegrationType.mongo]: MongoIntegrationConfig;
     [IntegrationType.mysql]: MySqlIntegrationConfig;
     [IntegrationType.oracledb]: OracleIntegrationConfig;
+    [IntegrationType.pinot]: PinotIntegrationConfig;
     [IntegrationType.mssql]: MssqlIntegrationConfig;
     [IntegrationType.postgres]: PostgresIntegrationConfig;
     [IntegrationType.cockroach]: CockroachIntegrationConfig;
@@ -87,6 +89,7 @@ export interface IntegrationSchemaTypes {
     [IntegrationType.mongo]: IntegrationDataSchema;
     [IntegrationType.mysql]: IntegrationDataSchema;
     [IntegrationType.oracledb]: IntegrationDataSchema;
+    [IntegrationType.pinot]: IntegrationDataSchema;
     [IntegrationType.mssql]: IntegrationDataSchema;
     [IntegrationType.postgres]: IntegrationDataSchema;
     [IntegrationType.cockroach]: IntegrationDataSchema;
@@ -98,7 +101,7 @@ export type IntegrationTypeWithConfig = keyof IntegrationConfigTypes;
 export type IntegrationSchemaKeys = keyof IntegrationSchemaTypes;
 export type IntegrationConfig = IntegrationConfigTypes[IntegrationTypeWithConfig];
 export type IntegrationSchema = IntegrationSchemaTypes[IntegrationSchemaKeys];
-export declare const DatabaseIntegrationTypes: readonly [IntegrationType.built_in_db, IntegrationType.mongo, IntegrationType.mysql, IntegrationType.mssql, IntegrationType.postgres, IntegrationType.cockroach, IntegrationType.snowflake, IntegrationType.oracledb];
+export declare const DatabaseIntegrationTypes: readonly [IntegrationType.built_in_db, IntegrationType.mongo, IntegrationType.mysql, IntegrationType.mssql, IntegrationType.postgres, IntegrationType.cockroach, IntegrationType.snowflake, IntegrationType.oracledb, IntegrationType.pinot];
 export type DatabaseIntegrationType = (typeof DatabaseIntegrationTypes)[number];
 export type DatabaseIntegrationConfig = IntegrationConfigTypes[DatabaseIntegrationType];
 export declare const ApiIntegrationTypes: readonly [IntegrationType.api, IntegrationType.graphql];

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

@@ -1,4 +1,4 @@
-import { AiAssistantContext, IntegrationId, OpenAiModelName } from '@squidcloud/common';
+import { AiAssistantChatOptions, AiAssistantContext, AiModelName, IntegrationId } from '@squidcloud/common';
 import { Observable } from 'rxjs';
 import { RpcManager } from './rpc.manager';
 import { SocketManager } from './socket.manager';
@@ -22,10 +22,11 @@ export declare class AiAssistantClient {
      *
      * @param profileId - The profile id.
      * @param prompt - The prompt.
+     * @param options - The options to send to the chat model.
      * @returns An observable that emits when a new response token is received. The emitted value is the entire response
      * that has been received so far.
      */
-    chat(profileId: string, prompt: string): Observable<string>;
+    chat(profileId: string, prompt: string, options?: AiAssistantChatOptions): Observable<string>;
     private handleChatResponse;
 }
 declare class AiAssistantProfileReference {
@@ -37,10 +38,11 @@ declare class AiAssistantProfileReference {
      * Sends a prompt to the current profile.
      *
      * @param prompt - The prompt.
+     * @param options - The chat options.
      * @returns An observable that emits when a new response token is received. The emitted value is the entire response
      * that has been received so far.
      */
-    chat(prompt: string): Observable<string>;
+    chat(prompt: string, options?: AiAssistantChatOptions): Observable<string>;
     /**
      * Retrieves a context reference for the current profile. A context reference can be used to add a new context entry
      * to the profile, or update/delete an existing entry context.
@@ -63,12 +65,12 @@ declare class AiAssistantProfileReference {
      * Adds a new profile to the assistant. This will result in an error if a profile already exists with the same id.
      *
      * @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.modelName - The name of the OpenAI model (`gpt-3.5, `gpt-4` or `claude-2`).
      * @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;
+        modelName: AiModelName;
         isPublic: boolean;
     }): Promise<void>;
     /**
@@ -76,12 +78,12 @@ declare class AiAssistantProfileReference {
      * current profile id.
      *
      * @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.modelName - The name of the OpenAI model (`gpt-3.5, `gpt-4` or `claude-2`).
      * @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?: AiModelName;
         isPublic?: boolean;
     }): Promise<void>;
     /**
@@ -105,6 +107,7 @@ declare class AiAssistantContextReference {
      * @param data An object containing options for creating the entry.
      * @param data.title - The title of the entry.
      * @param data.context - The context data.
+     * @param file - The file to insert.
      * @returns A promise that resolves when the context is successfully created.
      */
     insert(data: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@squidcloud/client",
-  "version": "1.0.124",
+  "version": "1.0.126",
   "description": "A typescript implementation of the Squid client",
   "main": "dist/cjs/index.js",
   "types": "dist/typescript-client/src/index.d.ts",