diff --git a/server/app/model/config/config.py b/server/app/model/config/config.py index 8703cf02..b6d98049 100644 --- a/server/app/model/config/config.py +++ b/server/app/model/config/config.py @@ -164,6 +164,10 @@ class ConfigInfo: "env_vars": [], "toolkit": "pptx_toolkit", }, + ConfigGroup.RAG.value: { + "env_vars": ["OPENAI_API_KEY"], + "toolkit": "rag_toolkit", + }, ConfigGroup.REDDIT.value: { "env_vars": [ "REDDIT_CLIENT_ID", diff --git a/server/app/type/config_group.py b/server/app/type/config_group.py index ea560aac..f6c8802c 100644 --- a/server/app/type/config_group.py +++ b/server/app/type/config_group.py @@ -40,6 +40,7 @@ class ConfigGroup(str, Enum): IMAGE_ANALYSIS = "Image Analysis" MCP_SEARCH = "MCP Search" PPTX = "PPTX" + RAG = "RAG" TERMINAL = "Terminal" @classmethod diff --git a/src/components/IntegrationList/index.tsx b/src/components/IntegrationList/index.tsx index 49c98bd9..e6776523 100644 --- a/src/components/IntegrationList/index.tsx +++ b/src/components/IntegrationList/index.tsx @@ -103,8 +103,11 @@ export default function IntegrationList({ console.log(item); const searchKey = isSelectMode ? 'EXA Search' : 'Search'; - if (item.key === searchKey || item.key === 'Lark') { - const mcp = createMcpFromItem(item, item.key === 'Lark' ? 15 : 13); + if (item.key === searchKey || item.key === 'Lark' || item.key === 'RAG') { + const mcp = createMcpFromItem( + item, + item.key === 'Lark' ? 15 : item.key === 'RAG' ? 16 : 13 + ); if (isSelectMode) { onShowEnvConfig?.(mcp); } else {