From 6ccac3f497da4cfb5389d1f81342a2dcabdf2f3e Mon Sep 17 00:00:00 2001
From: Henry Fontanier <h.fontanier@gmail.com>
Date: Wed, 27 Nov 2024 18:33:11 +0100
Subject: [PATCH] fix(kw-search): snowflake tables should upsert w/ mimetype
 and title (#8977)

Co-authored-by: Henry Fontanier <henry@dust.tt>
---
 connectors/src/connectors/snowflake/temporal/activities.ts | 2 ++
 connectors/src/lib/data_sources.ts                         | 6 ++++++
 2 files changed, 8 insertions(+)

diff --git a/connectors/src/connectors/snowflake/temporal/activities.ts b/connectors/src/connectors/snowflake/temporal/activities.ts
index 57f5066f6b94..5afc4dbad35c 100644
--- a/connectors/src/connectors/snowflake/temporal/activities.ts
+++ b/connectors/src/connectors/snowflake/temporal/activities.ts
@@ -163,6 +163,8 @@ export async function syncSnowflakeConnection(connectorId: ModelId) {
             `${table.databaseName}.${table.schemaName}`,
             table.databaseName,
           ],
+          title: table.name,
+          mimeType: "application/vnd.snowflake.table",
         });
         await table.update({
           lastUpsertedAt: new Date(),
diff --git a/connectors/src/lib/data_sources.ts b/connectors/src/lib/data_sources.ts
index 966a679fcea7..3f519eb984dd 100644
--- a/connectors/src/lib/data_sources.ts
+++ b/connectors/src/lib/data_sources.ts
@@ -566,6 +566,8 @@ export async function upsertTableFromConnectors({
   remoteDatabaseSecretId,
   loggerArgs,
   parents,
+  title,
+  mimeType,
 }: {
   dataSourceConfig: DataSourceConfig;
   tableId: string;
@@ -575,6 +577,8 @@ export async function upsertTableFromConnectors({
   remoteDatabaseSecretId: string | null;
   loggerArgs?: Record<string, string | number>;
   parents: string[];
+  title: string;
+  mimeType: string;
 }) {
   const localLogger = logger.child({ ...loggerArgs, tableId, tableName });
   const statsDTags = [
@@ -601,6 +605,8 @@ export async function upsertTableFromConnectors({
     table_id: tableId,
     remote_database_table_id: remoteDatabaseTableId,
     remote_database_secret_id: remoteDatabaseSecretId,
+    title,
+    mimeType,
   };
   const dustRequestConfig: AxiosRequestConfig = {
     headers: {