Skip to content

Commit ff9b754

Browse files
committed
feat: Modify Database Input Method in Streamlit App
- Changed user input from a text area to a selection box, providing SQL_PROMPTS options - Added SQL_PROMPTS module
1 parent 2c11d01 commit ff9b754

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

interface/streamlit_app.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import streamlit as st
22
from langchain_core.messages import HumanMessage
33
from llm_utils.graph import builder
4+
from langchain.chains.sql_database.prompt import SQL_PROMPTS
45

56
# Streamlit 앱 제목
67
st.title("Lang2SQL")
@@ -11,9 +12,10 @@
1112
value="고객 데이터를 기반으로 유니크한 유저 수를 카운트하는 쿼리",
1213
)
1314

14-
user_database_env = st.text_area(
15+
user_database_env = st.selectbox(
1516
"db 환경정보를 입력하세요:",
16-
value="duckdb",
17+
options=SQL_PROMPTS.keys(),
18+
index=0,
1719
)
1820

1921

0 commit comments

Comments
 (0)