feat: add decompose_queries field to get_docs_settings (#5947)

Author: sahil485

fern/apis/fai/openapi.json

@@ -5926,6 +5926,18 @@
             ],
             "title": "Discord Enabled",
             "description": "Whether Ask AI is enabled for discord"
+          },
+          "decompose_queries": {
+            "anyOf": [
+              {
+                "type": "boolean"
+              },
+              {
+                "type": "null"
+              }
+            ],
+            "title": "Decompose Queries",
+            "description": "Whether query decomposition is enabled"
           }
         },
         "type": "object",

packages/fai-sdk/src/api/types/GetSettingsResponse.ts

@@ -13,4 +13,6 @@ export interface GetSettingsResponse {
     slack_enabled?: boolean;
     /** Whether Ask AI is enabled for discord */
     discord_enabled?: boolean;
+    /** Whether query decomposition is enabled */
+    decompose_queries?: boolean;
 }

servers/fai/alembic/versions/add_decompose_queries_to_settings.py

@@ -0,0 +1,28 @@
+"""add decompose_queries column to settings table
+
+Revision ID: add_decompose_queries
+Revises: add_subqueries_to_queries
+Create Date: 2025-12-15 00:00:00.000000
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'add_decompose_queries'
+down_revision: Union[str, Sequence[str], None] = 'add_subqueries_to_queries'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    op.add_column('settings', sa.Column('decompose_queries', sa.Boolean(), nullable=False, server_default='false'))
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    op.drop_column('settings', 'decompose_queries')

servers/fai/src/fai/models/api/settings_api.py

@@ -22,6 +22,7 @@ class GetSettingsResponse(BaseModel):
     docs_enabled: bool | None = Field(None, description="Whether Ask AI is enabled for docs")
     slack_enabled: bool | None = Field(None, description="Whether Ask AI is enabled for slack")
     discord_enabled: bool | None = Field(None, description="Whether Ask AI is enabled for discord")
+    decompose_queries: bool | None = Field(None, description="Whether query decomposition is enabled")
 
 
 class ToggleAskAiResponse(BaseModel):

servers/fai/src/fai/models/db/settings_db.py

@@ -28,6 +28,7 @@ class SettingsDb(Base):
     slack_enabled = Column(Boolean, nullable=False, default=True)
     discord_enabled = Column(Boolean, nullable=False, default=True)
     is_preview = Column(Boolean, nullable=False, default=False)
+    decompose_queries = Column(Boolean, nullable=False, default=False)
 
     def to_api(self) -> Settings:
         return Settings(
@@ -40,4 +41,5 @@ def to_api(self) -> Settings:
             slack_enabled=self.slack_enabled,
             discord_enabled=self.discord_enabled,
             is_preview=self.is_preview,
+            decompose_queries=self.decompose_queries,
         )

servers/fai/src/fai/models/types/settings_types.py

@@ -11,3 +11,4 @@ class Settings(BaseModel):
     docs_enabled: bool = True
     slack_enabled: bool = True
     discord_enabled: bool = True
+    decompose_queries: bool = False

servers/fai/src/fai/routes/settings.py

@@ -78,10 +78,18 @@ async def get_docs_settings(
         job_id = existing_record.job_id if existing_record else None
 
         return GetSettingsResponse(
-            ask_ai_enabled=ask_ai_enabled, job_id=job_id, docs_enabled=existing_record.docs_enabled
+            ask_ai_enabled=ask_ai_enabled,
+            job_id=job_id,
+            docs_enabled=existing_record.docs_enabled,
+            decompose_queries=existing_record.decompose_queries,
         )
     except Exception:
-        return GetSettingsResponse(ask_ai_enabled=False, job_id=None, docs_enabled=existing_record.docs_enabled)
+        return GetSettingsResponse(
+            ask_ai_enabled=False,
+            job_id=None,
+            docs_enabled=existing_record.docs_enabled if existing_record else None,
+            decompose_queries=existing_record.decompose_queries if existing_record else None,
+        )
 
 
 @fai_app.get(