From da4aacc4dbcc63f297dfd1473d0650f5759971c8 Mon Sep 17 00:00:00 2001 From: Samantha Cox Date: Fri, 1 Nov 2024 00:42:25 -0700 Subject: [PATCH] remove pqapi api key - local only pqa --- .env.example | 2 -- mdagent/tools/maketools.py | 2 +- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/.env.example b/.env.example index fdee9af0..014b5807 100644 --- a/.env.example +++ b/.env.example @@ -5,7 +5,5 @@ # OpenAI API Key OPENAI_API_KEY=YOUR_OPENAI_API_KEY_GOES_HERE # pragma: allowlist secret -# PQA API Key to use LiteratureSearch tool (optional) -- it also requires OpenAI key -PQA_API_KEY=YOUR_PQA_API_KEY_GOES_HERE # pragma: allowlist secret # Optional: add TogetherAI, Fireworks, or Anthropic API key here to use their models diff --git a/mdagent/tools/maketools.py b/mdagent/tools/maketools.py index a7f514a1..b690d615 100644 --- a/mdagent/tools/maketools.py +++ b/mdagent/tools/maketools.py @@ -74,7 +74,7 @@ def make_all_tools( all_tools += [ ModifyBaseSimulationScriptTool(path_registry=path_instance, llm=llm), ] - if "OPENAI_API_KEY" in os.environ and "PQA_API_KEY" in os.environ: + if path_instance.ckpt_papers: all_tools += [Scholar2ResultLLM(llm=llm, path_registry=path_instance)] if human: all_tools += [agents.load_tools(["human"], llm)[0]]