From e665c8ca62b5464b9c4caa1f60328d2d49d62a12 Mon Sep 17 00:00:00 2001
From: patcher99 <patcher99@dokulabs.com>
Date: Thu, 14 Mar 2024 22:32:20 +0530
Subject: [PATCH] add timeout

---
 tests/anthropic.test.mjs | 2 +-
 tests/mistral.test.mjs   | 4 ++--
 tests/openai.test.mjs    | 6 +++---
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/tests/anthropic.test.mjs b/tests/anthropic.test.mjs
index b5e8320..3f34c5a 100644
--- 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);
 });
\ No newline at end of file
diff --git a/tests/mistral.test.mjs b/tests/mistral.test.mjs
index c3e77c7..669da28 100644
--- 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);
 });
\ No newline at end of file
diff --git a/tests/openai.test.mjs b/tests/openai.test.mjs
index 1792f4c..d5f2d40 100644
--- 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});