Sultan
/
Flowise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feature/export-import-4 (#4272) 

* feat: seperate assistant export-import into custom, openai and azure

* chore: update assistant names in export data to maintain consistency
This commit is contained in:
Ong Chung Yau 2025-04-09 16:28:51 +08:00 committed by GitHub
parent 3098c8e75f
commit 4277819c5f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 32 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
packages/server/src/services/export-import/index.ts
View File

@ -23,7 +23,9 @@ import variableService from '../variables'
type ExportInput = { type ExportInput = {
agentflow: boolean agentflow: boolean
assistant: boolean assistantCustom: boolean
assistantOpenAI: boolean
assistantAzure: boolean
chatflow: boolean chatflow: boolean
chat_message: boolean chat_message: boolean
chat_feedback: boolean chat_feedback: boolean
@ -35,8 +37,10 @@ type ExportInput = {
type ExportData = { type ExportData = {
AgentFlow: ChatFlow[] AgentFlow: ChatFlow[]
AssistantCustom: Assistant[]
AssistantFlow: ChatFlow[] AssistantFlow: ChatFlow[]
Assistant: Assistant[] AssistantOpenAI: Assistant[]
AssistantAzure: Assistant[]
ChatFlow: ChatFlow[] ChatFlow: ChatFlow[]
ChatMessage: ChatMessage[] ChatMessage: ChatMessage[]
ChatMessageFeedback: ChatMessageFeedback[] ChatMessageFeedback: ChatMessageFeedback[]
@ -77,9 +81,12 @@ const exportData = async (exportInput: ExportInput): Promise<{ FileDefaultName:
try { try {
let AgentFlow: ChatFlow[] = exportInput.agentflow === true ? await chatflowService.getAllChatflows('MULTIAGENT') : [] let AgentFlow: ChatFlow[] = exportInput.agentflow === true ? await chatflowService.getAllChatflows('MULTIAGENT') : []
let Assistant: Assistant[] = exportInput.assistant === true ? await assistantService.getAllAssistants() : [] let AssistantCustom: Assistant[] = exportInput.assistantCustom === true ? await assistantService.getAllAssistants('CUSTOM') : []
let AssistantFlow: ChatFlow[] = exportInput.assistantCustom === true ? await chatflowService.getAllChatflows('ASSISTANT') : []
let AssistantFlow: ChatFlow[] = exportInput.assistant === true ? await chatflowService.getAllChatflows('ASSISTANT') : [] let AssistantOpenAI: Assistant[] = exportInput.assistantOpenAI === true ? await assistantService.getAllAssistants('OPENAI') : []
let AssistantAzure: Assistant[] = exportInput.assistantAzure === true ? await assistantService.getAllAssistants('AZURE') : []
let ChatFlow: ChatFlow[] = exportInput.chatflow === true ? await chatflowService.getAllChatflows('CHATFLOW') : [] let ChatFlow: ChatFlow[] = exportInput.chatflow === true ? await chatflowService.getAllChatflows('CHATFLOW') : []
@ -103,8 +110,10 @@ const exportData = async (exportInput: ExportInput): Promise<{ FileDefaultName:
return { return {
FileDefaultName, FileDefaultName,
AgentFlow, AgentFlow,
AssistantCustom,
AssistantFlow, AssistantFlow,
Assistant, AssistantOpenAI,
AssistantAzure,
ChatFlow, ChatFlow,
ChatMessage, ChatMessage,
ChatMessageFeedback, ChatMessageFeedback,
@ -412,12 +421,16 @@ const importData = async (importData: ExportData) => {
importData.AgentFlow = reduceSpaceForChatflowFlowData(importData.AgentFlow) importData.AgentFlow = reduceSpaceForChatflowFlowData(importData.AgentFlow)
importData = await replaceDuplicateIdsForChatFlow(queryRunner, importData, importData.AgentFlow) importData = await replaceDuplicateIdsForChatFlow(queryRunner, importData, importData.AgentFlow)
} }
if (importData.AssistantCustom.length > 0)
importData = await replaceDuplicateIdsForAssistant(queryRunner, importData, importData.AssistantCustom)
if (importData.AssistantFlow.length > 0) { if (importData.AssistantFlow.length > 0) {
importData.AssistantFlow = reduceSpaceForChatflowFlowData(importData.AssistantFlow) importData.AssistantFlow = reduceSpaceForChatflowFlowData(importData.AssistantFlow)
importData = await replaceDuplicateIdsForChatFlow(queryRunner, importData, importData.AssistantFlow) importData = await replaceDuplicateIdsForChatFlow(queryRunner, importData, importData.AssistantFlow)
} }
if (importData.Assistant.length > 0) if (importData.AssistantOpenAI.length > 0)
importData = await replaceDuplicateIdsForAssistant(queryRunner, importData, importData.Assistant) importData = await replaceDuplicateIdsForAssistant(queryRunner, importData, importData.AssistantOpenAI)
if (importData.AssistantAzure.length > 0)
importData = await replaceDuplicateIdsForAssistant(queryRunner, importData, importData.AssistantAzure)
if (importData.ChatFlow.length > 0) { if (importData.ChatFlow.length > 0) {
importData.ChatFlow = reduceSpaceForChatflowFlowData(importData.ChatFlow) importData.ChatFlow = reduceSpaceForChatflowFlowData(importData.ChatFlow)
importData = await replaceDuplicateIdsForChatFlow(queryRunner, importData, importData.ChatFlow) importData = await replaceDuplicateIdsForChatFlow(queryRunner, importData, importData.ChatFlow)
@ -440,7 +453,9 @@ const importData = async (importData: ExportData) => {
if (importData.AgentFlow.length > 0) await queryRunner.manager.save(ChatFlow, importData.AgentFlow) if (importData.AgentFlow.length > 0) await queryRunner.manager.save(ChatFlow, importData.AgentFlow)
if (importData.AssistantFlow.length > 0) await queryRunner.manager.save(ChatFlow, importData.AssistantFlow) if (importData.AssistantFlow.length > 0) await queryRunner.manager.save(ChatFlow, importData.AssistantFlow)
if (importData.Assistant.length > 0) await queryRunner.manager.save(Assistant, importData.Assistant) if (importData.AssistantCustom.length > 0) await queryRunner.manager.save(Assistant, importData.AssistantCustom)
if (importData.AssistantOpenAI.length > 0) await queryRunner.manager.save(Assistant, importData.AssistantOpenAI)
if (importData.AssistantAzure.length > 0) await queryRunner.manager.save(Assistant, importData.AssistantAzure)
if (importData.ChatFlow.length > 0) await queryRunner.manager.save(ChatFlow, importData.ChatFlow) if (importData.ChatFlow.length > 0) await queryRunner.manager.save(ChatFlow, importData.ChatFlow)
if (importData.ChatMessage.length > 0) await queryRunner.manager.save(ChatMessage, importData.ChatMessage) if (importData.ChatMessage.length > 0) await queryRunner.manager.save(ChatMessage, importData.ChatMessage)
if (importData.ChatMessageFeedback.length > 0) if (importData.ChatMessageFeedback.length > 0)

8
packages/ui/src/layout/MainLayout/Header/ProfileSection/index.jsx
View File

@ -54,7 +54,9 @@ import { useNavigate } from 'react-router-dom'
const dataToExport = [ const dataToExport = [
'Agentflows', 'Agentflows',
'Assistants', 'Assistants Custom',
'Assistants OpenAI',
'Assistants Azure',
'Chatflows', 'Chatflows',
'Chat Messages', 'Chat Messages',
'Chat Feedbacks', 'Chat Feedbacks',
@ -254,7 +256,9 @@ const ProfileSection = ({ username, handleLogout }) => {
const onExport = (data) => { const onExport = (data) => {
const body = {} const body = {}
if (data.includes('Agentflows')) body.agentflow = true if (data.includes('Agentflows')) body.agentflow = true
if (data.includes('Assistants')) body.assistant = true if (data.includes('Assistants Custom')) body.assistantCustom = true
if (data.includes('Assistants OpenAI')) body.assistantOpenAI = true
if (data.includes('Assistants Azure')) body.assistantAzure = true
if (data.includes('Chatflows')) body.chatflow = true if (data.includes('Chatflows')) body.chatflow = true
if (data.includes('Chat Messages')) body.chat_message = true if (data.includes('Chat Messages')) body.chat_message = true
if (data.includes('Chat Feedbacks')) body.chat_feedback = true if (data.includes('Chat Feedbacks')) body.chat_feedback = true

4
packages/ui/src/utils/exportImport.js
View File

@ -79,7 +79,9 @@ export const exportData = (exportAllData) => {
return { return {
AgentFlow: sanitizeChatflow(exportAllData.AgentFlow), AgentFlow: sanitizeChatflow(exportAllData.AgentFlow),
AssistantFlow: sanitizeChatflow(exportAllData.AssistantFlow), AssistantFlow: sanitizeChatflow(exportAllData.AssistantFlow),
Assistant: sanitizeAssistant(exportAllData.Assistant), AssistantCustom: sanitizeAssistant(exportAllData.AssistantCustom),
AssistantOpenAI: sanitizeAssistant(exportAllData.AssistantOpenAI),
AssistantAzure: sanitizeAssistant(exportAllData.AssistantAzure),
ChatFlow: sanitizeChatflow(exportAllData.ChatFlow), ChatFlow: sanitizeChatflow(exportAllData.ChatFlow),
ChatMessage: exportAllData.ChatMessage, ChatMessage: exportAllData.ChatMessage,
ChatMessageFeedback: exportAllData.ChatMessageFeedback, ChatMessageFeedback: exportAllData.ChatMessageFeedback,