feat: added api key in vision APIs (#225)

* feat: added api key in vision APIs * chore: added changeset * fix: updated llm version
empirical-run · Nov 14, 2024 · a5eacf0 · a5eacf0
1 parent a63961d
commit a5eacf0
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 9 deletions.
diff --git a/.changeset/cool-pianos-lay.md b/.changeset/cool-pianos-lay.md
@@ -0,0 +1,5 @@
+---
+"appwright": patch
+---
+
+feat: added api key in vision APIs
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -36,7 +36,7 @@
   "license": "Apache-2.0",
   "description": "E2E mobile app testing done right, with the Playwright test runner",
   "dependencies": {
-    "@empiricalrun/llm": "^0.9.21",
+    "@empiricalrun/llm": "^0.9.25",
     "@ffmpeg-installer/ffmpeg": "^1.1.0",
     "@playwright/test": "^1.47.1",
     "appium": "^2.6.0",

diff --git a/src/vision/index.ts b/src/vision/index.ts
@@ -90,8 +90,8 @@ export class VisionProvider {
     options?: { useCache?: boolean },
   ): Promise<{ x: number; y: number }> {
     test.skip(
-      !process.env.VISION_MODEL_ENDPOINT,
-      "LLM vision based tap is not enabled. Set the VISION_MODEL_ENDPOINT environment variable to enable it",
+      !process.env.EMPIRICAL_API_KEY,
+      "LLM vision based tap is not enabled. Set the EMPIRICAL_API_KEY environment variable to enable it",
     );
     const base64Image = await this.webDriverClient.takeScreenshot();
     const coordinates = await getCoordinatesFor(prompt, base64Image, options);