From 4048c8181486867c814823228d72c6758c466840 Mon Sep 17 00:00:00 2001
From: patcher99 <patcher99@dokulabs.com>
Date: Sun, 24 Mar 2024 14:21:52 +0530
Subject: [PATCH] use cheaper models  in test

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

diff --git a/tests/anthropic.test.mjs b/tests/anthropic.test.mjs
index 4c2cca3..6d45d55 100644
--- 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
index 2e15b9d..1f51204 100644
--- 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,
     });