Feature: Custom Templates (#3169)

* New Feature: Custom Templates in the marketplace. * New Feature: Custom Templates in the marketplace. * Custom Template Delete and Shortcut in the dropdown menu * auto detect framework * minor ui fixes * adding custom template feature for tools * ui tool dialog save template --------- Co-authored-by: Henry <[email protected]>
FlowiseAI · Sep 16, 2024 · b02bdc7 · b02bdc7
1 parent 44b70ca
commit b02bdc7
Show file tree

Hide file tree

Showing 23 changed files with 1,213 additions and 166 deletions.
diff --git a/packages/server/src/Interface.ts b/packages/server/src/Interface.ts
@@ -273,5 +273,18 @@ export interface IApiKey {
     updatedDate: Date
 }
 
+export interface ICustomTemplate {
+    id: string
+    name: string
+    flowData: string
+    updatedDate: Date
+    createdDate: Date
+    description?: string
+    type?: string
+    badge?: string
+    framework?: string
+    usecases?: string
+}
+
 // DocumentStore related
 export * from './Interface.DocumentStore'
diff --git a/packages/server/src/controllers/marketplaces/index.ts b/packages/server/src/controllers/marketplaces/index.ts
@@ -1,5 +1,7 @@
 import { Request, Response, NextFunction } from 'express'
 import marketplacesService from '../../services/marketplaces'
+import { InternalFlowiseError } from '../../errors/internalFlowiseError'
+import { StatusCodes } from 'http-status-codes'
 
 // Get all templates for marketplaces
 const getAllTemplates = async (req: Request, res: Response, next: NextFunction) => {
@@ -11,6 +13,48 @@ const getAllTemplates = async (req: Request, res: Response, next: NextFunction)
     }
 }
 
+const deleteCustomTemplate = async (req: Request, res: Response, next: NextFunction) => {
+    try {
+        if (typeof req.params === 'undefined' || !req.params.id) {
+            throw new InternalFlowiseError(
+                StatusCodes.PRECONDITION_FAILED,
+                `Error: marketplacesService.deleteCustomTemplate - id not provided!`
+            )
+        }
+        const apiResponse = await marketplacesService.deleteCustomTemplate(req.params.id)
+        return res.json(apiResponse)
+    } catch (error) {
+        next(error)
+    }
+}
+
+const getAllCustomTemplates = async (req: Request, res: Response, next: NextFunction) => {
+    try {
+        const apiResponse = await marketplacesService.getAllCustomTemplates()
+        return res.json(apiResponse)
+    } catch (error) {
+        next(error)
+    }
+}
+
+const saveCustomTemplate = async (req: Request, res: Response, next: NextFunction) => {
+    try {
+        if ((!req.body && !(req.body.chatflowId || req.body.tool)) || !req.body.name) {
+            throw new InternalFlowiseError(
+                StatusCodes.PRECONDITION_FAILED,
+                `Error: marketplacesService.saveCustomTemplate - body not provided!`
+            )
+        }
+        const apiResponse = await marketplacesService.saveCustomTemplate(req.body)
+        return res.json(apiResponse)
+    } catch (error) {
+        next(error)
+    }
+}
+
 export default {
-    getAllTemplates
+    getAllTemplates,
+    getAllCustomTemplates,
+    saveCustomTemplate,
+    deleteCustomTemplate
 }
diff --git a/packages/server/src/database/entities/CustomTemplate.ts b/packages/server/src/database/entities/CustomTemplate.ts
@@ -0,0 +1,37 @@
+import { ICustomTemplate } from '../../Interface'
+import { Column, CreateDateColumn, Entity, PrimaryGeneratedColumn, UpdateDateColumn } from 'typeorm'
+
+@Entity('custom_template')
+export class CustomTemplate implements ICustomTemplate {
+    @PrimaryGeneratedColumn('uuid')
+    id: string
+
+    @Column()
+    name: string
+
+    @Column({ type: 'text' })
+    flowData: string
+
+    @Column({ nullable: true, type: 'text' })
+    description?: string
+
+    @Column({ nullable: true, type: 'text' })
+    badge?: string
+
+    @Column({ nullable: true, type: 'text' })
+    framework?: string
+
+    @Column({ nullable: true, type: 'text' })
+    usecases?: string
+
+    @Column({ nullable: true, type: 'text' })
+    type?: string
+
+    @Column({ type: 'timestamp' })
+    @CreateDateColumn()
+    createdDate: Date
+
+    @Column({ type: 'timestamp' })
+    @UpdateDateColumn()
+    updatedDate: Date
+}
diff --git a/packages/server/src/database/entities/index.ts b/packages/server/src/database/entities/index.ts
@@ -10,6 +10,7 @@ import { DocumentStoreFileChunk } from './DocumentStoreFileChunk'
 import { Lead } from './Lead'
 import { UpsertHistory } from './UpsertHistory'
 import { ApiKey } from './ApiKey'
