feat: add toolkits to settings

This commit is contained in:
Gabriel Almeida 2023-03-31 18:49:31 -03:00
commit 021ede2f72
3 changed files with 8 additions and 1 deletions

View file

@ -28,4 +28,8 @@ tools:
wrappers:
- RequestsWrapper
toolkits:
- OpenAPIToolkit
- JsonToolkit
dev: false

View file

@ -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

View file

@ -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