test: timeout for embeddings

EnjoinHQ · Nov 8, 2023 · 1208da4 · 1208da4 · vercel · Nov 8, 2023
1 parent 544a636
commit 1208da4
Showing 1 changed file with 24 additions and 16 deletions.
diff --git a/src/openai/embedding.integration.ts b/src/openai/embedding.integration.ts
@@ -8,22 +8,30 @@ const hopfield = hop.client(openai).provider(openAIClient);
 const embeddings = hopfield.embedding();
 const multipleEmbeddings = hopfield.embedding('text-embedding-ada-002', 3);
 
-test('should respond with embedding', async () => {
-  const response = await embeddings.get({
-    input: ['hopfield'],
-  });
+test(
+  'should respond with embedding',
+  async () => {
+    const response = await embeddings.get({
+      input: ['hopfield'],
+    });
 
-  expect(response.model).toMatchInlineSnapshot('"text-embedding-ada-002-v2"');
-  expect(response.data[0].embedding[0]).toBeCloseTo(-0.0073666335, 2);
-  expect(response.data[0].embedding[1535]).toBeCloseTo(-0.0013278616, 2);
-});
+    expect(response.model).toMatchInlineSnapshot('"text-embedding-ada-002-v2"');
+    expect(response.data[0].embedding[0]).toBeCloseTo(-0.0073666335, 2);
+    expect(response.data[0].embedding[1535]).toBeCloseTo(-0.0013278616, 2);
+  },
+  { timeout: 10_000 },
+);
 
-test('should respond with multiple embeddings', async () => {
-  const response = await multipleEmbeddings.get({
-    input: ['ready', 'set', 'hopfield'],
-  });
+test(
+  'should respond with multiple embeddings',
+  async () => {
+    const response = await multipleEmbeddings.get({
+      input: ['ready', 'set', 'hopfield'],
+    });
 
-  expect(response.model).toMatchInlineSnapshot('"text-embedding-ada-002-v2"');
-  expect(response.data[0].embedding[0]).toBeCloseTo(-0.009482461, 2);
-  expect(response.data[2].embedding[0]).toBeCloseTo(-0.0073666335, 2);
-});
+    expect(response.model).toMatchInlineSnapshot('"text-embedding-ada-002-v2"');
+    expect(response.data[0].embedding[0]).toBeCloseTo(-0.009482461, 2);
+    expect(response.data[2].embedding[0]).toBeCloseTo(-0.0073666335, 2);
+  },
+  { timeout: 10_000 },
+);