@superblocksteam/shared 0.9565.5 → 0.9566.0

package/src/org-features/index.ts

@@ -35,7 +35,8 @@ export enum OrgFeature {
   ScheduledJobs = 'ScheduledJobs',
   LegacyApps = 'LegacyApps',
   PublicApps = 'PublicApps',
-  ReactApps = 'ReactApps'
+  ReactApps = 'ReactApps',
+  Knowledge = 'Knowledge'
 }
 
 /**
@@ -68,6 +69,7 @@ export function orgHasFeature(orgVersion: OrgVersionType | undefined, feature: O
     // V1 orgs can also access React apps, but that's controlled by feature flags
     // checked separately - this function only handles org-version gating.
     case OrgFeature.ReactApps:
+    case OrgFeature.Knowledge:
       return orgVersion === OrgVersion.V2;
 
     default:

package/src/plugins/index.ts

@@ -64,6 +64,8 @@ import {
   SlackPlugin,
   SmtpPlugin,
   SnowflakePlugin,
+  SnowflakeCortexPlugin,
+  SnowflakePostgresPlugin,
   StabilityAiPlugin,
   StripePlugin,
   SuperblocksOcrPlugin,
@@ -144,6 +146,8 @@ export const CLOUD_PLUGIN_VERSIONS_LATEST: PluginExecutionVersions = {
   [SlackPlugin.id]: MAX_VERSION,
   [SmtpPlugin.id]: MAX_VERSION,
   [SnowflakePlugin.id]: MAX_VERSION,
+  [SnowflakeCortexPlugin.id]: MAX_VERSION,
+  [SnowflakePostgresPlugin.id]: MAX_VERSION,
   [StabilityAiPlugin.id]: MAX_VERSION,
   [StripePlugin.id]: MAX_VERSION,
   [SuperblocksOcrPlugin.id]: MAX_VERSION,
@@ -200,6 +204,7 @@ export const ORCHESTRATOR_PLUGIN_VERSIONS: SupportedPluginVersions = {
   [SalesforcePlugin.id]: ['0.0.2'],
   [SmtpPlugin.id]: ['0.0.1'],
   [SnowflakePlugin.id]: ['0.0.10'],
+  [SnowflakePostgresPlugin.id]: ['0.0.1'],
   [SuperblocksOcrPlugin.id]: ['0.0.1'],
   [WorkflowPlugin.id]: ['0.0.4']
 };

package/src/plugins/registry.ts

@@ -66,6 +66,8 @@ import {
   SlackPlugin,
   SmtpPlugin,
   SnowflakePlugin,
+  SnowflakeCortexPlugin,
+  SnowflakePostgresPlugin,
   StabilityAiPlugin,
   StripePlugin,
   SuperblocksOcrPlugin,
@@ -181,6 +183,8 @@ export const RegisteredPlugins = (googleSheetsClientId: string): PluginRegistry
     [SlackPlugin.id]: SlackPlugin,
     [SmtpPlugin.id]: SmtpPlugin,
     [SnowflakePlugin.id]: SnowflakePlugin,
+    [SnowflakeCortexPlugin.id]: SnowflakeCortexPlugin,
+    [SnowflakePostgresPlugin.id]: SnowflakePostgresPlugin,
     [StripePlugin.id]: StripePlugin,
     [SuperblocksOcrPlugin.id]: SuperblocksOcrPlugin,
     [TwilioPlugin.id]: TwilioPlugin,

package/src/plugins/templates/index.ts

@@ -63,6 +63,8 @@ export * from './shared/index.js';
 export * from './slack.js';
 export * from './smtp.js';
 export * from './snowflake.js';
+export * from './snowflakecortex.js';
+export * from './snowflakepostgres.js';
 export * from './stabilityai.js';
 export * from './stripe.js';
 export * from './superblocks-ocr.js';

package/src/plugins/templates/snowflake.ts

@@ -120,7 +120,7 @@ const BASE_CONNECTION_METHOD = {
 
 export const SnowflakePlugin: Plugin = {
   id: 'snowflake',
-  name: 'Snowflake',
+  name: 'Snowflake Warehouse',
   moduleName: 'SnowflakePlugin',
   modulePath: 'plugins/snowflake/SnowflakePlugin',
   iconLocation: `${PUBLIC_INTEGRATIONS_LOGO_URL}/snowflake.png`,

package/src/plugins/templates/snowflakecortex.ts

@@ -0,0 +1,15 @@
+import { ExtendedIntegrationPluginId, Plugin } from '../../types/index.js';
+import { PUBLIC_INTEGRATIONS_CDN_URL, PUBLIC_INTEGRATIONS_LOGO_URL } from './constants.js';
+import { getNativeOpenApiPlugin } from './shared/index.js';
+
+export const SnowflakeCortexPlugin: Plugin = getNativeOpenApiPlugin({
+  name: 'Snowflake Cortex',
+  id: ExtendedIntegrationPluginId.SNOWFLAKE_CORTEX,
+  openApiSpecURL: `${PUBLIC_INTEGRATIONS_CDN_URL}/openapi/snowflake-cortex.yaml`,
+  logoURL: `${PUBLIC_INTEGRATIONS_LOGO_URL}/snowflake.png`,
+  authMethods: {
+    apiKeyForm: [{ label: 'API key', header: 'Authorization', key: 'api-key', required: true, valuePrefix: 'Bearer ' }]
+  },
+  isStreamable: true,
+  serverURL: 'https://{SUPERBLOCKS_OPENAPI_TENANT_NAME}.snowflakecomputing.com'
+});

package/src/plugins/templates/snowflakepostgres.ts

@@ -0,0 +1,299 @@
+import {
+  DB_SQL_INITIAL_TEXT,
+  EditorLanguage,
+  FormComponentType,
+  FormItem,
+  InputDataType,
+  Plugin,
+  PluginResponseType,
+  PluginType
+} from '../../types/index.js';
+import { PARAMETERIZED_SQL_DESCRIPTION, PUBLIC_INTEGRATIONS_LOGO_URL, DOCS_BASE_URL } from './constants.js';
+import { CONNECTION_METHODS_AND_DISPLAY_NAMES, makeDropdownItem } from './shared/db.js';
+
+export const SnowflakePostgresPluginVersions = {
+  V1: '0.0.1'
+};
+
+const BASE_HOST = {
+  label: 'Host',
+  name: 'connection.host',
+  componentType: FormComponentType.INPUT_TEXT,
+  placeholder: 'xxxx.us-west-2.aws.postgres.snowflake.app',
+  rules: [{ required: true, message: 'Host is required' }]
+};
+
+const BASE_PORT = {
+  label: 'Port',
+  name: 'connection.port',
+  componentType: FormComponentType.INPUT_TEXT,
+  dataType: InputDataType.NUMBER,
+  initialValue: 5432,
+  rules: [{ required: true, message: 'Port is required' }]
+};
+
+const BASE_DATABASE_NAME = {
+  label: 'Database name',
+  name: 'connection.databaseName',
+  componentType: FormComponentType.INPUT_TEXT,
+  rules: [{ required: true, message: 'Database name is required' }]
+};
+
+const BASE_DATABASE_USERNAME = {
+  label: 'Database username',
+  name: 'connection.username',
+  componentType: FormComponentType.INPUT_TEXT
+};
+
+const BASE_DATABASE_PASSWORD = {
+  label: 'Database password',
+  name: 'connection.password',
+  componentType: FormComponentType.INPUT_TEXT,
+  dataType: InputDataType.PASSWORD
+};
+
+const BASE_SELFSIGNED_SSL = {
+  label: 'Use a self-signed SSL certificate',
+  name: 'connection.useSelfSignedSsl',
+  componentType: FormComponentType.CHECKBOX,
+  tooltip: {
+    markdownText: 'Use a self-signed SSL certificate'
+  }
+};
+
+const BASE_SERVER_CA = {
+  label: 'Server CA',
+  name: 'connection.ca',
+  componentType: FormComponentType.CODE_EDITOR,
+  language: EditorLanguage.TEXT,
+  placeholder:
+    '-----BEGIN CERTIFICATE-----\n' +
+    'MIIC/jCCAeYCCQDLwS4pIwJC3zANBgkqhkiG9w0BAQsFADBBMQswCQYDVQQGEwJV\n' +
+    '...\n' +
+    '-----END CERTIFICATE-----'
+};
+
+const BASE_CLIENT_KEY = {
+  label: 'Client key',
+  name: 'connection.key',
+  componentType: FormComponentType.CODE_EDITOR,
+  language: EditorLanguage.TEXT,
+  placeholder:
+    '-----BEGIN RSA PRIVATE KEY-----\n' +
+    'BAoMBFRlc3QwggEiMA0GCSqGSIb3DQEBAQUAA4IBDwAwggEKAoIBAQCpvozN05Ou\n' +
+    '...\n' +
+    '-----END RSA PRIVATE KEY-----'
+};
+
+const BASE_CLIENT_CERT = {
+  label: 'Client cert',
+  name: 'connection.cert',
+  componentType: FormComponentType.CODE_EDITOR,
+  language: EditorLanguage.TEXT,
+  placeholder:
+    '-----BEGIN CERTIFICATE-----\n' +
+    'Ur2LYWdrVjqlS/wJyVIze15FMf7sgl+RINsLbQDLwS4pIwJC3zANBgkqhkiG9w0B\n' +
+    '...\n' +
+    '-----END CERTIFICATE-----'
+};
+
+export const SnowflakePostgresPlugin: Plugin = {
+  id: 'snowflakepostgres',
+  name: 'Snowflake Postgres',
+  moduleName: 'SnowflakePostgresPlugin',
+  modulePath: 'plugins/snowflakepostgres/SnowflakePostgresPlugin',
+  iconLocation: `${PUBLIC_INTEGRATIONS_LOGO_URL}/snowflake.png`,
+  docsUrl: `${DOCS_BASE_URL}/snowflakepostgres`,
+  type: PluginType.DB,
+  responseType: PluginResponseType.TABLE,
+  hasRawRequest: true,
+  hasMetadata: true,
+  rawRequestName: 'Executed SQL',
+  datasourceTemplate: {
+    sections: [
+      {
+        name: 'main',
+        items: [
+          {
+            label: 'Display name',
+            name: 'name',
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            componentType: FormComponentType.INPUT_TEXT,
+            placeholder: 'Snowflake Postgres Production',
+            rules: [{ required: true, message: 'Display name is required' }]
+          },
+          {
+            label: 'Connection method',
+            name: 'connectionType',
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            componentType: FormComponentType.DROPDOWN,
+            initialValue: 'fields',
+            rules: [{ required: true }],
+            options: Object.entries(CONNECTION_METHODS_AND_DISPLAY_NAMES).map(([value, displayName]) =>
+              makeDropdownItem(value, displayName)
+            )
+          },
+          {
+            label: 'URI',
+            name: 'connectionUrl',
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            componentType: FormComponentType.DYNAMIC_INPUT_TEXT,
+            placeholder: `postgres://[:user]:[:password]@[:host]:[:port]/[:database]`,
+            tooltip: {
+              markdownText: 'PostgreSQL-compatible connection URI for your Snowflake Postgres instance'
+            },
+            rules: [{ required: true, message: 'URI is required' }],
+            display: {
+              show: {
+                connectionType: ['url']
+              }
+            }
+          },
+          // HOST
+          {
+            ...BASE_HOST,
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            display: {
+              show: {
+                connectionType: ['fields']
+              }
+            }
+          } as FormItem,
+          // PORT
+          {
+            ...BASE_PORT,
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            display: {
+              show: {
+                connectionType: ['fields']
+              }
+            }
+          } as FormItem,
+          // DATABASE NAME
+          {
+            ...BASE_DATABASE_NAME,
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            display: {
+              show: {
+                connectionType: ['fields']
+              }
+            }
+          } as FormItem,
+          // DATABASE USERNAME
+          {
+            ...BASE_DATABASE_USERNAME,
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            display: {
+              show: {
+                connectionType: ['fields']
+              }
+            }
+          } as FormItem,
+          // DATABASE PASSWORD
+          {
+            ...BASE_DATABASE_PASSWORD,
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            display: {
+              show: {
+                connectionType: ['fields']
+              }
+            }
+          } as FormItem,
+          // SSL is always enabled for Snowflake - no checkbox shown.
+          // Self-signed SSL certificate option
+          {
+            ...BASE_SELFSIGNED_SSL,
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            display: {
+              show: {
+                connectionType: ['fields']
+              }
+            }
+          } as FormItem,
+          // SERVER CA
+          {
+            ...BASE_SERVER_CA,
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            display: {
+              show: {
+                'connection.useSelfSignedSsl': ['true'],
+                connectionType: ['fields']
+              }
+            }
+          } as FormItem,
+          // CLIENT KEY
+          {
+            ...BASE_CLIENT_KEY,
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            display: {
+              show: {
+                'connection.useSelfSignedSsl': ['true'],
+                connectionType: ['fields']
+              }
+            }
+          } as FormItem,
+          // CLIENT CERT
+          {
+            ...BASE_CLIENT_CERT,
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            display: {
+              show: {
+                'connection.useSelfSignedSsl': ['true'],
+                connectionType: ['fields']
+              }
+            }
+          } as FormItem
+        ]
+      }
+    ]
+  },
+  actionTemplate: {
+    sections: [
+      {
+        name: 'main',
+        items: [
+          {
+            label: 'Parameters ($1, $2, ...)',
+            name: 'parameters',
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            ldFlag: 'clark.parameterized-apis.v2.enabled',
+            componentType: FormComponentType.DYNAMIC_INPUT_TEXT,
+            placeholder: '[param1, param2, ...]',
+            tooltip: {
+              markdownText:
+                'JavaScript expression that evaluates to an array of parameter values. Use `$1`, `$2`, etc. in your SQL query to reference these parameters.'
+            }
+          },
+          {
+            label: '', // Query
+            name: 'body',
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            componentType: FormComponentType.CODE_EDITOR,
+            language: EditorLanguage.SQL,
+            initialValue: DB_SQL_INITIAL_TEXT
+          }
+        ]
+      },
+      {
+        name: 'advanced:main',
+        items: [
+          {
+            label: 'Use parameterized SQL',
+            name: 'usePreparedSql',
+            startVersion: SnowflakePostgresPluginVersions.V1,
+            componentType: FormComponentType.SWITCH,
+            initialValue: true,
+            tooltip: {
+              markdownText: PARAMETERIZED_SQL_DESCRIPTION
+            },
+            display: {
+              show: {
+                parameters: ['undefined']
+              }
+            }
+          }
+        ]
+      }
+    ]
+  }
+};

package/src/types/ai/credit-pricing.ts

@@ -0,0 +1,87 @@
+export type AiCreditTokenRates = {
+  inputTokenCost: number;
+  outputTokenCost: number;
+  cacheReadTokenCost: number;
+  cacheWriteTokenCost: number;
+};
+
+export const AI_CREDIT_COST_MULTIPLIER = 2;
+
+export const AI_MODEL_CREDIT_RATES = {
+  'claude-haiku-4-5': {
+    inputTokenCost: 0.000001,
+    outputTokenCost: 0.000005,
+    cacheReadTokenCost: 0.0000001,
+    cacheWriteTokenCost: 0.00000125
+  },
+  'claude-opus-4-5': {
+    inputTokenCost: 0.000005,
+    outputTokenCost: 0.000025,
+    cacheReadTokenCost: 0.0000005,
+    cacheWriteTokenCost: 0.00000625
+  },
+  'claude-opus-4-6': {
+    inputTokenCost: 0.000005,
+    outputTokenCost: 0.000025,
+    cacheReadTokenCost: 0.0000005,
+    cacheWriteTokenCost: 0.00000625
+  },
+  'claude-sonnet-4-5': {
+    inputTokenCost: 0.000003,
+    outputTokenCost: 0.000015,
+    cacheReadTokenCost: 0.0000003,
+    cacheWriteTokenCost: 0.00000375
+  }
+} as const satisfies Record<string, AiCreditTokenRates>;
+
+export const AI_FALLBACK_CREDIT_RATES: AiCreditTokenRates = AI_MODEL_CREDIT_RATES['claude-sonnet-4-5'];
+
+const MODEL_MATCH_ORDER = ['claude-opus-4-6', 'claude-opus-4-5', 'claude-sonnet-4-5', 'claude-haiku-4-5'] as const;
+
+const normalizeModelForPricing = (rawModel: string): string | undefined => {
+  const normalized = rawModel.toLowerCase();
+  for (const modelKey of MODEL_MATCH_ORDER) {
+    if (normalized.includes(modelKey)) {
+      return modelKey;
+    }
+  }
+  return undefined;
+};
+
+export const resolveAiCreditRates = (
+  model: string
+): {
+  modelKey: string;
+  rates: AiCreditTokenRates;
+  usedFallback: boolean;
+} => {
+  const modelKey = normalizeModelForPricing(model);
+  if (!modelKey) {
+    return {
+      modelKey: 'fallback',
+      rates: AI_FALLBACK_CREDIT_RATES,
+      usedFallback: true
+    };
+  }
+
+  return {
+    modelKey,
+    rates: AI_MODEL_CREDIT_RATES[modelKey],
+    usedFallback: false
+  };
+};
+
+export const computeAiCreditUsage = (params: {
+  inputTokens: number;
+  outputTokens: number;
+  cachedReadTokens: number;
+  cachedWriteTokens: number;
+  rates: AiCreditTokenRates;
+}): number => {
+  const usdCost =
+    params.inputTokens * params.rates.inputTokenCost +
+    params.outputTokens * params.rates.outputTokenCost +
+    params.cachedReadTokens * params.rates.cacheReadTokenCost +
+    params.cachedWriteTokens * params.rates.cacheWriteTokenCost;
+  return usdCost * AI_CREDIT_COST_MULTIPLIER;
+};

package/src/types/ai/index.ts

@@ -12,6 +12,7 @@ export type AiChatMessageDto = {
  * System event trigger types for AI operations without prompt_id
  */
 export enum AiSystemTriggerType {
+  APP_NAME = 'app_name',
   COMMIT_MESSAGE = 'commit_message',
   FACT_METADATA = 'fact_metadata'
 }
