feat: add selectbox for users to choose config.example.yaml by LLM pr…

…ovider

wren-ai-service/streamlit-ui/app.py

@@ -1,5 +1,4 @@
-import constants as cst
-from config_loader import download_config, load_yaml_list, group_blocks
+from config_loader import load_config_yaml_blocks, group_blocks
 from session_state import ConfigState
 from ui_components import (
     render_llm_config, 
@@ -18,13 +17,7 @@
     initial_sidebar_state="expanded"  # 控制側邊欄的初始狀態
 )
 
-if not cst.CONFIG_IN_PATH.exists():
-    download_state, download_msg = download_config()
-    if not download_state:
-        st.error(download_msg)
-
-
-yaml_list = load_yaml_list(cst.CONFIG_IN_PATH)
+yaml_list = load_config_yaml_blocks()
 blocks = group_blocks(yaml_list)
 
 llm_block = blocks.get("llm", {})
@@ -43,6 +36,7 @@
 ConfigState.init(llm_block, embedder_block, document_store_block, pipeline_block)
 
 with col1:
+    st.subheader("LLM Configuration")
     # =====================
     # LLM Configuration UI
     # =====================

wren-ai-service/streamlit-ui/config_loader.py

@@ -1,19 +1,65 @@
-# config_loader.py
 import requests
 import yaml
+from session_state import ConfigState
 from pathlib import Path
 import constants as cst
 from typing import Any, Dict, List 
 
-def download_config():
+def load_config_yaml_blocks() -> List[Dict[str, Any]]:
+    """
+    嘗試從本地讀 config.yaml，若不存在則從 GitHub 讀取（不下載）。
+    """
+    if cst.CONFIG_IN_PATH.exists():
+        try:
+            return load_yaml_list(cst.CONFIG_IN_PATH)
+        except Exception as e:
+            print(f"❌ Failed to parse local config.yaml: {e}")
+            return []
+    else:
+        return fetch_yaml_from_url(cst.CONFIG_URL)
+
+def load_selected_example_yaml(example_name: str) -> List[Dict[str, Any]]:
+    url = cst.CONFIG_EXAMPLES_SELECTED_URL + example_name
+    return fetch_yaml_from_url(url)
+
+def fetch_yaml_from_url(url: str) -> List[Dict[str, Any]]:
     try:
-        response = requests.get(cst.CONFIG_URL, timeout=cst.REQUEST_TIMEOUT)
+        response = requests.get(url, timeout=cst.REQUEST_TIMEOUT)
         response.raise_for_status()
-        cst.CONFIG_IN_PATH.write_text(response.text, encoding='utf-8')
-        return True, "Download Successed"
-    except requests.RequestException as e:
-        return False, str(e)
+        config_list = list(yaml.safe_load_all(response.text))
+
+        if not config_list:
+            raise ValueError(f"⚠️ GitHub 回傳的 YAML 是空的，URL: {url}")
+
+        return config_list
+
+    except (requests.RequestException, ValueError, yaml.YAMLError) as e:
+        print(f"❌ Error loading config from {url}: {e}")
+        return []
 
+def extract_config_blocks(config_list: List[Dict[str, Any]]) -> Dict[str, Any]:
+    """
+    傳入 YAML List，回傳各 block 分類後的 dict
+    """
+    grouped = group_blocks(config_list)
+    return {
+        "llm": grouped.get("llm", {}),
+        "embedder": grouped.get("embedder", {}),
+        "document_store": grouped.get("document_store", {}),
+        "pipeline": grouped.get("pipeline", {})
+    }
+
+def load__selected_config_yaml_blocks(selected_examples) -> List[Dict[str, Any]]:
+        selected_url = cst.CONFIG_EXAMPLES_SELECTED_URL + selected_examples
+        # return selected_url
+        try:
+            response = requests.get(selected_url, timeout=cst.REQUEST_TIMEOUT)
+            response.raise_for_status()
+            return list(yaml.safe_load_all(response.text))
+        except requests.RequestException as e:
+            print(f"❌ Error loading config from GitHub: {e}")
+            return []
+
 def load_yaml_list(path: Path) -> List[Dict[str, Any]]:
     with path.open("r", encoding="utf-8") as f:
         return list(yaml.safe_load_all(f))
