srini047 commited on
Commit
db949df
1 Parent(s): 1d63504

Update ques_1.py

Browse files
Files changed (1) hide show
  1. ques_1.py +1 -1
ques_1.py CHANGED
@@ -7,7 +7,7 @@ from env import OPENAI_API_KEY
7
 
8
  dburi = "sqlite:///output.db"
9
  db = SQLDatabase.from_uri(dburi)
10
- llm = OpenAI(temperature=0, openai_api_key=OPENAI_API_KEY)
11
 
12
  toolkit = SQLDatabaseToolkit(db=db, llm=llm)
13
  agent_executor = create_sql_agent(
 
7
 
8
  dburi = "sqlite:///output.db"
9
  db = SQLDatabase.from_uri(dburi)
10
+ llm = OpenAI(temperature=0, openai_api_key=os.environ['OPENAI_API_KEY'])
11
 
12
  toolkit = SQLDatabaseToolkit(db=db, llm=llm)
13
  agent_executor = create_sql_agent(