use cheaper models in test

openlit · Mar 24, 2024 · 4048c81 · 4048c81
1 parent 921708d
commit 4048c81
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/anthropic.test.mjs b/tests/anthropic.test.mjs
@@ -12,7 +12,7 @@ describe('Anthropic Test', () => {
     DokuMetry.init({llm: anthropic, dokuUrl: process.env.DOKU_URL, apiKey: process.env.DOKU_TOKEN, environment: "dokumetry-testing", applicationName: "dokumetry-node-test", skipResp: false});
     try {
       const message = await anthropic.messages.create({
-        model: "claude-3-opus-20240229",
+        model: "claude-3-haiku-20240307",
         max_tokens: 1,
         messages: [{ role: "user", content: "How to monitor LLM Applications in one sentence?" }],
       });
@@ -30,7 +30,7 @@ describe('Anthropic Test', () => {
       var stream = await anthropic.messages.create({
         max_tokens: 1,
         messages: [{ role: 'user', content: 'How to monitor LLM Applications in one sentence?' }],
-        model: 'claude-3-opus-20240229',
+        model: 'claude-3-haiku-20240307',
         stream: true,
       });
       for await (const messageStreamEvent of stream) {

diff --git a/tests/mistral.test.mjs b/tests/mistral.test.mjs
@@ -23,7 +23,7 @@ describe('Mistral Test', () => {
   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});
     const message = await client.chat({
-      model: 'mistral-large-latest',
+      model: 'open-mistral-7b',
       messages: [{role: 'user', content: 'What is LLM Observability?'}],
       maxTokens: 1,
     });