@@ -33,3 +79,25 @@ def group_blocks(blocks: List[Dict[str, Any]]) -> Dict[str, Any]:
             save_blocks[key] = block
     return save_blocks
 
+def fetch_example_yaml_filenames() -> List[str]:
+    """從 GitHub 的 config_examples 目錄中取得所有 .yaml 檔案名稱（不載入內容）"""
+    try:
+        response = requests.get(cst.CONFIG_EXAMPLES_URL, timeout=cst.REQUEST_TIMEOUT)
+        response.raise_for_status()
+        file_list = response.json()
+        return [f["name"] for f in file_list if f["name"].endswith(".yaml")]
+    except requests.RequestException as e:
+        print(f"Error fetching config example filenames: {e}")
+        return []
+
+def apply_config_blocks(config_blocks: List[Dict[str, Any]]):
+    grouped = extract_config_blocks(config_blocks)
+
+    # 更新 ConfigState
+    ConfigState.init(
+        grouped["llm"],
+        grouped["embedder"],
+        grouped["document_store"],
+        grouped["pipeline"],
+        force=True
+    )

wren-ai-service/streamlit-ui/constants.py

@@ -18,6 +18,8 @@ def get_latest_config_version():
 # --- constant ---
 CONFIG_VERSION = get_latest_config_version()
 CONFIG_URL = f"https://raw.githubusercontent.com/Canner/WrenAI/{CONFIG_VERSION}/docker/config.example.yaml"
+CONFIG_EXAMPLES_URL = f"https://api.github.com/repos/Canner/WrenAI/contents/wren-ai-service/docs/config_examples?ref={CONFIG_VERSION}"
+CONFIG_EXAMPLES_SELECTED_URL = f"https://raw.githubusercontent.com/Canner/WrenAI/{CONFIG_VERSION}/wren-ai-service/docs/config_examples/"
 CONFIG_IN_PATH = Path("config.yaml")
 CONFIG_OUT_PATH = Path("generated_config.yaml")
 REQUEST_TIMEOUT = 10  # seconds

wren-ai-service/streamlit-ui/session_state.py

@@ -10,6 +10,7 @@ class ConfigState:
     PIPELINE_KEY = "pipeline"
     API_KEY = "api_key"
     API_KEY_FORM = "api_key_form"
+    EXAMPLE_CONFIG_NAMES_KEY = "example_yaml_names"
 
     @classmethod
     def init(cls, llm_block, embedder_block, document_store_block, pipeline_block,force=False):
@@ -19,6 +20,7 @@ def init(cls, llm_block, embedder_block, document_store_block, pipeline_block,fo
         cls.init_document_store(document_store_block, force=force)
         cls.init_pipeline(pipeline_block, force=force)
         cls.init_apikey()
+        cls.init_example_configs()
 
     @classmethod
     def init_llm_forms(cls, llm_block, force=False):
@@ -103,11 +105,15 @@ def init_pipeline(cls, pipeline_block, force=False):
         }
 
     @classmethod
-    def init_apikey(cls):
+    def init_apikey(cls, force=False):
         if cls.API_KEY not in st.session_state:
             st.session_state[cls.API_KEY] = {}
         if ConfigState.API_KEY_FORM not in st.session_state:
             st.session_state[cls.API_KEY_FORM] = []
 
+    @classmethod
+    def init_example_configs(cls, force=False):
+        from config_loader import fetch_example_yaml_filenames  # 避免循環 import，要晚點 import
+        st.session_state[cls.EXAMPLE_CONFIG_NAMES_KEY] = fetch_example_yaml_filenames()
     # 可以額外做 CRUD：新增 LLM、刪除 LLM、更新 Embedder...
 

