Fix tests to supply invocation param

robatwilliams · Dec 19, 2023 · a2d5e1a · a2d5e1a
1 parent 083e54d
commit a2d5e1a
Showing 1 changed file with 23 additions and 11 deletions.
diff --git a/src/functions/functions.test.mjs b/src/functions/functions.test.mjs
@@ -12,6 +12,7 @@ describe('CHAT_COMPLETE', () => {
     const completion = await chatComplete(
       [['user', 'Say hello']],
       [['API_KEY', 'someapikey']],
+      {},
     );
 
     const requestBody = JSON.parse(fetch.mock.calls[0].arguments[1].body);
@@ -27,7 +28,7 @@ describe('CHAT_COMPLETE', () => {
       mockResponseOk(makeCompletionResponse()),
     );
 
-    await chatComplete([['Say hello']], [['API_KEY', 'someapikey']]);
+    await chatComplete([['Say hello']], [['API_KEY', 'someapikey']], {});
 
     const requestBody = JSON.parse(fetch.mock.calls[0].arguments[1].body);
     assert.deepStrictEqual(requestBody.messages[1], {
@@ -47,6 +48,7 @@ describe('CHAT_COMPLETE', () => {
         [0, 0],
       ],
       [['API_KEY', 'someapikey']],
+      {},
     );
 
     const requestBody = JSON.parse(fetch.mock.calls[0].arguments[1].body);
@@ -64,6 +66,7 @@ describe('CHAT_COMPLETE', () => {
         ['API_KEY', 'someapikey'],
         ['temperature', 0.3],
       ],
+      {},
     );
 
     const requestBody = JSON.parse(fetch.mock.calls[0].arguments[1].body);
@@ -82,24 +85,33 @@ describe('CHAT_COMPLETE', () => {
         ['API_KEY', 'someapikey'],
         [0, 0],
       ],
+      {},
     );
 
     const requestBody = JSON.parse(fetch.mock.calls[0].arguments[1].body);
     assert(!('0' in requestBody));
   });
 
   it('throws an error when no API key is provided - key absent', () => {
-    assert.rejects(() => chatComplete([['Say hello']], []), {
-      code: '#VALUE!',
-      message: 'API_KEY is required',
-    });
+    assert.rejects(
+      () => chatComplete([['Say hello']], []),
+      {
+        code: '#VALUE!',
+        message: 'API_KEY is required',
+      },
+      {},
+    );
   });
 
   it('throws an error when no API key is provided - value cell empty', () => {
-    assert.rejects(() => chatComplete([['Say hello']], [['API_KEY', 0]]), {
-      code: '#VALUE!',
-      message: 'API_KEY is required',
-    });
+    assert.rejects(
+      () => chatComplete([['Say hello']], [['API_KEY', 0]]),
+      {
+        code: '#VALUE!',
+        message: 'API_KEY is required',
+      },
+      {},
+    );
   });
 
   it('throws an error for an API error with a provided message', (t) => {
@@ -122,7 +134,7 @@ describe('CHAT_COMPLETE', () => {
     );
 
     assert.rejects(
-      () => chatComplete([['Say hello']], [['API_KEY', 'someapikey']]),
+      () => chatComplete([['Say hello']], [['API_KEY', 'someapikey']], {}),
       {
         code: '#N/A',
         message: "API error: 0 is less than the minimum of 1 - 'n'",
@@ -151,7 +163,7 @@ describe('CHAT_COMPLETE', () => {
     );
 
     assert.rejects(
-      () => chatComplete([['Say hello']], [['API_KEY', 'someapikey']]),
+      () => chatComplete([['Say hello']], [['API_KEY', 'someapikey']], {}),
       {
         code: '#N/A',
         message: 'API error: 502 Bad Gateway',