Spaces:
Running
Running
Fix environment variable handling
Browse files- tools/legal_tools.py +5 -1
- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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):
|