From 021ede2f728d5e012060239f24d3240a36fb8b7b Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Fri, 31 Mar 2023 18:49:31 -0300
Subject: [PATCH] feat: add toolkits to settings

---
 src/backend/langflow/config.yaml                | 4 ++++
 src/backend/langflow/interface/toolkits/base.py | 3 ++-
 src/backend/langflow/settings.py                | 2 ++
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/backend/langflow/config.yaml b/src/backend/langflow/config.yaml
index b8c2be6da..114db79d4 100644
--- a/src/backend/langflow/config.yaml
+++ b/src/backend/langflow/config.yaml
@@ -28,4 +28,8 @@ tools:
 wrappers:
   - RequestsWrapper
 
+toolkits:
+ - OpenAPIToolkit
+ - JsonToolkit
+
 dev: false
diff --git a/src/backend/langflow/interface/toolkits/base.py b/src/backend/langflow/interface/toolkits/base.py
index 0af46e504..c78a9c051 100644
--- a/src/backend/langflow/interface/toolkits/base.py
+++ b/src/backend/langflow/interface/toolkits/base.py
@@ -4,6 +4,7 @@ from langchain.agents import agent_toolkits
 
 from langflow.interface.base import LangChainTypeCreator
 from langflow.interface.importing.utils import import_class, import_module
+from langflow.settings import settings
 from langflow.utils.util import build_template_from_class
 
 
@@ -33,7 +34,7 @@ class ToolkitCreator(LangChainTypeCreator):
                 )
                 # if toolkit_name is not lower case it is a class
                 for toolkit_name in agent_toolkits.__all__
-                if not toolkit_name.islower()
+                if not toolkit_name.islower() and toolkit_name in settings.toolkits
             }
 
         return self.type_dict
diff --git a/src/backend/langflow/settings.py b/src/backend/langflow/settings.py
index 092bea040..a8c2edc4a 100644
--- a/src/backend/langflow/settings.py
+++ b/src/backend/langflow/settings.py
@@ -13,6 +13,7 @@ class Settings(BaseSettings):
     tools: List[str] = []
     memories: List[str] = []
     wrappers: List[str] = []
+    toolkits: List[str] = []
     dev: bool = False
 
     class Config:
@@ -35,6 +36,7 @@ class Settings(BaseSettings):
         self.tools = new_settings.tools or []
         self.memories = new_settings.memories or []
         self.wrappers = new_settings.wrappers or []
+        self.toolkits = new_settings.toolkits or []
         self.dev = new_settings.dev or False