diff --git a/apps/docs/package.json b/apps/docs/package.json
index c59b1dcf3c4..ea2cbed2113 100644
--- a/apps/docs/package.json
+++ b/apps/docs/package.json
@@ -102,11 +102,13 @@
     "unist-builder": "^3.0.1",
     "unist-util-filter": "^4.0.1",
     "uuid": "^9.0.0",
-    "valtio": "^1.7.6"
+    "valtio": "^1.7.6",
+    "yargs": "^17.7.2"
   },
   "devDependencies": {
     "@types/node": "^17.0.12",
     "@types/react": "17.0.39",
+    "@types/yargs": "^17.0.24",
     "config": "*",
     "dotenv": "^16.0.3",
     "ejs": "^3.1.8",
diff --git a/apps/docs/scripts/search/generate-embeddings.ts b/apps/docs/scripts/search/generate-embeddings.ts
index 490861209d2..c78ef004243 100644
--- a/apps/docs/scripts/search/generate-embeddings.ts
+++ b/apps/docs/scripts/search/generate-embeddings.ts
@@ -5,13 +5,18 @@ import { Configuration, OpenAIApi } from 'openai'
 import { inspect } from 'util'
 import { v4 as uuidv4 } from 'uuid'
 import { fetchSources } from './sources'
+import yargs from 'yargs'
 
 dotenv.config()
 
 async function generateEmbeddings() {
-  // TODO: use better CLI lib like yargs
-  const args = process.argv.slice(2)
-  const shouldRefresh = args.includes('--refresh')
+  const argv = await yargs.option('refresh', {
+    alias: 'r',
+    description: 'Refresh data',
+    type: 'boolean',
+  }).argv
+
+  const shouldRefresh = argv.refresh
 
   const requiredEnvVars = [
     'NEXT_PUBLIC_SUPABASE_URL',