Update tests

heroku · Nov 12, 2024 · 6f22d85 · 6f22d85
1 parent 8bd8fc0
commit 6f22d85
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/test/commands/ai/models/call.test.ts b/test/commands/ai/models/call.test.ts
@@ -30,7 +30,7 @@ describe('ai:models:call', function () {
     process.env = {}
     sandbox = sinon.createSandbox()
     api = nock('https://api.heroku.com')
-    defaultInferenceApi = nock('https://inference.heroku.com')
+    defaultInferenceApi = nock('https://us.inference.heroku.com')
       .get('/available-models')
       .reply(200, availableModels)
   })

diff --git a/test/commands/ai/models/info.test.ts b/test/commands/ai/models/info.test.ts
@@ -17,7 +17,7 @@ describe('ai:models:info', function () {
     beforeEach(function () {
       process.env = {}
       api = nock('https://api.heroku.com:443')
-      herokuAI = nock('https://inference.heroku.com')
+      herokuAI = nock('https://us.inference.heroku.com')
     })
 
     afterEach(function () {
@@ -36,7 +36,7 @@ describe('ai:models:info', function () {
         .reply(200, {
           INFERENCE_KEY: 's3cr3t_k3y',
           INFERENCE_MODEL_ID: 'claude-3-haiku',
-          INFERENCE_URL: 'inference.heroku.com',
+          INFERENCE_URL: 'us.inference.heroku.com',
         })
       herokuAI
         .get('/models/claude-3-haiku')
@@ -88,7 +88,7 @@ describe('ai:models:info', function () {
         .reply(200, {
           INFERENCE_KEY: 's3cr3t_k3y',
           INFERENCE_MODEL_ID: 'claude-3-haiku',
-          INFERENCE_URL: 'inference.heroku.com',
+          INFERENCE_URL: 'us.inference.heroku.com',
         })
       herokuAI
         .get('/models/claude-3-haiku')
@@ -105,7 +105,7 @@ describe('ai:models:info', function () {
         .reply(200, {
           INFERENCE_KEY: 's3cr3t_k3y',
           INFERENCE_MODEL_ID: 'claude-3-haiku',
-          INFERENCE_URL: 'inference.heroku.com',
+          INFERENCE_URL: 'us.inference.heroku.com',
         })
       herokuAI
         .get('/models/claude-3-haiku')

diff --git a/test/commands/ai/models/list.test.ts b/test/commands/ai/models/list.test.ts
@@ -13,7 +13,7 @@ describe('ai:models:list', function () {
 
   beforeEach(function () {
     process.env = {}
-    herokuAI = nock('https://inference.heroku.com')
+    herokuAI = nock('https://us.inference.heroku.com')
   })
 
   afterEach(function () {

diff --git a/test/lib/base.test.ts b/test/lib/base.test.ts
@@ -84,7 +84,7 @@ describe('attempt a request using the Heroku AI client', function () {
 
   context('when the command doesn’t require a resource name', function () {
     it('makes a request to the default host', async function () {
-      const defaultApiHost = nock('https://inference.heroku.com')
+      const defaultApiHost = nock('https://us.inference.heroku.com')
         .get('/models')
         .reply(200, [])