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 |
|