Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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) #
|
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 = """
|