LLM/Embeddings/DB 프로파일(시크릿 포함) JSON 저장·사이드바 적용 + DB 섹션 CRUD 확장

Author: ehddnr301

interface/app_pages/components/db_selector.py

@@ -0,0 +1,48 @@
+import os
+import streamlit as st
+
+from interface.core.config import get_db_connections_registry, update_db_settings
+
+
+def render_sidebar_db_selector() -> None:
+    st.sidebar.markdown("### DB 연결")
+
+    registry = get_db_connections_registry()
+    names = [c.name for c in registry.connections]
+    if not names:
+        st.sidebar.warning("등록된 DB 프로파일이 없습니다. 설정 > DB에서 추가하세요.")
+        return
+
+    # 기본 선택: 세션 또는 ENV의 DB_TYPE과 일치하는 첫 프로파일
+    current_type = (
+        st.session_state.get("DB_TYPE") or os.getenv("DB_TYPE") or ""
+    ).lower()
+    default_index = 0
+    if current_type:
+        for idx, c in enumerate(registry.connections):
+            if c.type == current_type:
+                default_index = idx
+                break
+
+    sel_name = st.sidebar.selectbox(
+        "프로파일", options=names, index=default_index, key="sidebar_db_profile"
+    )
+    selected = next((c for c in registry.connections if c.name == sel_name), None)
+    if selected is None:
+        st.sidebar.error("선택한 프로파일을 찾을 수 없습니다.")
+        return
+
+    if st.sidebar.button("적용", key="sidebar_apply_db"):
+        try:
+            values = {
+                "host": selected.host,
+                "port": selected.port,
+                "user": selected.user,
+                "password": selected.password,
+                "database": selected.database,
+                "extra": selected.extra,
+            }
+            update_db_settings(db_type=selected.type, values=values, secrets={})
+            st.sidebar.success(f"DB 적용됨: {selected.name}")
+        except Exception as e:
+            st.sidebar.error(f"적용 실패: {e}")

interface/app_pages/components/embedding_selector.py

@@ -1,35 +1,81 @@
 import os
 import streamlit as st
 
-from interface.core.config import update_embedding_settings
-from interface.app_pages.settings_sections.llm_section import LLM_PROVIDERS
+from interface.core.config import (
+    update_embedding_settings,
+    get_embedding_registry,
+)
 
 
 def render_sidebar_embedding_selector() -> None:
     st.sidebar.markdown("### Embeddings 선택")
 
