Amodit commited on
Commit
9a2f0cb
·
1 Parent(s): a5e82c9

Fix environment variable handling

Browse files
Files changed (2) hide show
  1. tools/legal_tools.py +5 -1
  2. tools/scheme_tools.py +5 -1
tools/legal_tools.py CHANGED
@@ -6,7 +6,11 @@ from langchain.tools import tool
6
  from langchain_community.tools.tavily_search import TavilySearchResults
7
 
8
  load_dotenv()
9
- os.environ["TAVILY_API_KEY"] = os.getenv("TAVILY_API_KEY")
 
 
 
 
10
 
11
  @tool
12
  def legal_search(query: str):
 
6
  from langchain_community.tools.tavily_search import TavilySearchResults
7
 
8
  load_dotenv()
9
+
10
+ # Safely set environment variable only if it exists
11
+ tavily_key = os.getenv("TAVILY_API_KEY")
12
+ if tavily_key:
13
+ os.environ["TAVILY_API_KEY"] = tavily_key
14
 
15
  @tool
16
  def legal_search(query: str):
tools/scheme_tools.py CHANGED
@@ -6,7 +6,11 @@ from langchain.tools import tool
6
  from langchain_community.tools.tavily_search import TavilySearchResults
7
 
8
  load_dotenv()
9
- os.environ["TAVILY_API_KEY"] = os.getenv("TAVILY_API_KEY")
 
 
 
 
10
 
11
  @tool
12
  def scheme_search(query: str):
 
6
  from langchain_community.tools.tavily_search import TavilySearchResults
7
 
8
  load_dotenv()
9
+
10
+ # Safely set environment variable only if it exists
11
+ tavily_key = os.getenv("TAVILY_API_KEY")
12
+ if tavily_key:
13
+ os.environ["TAVILY_API_KEY"] = tavily_key
14
 
15
  @tool
16
  def scheme_search(query: str):