Merge pull request #643 from Kiln-AI/main

Docs update for v0.21 release

Author: scosman

.github/workflows/check_api_bindings.yml

@@ -0,0 +1,69 @@
+name: Check API Bindings
+
+on:
+  push:
+
+jobs:
+  check-api-bindings:
+    name: Check API Schema Bindings
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v3
+        with:
+          enable-cache: true
+
+      - name: Set up Python
+        run: uv python install
+
+      - name: Install the project
+        run: uv sync --all-extras --dev
+
+      - name: Set up Node.js
+        uses: actions/setup-node@v4
+        with:
+          node-version: "24"
+          cache: "npm"
+          cache-dependency-path: app/web_ui/package-lock.json
+
+      - name: Install Node.js dependencies
+        run: npm ci
+        working-directory: app/web_ui
+
+      - name: Mock Studio Web UI
+        run: mkdir -p ./app/web_ui/build && echo "test" > ./app/web_ui/build/index.html
+
+      - name: Check API schema bindings
+        run: |
+          # Start dev server in background
+          uv run python -m app.desktop.dev_server &
+          DEV_SERVER_PID=$!
+
+          # Wait for server to be ready
+          echo "Waiting for dev server to start..."
+          for i in {1..30}; do
+            if curl -s -f http://localhost:8757/openapi.json > /dev/null 2>&1; then
+              echo "Dev server is ready"
+              break
+            fi
+            if [ $i -eq 30 ]; then
+              echo "Dev server failed to start within 30 seconds"
+              kill $DEV_SERVER_PID || true
+              exit 1
+            fi
+            sleep 1
+          done
+
+          # Change to the correct directory and run the schema check
+          cd app/web_ui/src/lib
+          ./check_schema.sh
+
+          # Stop dev server
+          kill $DEV_SERVER_PID || true
+          # Wait a moment for graceful shutdown
+          sleep 2
+          # Force kill if still running
+          kill -9 $DEV_SERVER_PID 2>/dev/null || true

.github/workflows/windows_release_build.yml

@@ -49,7 +49,7 @@ jobs:
           timestamp-digest: SHA256
 
       - name: Build Windows Installer
-        uses: Minionguyjpro/Inno-Setup-Action@v1.2.5
+        uses: Minionguyjpro/Inno-Setup-Action@v1.2.7
         with:
           path: ./app/desktop/WinInnoSetup.iss
 

.gitignore

@@ -8,6 +8,7 @@ __pycache__/
 .coverage
 **/.venv
 **/*.egg-info
+node_modules/
 
 libs/core/docs
 libs/core/build

README.md

@@ -39,6 +39,7 @@
 - 📊 [**Evals**](https://docs.kiln.tech/docs/evaluations): Evaluate the quality of your models/tasks using state of the art evaluators.
 - 🎛️ [**Fine Tuning**](https://docs.kiln.tech/docs/fine-tuning-guide): Zero-code fine-tuning for Llama, GPT-4o, and more. Automatic serverless deployment of models.
 - 🤖 [**Synthetic Data Generation**](https://docs.kiln.tech/docs/synthetic-data-generation): Generate eval datasets or fine-tuning data with our interactive visual tooling.
+- 🛠 [**Tools & MCP**](https://docs.kiln.tech/docs/tools-and-mcp): Connect powerful tools to your Kiln tasks
 - 🧠 [**Reasoning Models**](https://docs.kiln.tech/docs/guide-train-a-reasoning-model): Train or distill your own custom reasoning models.
 - 📝 [**Prompt Generation**](https://docs.kiln.tech/docs/prompts): Automatically generate prompts including chain-of-thought, few-shot, and multi-shot, and more.
 - 🌐 [**Comprehensive Model Support**](https://kiln.tech/model_library): Skip the guesswork — we've tested over 100 models' capabilities. Use any model via Ollama, OpenAI, OpenRouter, Fireworks, Groq, AWS, any OpenAI compatible API, and more.

app/desktop/WinInnoSetup.iss

@@ -3,7 +3,7 @@
 
 #define MyAppPath "build\dist\Kiln"
 #define MyAppName "Kiln"
-#define MyAppVersion "0.20.0"
+#define MyAppVersion "0.20.1"
 #define MyAppPublisher "Chesterfield Laboratories Inc"
 #define MyAppURL "https://kiln.tech"
 #define MyAppExeName "Kiln.exe"

app/desktop/desktop.py

@@ -10,6 +10,7 @@
 import tkinter as tk
 import webbrowser
 
+from kiln_ai.utils.config import Config
 from PIL import Image
 
 # Unused, but needed for pyinstaller to not miss this import
@@ -19,6 +20,12 @@
 from app.desktop.custom_tray import KilnMenuItem, KilnTray
 from app.desktop.desktop_server import ThreadedServer, server_config
 
+# Set writeable cache directories as soon as we start
+os.environ["LLAMA_INDEX_CACHE_DIR"] = os.path.join(
+    Config.settings_dir(), "cache", "llama_index_cache"
+)
+os.environ["NLTK_DATA"] = os.path.join(Config.settings_dir(), "cache", "nltk_data")
+
 
 class DesktopApp:
     def __init__(self, port: int = 8757):

app/desktop/studio_server/data_gen_api.py

@@ -155,6 +155,21 @@ async def generate_samples(
 
     @app.post("/api/projects/{project_id}/tasks/{task_id}/save_sample")
     async def save_sample(
+        project_id: str,
+        task_id: str,
+        task_run: TaskRun,
+    ) -> TaskRun:
+        """
+        Save a sample generated by the generate_sample endpoint.
+        """
+        task = task_from_id(project_id, task_id)
+        # Set parent to task to ensure the correct path is used
+        task_run.parent = task
+        task_run.save_to_file()
+        return task_run
+
+    @app.post("/api/projects/{project_id}/tasks/{task_id}/generate_sample")
+    async def generate_sample(
         project_id: str,
         task_id: str,
         sample: DataGenSaveSamplesApiInput,
@@ -212,7 +227,6 @@ async def save_sample(
             tags.extend(sample.tags)
         run.tags = tags
 
-        run.save_to_file()
         return run