@superblocksteam/sdk-api 0.0.6 → 0.0.7

package/src/integrations/registry.ts

@@ -5,17 +5,8 @@
  * and provides factory functions for creating clients.
  */
 
-import type { IntegrationConfig, IntegrationClientImpl } from "./types.js";
-import { PostgresClientImpl } from "./postgres/index.js";
-import { SlackClientImpl } from "./slack/index.js";
-import { OpenAIClientImpl } from "./openai_v2/index.js";
-import { AnthropicClientImpl } from "./anthropic/index.js";
-import { StripeClientImpl } from "./stripe/index.js";
-import { GitHubClientImpl } from "./github/index.js";
-import { NotionClientImpl } from "./notion/index.js";
-import { SnowflakeClientImpl } from "./snowflake/index.js";
-import { SnowflakeCortexClientImpl } from "./snowflakecortex/index.js";
 import { AirtableClientImpl } from "./airtable/index.js";
+import { AnthropicClientImpl } from "./anthropic/index.js";
 import { AsanaClientImpl } from "./asana/index.js";
 import { BitbucketClientImpl } from "./bitbucket/index.js";
 import { BoxClientImpl } from "./box/index.js";
@@ -28,53 +19,56 @@ import { ElasticSearchClientImpl } from "./elasticsearch/index.js";
 import { FireworksClientImpl } from "./fireworks/index.js";
 import { FrontClientImpl } from "./front/index.js";
 import { GeminiClientImpl } from "./gemini/index.js";
+import { GitHubClientImpl } from "./github/index.js";
 import { GoogleAnalyticsClientImpl } from "./googleanalytics/index.js";
 import { GoogleDriveClientImpl } from "./googledrive/index.js";
+import { GraphQLClientImpl } from "./graphql/index.js";
 import { GroqClientImpl } from "./groq/index.js";
 import { HubSpotClientImpl } from "./hubspot/index.js";
 import { IntercomClientImpl } from "./intercom/index.js";
 import { JiraClientImpl } from "./jira/index.js";
 import { LaunchDarklyClientImpl } from "./launchdarkly/index.js";
 import { MistralClientImpl } from "./mistral/index.js";
+import { NotionClientImpl } from "./notion/index.js";
+import { OpenAIClientImpl } from "./openai_v2/index.js";
 import { PagerDutyClientImpl } from "./pagerduty/index.js";
 import { PerplexityClientImpl } from "./perplexity/index.js";
+import { PostgresClientImpl } from "./postgres/index.js";
 import { SegmentClientImpl } from "./segment/index.js";
 import { SendGridClientImpl } from "./sendgrid/index.js";
+import { SlackClientImpl } from "./slack/index.js";
+import { SnowflakeClientImpl } from "./snowflake/index.js";
+import { SnowflakeCortexClientImpl } from "./snowflakecortex/index.js";
 import { StabilityAIClientImpl } from "./stabilityai/index.js";
+import { StripeClientImpl } from "./stripe/index.js";
 import { TwilioClientImpl } from "./twilio/index.js";
+import type { IntegrationConfig, IntegrationClientImpl } from "./types.js";
 import { ZendeskClientImpl } from "./zendesk/index.js";
 import { ZoomClientImpl } from "./zoom/index.js";
-import { GraphQLClientImpl } from "./graphql/index.js";
 // SQL Database clients
-import { MySQLClientImpl } from "./mysql/index.js";
+import { AthenaClientImpl } from "./athena/index.js";
+import { BigQueryClientImpl } from "./bigquery/index.js";
+import { CockroachDBClientImpl } from "./cockroachdb/index.js";
+import { DatabricksClientImpl } from "./databricks/index.js";
 import { MariaDBClientImpl } from "./mariadb/index.js";
 import { MSSQLClientImpl } from "./mssql/index.js";
-import { CockroachDBClientImpl } from "./cockroachdb/index.js";
+import { MySQLClientImpl } from "./mysql/index.js";
 import { OracleDBClientImpl } from "./oracledb/index.js";
 import { RedshiftClientImpl } from "./redshift/index.js";
