Spaces:
Runtime error
Runtime error
abtsousa commited on
Commit ·
9465031
1
Parent(s): e214909
Add tools to the toolset
Browse files- tools/__init__.py +3 -0
tools/__init__.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
from .wikipedia import wiki_search
|
| 2 |
from .search import web_search
|
| 3 |
from .code_interpreter import execute_code_multilang
|
|
|
|
| 4 |
from langchain_core.tools import BaseTool
|
| 5 |
|
| 6 |
def get_all_tools() -> list[BaseTool]:
|
|
@@ -16,6 +17,8 @@ def get_all_tools() -> list[BaseTool]:
|
|
| 16 |
execute_code_multilang
|
| 17 |
]
|
| 18 |
|
|
|
|
|
|
|
| 19 |
return tools
|
| 20 |
|
| 21 |
def list_tools() -> str:
|
|
|
|
| 1 |
from .wikipedia import wiki_search
|
| 2 |
from .search import web_search
|
| 3 |
from .code_interpreter import execute_code_multilang
|
| 4 |
+
from .files import file_management_toolkit
|
| 5 |
from langchain_core.tools import BaseTool
|
| 6 |
|
| 7 |
def get_all_tools() -> list[BaseTool]:
|
|
|
|
| 17 |
execute_code_multilang
|
| 18 |
]
|
| 19 |
|
| 20 |
+
tools.extend(file_management_toolkit.get_tools()) # Add file management tools
|
| 21 |
+
|
| 22 |
return tools
|
| 23 |
|
| 24 |
def list_tools() -> str:
|