@@ -44,3 +45,6 @@ export interface AiWorkflowEvent {
   promptId: string;
   attributes: AiWorkflowAttributes;
 }
+
+export * from './credit-pricing.js';
+export * from './quota-paywall.js';

package/src/types/ai/quota-paywall.ts

@@ -0,0 +1,62 @@
+export type AiQuotaPaywallReason = 'token_limit_exceeded' | 'trial_expired';
+
+export const getAiQuotaPaywallReasonFromMessage = (message: string): AiQuotaPaywallReason | undefined => {
+  const normalizedMessage = message.toLowerCase();
+
+  if (
+    normalizedMessage.includes('trial_expired') ||
+    normalizedMessage.includes('ai trial period has ended') ||
+    normalizedMessage.includes('free trial has ended')
+  ) {
+    return 'trial_expired';
+  }
+
+  if (normalizedMessage.includes('token_limit_exceeded') || normalizedMessage.includes('ai usage limit')) {
+    return 'token_limit_exceeded';
+  }
+
+  return undefined;
+};
+
+export const AI_QUOTA_PAYWALL_STRIPE_CHECKOUT_URL = 'https://buy.stripe.com/7sYeVc8XNfl3fZ40FT8EM02';
+
+export const AI_QUOTA_PAYWALL_CONTACT_SALES_EMAIL = 'sales@superblockshq.com';
+
+export const AI_QUOTA_PAYWALL_CONTACT_SALES_MAILTO = 'mailto:sales@superblockshq.com?subject=Enterprise%20inquiry';
+
+export const AI_QUOTA_PAYWALL_REASON_CONFIG: Record<AiQuotaPaywallReason, { title: string; subtitle: string }> = {
+  token_limit_exceeded: {
+    title: 'Your free trial has ended',
+    subtitle: 'Upgrade your plan to continue using Clark AI and other AI-powered features.'
+  },
+  trial_expired: {
+    title: 'Your free trial has ended',
+    subtitle: 'Upgrade to keep building with AI.'
+  }
+};
+
+export const AI_QUOTA_PAYWALL_TEAMS_FEATURES = [
+  '200 AI Credits per AI Builder / month',
+  'Clark AI agent for app building',
+  'Clark User Memory',
+  'AI Credit Packs add-on available'
+];
+
+export const AI_QUOTA_PAYWALL_ENTERPRISE_FEATURES = [
+  '200 AI Credits per AI Builder / month',
+  'Org-level credit top-ups, shared across team',
+  'Clark AI agent for app building',
+  'Clark Integration and Organization Memory'
+];
+
+export const AI_QUOTA_PAYWALL_TEAMS_PLATFORM_FEATURES = ['Staging and production environments', '50+ integrations'];
+
+export const AI_QUOTA_PAYWALL_ENTERPRISE_PLATFORM_FEATURES = [
+  'Source Control via GitHub, GitLab, BitBucket, Azure DevOps',
+  'Custom app hosting with volume pricing',
+  'SSO / SAML / OIDC',
+  'VPC Deployment',
+  'Audit Logs',
+  'Dedicated support',
+  '99.9% SLA'
+];