+import { CustomTemplate } from './CustomTemplate'
 
 export const entities = {
     ChatFlow,
@@ -23,5 +24,6 @@ export const entities = {
     DocumentStoreFileChunk,
     Lead,
     UpsertHistory,
-    ApiKey
+    ApiKey,
+    CustomTemplate
 }
diff --git a/packages/server/src/database/migrations/mariadb/1725629836652-AddCustomTemplate.ts b/packages/server/src/database/migrations/mariadb/1725629836652-AddCustomTemplate.ts
@@ -0,0 +1,25 @@
+import { MigrationInterface, QueryRunner } from 'typeorm'
+
+export class AddCustomTemplate1725629836652 implements MigrationInterface {
+    public async up(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(
+            `CREATE TABLE IF NOT EXISTS \`custom_template\` (
+                \`id\` varchar(36) NOT NULL,
+                \`name\` varchar(255) NOT NULL,
+                \`flowData\` text NOT NULL,
+                \`description\` varchar(255) DEFAULT NULL,
+                \`badge\` varchar(255) DEFAULT NULL,
+                \`framework\` varchar(255) DEFAULT NULL,
+                \`usecases\` varchar(255) DEFAULT NULL,
+                \`type\` varchar(30) DEFAULT NULL,
+                \`createdDate\` datetime(6) NOT NULL DEFAULT CURRENT_TIMESTAMP(6),
+                \`updatedDate\` datetime(6) NOT NULL DEFAULT CURRENT_TIMESTAMP(6) ON UPDATE CURRENT_TIMESTAMP(6),
+                PRIMARY KEY (\`id\`)
+              ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci;`
+        )
+    }
+
+    public async down(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(`DROP TABLE custom_template`)
+    }
+}
diff --git a/packages/server/src/database/migrations/mariadb/index.ts b/packages/server/src/database/migrations/mariadb/index.ts
@@ -23,6 +23,7 @@ import { AddTypeToChatFlow1716300000000 } from './1716300000000-AddTypeToChatFlo
 import { AddApiKey1720230151480 } from './1720230151480-AddApiKey'
 import { AddActionToChatMessage1721078251523 } from './1721078251523-AddActionToChatMessage'
 import { LongTextColumn1722301395521 } from './1722301395521-LongTextColumn'
+import { AddCustomTemplate1725629836652 } from './1725629836652-AddCustomTemplate'
 
 export const mariadbMigrations = [
     Init1693840429259,
@@ -49,5 +50,6 @@ export const mariadbMigrations = [
     AddTypeToChatFlow1716300000000,
     AddApiKey1720230151480,
     AddActionToChatMessage1721078251523,
-    LongTextColumn1722301395521
+    LongTextColumn1722301395521,
+    AddCustomTemplate1725629836652
 ]
diff --git a/packages/server/src/database/migrations/mysql/1725629836652-AddCustomTemplate.ts b/packages/server/src/database/migrations/mysql/1725629836652-AddCustomTemplate.ts
@@ -0,0 +1,25 @@
+import { MigrationInterface, QueryRunner } from 'typeorm'
+
+export class AddCustomTemplate1725629836652 implements MigrationInterface {
+    public async up(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(
+            `CREATE TABLE IF NOT EXISTS \`custom_template\` (
+                \`id\` varchar(36) NOT NULL,
+                \`name\` varchar(255) NOT NULL,
+                \`flowData\` text NOT NULL,
+                \`description\` varchar(255) DEFAULT NULL,
+                \`badge\` varchar(255) DEFAULT NULL,
+                \`framework\` varchar(255) DEFAULT NULL,
+                \`usecases\` varchar(255) DEFAULT NULL,
+                \`type\` varchar(30) DEFAULT NULL,
+                \`createdDate\` datetime(6) NOT NULL DEFAULT CURRENT_TIMESTAMP(6),
+                \`updatedDate\` datetime(6) NOT NULL DEFAULT CURRENT_TIMESTAMP(6) ON UPDATE CURRENT_TIMESTAMP(6),
+                PRIMARY KEY (\`id\`)
+              ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci;`
+        )
+    }
+
+    public async down(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(`DROP TABLE custom_template`)
+    }
+}
diff --git a/packages/server/src/database/migrations/mysql/index.ts b/packages/server/src/database/migrations/mysql/index.ts
@@ -24,6 +24,7 @@ import { AddTypeToChatFlow1716300000000 } from './1716300000000-AddTypeToChatFlo
 import { AddApiKey1720230151480 } from './1720230151480-AddApiKey'
 import { AddActionToChatMessage1721078251523 } from './1721078251523-AddActionToChatMessage'
 import { LongTextColumn1722301395521 } from './1722301395521-LongTextColumn'
+import { AddCustomTemplate1725629836652 } from './1725629836652-AddCustomTemplate'
 
 export const mysqlMigrations = [
     Init1693840429259,
@@ -51,5 +52,6 @@ export const mysqlMigrations = [
     AddVectorStoreConfigToDocStore1715861032479,
     AddApiKey1720230151480,
     AddActionToChatMessage1721078251523,
-    LongTextColumn1722301395521
+    LongTextColumn1722301395521,
+    AddCustomTemplate1725629836652
 ]
diff --git a/packages/server/src/database/migrations/postgres/1725629836652-AddCustomTemplate.ts b/packages/server/src/database/migrations/postgres/1725629836652-AddCustomTemplate.ts
@@ -0,0 +1,25 @@
+import { MigrationInterface, QueryRunner } from 'typeorm'
+
+export class AddCustomTemplate1725629836652 implements MigrationInterface {
+    public async up(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(
+            `CREATE TABLE IF NOT EXISTS custom_template (
+                id uuid NOT NULL DEFAULT uuid_generate_v4(),
+                "name" varchar NOT NULL,
+                "flowData" text NOT NULL,
+                "description" varchar NULL,
+                "badge" varchar NULL,
+                "framework" varchar NULL,
+                "usecases" varchar NULL,
+                "type" varchar NULL,
+                "createdDate" timestamp NOT NULL DEFAULT now(),
+                "updatedDate" timestamp NOT NULL DEFAULT now(),
+                CONSTRAINT "PK_3c7cea7d087ac4b91764574cdbf" PRIMARY KEY (id)
+            );`
+        )
+    }
+
+    public async down(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(`DROP TABLE custom_template`)
+    }
+}
diff --git a/packages/server/src/database/migrations/postgres/index.ts b/packages/server/src/database/migrations/postgres/index.ts
@@ -24,6 +24,7 @@ import { AddAgentReasoningToChatMessage1714679514451 } from './1714679514451-Add
 import { AddTypeToChatFlow1716300000000 } from './1716300000000-AddTypeToChatFlow'
 import { AddApiKey1720230151480 } from './1720230151480-AddApiKey'
 import { AddActionToChatMessage1721078251523 } from './1721078251523-AddActionToChatMessage'
+import { AddCustomTemplate1725629836652 } from './1725629836652-AddCustomTemplate'
 
 export const postgresMigrations = [
     Init1693891895163,
@@ -51,5 +52,6 @@ export const postgresMigrations = [
     AddTypeToChatFlow1716300000000,
     AddVectorStoreConfigToDocStore1715861032479,
     AddApiKey1720230151480,
-    AddActionToChatMessage1721078251523
+    AddActionToChatMessage1721078251523,
+    AddCustomTemplate1725629836652
 ]
diff --git a/packages/server/src/database/migrations/sqlite/1725629836652-AddCustomTemplate.ts b/packages/server/src/database/migrations/sqlite/1725629836652-AddCustomTemplate.ts
@@ -0,0 +1,23 @@
+import { MigrationInterface, QueryRunner } from 'typeorm'
+
+export class AddCustomTemplate1725629836652 implements MigrationInterface {
+    public async up(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(
+            `CREATE TABLE IF NOT EXISTS "custom_template" (
+                "id" varchar PRIMARY KEY NOT NULL, 
+                "name" varchar NOT NULL, 
+                "flowData" text NOT NULL, 
+                "description" varchar, 
+                "badge" varchar, 
+                "framework" varchar, 
+                "usecases" varchar, 
+                "type" varchar, 
+                "updatedDate" datetime NOT NULL DEFAULT (datetime('now')),
+                "createdDate" datetime NOT NULL DEFAULT (datetime('now')));`
+        )
+    }
+
+    public async down(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(`DROP TABLE IF EXISTS "custom_template";`)
+    }
+}
diff --git a/packages/server/src/database/migrations/sqlite/index.ts b/packages/server/src/database/migrations/sqlite/index.ts
@@ -23,6 +23,7 @@ import { AddAgentReasoningToChatMessage1714679514451 } from './1714679514451-Add
 import { AddTypeToChatFlow1716300000000 } from './1716300000000-AddTypeToChatFlow'
 import { AddApiKey1720230151480 } from './1720230151480-AddApiKey'
 import { AddActionToChatMessage1721078251523 } from './1721078251523-AddActionToChatMessage'
+import { AddCustomTemplate1725629836652 } from './1725629836652-AddCustomTemplate'
 
 export const sqliteMigrations = [
     Init1693835579790,
@@ -49,5 +50,6 @@ export const sqliteMigrations = [
     AddTypeToChatFlow1716300000000,
     AddVectorStoreConfigToDocStore1715861032479,
     AddApiKey1720230151480,
-    AddActionToChatMessage1721078251523
+    AddActionToChatMessage1721078251523,
+    AddCustomTemplate1725629836652
 ]
diff --git a/packages/server/src/routes/marketplaces/index.ts b/packages/server/src/routes/marketplaces/index.ts
@@ -5,4 +5,12 @@ const router = express.Router()
 // READ
 router.get('/templates', marketplacesController.getAllTemplates)
 
+router.post('/custom', marketplacesController.saveCustomTemplate)
+
+// READ
+router.get('/custom', marketplacesController.getAllCustomTemplates)
+
+// DELETE
+router.delete(['/', '/custom/:id'], marketplacesController.deleteCustomTemplate)
+
 export default router