fix type tests

SAP · Aug 28, 2024 · 7fa347a · 7fa347a
1 parent f8b90df
commit 7fa347a
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/tests/type-tests/test/openai.test-d.ts b/tests/type-tests/test/openai.test-d.ts
@@ -12,18 +12,24 @@ expectType<OpenAiClient>(client);
  * Chat Completion.
  */
 expectType<Promise<OpenAiChatCompletionOutput>>(
-  client.chatCompletion('gpt-4', {
-    messages: [{ role: 'user', content: 'test prompt' }]
-  })
+  client.chatCompletion(
+    {
+      messages: [{ role: 'user', content: 'test prompt' }]
+    },
+    'gpt-4'
+  )
 );
 
 /**
  * Embeddings.
  */
 expectType<Promise<OpenAiEmbeddingOutput>>(
-  client.embeddings('text-embedding-ada-002', {
-    input: 'test input'
-  })
+  client.embeddings(
+    {
+      input: 'test input'
+    },
+    'text-embedding-ada-002'
+  )
 );
 
-expectError<any>(client.embeddings('gpt-35-turbo', { input: 'test input' }));
+expectError<any>(client.embeddings({ input: 'test input' }, 'gpt-35-turbo'));