budibase/scripts/syncLiteLLMVersion.js

#!/usr/bin/env node
const fs = require("fs")
const path = require("path")

const repoRoot = path.resolve(__dirname, "..")
const versionConfigPath = path.join(repoRoot, "hosting", "litellm-version.json")

function readJson(filePath) {
  return JSON.parse(fs.readFileSync(filePath, "utf8"))
}

function updateFile(filePath, updater) {
  const original = fs.readFileSync(filePath, "utf8")
  const updated = updater(original)

  return {
    changed: original !== updated,
    updated,
  }
}

function required(value, label) {
  if (!value || typeof value !== "string") {
    throw new Error(`Invalid ${label}`)
  }
  return value
}

function replaceRequired(content, pattern, replacement, filePath) {
  if (!pattern.test(content)) {
    throw new Error(`Could not find target pattern in ${filePath}`)
  }
  return content.replace(pattern, replacement)
}

function main() {
  const checkOnly = process.argv.includes("--check")
  const versionConfig = readJson(versionConfigPath)
  const version = required(versionConfig.version, "version")
  const channel = required(versionConfig.channel, "channel")

  const imageTag = `main-v${version}-${channel}`

  const dockerHubImagePattern =
    /image:\s*docker\.io\/litellm\/litellm:main-v(\d+\.\d+\.\d+)-([a-zA-Z0-9]+)((?:\.[a-zA-Z0-9]+)*)/
  const dockerHubImageReplacement = (
    _match,
    existingVersion,
    existingChannel,
    suffix
  ) => {
    const isSameRelease =
      existingVersion === version && existingChannel === channel
    return `image: docker.io/litellm/litellm:${imageTag}${
      isSameRelease ? suffix : ""
    }`
  }

  const updates = [
    {
      path: path.join(repoRoot, "hosting", "single", "Dockerfile"),
      update: content =>
        replaceRequired(
          content,
          /ARG LITELLM_PYPI_VERSION=.*/,
          `ARG LITELLM_PYPI_VERSION=${version}`,
          "hosting/single/Dockerfile"
        ),
    },
    {
      path: path.join(repoRoot, "hosting", "docker-compose.yaml"),
      update: content =>
        replaceRequired(
          content,
          dockerHubImagePattern,
          dockerHubImageReplacement,
          "hosting/docker-compose.yaml"
        ),
    },
    {
      path: path.join(repoRoot, "hosting", "docker-compose.build.yaml"),
      update: content =>
        replaceRequired(
          content,
          dockerHubImagePattern,
          dockerHubImageReplacement,
          "hosting/docker-compose.build.yaml"
        ),
    },
    {
      path: path.join(repoRoot, "hosting", "docker-compose.dev.yaml"),
      update: content =>
        replaceRequired(
          content,
          dockerHubImagePattern,
          dockerHubImageReplacement,
          "hosting/docker-compose.dev.yaml"
        ),
    },
    {
      path: path.join(repoRoot, "charts", "budibase", "values.yaml"),
      update: content =>
        replaceRequired(
          content,
          /image:\s*ghcr\.io\/berriai\/litellm:[^\s\n]+/,
          `image: ghcr.io/berriai/litellm:${imageTag}`,
          "charts/budibase/values.yaml"
        ),
    },
  ]

  const changed = []
  for (const item of updates) {
    const result = updateFile(item.path, item.update)
    if (result.changed) {
      if (!checkOnly) {
        fs.writeFileSync(item.path, result.updated)
      }
      changed.push(path.relative(repoRoot, item.path))
    }
  }

  console.log(
    `LiteLLM version ${
      checkOnly ? "check" : "sync"
    }: version=${version}, channel=${channel}`
  )
  if (changed.length > 0) {
    const header = checkOnly ? "Out-of-sync files" : "Updated files"
    console.log(`${header}:\n${changed.join("\n")}`)
    if (checkOnly) {
      process.exit(1)
    }
  } else {
    console.log(checkOnly ? "All files are in sync" : "No files changed")
  }
}

main()