-    default_emb = (
-        (
-            st.session_state.get("EMBEDDING_PROVIDER")
-            or os.getenv("EMBEDDING_PROVIDER")
-            or "openai"
+    e_reg = get_embedding_registry()
+    if not e_reg.profiles:
+        st.sidebar.info(
+            "저장된 Embeddings 프로파일이 없습니다. 설정 > LLM에서 저장하세요."
         )
+        # fallback: 간단 공급자 선택 유지
+        default_emb = (
+            (
+                st.session_state.get("EMBEDDING_PROVIDER")
+                or os.getenv("EMBEDDING_PROVIDER")
+                or "openai"
+            )
+        ).lower()
+        selected = st.sidebar.selectbox(
+            "Embeddings 공급자",
+            options=["openai", "azure", "bedrock", "gemini", "ollama", "huggingface"],
+            index=(
+                ["openai", "azure", "bedrock", "gemini", "ollama", "huggingface"].index(
+                    default_emb
+                )
+                if default_emb
+                in {"openai", "azure", "bedrock", "gemini", "ollama", "huggingface"}
+                else 0
+            ),
+            key="sidebar_embedding_provider_fallback",
+        )
+        if selected != default_emb:
+            try:
+                update_embedding_settings(provider=selected, values={})
+                st.sidebar.success(
+                    f"Embeddings 공급자가 '{selected}'로 변경되었습니다."
+                )
+            except Exception as e:
+                st.sidebar.error(f"Embeddings 공급자 변경 실패: {e}")
+        return
+
+    e_names = [p.name for p in e_reg.profiles]
+    current_emb_provider = (
+        st.session_state.get("EMBEDDING_PROVIDER")
+        or os.getenv("EMBEDDING_PROVIDER")
+        or ""
     ).lower()
-    try:
-        default_idx = LLM_PROVIDERS.index(default_emb)
-    except ValueError:
-        default_idx = 0
-
-    selected = st.sidebar.selectbox(
-        "Embeddings 공급자",
-        options=LLM_PROVIDERS,
-        index=default_idx,
-        key="sidebar_embedding_provider",
+    e_default_index = 0
+    if current_emb_provider:
+        for idx, p in enumerate(e_reg.profiles):
+            if p.provider == current_emb_provider:
+                e_default_index = idx
+                break
+
+    e_sel_name = st.sidebar.selectbox(
+        "Embeddings 프로파일",
+        options=e_names,
+        index=e_default_index,
+        key="sidebar_embedding_profile",
     )
 
-    if selected != default_emb:
+    e_selected = next((p for p in e_reg.profiles if p.name == e_sel_name), None)
+    if e_selected is None:
+        st.sidebar.error("선택한 Embeddings 프로파일을 찾을 수 없습니다.")
+        return
+
+    if st.sidebar.button("Embeddings 적용", key="sidebar_apply_embedding_profile"):
         try:
-            update_embedding_settings(provider=selected, values={})
-            st.sidebar.success(f"Embeddings 공급자가 '{selected}'로 변경되었습니다.")
+            update_embedding_settings(
+                provider=e_selected.provider, values=e_selected.fields
+            )
+            st.sidebar.success(f"Embeddings 프로파일 적용됨: {e_selected.name}")
         except Exception as e:
-            st.sidebar.error(f"Embeddings 공급자 변경 실패: {e}")
+            st.sidebar.error(f"Embeddings 프로파일 적용 실패: {e}")

interface/app_pages/components/llm_selector.py

@@ -1,31 +1,77 @@
 import os
 import streamlit as st
 
-from interface.core.config import update_llm_settings
-from interface.app_pages.settings_sections.llm_section import LLM_PROVIDERS
+from interface.core.config import (
+    update_llm_settings,
+    get_llm_registry,
+)
 
 
 def render_sidebar_llm_selector() -> None:
     st.sidebar.markdown("### LLM 선택")
 
-    default_llm = (
-        (st.session_state.get("LLM_PROVIDER") or os.getenv("LLM_PROVIDER") or "openai")
+    reg = get_llm_registry()
+    if not reg.profiles:
+        st.sidebar.info(
+            "저장된 LLM 프로파일이 없습니다. 설정 > LLM에서 프로파일을 저장하세요."
+        )
+        # 기존 방식 fallback
+        default_llm = (
+            (
+                st.session_state.get("LLM_PROVIDER")
+                or os.getenv("LLM_PROVIDER")
+                or "openai"
+            )
+        ).lower()
+        selected_provider = st.sidebar.selectbox(
+            "LLM 공급자",
+            options=["openai", "azure", "bedrock", "gemini", "ollama", "huggingface"],
+            index=(
+                ["openai", "azure", "bedrock", "gemini", "ollama", "huggingface"].index(
+                    default_llm
+                )
+                if default_llm
+                in {"openai", "azure", "bedrock", "gemini", "ollama", "huggingface"}
+                else 0
+            ),
+            key="sidebar_llm_provider_fallback",
+        )
+        if selected_provider != default_llm:
+            try:
+                update_llm_settings(provider=selected_provider, values={})
+                st.sidebar.success(
+                    f"LLM 공급자가 '{selected_provider}'로 변경되었습니다."
+                )
+            except Exception as e:
+                st.sidebar.error(f"LLM 공급자 변경 실패: {e}")
+        return
+
+    names = [p.name for p in reg.profiles]
+    # 기본 선택: 세션의 LLM_PROVIDER와 같은 provider를 가진 첫 프로파일
+    current_provider = (
+        st.session_state.get("LLM_PROVIDER") or os.getenv("LLM_PROVIDER") or ""
     ).lower()
-    try:
-        default_idx = LLM_PROVIDERS.index(default_llm)
-    except ValueError:
-        default_idx = 0
-
-    selected = st.sidebar.selectbox(
-        "LLM 공급자",
-        options=LLM_PROVIDERS,
-        index=default_idx,
-        key="sidebar_llm_provider",
+    default_index = 0
+    if current_provider:
+        for idx, p in enumerate(reg.profiles):
+            if p.provider == current_provider:
+                default_index = idx
+                break
+
+    sel_name = st.sidebar.selectbox(
+        "LLM 프로파일", options=names, index=default_index, key="sidebar_llm_profile"
     )
+    selected = next((p for p in reg.profiles if p.name == sel_name), None)
+    if selected is None:
+        st.sidebar.error("선택한 LLM 프로파일을 찾을 수 없습니다.")
+        return
 
-    if selected != default_llm:
+    if st.sidebar.button("적용", key="sidebar_apply_llm_profile"):
         try:
-            update_llm_settings(provider=selected, values={})
-            st.sidebar.success(f"LLM 공급자가 '{selected}'로 변경되었습니다.")
+            # provider 설정 + 프로파일의 비민감 필드만 적용
+            update_llm_settings(provider=selected.provider, values=selected.fields)
+            st.sidebar.success(f"LLM 프로파일 적용됨: {selected.name}")
         except Exception as e:
-            st.sidebar.error(f"LLM 공급자 변경 실패: {e}")
+            st.sidebar.error(f"LLM 프로파일 적용 실패: {e}")
+
+    # Embeddings 관련 UI는 embedding_selector.py에서 처리

interface/app_pages/lang2sql.py

@@ -28,6 +28,7 @@
 from interface.app_pages.components.embedding_selector import (
     render_sidebar_embedding_selector,
 )
+from interface.app_pages.components.db_selector import render_sidebar_db_selector
 
 TITLE = "Lang2SQL"
 DEFAULT_QUERY = "고객 데이터를 기반으로 유니크한 유저 수를 카운트하는 쿼리"
@@ -50,6 +51,7 @@
 render_sidebar_data_source_selector(config)
 render_sidebar_llm_selector()
 render_sidebar_embedding_selector()
+render_sidebar_db_selector()
 
 st.sidebar.markdown("### 워크플로우 선택")
 use_enriched = st.sidebar.checkbox(

interface/app_pages/settings.py

@@ -9,6 +9,7 @@
     render_data_source_section,
 )
 from interface.app_pages.settings_sections.llm_section import render_llm_section
+from interface.app_pages.settings_sections.db_section import render_db_section
 
 
 st.title("⚙️ 설정")
@@ -24,7 +25,7 @@
     render_llm_section(config)
 
 with tabs[2]:
-    st.info("DB 연결 설정은 곧 제공됩니다.")
+    render_db_section()
 
 with tabs[3]:
     st.info("디바이스 설정은 곧 제공됩니다.")