-import { AthenaClientImpl } from "./athena/index.js";
-import { DatabricksClientImpl } from "./databricks/index.js";
-import { BigQueryClientImpl } from "./bigquery/index.js";
 // REST API clients
-import { MongoDBClientImpl } from "./mongodb/index.js";
-import { DynamoDBClientImpl } from "./dynamodb/index.js";
 import { CosmosDBClientImpl } from "./cosmosdb/index.js";
-import { CouchbaseClientImpl } from "./couchbase/index.js";
-import { S3ClientImpl } from "./s3/index.js";
+import { DynamoDBClientImpl } from "./dynamodb/index.js";
 import { GCSClientImpl } from "./gcs/index.js";
 import { GoogleSheetsClientImpl } from "./gsheets/index.js";
-import { KafkaClientImpl } from "./kafka/index.js";
-import { KinesisClientImpl } from "./kinesis/index.js";
-import { SalesforceClientImpl } from "./salesforce/index.js";
-import { RedisClientImpl } from "./redis/index.js";
-import { SuperblocksOCRClientImpl } from "./superblocks-ocr/index.js";
 import { LakebaseClientImpl } from "./lakebase/index.js";
-import { SnowflakePostgresClientImpl } from "./snowflakepostgres/index.js";
+import { MongoDBClientImpl } from "./mongodb/index.js";
+import { S3ClientImpl } from "./s3/index.js";
+import { SalesforceClientImpl } from "./salesforce/index.js";
 import { SmtpClientImpl } from "./smtp/index.js";
+import { SnowflakePostgresClientImpl } from "./snowflakepostgres/index.js";
+import { SuperblocksOCRClientImpl } from "./superblocks-ocr/index.js";
 // REST API plugins
 import { RestApiIntegrationPluginClientImpl } from "./restapiintegration/index.js";
-// Code execution clients
-import { PythonClientImpl } from "./python/index.js";
 
 /**
  * Supported plugin IDs mapped to their client constructors.
@@ -133,18 +127,11 @@ export const SUPPORTED_PLUGINS = {
   mongodb: "mongodb",
   dynamodb: "dynamodb",
   cosmosdb: "cosmosdb",
-  couchbase: "couchbase",
   s3: "s3",
   gcs: "gcs",
   gsheets: "gsheets",
-  kafka: "kafka",
-  kinesis: "kinesis",
   salesforce: "salesforce",
-  redis: "redis",
   "superblocks-ocr": "superblocks-ocr",
-  // Kafka-compatible streaming
-  confluent: "confluent",
-  redpanda: "redpanda",
   // SQL databases (continued)
   lakebase: "lakebase",
   snowflakepostgres: "snowflakepostgres",
@@ -152,13 +139,19 @@ export const SUPPORTED_PLUGINS = {
   smtp: "smtp",
   // REST API plugins
   restapiintegration: "restapiintegration",
-  // Code execution
-  python: "python",
 } as const;
 
 export type SupportedPluginId =
   (typeof SUPPORTED_PLUGINS)[keyof typeof SUPPORTED_PLUGINS];
 
+/**
+ * Flat list of SDK-supported plugin IDs.
+ * Useful for compatibility checks where only the ID set matters.
+ */
+export const SUPPORTED_PLUGIN_IDS = Object.values(
+  SUPPORTED_PLUGINS,
+) as SupportedPluginId[];
+
 /**
  * Optional metadata attached to an integration call for diagnostics.
  *
@@ -183,11 +176,8 @@ export interface TraceMetadata {
  * Accepts a plugin request object that matches the protobuf Plugin definition
  * for the specific integration type (e.g., postgresql.v1.Plugin, restapi.v1.Plugin).
  *
- * For language plugins (Python, JavaScript), the bindings parameter should be passed
- * to the API execution request as `input` or `inputs` field for binding resolution.
- *
  * @param request - Plugin-specific request object matching the proto schema
- * @param bindings - Optional bindings data for binding resolution (e.g., Python/JavaScript bindings)
+ * @param bindings - Optional bindings data for binding resolution
  * @param metadata - Optional trace metadata (label, description) for diagnostics
  * @returns Promise resolving to the operation result
  */
