Kiranontimitta commited on
Commit
dec8c35
·
verified ·
1 Parent(s): dae7884

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,15 +6,15 @@ from langchain.chat_models import ChatOpenAI
6
  from langchain.prompts import PromptTemplate
7
  from langchain.chains import LLMChain
8
 
9
- # Set OpenAI API Key
10
- openai.api_key = "sk-O7esHSo2XAWm-GXUGXp7_P9l4qXrQMn0CIGzs34ojLT3BlbkFJeXGSSvywppRTAvyT0zZkmZLZsj5cg7XkAkBTh8ZxoA"
11
 
12
  # Database connection
13
  DATABASE_URL = "sqlite:///Sakila.db" # Replace with your DB path or connection string
14
  engine = create_engine(DATABASE_URL)
15
 
16
  # Set up LangChain components
17
- llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0.5) # OpenAI's Chat model for LLM
18
 
19
  # Custom prompt to generate SQL query
20
  prompt_template = """
 
6
  from langchain.prompts import PromptTemplate
7
  from langchain.chains import LLMChain
8
 
9
+ # Set OpenAI API Key directly here if not set as an environment variable
10
+ OPENAI_API_KEY = "sk-O7esHSo2XAWm-GXUGXp7_P9l4qXrQMn0CIGzs34ojLT3BlbkFJeXGSSvywppRTAvyT0zZkmZLZsj5cg7XkAkBTh8ZxoA"
11
 
12
  # Database connection
13
  DATABASE_URL = "sqlite:///Sakila.db" # Replace with your DB path or connection string
14
  engine = create_engine(DATABASE_URL)
15
 
16
  # Set up LangChain components
17
+ llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0.5, openai_api_key=OPENAI_API_KEY) # Pass API key here
18
 
19
  # Custom prompt to generate SQL query
20
  prompt_template = """