@squidcloud/client 1.0.197 → 1.0.198

package/dist/cjs/index.js

@@ -9485,6 +9485,7 @@ __webpack_require__.r(__webpack_exports__);
 
 // EXPORTS
 __webpack_require__.d(__webpack_exports__, {
+  AI_IMAGE_GENERATION_MODEL_NAMES: () => (/* reexport */ AI_IMAGE_GENERATION_MODEL_NAMES),
   AI_MODEL_NAMES: () => (/* reexport */ AI_MODEL_NAMES),
   ALL_OPERATORS: () => (/* reexport */ ALL_OPERATORS),
   AiAssistantClient: () => (/* reexport */ AiAssistantClient),
@@ -9496,6 +9497,7 @@ __webpack_require__.d(__webpack_exports__, {
   AiChatbotMutationContext: () => (/* reexport */ AiChatbotMutationContext),
   AiChatbotProfileReference: () => (/* reexport */ AiChatbotProfileReference),
   AiClient: () => (/* reexport */ AiClient),
+  AiImageClient: () => (/* reexport */ AiImageClient),
   ApiCallContext: () => (/* reexport */ ApiCallContext),
   ApiClient: () => (/* reexport */ ApiClient),
   ApiKeysSecretClient: () => (/* reexport */ ApiKeysSecretClient),
@@ -11027,8 +11029,26 @@ class AiChatbotClientFactory {
     }
 }
 
+;// CONCATENATED MODULE: ./src/ai-image-client.ts
+class AiImageClient {
+    /** @internal */
+    constructor(rpcManager, integrationId) {
+        this.rpcManager = rpcManager;
+        this.integrationId = integrationId;
+    }
+    async generate(prompt, options) {
+        const request = {
+            prompt,
+            options,
+            integrationId: this.integrationId,
+        };
+        return this.rpcManager.post('ai/image/generate', request);
+    }
+}
+
 ;// CONCATENATED MODULE: ./src/ai.types.ts
 
+
 /**
  * AiClient class serves as a facade for interacting with different AI services.
  * It provides simplified access to AI chatbot and assistant functionalities
@@ -11056,6 +11076,13 @@ class AiClient {
     assistant() {
         return this.aiAssistantClient;
     }
+    /**
+     * Retrieves an AI image client for a specific AI integration.
+     * @param integrationId - The identifier for the AI integration.
+     */
+    image(integrationId) {
+        return new AiImageClient(this.rpcManager, integrationId);
+    }
     /**
      * Executes an AI query using a specific DB integration, sending a prompt to the AI and returning its response.
      * This function allows for direct interaction with the AI's capabilities by sending text prompts and receiving
@@ -11768,6 +11795,8 @@ const AI_MODEL_NAMES = [
     'claude-3-haiku-20240307',
     'gpt-4-turbo-preview',
 ];
+/** The supported AI image generation model names. */
+const AI_IMAGE_GENERATION_MODEL_NAMES = ['dall-e-3'];
 
 ;// CONCATENATED MODULE: ../internal-common/src/public-types/api-call.public-context.ts
 /** The context of an API call. */
@@ -15433,7 +15462,12 @@ class DistributedLockManager {
             lock.release();
         });
     }
-    async lock(mutex) {
+    /**
+     * Acquires a lock.
+     * @param mutex The mutex to lock.
+     * @param timeoutMillis The maximum time to wait for the lock to be acquired on the server.
+     */
+    async lock(mutex, timeoutMillis = 1000) {
         /**
          * Wait up to 2 seconds for the connection to be ready and if it is not, consider the client to be not connected.
          * This is useful because right after the Squid client is created, the socket is still not connected but the user
@@ -15449,11 +15483,12 @@ class DistributedLockManager {
             type: 'acquireLock',
             payload: {
                 mutex,
+                timeoutMillis,
                 clientRequestId,
             },
         };
         this.socketManager.sendMessage(acquireLockMessage);
-        const result = await (0,external_rxjs_.firstValueFrom)((0,external_rxjs_.race)((0,external_rxjs_.timer)(5000).pipe((0,external_rxjs_.take)(1), map(() => {
+        const result = await (0,external_rxjs_.firstValueFrom)((0,external_rxjs_.race)((0,external_rxjs_.timer)(timeoutMillis + 4000).pipe((0,external_rxjs_.take)(1), map(() => {
             return {
                 payload: {
                     error: 'TIMEOUT_GETTING_LOCK',
@@ -34124,6 +34159,7 @@ Squid.squidInstancesMap = {};
 
 
 
+
 
 
 })();

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

@@ -1,6 +1,19 @@
 /** The supported AI model names. */
 export declare const AI_MODEL_NAMES: readonly ["gpt-3.5-turbo", "gpt-3.5-turbo-1106", "gpt-4", "claude-3-opus-20240229", "claude-3-sonnet-20240229", "claude-3-haiku-20240307", "gpt-4-turbo-preview"];
+/** The supported AI image generation model names. */
+export declare const AI_IMAGE_GENERATION_MODEL_NAMES: readonly ["dall-e-3"];
 export type AiModelName = (typeof AI_MODEL_NAMES)[number];
+export type AiImageGenerationModelName = (typeof AI_IMAGE_GENERATION_MODEL_NAMES)[number];
+export type AiGenerateImageOptions = DallEOptions;
+export interface BaseAiGenerateImageOptions {
+    modelName: AiImageGenerationModelName;
+}
+export interface DallEOptions extends BaseAiGenerateImageOptions {
+    modelName: 'dall-e-3';
+    quality?: 'hd' | 'standard';
+    size?: '256x256' | '512x512' | '1024x1024' | '1792x1024' | '1024x1792';
+    numberOfImagesToGenerate?: 1;
+}
 /** The possible sources for the LLM provider API key. */
 export type ApiKeySource = 'user' | 'system';
 export type OpenAiResponseFormat = 'text' | 'json_object';

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

@@ -0,0 +1,6 @@
+import { AiGenerateImageOptions } from './public-types';
+export declare class AiImageClient {
+    private readonly rpcManager;
+    private readonly integrationId;
+    generate(prompt: string, options: AiGenerateImageOptions): Promise<string>;
+}

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

@@ -1,6 +1,7 @@
 import { AiChatbotClient } from './ai-chatbot-client';
 import { AiAssistantClient } from './ai-assistant-client';
 import { IntegrationId } from './public-types';
+import { AiImageClient } from './ai-image-client';
 export interface ExecuteAiQueryRequest {
     integrationId: IntegrationId;
     prompt: string;
@@ -32,6 +33,11 @@ export declare class AiClient {
      * @returns An instance of AiAssistantClient.
      */
     assistant(): AiAssistantClient;
+    /**
+     * Retrieves an AI image client for a specific AI integration.
+     * @param integrationId - The identifier for the AI integration.
+     */
+    image(integrationId: IntegrationId): AiImageClient;
     /**
      * Executes an AI query using a specific DB integration, sending a prompt to the AI and returning its response.
      * This function allows for direct interaction with the AI's capabilities by sending text prompts and receiving

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

@@ -1,6 +1,7 @@
 export * from './ai-assistant-client';
 export * from './ai-chatbot-client.factory';
 export * from './ai-chatbot-client';
+export * from './ai-image-client';
 export * from './ai.types';
 export * from './api-client';
 export * from './api.manager';

package/package.json

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