update mistral

auto-imports.d.ts

@@ -5,17 +5,17 @@
 // Generated by unplugin-auto-import
 export {}
 declare global {
-  const afterAll: typeof import("vitest")["afterAll"];
-  const afterEach: typeof import("vitest")["afterEach"];
-  const assert: typeof import("vitest")["assert"];
-  const beforeAll: typeof import("vitest")["beforeAll"];
-  const beforeEach: typeof import("vitest")["beforeEach"];
-  const chai: typeof import("vitest")["chai"];
-  const describe: typeof import("vitest")["describe"];
-  const expect: typeof import("vitest")["expect"];
-  const it: typeof import("vitest")["it"];
-  const suite: typeof import("vitest")["suite"];
-  const test: typeof import("vitest")["test"];
-  const vi: typeof import("vitest")["vi"];
-  const vitest: typeof import("vitest")["vitest"];
+  const afterAll: typeof import('vitest')['afterAll']
+  const afterEach: typeof import('vitest')['afterEach']
+  const assert: typeof import('vitest')['assert']
+  const beforeAll: typeof import('vitest')['beforeAll']
+  const beforeEach: typeof import('vitest')['beforeEach']
+  const chai: typeof import('vitest')['chai']
+  const describe: typeof import('vitest')['describe']
+  const expect: typeof import('vitest')['expect']
+  const it: typeof import('vitest')['it']
+  const suite: typeof import('vitest')['suite']
+  const test: typeof import('vitest')['test']
+  const vi: typeof import('vitest')['vi']
+  const vitest: typeof import('vitest')['vitest']
 }

package.json

@@ -26,7 +26,7 @@
   "dependencies": {
     "@anthropic-ai/sdk": "0.21.0",
     "@google/generative-ai": "0.12.0",
-    "@mistralai/mistralai": "0.1.3",
+    "@mistralai/mistralai": "0.4.0",
     "@sinclair/typebox": "0.32.30",
     "@stdlib/stats-pcorrtest": "0.2.1",
     "cohere-ai": "7.10.2",

src/lib/models/mistral.ts

@@ -1,6 +1,6 @@
 import MistralClient, {
   ChatCompletionResponse,
-  ToolChoice,
+  ChatRequest,
 } from "@mistralai/mistralai";
 import { Model, ModelTool, ModelResponse, ModelPricing } from "./model";
 import logger from "../logger";
@@ -40,7 +40,7 @@ const buildModel = (
   pricing?: ModelPricing
 ) => {
   const makeRequest = async function (prompt: string, toolParams: ModelTool) {
-    const req = {
+    const req: ChatRequest = {
       model: modelId,
       messages: [
         { role: "system", content: "You are a helpful assistant." },
@@ -49,7 +49,7 @@ const buildModel = (
           content: prompt,
         },
       ],
-      toolChoice: "any" as ToolChoice,
+      toolChoice: "any",
       tools: [
         {
           type: "function",

src/scripts/testMistral.ts

@@ -1,4 +1,4 @@
-import MistralClient, { ToolChoice } from "@mistralai/mistralai";
+import MistralClient from "@mistralai/mistralai";
 import "dotenv/config";
 const apiKey = process.env.MISTRAL_API_KEY;
 
@@ -16,7 +16,7 @@ const run = async () => {
           "Give me a list of 10 French cheeses so that I could choose the best one.",
       },
     ],
-    toolChoice: "any" as ToolChoice,
+    toolChoice: "any",
     tools: [
       {
         type: "function",