diff --git a/chemcrow/__init__.py b/chemcrow/__init__.py index eeda285..78db58b 100644 --- a/chemcrow/__init__.py +++ b/chemcrow/__init__.py @@ -1,6 +1,6 @@ -from .tools.databases import * +#from .tools.databases import * from .tools.rdkit import * -from .tools.search import * -from .frontend import * +#from .tools.search import * +#from .frontend import * from .agents import * from .version import __version__ diff --git a/chemcrow/agents/tools.py b/chemcrow/agents/tools.py index c59e5ec..fe33103 100644 --- a/chemcrow/agents/tools.py +++ b/chemcrow/agents/tools.py @@ -11,17 +11,9 @@ def make_tools( api_keys: dict = {}, verbose=True ): - serp_key = api_keys.get('SERP_API_KEY') or os.getenv("SERP_API_KEY") - rxn4chem_api_key = api_keys.get('RXN4CHEM_API_KEY') or os.getenv("RXN4CHEM_API_KEY") - all_tools = agents.load_tools([ - "python_repl", - "ddg-search", - "wikipedia", - #"human" - ]) - all_tools += [ + all_tools = [ Query2SMILES(), Query2CAS(), PatentCheck(), @@ -32,7 +24,6 @@ def make_tools( SafetySummary(llm=llm), #LitSearch(llm=llm, verbose=verbose), ] - if rxn4chem_api_key: - all_tools.append(RXNPredict(rxn4chem_api_key)) - return all_tools \ No newline at end of file + + return all_tools diff --git a/chemcrow/tools/__init__.py b/chemcrow/tools/__init__.py index a8cd1d5..5192f73 100644 --- a/chemcrow/tools/__init__.py +++ b/chemcrow/tools/__init__.py @@ -1,5 +1,5 @@ from .databases import * from .rdkit import * -from .search import * +#from .search import * from .rxn4chem import * from .safety import SafetySummary, ExplosiveCheck diff --git a/chemcrow/tools/search.py b/chemcrow/tools/search.py index 022d0c1..02fc9cf 100644 --- a/chemcrow/tools/search.py +++ b/chemcrow/tools/search.py @@ -3,7 +3,7 @@ from typing import Optional import langchain -import paperqa +#import paperqa import paperscraper from langchain import SerpAPIWrapper from langchain.base_language import BaseLanguageModel