package/src/types/application/index.ts

@@ -506,6 +506,8 @@ export type PostApplicationCreateRequestBody = {
   createUsingReact?: boolean;
   /** which template to use when creating a React/code-mode application (e.g. 'shadcn-demo-app', 'blank-app') */
   templateName?: string;
+  /** When provided, server generates an AI app name from this prompt */
+  initialPrompt?: string;
 };
 export interface ApplicationPageClonePayload {
   applicationId: string;

package/src/types/billing/billing.ts

@@ -3,7 +3,9 @@ export enum BillingPlan {
   FREE = 'FREE',
   STARTER = 'STARTER',
   PRO = 'PRO',
-  ENTERPRISE = 'ENTERPRISE'
+  ENTERPRISE = 'ENTERPRISE',
+  TEAMS = 'TEAMS',
+  POC = 'POC'
 }
 
 export type Billing = {

package/src/types/datasource/index.ts

@@ -453,6 +453,24 @@ export type LakebaseDatasourceConfiguration = {
     oauthType?: IntegrationAuthType.OAUTH2_TOKEN_EXCHANGE;
   };
 
+// Snowflake Postgres is Snowflake's Postgres-compatible database endpoint
+export type SnowflakePostgresDatasourceConfiguration = {
+  name?: string;
+  connection?: {
+    host?: string;
+    port?: number;
+    databaseName?: string;
+    username?: string;
+    password?: string;
+    useSelfSignedSsl?: boolean;
+    ca?: string;
+    key?: string;
+    cert?: string;
+  };
+  connectionType?: 'fields' | 'url';
+  connectionUrl?: string;
+};
+
 export type OracleDbDatasourceConfiguration = Pick<OracleDbPlugin.Plugin, 'name' | 'connection' | 'dynamicWorkflowConfiguration'> & {
   connectionType?: 'fields' | 'url';
   connectionUrl?: string;
@@ -501,6 +519,7 @@ export type DatasourceConfiguration = (
   | OracleDbDatasourceConfiguration
   | DatabricksDatasourceConfiguration
   | LakebaseDatasourceConfiguration
+  | SnowflakePostgresDatasourceConfiguration
   | CouchbaseDatasourceConfiguration
   | SecretStore
 ) &
@@ -537,6 +556,8 @@ export type DatasourceConfigurationByType = {
   s3?: S3DatasourceConfiguration & TemporaryFields;
   smtp?: SmtpDatasourceConfiguration & TemporaryFields;
   snowflake?: SnowflakeDatasourceConfiguration & TemporaryFields;
+  snowflakecortex?: RestApiIntegrationDatasourceConfiguration & TemporaryFields;
+  snowflakepostgres?: SnowflakePostgresDatasourceConfiguration & TemporaryFields;
   salesforce?: SalesforceDatasourceConfiguration & TemporaryFields;
   'superblocks-ocr'?: SuperblocksOcrDatasourceConfiguration & TemporaryFields;
   workflow?: WorkflowDatasourceConfiguration & TemporaryFields;

package/src/types/plugin/integration.ts

@@ -43,6 +43,7 @@ export enum ExtendedIntegrationPluginId {
   COHERE = 'cohere',
   GEMINI = 'gemini',
   OPENAI_V2 = 'openai_v2',
+  SNOWFLAKE_CORTEX = 'snowflakecortex',
   // Not native OpenAPI-backed integrations, but leveraging the same auth system
   GOOGLE_SHEETS_PLUGIN_ID = 'gsheets',
   SNOWFLAKE = 'snowflake',
@@ -91,7 +92,8 @@ export const ExtendedIntegrationPluginMap: Record<string, string> = {
   [ExtendedIntegrationPluginId.STABILITYAI]: ExtendedIntegrationPluginId.REST_API,
   [ExtendedIntegrationPluginId.COHERE]: ExtendedIntegrationPluginId.REST_API,
   [ExtendedIntegrationPluginId.GEMINI]: ExtendedIntegrationPluginId.REST_API,
-  [ExtendedIntegrationPluginId.OPENAI_V2]: ExtendedIntegrationPluginId.REST_API
+  [ExtendedIntegrationPluginId.OPENAI_V2]: ExtendedIntegrationPluginId.REST_API,
+  [ExtendedIntegrationPluginId.SNOWFLAKE_CORTEX]: ExtendedIntegrationPluginId.REST_API
 };
 
 export const EXTENDED_INTEGRATION_PLUGIN_IDS = Object.values(ExtendedIntegrationPluginId).map((val) => val as string);

package/src/types/user/survey.ts

@@ -1,5 +1,8 @@
 export enum SurveyToShow {
   NO_SURVEY = 'NO_SURVEY',
+  SURVEY_FOR_NEW_USER = 'SURVEY_FOR_NEW_USER',
+  /** @deprecated Use SURVEY_FOR_NEW_USER instead */
   SURVEY_FOR_NEW_USER_NEW_ORG = 'SURVEY_FOR_NEW_USER_NEW_ORG',
+  /** @deprecated Use SURVEY_FOR_NEW_USER instead */
   SURVEY_FOR_NEW_USER_EXISTING_ORG = 'SURVEY_FOR_NEW_USER_EXISTING_ORG'
 }