feat(fai-chat): add support for conditoinal query decomposition per enablement (#5941)

Author: sahil485

fern/apis/fai/openapi.json

@@ -3728,10 +3728,10 @@
             "required": true,
             "schema": {
               "type": "string",
-              "description": "The GitHub repository to install the Fern Writer for. Must have the `fern-api` bot installed. Must be in the format `owner/repo`.Example: `fern-api/docs`",
+              "description": "The GitHub repository to install the Fern Writer for. Must have the `fern-api` bot installed. Must be in the format `owner/repo`. Example: `fern-api/docs`",
               "title": "Github Repo"
             },
-            "description": "The GitHub repository to install the Fern Writer for. Must have the `fern-api` bot installed. Must be in the format `owner/repo`.Example: `fern-api/docs`"
+            "description": "The GitHub repository to install the Fern Writer for. Must have the `fern-api` bot installed. Must be in the format `owner/repo`. Example: `fern-api/docs`"
           }
         ],
         "responses": {
@@ -4012,51 +4012,6 @@
         ]
       }
     },
-    "/upstash/qstash/failure-callback": {
-      "post": {
-        "tags": [
-          "Upstash"
-        ],
-        "summary": "Qstash Failure Callback",
-        "description": "Callback endpoint for Upstash QStash to notify about failed requests.\nLogs the failure details and posts to Slack #search-notifs channel.",
-        "operationId": "qstash_failure_callback",
-        "requestBody": {
-          "content": {
-            "application/json": {
-              "schema": {
-                "$ref": "#/components/schemas/QStashFailureCallback"
-              }
-            }
-          },
-          "required": true
-        },
-        "responses": {
-          "200": {
-            "description": "Successful Response",
-            "content": {
-              "application/json": {
-                "schema": {
-                  "$ref": "#/components/schemas/QStashFailureCallbackResponse"
-                }
-              }
-            }
-          },
-          "422": {
-            "description": "Validation Error",
-            "content": {
-              "application/json": {
-                "schema": {
-                  "$ref": "#/components/schemas/HTTPValidationError"
-                }
-              }
-            }
-          }
-        },
-        "x-fern-audiences": [
-          "internal"
-        ]
-      }
-    },
     "/sources/website/{domain}/index": {
       "post": {
         "tags": [
@@ -6665,59 +6620,6 @@
         ],
         "title": "PostChatCompletionResponse"
       },
-      "QStashFailureCallback": {
-        "properties": {
-          "dlqId": {
-            "type": "string",
-            "title": "Dlqid"
-          },
-          "url": {
-            "type": "string",
-            "title": "Url"
-          },
-          "status": {
-            "type": "integer",
-            "title": "Status"
-          },
-          "sourceHeader": {
-            "additionalProperties": {
-              "anyOf": [
-                {
-                  "type": "string"
-                },
-                {
-                  "items": {
-                    "type": "string"
-                  },
-                  "type": "array"
-                }
-              ]
-            },
-            "type": "object",
-            "title": "Sourceheader"
-          }
-        },
-        "type": "object",
-        "required": [
-          "dlqId",
-          "url",
-          "status"
-        ],
-        "title": "QStashFailureCallback"
-      },
-      "QStashFailureCallbackResponse": {
-        "properties": {
-          "success": {
-            "type": "boolean",
-            "title": "Success"
-          }
-        },
-        "type": "object",
-        "required": [
-          "success"
-        ],
-        "title": "QStashFailureCallbackResponse"
-      },
       "Query": {
         "properties": {
           "query_id": {
@@ -6759,6 +6661,20 @@
               }
             ],
             "title": "Time To First Token"
+          },
+          "subqueries": {
+            "anyOf": [
+              {
+                "items": {
+                  "type": "string"
+                },
+                "type": "array"
+              },
+              {
+                "type": "null"
+              }
+            ],
+            "title": "Subqueries"
           }
         },
         "type": "object",

packages/fai-sdk/src/Client.ts

