lakshmivairamani commited on
Commit
fffe3d2
1 Parent(s): 1d72851

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,8 +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
- 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:
 
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: