From 6309cab9fb89a86c375039c760df4e001366b14b Mon Sep 17 00:00:00 2001
From: Aubin <aubin@dust.tt>
Date: Tue, 7 Jan 2025 16:31:55 +0100
Subject: [PATCH] add Snowflake mime types

---
 .../src/connectors/snowflake/temporal/activities.ts    |  4 ++--
 types/src/shared/internal_mime_types.ts                | 10 +++++++++-
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/connectors/src/connectors/snowflake/temporal/activities.ts b/connectors/src/connectors/snowflake/temporal/activities.ts
index 753f8c67f8fe2..450cafc7edcb4 100644
--- 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
index 37785eaa53f02..4e63326135760 100644
--- a/types/src/shared/internal_mime_types.ts
+++ b/types/src/shared/internal_mime_types.ts
@@ -63,6 +63,13 @@ 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
@@ -70,4 +77,5 @@ export type DustMimeType =
   | IntercomMimeType
   | MicrosoftMimeType
   | NotionMimeType
-  | SlackMimeType;
+  | SlackMimeType
+  | SnowflakeMimeType;