@@ -197,21 +187,6 @@ export type QueryExecutor = (
   metadata?: TraceMetadata,
 ) => Promise<unknown>;
 
-/**
- * Type for the streaming query executor function provided by the orchestrator.
- *
- * Similar to QueryExecutor but returns an AsyncIterable for streaming responses.
- * Used for streaming operations like LLM token streaming.
- *
- * @param request - Plugin-specific request object matching the proto schema
- * @param metadata - Optional trace metadata (label, description) for diagnostics
- * @returns AsyncIterable yielding raw chunks from the stream
- */
-export type StreamingQueryExecutor = (
-  request: Record<string, unknown>,
-  metadata?: TraceMetadata,
-) => AsyncIterable<unknown>;
-
 /**
  * Options for creating an integration client.
  */
@@ -221,12 +196,6 @@ export interface CreateClientOptions {
 
   /** Function to execute queries/operations against the integration */
   executeQuery: QueryExecutor;
-
-  /**
-   * Optional function to execute streaming queries/operations.
-   * If provided, enables streaming support for compatible clients.
-   */
-  executeStreamingQuery?: StreamingQueryExecutor;
 }
 
 /**
@@ -255,189 +224,125 @@ export interface CreateClientOptions {
 export function createClient(
   options: CreateClientOptions,
 ): IntegrationClientImpl {
-  const { config, executeQuery, executeStreamingQuery } = options;
+  const { config, executeQuery } = options;
 
   switch (config.pluginId) {
     case SUPPORTED_PLUGINS.postgres:
       return new PostgresClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.slack:
-      return new SlackClientImpl(config, executeQuery, executeStreamingQuery);
+      return new SlackClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.openai_v2:
-      return new OpenAIClientImpl(config, executeQuery, executeStreamingQuery);
+      return new OpenAIClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.anthropic:
-      return new AnthropicClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new AnthropicClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.stripe:
-      return new StripeClientImpl(config, executeQuery, executeStreamingQuery);
+      return new StripeClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.github:
-      return new GitHubClientImpl(config, executeQuery, executeStreamingQuery);
+      return new GitHubClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.notion:
-      return new NotionClientImpl(config, executeQuery, executeStreamingQuery);
+      return new NotionClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.snowflake:
       return new SnowflakeClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.snowflakecortex:
-      return new SnowflakeCortexClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new SnowflakeCortexClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.airtable:
-      return new AirtableClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new AirtableClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.asana:
-      return new AsanaClientImpl(config, executeQuery, executeStreamingQuery);
+      return new AsanaClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.bitbucket:
-      return new BitbucketClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new BitbucketClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.box:
-      return new BoxClientImpl(config, executeQuery, executeStreamingQuery);
+      return new BoxClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.circleci:
-      return new CircleCIClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new CircleCIClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.cohere:
-      return new CohereClientImpl(config, executeQuery, executeStreamingQuery);
+      return new CohereClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.confluence:
-      return new ConfluenceClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new ConfluenceClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.datadog:
-      return new DatadogClientImpl(config, executeQuery, executeStreamingQuery);
+      return new DatadogClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.dropbox:
-      return new DropboxClientImpl(config, executeQuery, executeStreamingQuery);
+      return new DropboxClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.elasticsearch:
-      return new ElasticSearchClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new ElasticSearchClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.fireworks:
-      return new FireworksClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new FireworksClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.front:
-      return new FrontClientImpl(config, executeQuery, executeStreamingQuery);
+      return new FrontClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.gemini:
-      return new GeminiClientImpl(config, executeQuery, executeStreamingQuery);
+      return new GeminiClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.googleanalytics:
-      return new GoogleAnalyticsClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new GoogleAnalyticsClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.googledrive:
-      return new GoogleDriveClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new GoogleDriveClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.graphqlintegration:
       return new GraphQLClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.groq:
-      return new GroqClientImpl(config, executeQuery, executeStreamingQuery);
+      return new GroqClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.hubspot:
-      return new HubSpotClientImpl(config, executeQuery, executeStreamingQuery);
+      return new HubSpotClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.intercom:
-      return new IntercomClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new IntercomClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.jira:
-      return new JiraClientImpl(config, executeQuery, executeStreamingQuery);
+      return new JiraClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.launchdarkly:
-      return new LaunchDarklyClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new LaunchDarklyClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.mistral:
-      return new MistralClientImpl(config, executeQuery, executeStreamingQuery);
+      return new MistralClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.pagerduty:
-      return new PagerDutyClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new PagerDutyClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.perplexity:
-      return new PerplexityClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new PerplexityClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.segment:
-      return new SegmentClientImpl(config, executeQuery, executeStreamingQuery);
+      return new SegmentClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.sendgrid:
-      return new SendGridClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new SendGridClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.stabilityai:
-      return new StabilityAIClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
+      return new StabilityAIClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.twilio:
-      return new TwilioClientImpl(config, executeQuery, executeStreamingQuery);
+      return new TwilioClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.zendesk:
-      return new ZendeskClientImpl(config, executeQuery, executeStreamingQuery);
+      return new ZendeskClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.zoom:
-      return new ZoomClientImpl(config, executeQuery, executeStreamingQuery);
+      return new ZoomClientImpl(config, executeQuery);
 
     // SQL databases
     case SUPPORTED_PLUGINS.mysql:
@@ -477,9 +382,6 @@ export function createClient(
     case SUPPORTED_PLUGINS.cosmosdb:
       return new CosmosDBClientImpl(config, executeQuery);
 
-    case SUPPORTED_PLUGINS.couchbase:
-      return new CouchbaseClientImpl(config, executeQuery);
-
     case SUPPORTED_PLUGINS.s3:
       return new S3ClientImpl(config, executeQuery);
 
@@ -489,26 +391,12 @@ export function createClient(
     case SUPPORTED_PLUGINS.gsheets:
       return new GoogleSheetsClientImpl(config, executeQuery);
 
-    case SUPPORTED_PLUGINS.kafka:
-      return new KafkaClientImpl(config, executeQuery);
-
-    case SUPPORTED_PLUGINS.kinesis:
-      return new KinesisClientImpl(config, executeQuery);
-
     case SUPPORTED_PLUGINS.salesforce:
       return new SalesforceClientImpl(config, executeQuery);
 
-    case SUPPORTED_PLUGINS.redis:
-      return new RedisClientImpl(config, executeQuery);
-
     case SUPPORTED_PLUGINS["superblocks-ocr"]:
       return new SuperblocksOCRClientImpl(config, executeQuery);
 
-    // Confluent and Redpanda use the same Kafka plugin proto
-    case SUPPORTED_PLUGINS.confluent:
-    case SUPPORTED_PLUGINS.redpanda:
-      return new KafkaClientImpl(config, executeQuery);
-
     case SUPPORTED_PLUGINS.lakebase:
       return new LakebaseClientImpl(config, executeQuery);
 
@@ -519,14 +407,7 @@ export function createClient(
       return new SmtpClientImpl(config, executeQuery);
 
     case SUPPORTED_PLUGINS.restapiintegration:
-      return new RestApiIntegrationPluginClientImpl(
-        config,
-        executeQuery,
-        executeStreamingQuery,
-      );
-
-    case SUPPORTED_PLUGINS.python:
-      return new PythonClientImpl(config, executeQuery);
+      return new RestApiIntegrationPluginClientImpl(config, executeQuery);
 
     default:
       throw new Error(