fix(kw-search): snowflake tables should upsert w/ mimetype and title (#…

…8977) Co-authored-by: Henry Fontanier <[email protected]>
dust-tt · Nov 27, 2024 · 6ccac3f · 6ccac3f
1 parent adeb33c
commit 6ccac3f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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
@@ -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: {