Make quality

huggingface · Apr 29, 2024 · e184ace · e184ace
1 parent 91d25ad
commit e184ace
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/agents/__init__.py b/src/transformers/agents/__init__.py
@@ -24,7 +24,7 @@
 
 
 _import_structure = {
-    "agents": ["Agent", "CodeAgent", "ReactAgent", "ReactCodeAgent", "ReactJSONAgent"],
+    "agents": ["Agent", "CodeAgent", "ReactAgent", "ReactCodeAgent", "ReactJSONAgent", "Toolbox"],
     "llm_engine": ["HfEngine"],
     "tools": ["PipelineTool", "Tool", "launch_gradio_demo", "load_tool"],
 }
@@ -43,7 +43,7 @@
     _import_structure["translation"] = ["TranslationTool"]
 
 if TYPE_CHECKING:
-    from .agents import Agent, CodeAgent, ReactAgent, ReactCodeAgent, ReactJSONAgent
+    from .agents import Agent, CodeAgent, ReactAgent, ReactCodeAgent, ReactJSONAgent, Toolbox
     from .llm_engine import HfEngine
     from .tools import PipelineTool, Tool, launch_gradio_demo, load_tool