add timeout

openlit · Mar 14, 2024 · e665c8c · e665c8c
1 parent 1c2a9e3
commit e665c8c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/tests/anthropic.test.mjs b/tests/anthropic.test.mjs
@@ -16,5 +16,5 @@ describe('Anthropic Test', () => {
     });
 
     expect(message.type).to.equal('message');
-  }).timeout(10000);
+  }).timeout(30000);
 });
diff --git a/tests/mistral.test.mjs b/tests/mistral.test.mjs
@@ -18,7 +18,7 @@ describe('Mistral Test', () => {
     });
 
     expect(message.object).to.equal('list');
-  }).timeout(10000);
+  }).timeout(30000);
 
   it('should return a response with object as "chat.completion"', async () => {
     DokuMetry.init({llm: client, dokuUrl: process.env.DOKU_URL, apiKey: process.env.DOKU_TOKEN, environment: "dokumetry-testing", applicationName: "dokumetry-node-test", skipResp: false});
@@ -28,5 +28,5 @@ describe('Mistral Test', () => {
     });
 
     expect(message.object).to.equal('chat.completion');
-  }).timeout(10000);
+  }).timeout(30000);
 });
diff --git a/tests/openai.test.mjs b/tests/openai.test.mjs
@@ -20,7 +20,7 @@ describe('OpenAI Test', () => {
     });
 
     expect(chatCompletion.object).to.equal('chat.completion');
-  });
+  }).timeout(30000);;
 
   it('should return a response with object as "text_completion"', async () => {
     await DokuMetry.init({llm: openai, dokuUrl: process.env.DOKU_URL, apiKey: process.env.DOKU_TOKEN, environment: "dokumetry-testing", applicationName: "dokumetry-node-test", skipResp: false});
@@ -31,7 +31,7 @@ describe('OpenAI Test', () => {
     });
 
     expect(completion.object).to.equal('text_completion');
-  });
+  }).timeout(30000);;
 
   it('should return a response with object as "embedding"', async () => {
     await DokuMetry.init({llm: openai, dokuUrl: process.env.DOKU_URL, apiKey: process.env.DOKU_TOKEN, environment: "dokumetry-testing", applicationName: "dokumetry-node-test", skipResp: false});
@@ -42,7 +42,7 @@ describe('OpenAI Test', () => {
     });
 
     expect(embeddings.data[0].object).to.equal('embedding');
-  });
+  }).timeout(30000);;
 
   it('should return a response with object as "fine_tuning.job"', async () => {
     await DokuMetry.init({llm: openai, dokuUrl: process.env.DOKU_URL, apiKey: process.env.DOKU_TOKEN, environment: "dokumetry-testing", applicationName: "dokumetry-node-test", skipResp: false});