Skip to content

Commit

Permalink
Revert "turn back on features"
Browse files Browse the repository at this point in the history
This reverts commit 064adf1.
  • Loading branch information
avanteijlingen committed Sep 26, 2023
1 parent 064adf1 commit e0bef21
Show file tree
Hide file tree
Showing 4 changed files with 8 additions and 17 deletions.
6 changes: 3 additions & 3 deletions chemcrow/__init__.py
Original file line number Diff line number Diff line change
@@ -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__
15 changes: 3 additions & 12 deletions chemcrow/agents/tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(),
Expand All @@ -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

return all_tools
2 changes: 1 addition & 1 deletion chemcrow/tools/__init__.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from .databases import *
from .rdkit import *
from .search import *
#from .search import *
from .rxn4chem import *
from .safety import SafetySummary, ExplosiveCheck
2 changes: 1 addition & 1 deletion chemcrow/tools/search.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit e0bef21

Please sign in to comment.