arbiter-cli 0.1.0

package/index.mjs

@@ -0,0 +1,129 @@
+#!/usr/bin/env node
+
+/**
+ * Arbiter CLI
+ *
+ * Usage:
+ *   npx arbiter-cli init          Set up Arbiter in current project
+ *   npx arbiter-cli status        Check connection to Arbiter
+ *   npx arbiter-cli stats         View your savings
+ */
+
+import { readFileSync, writeFileSync, existsSync, appendFileSync } from 'fs'
+import { resolve } from 'path'
+
+const ARBITER_URL = 'https://arbiter-proxy-long-summit-6283.fly.dev'
+const ENV_LINE = `OPENAI_BASE_URL=${ARBITER_URL}/v1`
+
+const args = process.argv.slice(2)
+const command = args[0] || 'init'
+
+const colors = {
+  green: (s) => `\x1b[32m${s}\x1b[0m`,
+  cyan: (s) => `\x1b[36m${s}\x1b[0m`,
+  dim: (s) => `\x1b[2m${s}\x1b[0m`,
+  bold: (s) => `\x1b[1m${s}\x1b[0m`,
+  yellow: (s) => `\x1b[33m${s}\x1b[0m`,
+}
+
+console.log('')
+console.log(colors.bold('  Arbiter') + colors.dim(' — AI inference optimization'))
+console.log('')
+
+if (command === 'init') {
+  init()
+} else if (command === 'status') {
+  await status()
+} else if (command === 'stats') {
+  await stats()
+} else {
+  console.log(`  Commands:`)
+  console.log(`    ${colors.cyan('init')}     Set up Arbiter in current project`)
+  console.log(`    ${colors.cyan('status')}   Check connection to Arbiter`)
+  console.log(`    ${colors.cyan('stats')}    View your cost savings`)
+  console.log('')
+}
+
+function init() {
+  const envPath = resolve(process.cwd(), '.env')
+  const gitignorePath = resolve(process.cwd(), '.gitignore')
+
+  // Check if already configured
+  if (existsSync(envPath)) {
+    const content = readFileSync(envPath, 'utf-8')
+    if (content.includes('OPENAI_BASE_URL') && content.includes('arbiter')) {
+      console.log(colors.green('  ✓ Already configured.'))
+      console.log(colors.dim(`    ${envPath}`))
+      console.log('')
+      return
+    }
+  }
+
+  // Add to .env
+  const line = `\n# Arbiter — route all OpenAI calls through the optimizer\n${ENV_LINE}\n`
+
+  if (existsSync(envPath)) {
+    appendFileSync(envPath, line)
+    console.log(colors.green('  ✓ Added to existing .env'))
+  } else {
+    writeFileSync(envPath, `# Arbiter — route all OpenAI calls through the optimizer\n${ENV_LINE}\nOPENAI_API_KEY=sk-your-openrouter-key\n`)
+    console.log(colors.green('  ✓ Created .env'))
+  }
+
+  // Make sure .env is in .gitignore
+  if (existsSync(gitignorePath)) {
+    const gi = readFileSync(gitignorePath, 'utf-8')
+    if (!gi.includes('.env')) {
+      appendFileSync(gitignorePath, '\n.env\n')
+      console.log(colors.green('  ✓ Added .env to .gitignore'))
+    }
+  }
+
+  console.log('')
+  console.log(`  ${colors.bold('Done.')} All OpenAI SDK calls in this project now route through Arbiter.`)
+  console.log('')
+  console.log(colors.dim('  No code changes needed. The OpenAI SDK reads OPENAI_BASE_URL automatically.'))
+  console.log(colors.dim(`  Proxy: ${ARBITER_URL}/v1`))
+  console.log(colors.dim('  Dashboard: ' + ARBITER_URL + '/dashboard'))
+  console.log('')
+}
+
+async function status() {
+  try {
+    const res = await fetch(`${ARBITER_URL}/health`)
+    const data = await res.json()
+    console.log(colors.green('  ✓ Connected'))
+    console.log(colors.dim(`    Status: ${data.status}`))
+    console.log(colors.dim(`    Version: ${data.version}`))
+    console.log(colors.dim(`    Models: ${data.models}`))
+    console.log(colors.dim(`    Requests served: ${data.requests}`))
+  } catch (e) {
+    console.log(colors.yellow('  ✗ Cannot reach Arbiter'))
+    console.log(colors.dim(`    ${e.message}`))
+  }
+  console.log('')
+}
+
+async function stats() {
+  try {
+    const res = await fetch(`${ARBITER_URL}/api/analytics`)
+    const data = await res.json()
+    console.log(`  ${colors.bold('Savings Report')}`)
+    console.log('')
+    console.log(`    Requests routed:  ${colors.cyan(data.total_requests)}`)
+    console.log(`    Actual cost:      ${colors.cyan('$' + data.total_actual_cost.toFixed(4))}`)
+    console.log(`    Would have cost:  ${colors.dim('$' + data.total_baseline_cost.toFixed(4))}`)
+    console.log(`    ${colors.green('Saved: $' + data.total_savings.toFixed(4) + ' (' + data.savings_pct + '%)')}`)
+    console.log('')
+    if (data.models_used) {
+      console.log(`    Models used:`)
+      for (const [model, count] of Object.entries(data.models_used)) {
+        console.log(`      ${colors.dim(model.split('/').pop())}: ${count} requests`)
+      }
+    }
+  } catch (e) {
+    console.log(colors.yellow('  ✗ Cannot reach Arbiter'))
+    console.log(colors.dim(`    ${e.message}`))
+  }
+  console.log('')
+}

package/package.json

@@ -0,0 +1,12 @@
+{
+  "name": "arbiter-cli",
+  "version": "0.1.0",
+  "description": "Set up Arbiter in any project with one command",
+  "bin": {
+    "arbiter": "./index.mjs"
+  },
+  "type": "module",
+  "keywords": ["llm", "proxy", "openai", "cost", "optimization"],
+  "author": "Muhammad Saqib",
+  "license": "MIT"
+}