[Azure] Update Batch API (#871)

openai · May 31, 2024 · a70f641 · a70f641
1 parent ed4219a
commit a70f641
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -492,7 +492,6 @@ const _deployments_endpoints = new Set([
  '/audio/translations',
  '/audio/speech',
  '/images/generations',
- '/batches',
 ]);
 
 const API_KEY_SENTINEL = '<Missing Key>';

diff --git a/tests/lib/azure.test.ts b/tests/lib/azure.test.ts
@@ -290,15 +290,15 @@ describe('azure request building', () => {
  fetch: testFetch,
  });
 
- test('handles Batch', async () => {
+ test('handles batch', async () => {
  expect(
  await client.batches.create({
  completion_window: '24h',
  endpoint: '/v1/chat/completions',
  input_file_id: 'file-id',
  }),
  ).toStrictEqual({
- url: `https://example.com/openai/deployments/${deployment}/batches?api-version=${apiVersion}`,
+ url: `https://example.com/openai/batches?api-version=${apiVersion}`,
  });
  });
 
@@ -423,7 +423,7 @@ describe('azure request building', () => {
  fetch: testFetch,
  });
 
- test('Batch is not handled', async () => {
+ test('handles batch', async () => {
  expect(
  await client.batches.create({
  completion_window: '24h',