WebashalarForML commited on
Commit
79afa06
·
verified ·
1 Parent(s): f69d96d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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", "sqlite:///employee.db")
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