lakshmivairamani commited on
Commit
778868d
·
verified ·
1 Parent(s): 2141c97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -34,7 +34,8 @@ db_uri = 'mysql+mysqlconnector://redmindgen:51([email protected]
34
  db = SQLDatabase.from_uri(db_uri)
35
 
36
  # LLM setup
37
- llm = ChatOpenAI(model="gpt-3.5-turbo-0125")
 
38
 
39
  # Define the SQL query generation tool
40
  template_query_generation = """Based on the table schema below, write a SQL query that would answer the user's question:
 
34
  db = SQLDatabase.from_uri(db_uri)
35
 
36
  # LLM setup
37
+ #llm = ChatOpenAI(model="gpt-3.5-turbo-0125")
38
+ llm = OpenAI(temperature=0, api_key= os.environ['OPEN_AI_API'], model_name='gpt-3.5-turbo')
39
 
40
  # Define the SQL query generation tool
41
  template_query_generation = """Based on the table schema below, write a SQL query that would answer the user's question: