@onereach/idw-init-account-resources 0.19.3 → 0.19.5

package/src/assets/db_migrations/20240723101405-add-label-to-feed.ts

@@ -1,9 +1,12 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
 ALTER TABLE "feed"
 ADD COLUMN "label" VARCHAR;
@@ -18,7 +21,7 @@ CREATE INDEX "feed_label_idx" ON "feed" (label);
      */
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
 ALTER TABLE "feed"
 DROP COLUMN "label";

package/src/assets/db_migrations/20240819183250-create_reporting_tables.ts

@@ -1,10 +1,13 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
 CREATE SCHEMA if not exists idw_reporting;
 
@@ -25,7 +28,7 @@ PRIMARY KEY (id));
      */
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
     drop table idw_reporting.user_skill_daily_metrics;
     drop schema idw_reporting cascade;

package/src/assets/db_migrations/20240828121206-add_user_question_table.ts

@@ -1,10 +1,13 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
 CREATE TABLE if not exists idw_reporting.user_questions (
 question_id uuid NOT NULL,
@@ -24,7 +27,7 @@ PRIMARY KEY (question_id));
      */
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
     drop table idw_reporting.user_questions;
     `);

package/src/assets/db_migrations/20240926094736-add_autogenerated_id_to_reporting_tables.ts

@@ -1,10 +1,13 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE idw_reporting.user_skill_daily_metrics
       ALTER COLUMN id SET DEFAULT gen_random_uuid();
@@ -20,7 +23,7 @@ module.exports = {
      */
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
     ALTER TABLE idw_reporting.user_skill_daily_metrics
     ALTER COLUMN id DROP DEFAULT;

package/src/assets/db_migrations/20241018163528-create_feed_comments_table.ts

@@ -1,9 +1,12 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & { context: { sequelize: Sequelize; }; }) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
     CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
 
@@ -34,7 +37,7 @@ module.exports = {
     COMMENT ON COLUMN feed_comments.created_by IS 'id of the person who created the comment';`);
   },
 
-  async down(queryInterface: QueryInterface & { context: { sequelize: Sequelize; }; }) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       DROP TABLE IF EXISTS "feed_comments"
     `);

package/src/assets/db_migrations/20241031103134-feed_comments_cascade_delete.ts

@@ -1,9 +1,12 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & { context: { sequelize: Sequelize; }; }) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE feed_comments
       DROP CONSTRAINT feed_comments_feed_id_fkey,
@@ -15,7 +18,7 @@ module.exports = {
     `);
   },
 
-  async down(queryInterface: QueryInterface & { context: { sequelize: Sequelize; }; }) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE feed_comments
       DROP CONSTRAINT feed_comments_feed_id_fkey,

package/src/assets/db_migrations/20241106115317-add-skill-id-to-feed.ts

@@ -1,16 +1,19 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE "feed"
       ADD COLUMN IF NOT EXISTS "skill_id" varchar;
     `);
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE "feed"
       DROP COLUMN "skill_id";

package/src/assets/db_migrations/20241108215450-create_model_datasource_tables.ts

@@ -1,10 +1,13 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
 
@@ -73,7 +76,7 @@ module.exports = {
       `);
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     // Drop triggers and function
     await queryInterface.context.sequelize.query(`
       DROP TRIGGER IF EXISTS set_updated_at_models ON models;

package/src/assets/db_migrations/20241217131049-knowledge_auto_id.ts

@@ -1,10 +1,13 @@
 'use strict';
 
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
       ALTER TABLE knowledge
@@ -12,7 +15,7 @@ module.exports = {
     `);
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE knowledge
       ALTER COLUMN knowledge_id DROP DEFAULT;

package/src/assets/db_migrations/20241217131552-knowledge_add_domain_and_expiration_columns.ts

@@ -1,10 +1,13 @@
 'use strict';
 
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     const sequelize = queryInterface.context.sequelize;
 
     const [results] = await sequelize.query(`
@@ -31,7 +34,7 @@ module.exports = {
     `);
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE knowledge
       DROP COLUMN IF EXISTS domain,

package/src/assets/db_migrations/20241217132005-knowledge_auto_update_dates.ts

@@ -1,10 +1,13 @@
 'use strict';
 
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       CREATE OR REPLACE FUNCTION update_updated_at()
       RETURNS TRIGGER AS $$
@@ -25,7 +28,7 @@ module.exports = {
     `);
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       DROP FUNCTION IF EXISTS update_updated_at();
       ALTER TABLE knowledge

package/src/assets/db_migrations/20241224163528-create_feed_reactions.ts

@@ -1,9 +1,12 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & { context: { sequelize: Sequelize; }; }) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
     -- Table Definition
     CREATE TABLE IF NOT EXISTS feed_reactions (
@@ -20,7 +23,7 @@ module.exports = {
 `);
   },
 
-  async down(queryInterface: QueryInterface & { context: { sequelize: Sequelize; }; }) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       DROP TABLE IF EXISTS "feed_reactions"
     `);

package/src/assets/db_migrations/20250103134909-create-idw-graph.ts

@@ -1,6 +1,9 @@
 'use strict';
 import { Graph, Graphs } from '@or-sdk/graph';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 type DiscovyreService = {
   serviceKey: string;
   url: string;
@@ -8,7 +11,7 @@ type DiscovyreService = {
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & { context: { sequelize: Sequelize; _orToken: string; discoveryUrl: string;}; }) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     const { discoveryUrl, _orToken } = queryInterface.context;
 
     try {

package/src/assets/db_migrations/20250109103921-fix-broken-updated-at-function.ts

@@ -1,10 +1,13 @@
 'use strict';
 
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       CREATE OR REPLACE FUNCTION update_updated_at()
       RETURNS TRIGGER AS $$
@@ -31,7 +34,7 @@ module.exports = {
     `);
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       DROP FUNCTION IF EXISTS update_updated_at();
       DROP FUNCTION IF EXISTS update_date_modified();

