@squidcloud/client 1.0.150 → 1.0.152

package/dist/cjs/index.js

@@ -35514,7 +35514,13 @@ const AiChatbotChatRequestSchema = {
 
 ;// CONCATENATED MODULE: ../common/src/ai-chatbot.types.ts
 /** The supported AI model names. */
-const AI_MODEL_NAMES = (/* unused pure expression or super */ null && (['gpt-3.5-turbo', 'gpt-4', 'claude-2', 'gpt-4-1106-preview']));
+const AI_MODEL_NAMES = (/* unused pure expression or super */ null && ([
+    'gpt-3.5-turbo',
+    'gpt-3.5-turbo-1106',
+    'gpt-4',
+    'claude-2',
+    'gpt-4-1106-preview',
+]));
 /** @internal */
 const AiModelData = {
     'gpt-4': {
@@ -35523,6 +35529,9 @@ const AiModelData = {
     'gpt-3.5-turbo': {
         tokens: 4096,
     },
+    'gpt-3.5-turbo-1106': {
+        tokens: 16385,
+    },
     'claude-2': {
         tokens: 100000,
     },
@@ -41749,10 +41758,15 @@ class AuthManager {
     async getAuthData() {
         var _a;
         return {
-            token: this.authProvider && (await this.authProvider.getToken()),
+            token: await this.getTokenFromAuthProvider(),
             integrationId: (_a = this.authProvider) === null || _a === void 0 ? void 0 : _a.integrationId,
         };
     }
+    async getTokenFromAuthProvider() {
+        var _a;
+        const promiseOrPrimitive = (_a = this.authProvider) === null || _a === void 0 ? void 0 : _a.getToken();
+        return typeof promiseOrPrimitive === 'object' ? await promiseOrPrimitive : promiseOrPrimitive;
+    }
     getApiKey() {
         return this.apiKey;
     }
@@ -41766,7 +41780,7 @@ class AuthManager {
         if (this.apiKey) {
             return { type: 'ApiKey', token: this.apiKey };
         }
-        const token = this.authProvider && (await this.authProvider.getToken());
+        const token = await this.getTokenFromAuthProvider();
         if (!token)
             return undefined;
         return { type: 'Bearer', token, integrationId: (_a = this.authProvider) === null || _a === void 0 ? void 0 : _a.integrationId };
@@ -57645,6 +57659,14 @@ class QuerySender {
     }
     async processQueryBatch() {
         const queryRequestAndSubjects = this.pendingQueryRequests.splice(0);
+        /**
+         * There is a race condition where the batch timeout is triggered but the safeToSendQueriesToServer is not yet
+         * true. In this case this function will be invoked again once the safeToSendQueriesToServer is true within the
+         * setTimeout handler, and it may cause the queryRequestAndSubjects to be empty. In this case we do not want to send
+         * an empty batch.
+         */
+        if (!queryRequestAndSubjects.length)
+            return;
         const pendingQueryRequests = queryRequestAndSubjects.map(({ queryRequest }) => queryRequest);
         const responseSubjects = queryRequestAndSubjects.map(({ responseSubject }) => responseSubject);
         this.inflightQueriesCount.next(this.inflightQueriesCount.value + pendingQueryRequests.length);

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

@@ -1,5 +1,5 @@
 /** The supported AI model names. */
-export declare const AI_MODEL_NAMES: readonly ["gpt-3.5-turbo", "gpt-4", "claude-2", "gpt-4-1106-preview"];
+export declare const AI_MODEL_NAMES: readonly ["gpt-3.5-turbo", "gpt-3.5-turbo-1106", "gpt-4", "claude-2", "gpt-4-1106-preview"];
 export type AiModelName = (typeof AI_MODEL_NAMES)[number];
 /** The possible sources for the LLM provider API key. */
 export type ApiKeySource = 'user' | 'system';
@@ -18,10 +18,12 @@ export interface AiChatbotFileContext extends AiChatbotContextBase {
     type: 'file';
 }
 export type AiChatbotContext = AiChatbotTextContext | AiChatbotUrlContext | AiChatbotFileContext;
+export type OpenAiResponseFormat = 'text' | 'json_object';
 export interface AiChatbotChatOptions {
     maxTokens?: number;
     chatId?: string;
     disableHistory?: boolean;
     includeReference?: boolean;
+    responseFormat?: OpenAiResponseFormat;
 }
 export {};

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

@@ -11,6 +11,8 @@ export interface ExecuteAiQueryResponse {
     answer: string;
     explanation?: string;
     executedQuery?: string;
+    queryMarkdownType?: string;
+    success: boolean;
 }
 /**
  * AiClient class serves as a facade for interacting with different AI services.

package/dist/typescript-client/src/auth.manager.d.ts

@@ -16,6 +16,7 @@ export declare class AuthManager {
      */
     setAuthProvider(authProvider: SquidAuthProvider): void;
     getAuthData(): Promise<AuthData>;
+    private getTokenFromAuthProvider;
     getApiKey(): ApiKey | undefined;
     /**
      * Returns a valid AuthToken.

package/dist/typescript-client/src/squid.d.ts

@@ -56,10 +56,10 @@ export interface SquidOptions {
 /** Authentication data provider for Squid requests. */
 export interface SquidAuthProvider {
     /**
-     * Returns a valid AccessToken.
+     * Returns a valid AccessToken or undefined if there is no active authorized session.
      * Called by Squid every time a Squid client makes requests to the Squid backend.
      */
-    getToken(): Promise<string | undefined>;
+    getToken(): Promise<string | undefined> | string | undefined;
     /**
      * Optional Auth integration id.
      * Sent as a part of all Squid requests to the backend.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@squidcloud/client",
-  "version": "1.0.150",
+  "version": "1.0.152",
   "description": "A typescript implementation of the Squid client",
   "main": "dist/cjs/index.js",
   "types": "dist/typescript-client/src/index.d.ts",
@@ -33,7 +33,7 @@
   "license": "ISC",
   "dependencies": {
     "@apollo/client": "^3.7.4",
-    "@squidcloud/common": "1.0.150",
+    "@squidcloud/common": "1.0.152",
     "@supercharge/promise-pool": "^2.3.2",
     "axios": "^1.6.2",
     "cross-fetch": "^3.1.5",