feat(wren-ai-service): create Streamlit UI for configuring LLM models

.github/workflows/setup-provider-image.yaml

@@ -0,0 +1,130 @@
+name: Providers Setup Release Image
+
+on:
+  workflow_dispatch:
+    inputs:
+      tag_name:
+        description: 'Docker image tag name (optional)'
+        type: string
+
+
+# permissions:
+#   contents: read
+#   packages: write
+
+env:
+  REGISTRY: ghcr.io
+  PROVIDER_UI_IMAGE: ${{ github.repository_owner }}/wren-setup-provider-ui
+
+
+jobs:
+  build-image:
+    outputs:
+      tag_name: ${{ steps.prep-tag.outputs.TAG_NAME }}
+
+    strategy:
+      fail-fast: false
+      matrix:
+        arch:
+          - runner: ubuntu-latest         
+            platform: linux/amd64
+          - runner: ubuntu-24.04-arm      
+            platform: linux/arm64
+
+    runs-on: ${{ matrix.arch.runner }}
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Login to GHCR
+        uses: docker/login-action@v3
+        with:
+          registry: ${{ env.REGISTRY }}
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Prepare tag
+        id: prep-tag
+        run: |
+          if [ -n "${{ github.event.inputs.tag_name }}" ]; then
+            TAG="${{ github.event.inputs.tag_name }}"
+          else
+            TAG="commit-$(git rev-parse --short HEAD)"
+          fi
+          echo "TAG_NAME=$TAG" >> "$GITHUB_OUTPUT"
+          echo "TAG=$TAG"       >> "$GITHUB_ENV"
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Prepare platform env
+        run: |
+          echo "PLATFORM_PAIR=${{ matrix.arch.platform // '/' '-'}}" >> "$GITHUB_ENV"
+
+      - name: Build and push by digest
+        id: build
+        uses: docker/build-push-action@v6
+        with:
+          context: ./wren-ai-service/tools/providers-setup
+          file: ./wren-ai-service/tools/providers-setup/Dockerfile
+          platforms: ${{ matrix.arch.platform }}
+          labels: ${{ env.REGISTRY }}/${{ env.PROVIDER_UI_IMAGE }}
+          outputs: |
+            type=image,\
+            name=${{ env.REGISTRY }}/${{ env.PROVIDER_UI_IMAGE }},\
+            push-by-digest=true,\
+            name-canonical=true,\
+            push=true
+
+      - name: Export digest
+        run: |
+          mkdir -p /tmp/digests
+          digest="${{ steps.build.outputs.digest }}"
+          touch "/tmp/digests/${digest#sha256:}"
+
+      - name: Upload digest artifact
+        uses: actions/upload-artifact@v4
+        with:
+          name: digests-${{ env.PLATFORM_PAIR }}
+          path: /tmp/digests/*
+          if-no-files-found: error
+          retention-days: 1
+
+  merge:
+    needs: [build-image]
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Download digests
+        uses: actions/download-artifact@v4
+        with:
+          path: /tmp/digests
+          pattern: digests-*
+          merge-multiple: true
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Docker meta (tag list)
+        id: meta
+        uses: docker/metadata-action@v5
+        with:
+          images: ${{ env.REGISTRY }}/${{ env.PROVIDER_UI_IMAGE }}
+          tags: |
+            ${{ needs.build-image.outputs.tag_name }}
+
+      - name: Login to GHCR
+        uses: docker/login-action@v3
+        with:
+          registry: ${{ env.REGISTRY }}
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Create & push manifest list
+        working-directory: /tmp/digests
+        run: |
+          TAGS=$(jq -cr '.tags | map("--tag " + .) | join(" ")' <<< "$DOCKER_METADATA_OUTPUT_JSON")
+          docker buildx imagetools create \
+          $(printf '${{ env.REGISTRY }}/${{ env.PROVIDER_UI_IMAGE }}@sha256:%s ' *) \
+          $TAGS
+

wren-ai-service/tools/providers-setup/Dockerfile

@@ -0,0 +1,58 @@
+# Use official slim Python 3.12 base image
+FROM python:3.12.0-slim
+
+# -------------------------------
+# System Dependencies for Poetry
+# -------------------------------
+# Install minimal system packages: curl (for downloading), build tools (for native extensions)
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    curl build-essential gcc \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+# -------------------------------
+# Install Poetry (Python package manager)
+# -------------------------------
+ENV POETRY_VERSION=1.8.2
+ENV POETRY_HOME="/opt/poetry"
+ENV PATH="$POETRY_HOME/bin:$PATH"
+ENV POETRY_VIRTUALENVS_CREATE=false  
+# Don't use virtualenvs inside the container
+
+RUN curl -sSL https://install.python-poetry.org | python3 - \
+    && ln -s $POETRY_HOME/bin/poetry /usr/local/bin/poetry
+
+# -------------------------------
+# Set working directory for app
+# -------------------------------
+WORKDIR /app
+
+# -------------------------------
+# Install Python dependencies via Poetry
+# -------------------------------
+# Copy only dependency files first to leverage Docker layer caching
+COPY pyproject.toml poetry.lock ./
+RUN poetry install --no-interaction --no-ansi
+
+# -------------------------------
+# Copy remaining app code
+# -------------------------------
+COPY . .
+
+# -------------------------------
+# Environment variables for Streamlit
+# -------------------------------
+ENV PYTHONUNBUFFERED=1
+ENV STREAMLIT_SERVER_HEADLESS=true
+ENV STREAMLIT_SERVER_PORT=8501
+ENV STREAMLIT_SERVER_ENABLECORS=false
+
+# -------------------------------
+# Expose Streamlit port
+# -------------------------------
+EXPOSE 8501
+
+# -------------------------------
+# Default command to run the Streamlit app
+# -------------------------------
+CMD ["streamlit", "run", "app.py"]

wren-ai-service/tools/providers-setup/app.py

@@ -0,0 +1,89 @@
+from config_loader import load_config_yaml_blocks, group_blocks
+from session_state import ConfigState
+from ui_components import (
+    render_llm_config, 
+    render_embedder_config,  
+    render_import_yaml, 
+    render_pipeline_config,
+    render_preview,
+    render_apikey,
+    render_generate_button
+)
+import streamlit as st
+
+# Set Streamlit page layout
+st.set_page_config(
+    layout="wide",                # Use a wide layout for better horizontal space
+    initial_sidebar_state="expanded"  # Expand sidebar by default
+)
+
+# Load and group configuration blocks from YAML
+yaml_list = load_config_yaml_blocks()
+blocks = group_blocks(yaml_list)
+
+# Retrieve individual configuration sections
+llm_block = blocks.get("llm", {})
+embedder_block = blocks.get("embedder", {})
+document_store_block = blocks.get("document_store", {})
+engine_blocks = blocks.get("engine", [])
+pipeline_block = blocks.get("pipeline", {})
+settings_block = blocks.get("settings", {})
+
+
+# Validate required blocks (type + content)
+missing_blocks = []
+
+if not isinstance(llm_block, dict) or not llm_block:
+    missing_blocks.append("LLM")
+if not isinstance(embedder_block, dict) or not embedder_block:
+    missing_blocks.append("Embedder")
+if not isinstance(document_store_block, dict) or not document_store_block:
+    missing_blocks.append("Document Store")
+if not isinstance(pipeline_block, dict) or not pipeline_block:
+    missing_blocks.append("Pipeline")
+
+if missing_blocks:
+    st.warning(
+        f"⚠️ Missing or empty configuration blocks: {', '.join(missing_blocks)}. "
+        "Default values will be used where applicable."
+    )
+
+# Initialize session state with default or imported config values
+ConfigState.init(llm_block, embedder_block, document_store_block, pipeline_block)
+
+# ----------------------
+# Streamlit UI rendering
+# ----------------------
+st.title("Custom Provider Config Generator")
+
+# Layout: two columns – left for inputs, right for preview/export
+col1, col2 = st.columns([1.5, 1])  
+
+with col1:
+
+    # API key input section
+    st.subheader("API_KEY Configuration")
+    render_apikey()
+
+    # Upload and parse YAML file into session state
+    st.subheader("LLM Configuration") 
+    render_import_yaml()
+
+    # LLM model configuration UI
+    render_llm_config()
+
+    # Embedding model configuration UI
+    st.subheader("Embedder Configuration")
+    render_embedder_config()
+
+    # Pipeline flow configuration UI
+    st.subheader("Pipeline Configuration")
+    render_pipeline_config()
+
+    # Generate config.yaml and save configuration button
+    render_generate_button(engine_blocks, settings_block)
+
+with col2:
+    # Final preview and export of the combined configuration as YAML
+    render_preview(engine_blocks, settings_block)
+