Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ load_dotenv()
|
|
| 32 |
|
| 33 |
# Instead of hardcoding the DB URI, get it from an environment variable.
|
| 34 |
# This lets you plug in any single DB by changing the DATABASE_URI environment variable.
|
| 35 |
-
DATABASE_URI = os.getenv("DATABASE_URI", "
|
| 36 |
|
| 37 |
GROQ_API_KEY = os.getenv("GROQ_API_KEY")
|
| 38 |
os.environ["GROQ_API_KEY"] = GROQ_API_KEY
|
|
@@ -205,6 +205,7 @@ def create_agent_app(db_path: str):
|
|
| 205 |
workflow.add_edge("correct_query", "execute_query")
|
| 206 |
workflow.add_edge("execute_query", "query_gen")
|
| 207 |
|
|
|
|
| 208 |
# Compile and return the agent application workflow.
|
| 209 |
return workflow.compile()
|
| 210 |
|
|
|
|
| 32 |
|
| 33 |
# Instead of hardcoding the DB URI, get it from an environment variable.
|
| 34 |
# This lets you plug in any single DB by changing the DATABASE_URI environment variable.
|
| 35 |
+
DATABASE_URI = os.getenv("DATABASE_URI", "employee.db")
|
| 36 |
|
| 37 |
GROQ_API_KEY = os.getenv("GROQ_API_KEY")
|
| 38 |
os.environ["GROQ_API_KEY"] = GROQ_API_KEY
|
|
|
|
| 205 |
workflow.add_edge("correct_query", "execute_query")
|
| 206 |
workflow.add_edge("execute_query", "query_gen")
|
| 207 |
|
| 208 |
+
|
| 209 |
# Compile and return the agent application workflow.
|
| 210 |
return workflow.compile()
|
| 211 |
|