add Snowflake mime types

dust-tt · Jan 7, 2025 · 6309cab · 6309cab
1 parent 7e55465
commit 6309cab
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/connectors/src/connectors/snowflake/temporal/activities.ts b/connectors/src/connectors/snowflake/temporal/activities.ts
@@ -1,5 +1,5 @@
 import type { ModelId } from "@dust-tt/types";
-import { isSnowflakeCredentials } from "@dust-tt/types";
+import { isSnowflakeCredentials, SNOWFLAKE_MIME_TYPES } from "@dust-tt/types";
 
 import {
   connectToSnowflake,
@@ -172,7 +172,7 @@ export async function syncSnowflakeConnection(connectorId: ModelId) {
           ],
           parentId: `${table.databaseName}.${table.schemaName}`,
           title: table.name,
-          mimeType: "application/vnd.snowflake.table",
+          mimeType: SNOWFLAKE_MIME_TYPES.TABLE,
         });
         await table.update({
           lastUpsertedAt: new Date(),

diff --git a/types/src/shared/internal_mime_types.ts b/types/src/shared/internal_mime_types.ts
@@ -63,11 +63,19 @@ export const SLACK_MIME_TYPES = {
 export type SlackMimeType =
   (typeof SLACK_MIME_TYPES)[keyof typeof SLACK_MIME_TYPES];
 
+export const SNOWFLAKE_MIME_TYPES = {
+  TABLE: "application/vnd.snowflake.table",
+};
+
+export type SnowflakeMimeType =
+  (typeof SNOWFLAKE_MIME_TYPES)[keyof typeof SNOWFLAKE_MIME_TYPES];
+
 export type DustMimeType =
   | ConfluenceMimeType
   | GithubMimeType
   | GoogleDriveMimeType
   | IntercomMimeType
   | MicrosoftMimeType
   | NotionMimeType
-  | SlackMimeType;
+  | SlackMimeType
+  | SnowflakeMimeType;