test: fix check host (#829)

run-llama · May 10, 2024 · 68a508f · 68a508f
1 parent 6281fc8
commit 68a508f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/core/tests/vitest.setup.ts b/packages/core/tests/vitest.setup.ts
@@ -11,8 +11,8 @@ globalThis.fetch = function fetch(...args: Parameters<typeof originalFetch>) {
       url = url.toString();
     }
   }
-  const parsedUrl = new URL(url);
-  if (parsedUrl.hostname.includes("openai.com")) {
+  const { host } = new URL(url);
+  if (host.endsWith("openai.com")) {
     // todo: mock api using https://mswjs.io
     throw new Error(
       "Make sure to return a mock response for OpenAI API requests in your test.",