@@ -24,7 +24,6 @@ import { Settings } from "./api/resources/settings/client/Client.js";
 import { SlackAskFern } from "./api/resources/slackAskFern/client/Client.js";
 import { SlackScribe } from "./api/resources/slackScribe/client/Client.js";
 import { Sources } from "./api/resources/sources/client/Client.js";
-import { Upstash } from "./api/resources/upstash/client/Client.js";
 import { Website } from "./api/resources/website/client/Client.js";
 
 export declare namespace FernAIClient {
@@ -72,7 +71,6 @@ export class FernAIClient {
     protected _slackAskFern: SlackAskFern | undefined;
     protected _slackScribe: SlackScribe | undefined;
     protected _sources: Sources | undefined;
-    protected _upstash: Upstash | undefined;
     protected _website: Website | undefined;
 
     constructor(_options: FernAIClient.Options = {}) {
@@ -165,10 +163,6 @@ export class FernAIClient {
         return (this._sources ??= new Sources(this._options));
     }
 
-    public get upstash(): Upstash {
-        return (this._upstash ??= new Upstash(this._options));
-    }
-
     public get website(): Website {
         return (this._website ??= new Website(this._options));
     }

packages/fai-sdk/src/api/resources/index.ts

@@ -18,7 +18,6 @@ export * as sdks from "./sdks/index.js";
 export * as slackAskFern from "./slackAskFern/index.js";
 export * as slackScribe from "./slackScribe/index.js";
 export * as sources from "./sources/index.js";
-export * as upstash from "./upstash/index.js";
 export * as website from "./website/index.js";
 export * from "./analytics/client/requests/index.js";
 export * from "./chat/client/requests/index.js";
@@ -37,5 +36,4 @@ export * from "./settings/client/requests/index.js";
 export * from "./slackAskFern/client/requests/index.js";
 export * from "./slackScribe/client/requests/index.js";
 export * from "./sources/client/requests/index.js";
-export * from "./upstash/client/requests/index.js";
 export * from "./website/client/requests/index.js";

packages/fai-sdk/src/api/resources/settings/client/Client.ts

@@ -483,13 +483,6 @@ export class Settings {
     /**
      * Manually trigger reindex for an already enabled Ask AI setup.
      *
-     * Args:
-     *     domain: The domain to reindex
-     *     org_name: Organization name (unused, kept for backwards compatibility)
-     *     incremental: If True, only reindex changed content based on content hashes
-     *     db: Database session
-     *     _: Token verification
-     *
      * @param {FernAI.ReindexAskAiRequest} request
      * @param {Settings.RequestOptions} requestOptions - Request-specific configuration.
      *
@@ -498,8 +491,7 @@ export class Settings {
      * @example
      *     await client.settings.reindexAskAi({
      *         domain: "domain",
-     *         org_name: "org_name",
-     *         incremental: true
+     *         org_name: "org_name"
      *     })
      */
     public reindexAskAi(
@@ -513,17 +505,13 @@ export class Settings {
         request: FernAI.ReindexAskAiRequest,
         requestOptions?: Settings.RequestOptions,
     ): Promise<core.WithRawResponse<FernAI.ToggleAskAiResponse>> {
-        const { domain, org_name: orgName, incremental } = request;
+        const { domain, org_name: orgName } = request;
         const _queryParams: Record<string, string | string[] | object | object[] | null> = {};
         _queryParams["domain"] = domain;
         if (orgName != null) {
             _queryParams["org_name"] = orgName;
         }
 
-        if (incremental != null) {
-            _queryParams["incremental"] = incremental.toString();
-        }
-
         let _headers: core.Fetcher.Args["headers"] = mergeHeaders(
             this._options?.headers,
             mergeOnlyDefinedHeaders({ Authorization: await this._getAuthorizationHeader() }),
@@ -665,92 +653,6 @@ export class Settings {
         }
     }
 
-    /**
-     * Handle callback from Upstash QStash when preview reindex completes.
-     *
-     * @param {FernAI.UpstashCallbackRequest} request
-     * @param {Settings.RequestOptions} requestOptions - Request-specific configuration.
-     *
-     * @throws {@link FernAI.UnprocessableEntityError}
-     *
-     * @example
-     *     await client.settings.reindexPreviewCallback({
-     *         status: 1,
-     *         sourceMessageId: "sourceMessageId"
-     *     })
-     */
-    public reindexPreviewCallback(
-        request: FernAI.UpstashCallbackRequest,
-        requestOptions?: Settings.RequestOptions,
-    ): core.HttpResponsePromise<unknown> {
-        return core.HttpResponsePromise.fromPromise(this.__reindexPreviewCallback(request, requestOptions));
-    }
-
-    private async __reindexPreviewCallback(
-        request: FernAI.UpstashCallbackRequest,
-        requestOptions?: Settings.RequestOptions,
-    ): Promise<core.WithRawResponse<unknown>> {
-        let _headers: core.Fetcher.Args["headers"] = mergeHeaders(
-            this._options?.headers,
-            mergeOnlyDefinedHeaders({ Authorization: await this._getAuthorizationHeader() }),
-            requestOptions?.headers,
-        );
-        const _response = await core.fetcher({
-            url: core.url.join(
-                (await core.Supplier.get(this._options.baseUrl)) ??
-                    (await core.Supplier.get(this._options.environment)) ??
-                    environments.FernAIEnvironment.Production,
-                "settings/reindex-preview-callback",
-            ),
-            method: "POST",
-            headers: _headers,
-            contentType: "application/json",
-            queryParameters: requestOptions?.queryParams,
-            requestType: "json",
-            body: request,
-            timeoutMs: requestOptions?.timeoutInSeconds != null ? requestOptions.timeoutInSeconds * 1000 : 60000,
-            maxRetries: requestOptions?.maxRetries,
-            abortSignal: requestOptions?.abortSignal,
-        });
-        if (_response.ok) {
-            return { data: _response.body, rawResponse: _response.rawResponse };
-        }
-
-        if (_response.error.reason === "status-code") {
-            switch (_response.error.statusCode) {
-                case 422:
-                    throw new FernAI.UnprocessableEntityError(
-                        _response.error.body as FernAI.HttpValidationError,
-                        _response.rawResponse,
-                    );
-                default:
-                    throw new errors.FernAIError({
-                        statusCode: _response.error.statusCode,
-                        body: _response.error.body,
-                        rawResponse: _response.rawResponse,
-                    });
-            }
-        }
-
-        switch (_response.error.reason) {
-            case "non-json":
-                throw new errors.FernAIError({
-                    statusCode: _response.error.statusCode,
-                    body: _response.error.rawBody,
-                    rawResponse: _response.rawResponse,
-                });
-            case "timeout":
-                throw new errors.FernAITimeoutError(
-                    "Timeout exceeded when calling POST /settings/reindex-preview-callback.",
-                );
-            case "unknown":
-                throw new errors.FernAIError({
-                    message: _response.error.errorMessage,
-                    rawResponse: _response.rawResponse,
-                });
-        }
-    }
-
     /**
      * Set the job_id for a domain when reindex starts processing.
      *

packages/fai-sdk/src/api/resources/settings/client/requests/ReindexAskAiRequest.ts

@@ -6,12 +6,10 @@
  * @example
  *     {
  *         domain: "domain",
- *         org_name: "org_name",
- *         incremental: true
+ *         org_name: "org_name"
  *     }
  */
 export interface ReindexAskAiRequest {
     domain: string;
     org_name?: string;
-    incremental?: boolean;
 }

packages/fai-sdk/src/api/resources/settings/client/requests/UpstashCallbackRequest.ts

@@ -5,6 +5,5 @@ export { type EnableAskAiRequest } from "./EnableAskAiRequest.js";
 export { type ToggleAskAiRequest } from "./ToggleAskAiRequest.js";
 export { type ReindexAskAiRequest } from "./ReindexAskAiRequest.js";
 export { type GetToggleStatusRequest } from "./GetToggleStatusRequest.js";
-export { type UpstashCallbackRequest } from "./UpstashCallbackRequest.js";
 export { type SetJobIdRequest } from "./SetJobIdRequest.js";
 export { type ReindexCallbackRequest } from "./ReindexCallbackRequest.js";