wren-ai-service/streamlit-ui/ui_components.py

@@ -1,11 +1,32 @@
 import streamlit as st
 import uuid
 from session_state import ConfigState
-from config_loader import group_blocks
+from config_loader import load_selected_example_yaml, apply_config_blocks
 from dry_run_test import llm_completion_test, llm_embedding_test
 import yaml
 import os
 
+def render_import_yaml():
+    with st.expander("Import Configuration YAML", expanded=False):
+        # IMPORT LATESET VERSION OF CONFIG.EXAMPLES.YAML FORM GITHUB
+        example_names = ["None"] + st.session_state.get("example_yaml_names", [])
+        selected_examples_yaml  = st.selectbox("Select config.examples.yaml from WrenAI", options=example_names)
+
+        if selected_examples_yaml != "None" and st.button("Import.yaml", key="import__examples_yaml"):
+            blocks = load_selected_example_yaml(selected_examples_yaml)
+            apply_config_blocks(blocks)
+            st.success("YAML 匯入成功，設定已更新。")
+
+        uploaded_file = st.file_uploader("Choose your own YAML file", type=["yaml", "yml"])
+        if uploaded_file is not None and st.button("Import.yaml", key="import_own_yaml"):
+            try:
+                blocks = list(yaml.safe_load_all(uploaded_file))
+                apply_config_blocks(blocks)
+            except Exception as e:
+                st.error(f"匯入 YAML 檔案時發生錯誤: {e}")
+
+
+
 def render_apikey():
     with st.expander("API Key", expanded=False):
 
@@ -20,9 +41,9 @@ def render_apikey():
 
             with kcol:
                 if apikey.get("is_saved"):
-                    st.text_input("apikey_service", key=f"api_key_{apikey['id']}", value=apikey["key"], disabled=True)
+                    st.text_input("apikey_service (LLMAI_API_KEY)", key=f"api_key_{apikey['id']}", value=apikey["key"], disabled=True)
                 else:
-                    apikey["key"] = st.text_input("apikey_service", key=f"api_key_{apikey['id']}", value=apikey["key"])
+                    apikey["key"] = st.text_input("apikey_service (LLMAI_API_KEY)", key=f"api_key_{apikey['id']}", value=apikey["key"])
 
             with vcol:
                 if apikey.get("is_saved"):
@@ -65,35 +86,6 @@ def render_apikey():
                 item["is_saved"] = True
             st.rerun()
 
-def render_import_yaml():
-    st.subheader("LLM Configuration")
-    # IMPORT YAML
-    uploaded_file = st.file_uploader("Choose a YAML file", type=["yaml", "yml"])
-
-    if uploaded_file is not None:
-        if st.button("Import.yaml", key="import_yaml"):
-            try:
-                # 解析使用者上傳的 yaml 檔案
-                user_config_list = list(yaml.safe_load_all(uploaded_file))
-                user_config_block = group_blocks(user_config_list)  # 將 YAML 轉換為字典格式
-
-                # 用於更新的暫存區
-                user_llm_block = user_config_block.get("llm", {})
-                user_embedder_block = user_config_block.get("embedder", {})
-                user_document_store_block = user_config_block.get("document_store", {})
-                user_pipeline_block = user_config_block.get("pipeline", {})
-
-                # 僅在有新資料時才更新對應的 block
-                if user_llm_block: llm_block = user_llm_block
-                if user_embedder_block: embedder_block = user_embedder_block
-                if user_document_store_block: document_store_block = user_document_store_block
-                if user_pipeline_block: pipeline_block = user_pipeline_block
-
-                ConfigState.init(llm_block, embedder_block, document_store_block, pipeline_block,force=True)  # 強制重新初始化 Session State
-                st.success("YAML 匯入成功，設定已更新。")
-            except Exception as e:
-                st.error(f"匯入 YAML 檔案時發生錯誤: {e}")
-
 def render_llm_config():
 
     # 用一個 dict 來暫存每個 form_id 對應的 title