From 84af476c75589baa4a260d78d193d4d104a4ccd5 Mon Sep 17 00:00:00 2001
From: Guy Stone <GuyStoneExtra@gmail.com>
Date: Fri, 24 Jan 2025 22:41:51 +0000
Subject: [PATCH] Add run_demo.sh for llm_routing demo (#381)

* Create run_demo.sh for llm_routing demo

* Update run_demo.sh
---
 demos/llm_routing/run_demo.sh | 47 +++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 demos/llm_routing/run_demo.sh

diff --git a/demos/llm_routing/run_demo.sh b/demos/llm_routing/run_demo.sh
new file mode 100644
index 00000000..c0eafee6
--- /dev/null
+++ b/demos/llm_routing/run_demo.sh
@@ -0,0 +1,47 @@
+#!/bin/bash
+set -e
+
+# Function to start the demo
+start_demo() {
+  # Step 1: Check if .env file exists
+  if [ -f ".env" ]; then
+    echo ".env file already exists. Skipping creation."
+  else
+    # Step 2: Create `.env` file and set OpenAI key
+    if [ -z "$OPENAI_API_KEY" ]; then
+      echo "Error: OPENAI_API_KEY environment variable is not set for the demo."
+      exit 1
+    fi
+
+    echo "Creating .env file..."
+    echo "OPENAI_API_KEY=$OPENAI_API_KEY" > .env
+    echo ".env file created with OPENAI_API_KEY."
+  fi
+
+  # Step 3: Start Arch
+  echo "Starting Arch with arch_config.yaml..."
+  archgw up arch_config.yaml
+
+  # Step 4: Start LLM Routing
+  echo "Starting LLM Routing using Docker Compose..."
+  docker compose up -d  # Run in detached mode
+}
+
+# Function to stop the demo
+stop_demo() {
+  # Step 1: Stop Docker Compose services
+  echo "Stopping LLM Routing using Docker Compose..."
+  docker compose down
+
+  # Step 2: Stop Arch
+  echo "Stopping Arch..."
+  archgw down
+}
+
+# Main script logic
+if [ "$1" == "down" ]; then
+  stop_demo
+else
+  # Default action is to bring the demo up
+  start_demo
+fi