fm CLI and Python SDK for Apple Foundation Models

#!/usr/bin/env bash
# Requires macOS 27+ with the fm CLI pre-installed.
# Usage: ./sort_assets.sh /path/to/assets /path/to/backup /path/to/archive

set -euo pipefail

WORKING_DIR="${1:?Usage: $0 <working_dir> <backup_dir> <archive_dir>}"
BACKUP_DIR="${2:?}"
ARCHIVE_DIR="${3:?}"

mkdir -p "$BACKUP_DIR" "$ARCHIVE_DIR"

# Build a newline-separated file list
FILE_LIST=$(ls "$WORKING_DIR")

# Define the structured output schema once
SCHEMA=$(fm schema object \
  --property finalFiles array string \
  --property draftFiles array string)

# Ask the on-device model to classify the files
RESULT=$(fm respond \
  --schema "$SCHEMA" \
  --prompt "You are a file organiser. Given this list of asset filenames, \
classify each as either a final deliverable or a draft/work-in-progress. \
Return two arrays: finalFiles and draftFiles.\n\nFiles:\n$FILE_LIST")

# Parse the JSON arrays produced by the model
FINAL_FILES=$(echo "$RESULT" | python3 -c \
  "import sys,json; d=json.load(sys.stdin); print('\\n'.join(d['finalFiles']))")
DRAFT_FILES=$(echo "$RESULT" | python3 -c \
  "import sys,json; d=json.load(sys.stdin); print('\\n'.join(d['draftFiles']))")

# Copy final files to backup
while IFS= read -r file; do
  [[ -z "$file" ]] && continue
  cp "$WORKING_DIR/$file" "$BACKUP_DIR/$file"
  echo "Backed up final file: $file"
done <<< "$FINAL_FILES"

# Move draft files to archive
while IFS= read -r file; do
  [[ -z "$file" ]] && continue
  mv "$WORKING_DIR/$file" "$ARCHIVE_DIR/$file"
  echo "Archived draft file: $file"
done <<< "$DRAFT_FILES"

echo "Done. Backup: $BACKUP_DIR | Archive: $ARCHIVE_DIR"