correct tool name

ur-whitelab · Dec 8, 2023 · b2dd5a1 · b2dd5a1
1 parent d1b2bcf
commit b2dd5a1
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 7 deletions.
diff --git a/chemcrow/agents/tools.py b/chemcrow/agents/tools.py
@@ -37,7 +37,7 @@ def make_tools(
     if rxn4chem_api_key:
         all_tools += [
             RXNPredict(rxn4chem_api_key),
-            RXNPlanner(rxn4chem_api_key, openai_api_key)
+            RXNRetrosynthesis(rxn4chem_api_key, openai_api_key)
         ]
 
     return all_tools
diff --git a/chemcrow/tools/__init__.py b/chemcrow/tools/__init__.py
@@ -1,5 +1,7 @@
-from .databases import *
-from .rdkit import *
-from .search import *
-from .rxn4chem import *
-from .safety import SafetySummary, ExplosiveCheck
+"""load all tools."""
+
+from .databases import *  # noqa
+from .rdkit import *  # noqa
+from .search import *  # noqa
+from .rxn4chem import *  # noqa
+from .safety import SafetySummary, ExplosiveCheck  # noqa
diff --git a/chemcrow/tools/rxn4chem.py b/chemcrow/tools/rxn4chem.py
@@ -11,6 +11,11 @@
 from langchain.chat_models import ChatOpenAI
 from langchain.schema import HumanMessage
 
+__all__ = [
+    "RXNPredict",
+    "RXNRetrosynthesis"
+]
+
 
 class RXN4Chem(BaseTool):
     """Wrapper for RXN4Chem functionalities."""

diff --git a/chemcrow/version.py b/chemcrow/version.py
@@ -1,3 +1,3 @@
 """Package version"""
 
-__version__ = "0.3.18"
+__version__ = "0.3.19"