package/src/assets/db_migrations/20250113094030-knowledge_statuses.ts

@@ -1,10 +1,13 @@
 'use strict';
 
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
        UPDATE knowledge
        SET status = 'published'
@@ -18,7 +21,7 @@ module.exports = {
     `);
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       UPDATE knowledge
       SET status = 'Trained'

package/src/assets/db_migrations/20250113094602-drop_knowledge_categories.ts

@@ -1,17 +1,20 @@
 'use strict';
 
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
        ALTER TABLE knowledge
        DROP COLUMN selected_categories
     `);
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE knowledge
       ADD COLUMN IF NOT EXISTS selected_categories jsonb;

package/src/assets/db_migrations/20250113132203-create_feed_read.ts

@@ -1,9 +1,12 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       CREATE TABLE IF NOT EXISTS feed_read_history (
           user_id uuid,
@@ -14,7 +17,7 @@ module.exports = {
     `);
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       DROP TABLE IF EXISTS feed_read_history
     `);

package/src/assets/db_migrations/20250115172228-knowledge_domains.ts

@@ -1,10 +1,13 @@
 'use strict';
 
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     // Drop constraint to remove a table
     await queryInterface.context.sequelize.query(`
         ALTER TABLE knowledge
@@ -32,7 +35,7 @@ module.exports = {
     `);
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
         -- TODO: Implement undo script
     `);

package/src/assets/db_migrations/20250122131849-feed_post_video.ts

@@ -1,9 +1,12 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE "feed"
       ADD COLUMN IF NOT EXISTS "video_source" varchar,
@@ -11,7 +14,7 @@ module.exports = {
     `);
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE "feed"
       DROP COLUMN "video_source",

package/src/assets/db_migrations/20250214132209-create_feedback_table.ts

@@ -1,9 +1,12 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & { context: { sequelize: Sequelize; }; }) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
     CREATE TABLE IF NOT EXISTS feedback (
       id uuid default gen_random_uuid(), 
@@ -18,7 +21,7 @@ module.exports = {
 `);
   },
 
-  async down(queryInterface: QueryInterface & { context: { sequelize: Sequelize; }; }) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       DROP TABLE IF EXISTS feedback
     `);

package/src/assets/db_migrations/20250311105958-delete-knowledge-approved-status.ts

@@ -1,10 +1,13 @@
 'use strict';
 
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
        UPDATE knowledge
        SET status = 'new'
@@ -12,7 +15,7 @@ module.exports = {
     `);
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
         -- TODO: There is unreal to implement down migration
     `);

package/src/assets/db_migrations/20250401092323-datasource-search-order.ts

@@ -1,10 +1,13 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE models
       DROP COLUMN IF EXISTS search_order,
@@ -15,9 +18,9 @@ module.exports = {
     `);
   },
 
-  async down(queryInterface: QueryInterface & { context: { sequelize: Sequelize; }; }) {
-    await queryInterface.context.sequelize.query(
-      'ALTER TABLE datasources DROP COLUMN IF EXISTS search_order;'
-    );
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
+    await queryInterface.context.sequelize.query(`
+      ALTER TABLE datasources DROP COLUMN IF EXISTS search_order;
+    `);
   },
 };

package/src/assets/db_migrations/20250402100448-add-file-url-to-knowledge.ts

@@ -1,9 +1,12 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     const tableName = 'knowledge';
     const docType = 'doc';
     const textType = 'text';
@@ -36,7 +39,7 @@ module.exports = {
     `);
   },
 
-  async down(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE "knowledge"
       DROP COLUMN IF EXISTS "file_url",

package/src/assets/db_migrations/20250423100508-leave-feedback-drop-user-not-null.ts

@@ -1,9 +1,12 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & { context: { sequelize: Sequelize; }; }) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
     ALTER TABLE feedback
     ALTER COLUMN user_id DROP NOT NULL;

package/src/assets/db_migrations/20250520103623-datasources-access.ts

@@ -1,10 +1,13 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE models
       ADD COLUMN IF NOT EXISTS for_user_emails JSONB;
@@ -15,7 +18,7 @@ module.exports = {
     `);
   },
 
-  async down(queryInterface: QueryInterface & { context: { sequelize: Sequelize; }; }) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE models
       DROP COLUMN IF EXISTS for_user_emails;

package/src/assets/db_migrations/20250618135125-datasources-is-llm.ts

@@ -1,17 +1,20 @@
 'use strict';
-import { QueryInterface, Sequelize } from 'sequelize';
+import { QueryInterface } from 'sequelize';
+import { MigrationParams } from 'umzug';
+
+import { OrMigrationContext } from '../../types';
 
 
 /** @type {import('sequelize-cli').Migration} */
 module.exports = {
-  async up(queryInterface: QueryInterface & {context: {sequelize: Sequelize;};}) {
+  async up(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE datasources
       ADD COLUMN IF NOT EXISTS is_llm BOOLEAN;
     `);
   },
 
-  async down(queryInterface: QueryInterface & { context: { sequelize: Sequelize; }; }) {
+  async down(queryInterface: MigrationParams<QueryInterface & OrMigrationContext>) {
     await queryInterface.context.sequelize.query(`
       ALTER TABLE datasources
       DROP COLUMN IF EXISTS is_llm;