feat: add toolkits to settings
This commit is contained in:
parent
9ef1078786
commit
021ede2f72
3 changed files with 8 additions and 1 deletions
|
|
@ -28,4 +28,8 @@ tools:
|
||||||
wrappers:
|
wrappers:
|
||||||
- RequestsWrapper
|
- RequestsWrapper
|
||||||
|
|
||||||
|
toolkits:
|
||||||
|
- OpenAPIToolkit
|
||||||
|
- JsonToolkit
|
||||||
|
|
||||||
dev: false
|
dev: false
|
||||||
|
|
|
||||||
|
|
@ -4,6 +4,7 @@ from langchain.agents import agent_toolkits
|
||||||
|
|
||||||
from langflow.interface.base import LangChainTypeCreator
|
from langflow.interface.base import LangChainTypeCreator
|
||||||
from langflow.interface.importing.utils import import_class, import_module
|
from langflow.interface.importing.utils import import_class, import_module
|
||||||
|
from langflow.settings import settings
|
||||||
from langflow.utils.util import build_template_from_class
|
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
|
# if toolkit_name is not lower case it is a class
|
||||||
for toolkit_name in agent_toolkits.__all__
|
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
|
return self.type_dict
|
||||||
|
|
|
||||||
|
|
@ -13,6 +13,7 @@ class Settings(BaseSettings):
|
||||||
tools: List[str] = []
|
tools: List[str] = []
|
||||||
memories: List[str] = []
|
memories: List[str] = []
|
||||||
wrappers: List[str] = []
|
wrappers: List[str] = []
|
||||||
|
toolkits: List[str] = []
|
||||||
dev: bool = False
|
dev: bool = False
|
||||||
|
|
||||||
class Config:
|
class Config:
|
||||||
|
|
@ -35,6 +36,7 @@ class Settings(BaseSettings):
|
||||||
self.tools = new_settings.tools or []
|
self.tools = new_settings.tools or []
|
||||||
self.memories = new_settings.memories or []
|
self.memories = new_settings.memories or []
|
||||||
self.wrappers = new_settings.wrappers or []
|
self.wrappers = new_settings.wrappers or []
|
||||||
|
self.toolkits = new_settings.toolkits or []
|
||||||
self.dev = new_settings.dev or False
|
self.dev = new_settings.dev or False
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue