Update app.py
Browse files
app.py
CHANGED
|
@@ -17,14 +17,17 @@ def _set_if_undefined(var: str):
|
|
| 17 |
os.environ[var] = userdata.get(var)
|
| 18 |
|
| 19 |
_set_if_undefined("TAVILY_API_KEY")
|
| 20 |
-
|
| 21 |
|
|
|
|
|
|
|
|
|
|
| 22 |
# --- Basic Agent Definition ---
|
| 23 |
# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
|
| 24 |
class BasicAgent:
|
| 25 |
def __init__(self):
|
| 26 |
print("BasicAgent initialized.")
|
| 27 |
-
|
| 28 |
|
| 29 |
# Generate the chat interface, including the tools
|
| 30 |
llm = HuggingFaceEndpoint(
|
|
|
|
| 17 |
os.environ[var] = userdata.get(var)
|
| 18 |
|
| 19 |
_set_if_undefined("TAVILY_API_KEY")
|
| 20 |
+
_set_if_undefined("HUGGINGFACEHUB_API_TOKEN")
|
| 21 |
|
| 22 |
+
TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")
|
| 23 |
+
HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
|
| 24 |
+
|
| 25 |
# --- Basic Agent Definition ---
|
| 26 |
# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
|
| 27 |
class BasicAgent:
|
| 28 |
def __init__(self):
|
| 29 |
print("BasicAgent initialized.")
|
| 30 |
+
|
| 31 |
|
| 32 |
# Generate the chat interface, including the tools
|
| 33 |
llm = HuggingFaceEndpoint(
|