modelmix 3.0.4 → 3.0.8

package/index.js

@@ -237,7 +237,7 @@ class MessageHandler {
         this.options.response_format = { type: "json_object" };
         if (schemaExample) {
             const schema = generateJsonSchema(schemaExample, schemaDescription);
-            this.addText("Output expected JSON Schema: \n```\n" + JSON.stringify(schema, null, 2) + "\n```");
+            this.addText("Output expected JSON Schema: \n```\n" + JSON.stringify(schema) + "\n```");
         }
         return JSON.parse(await this.message());
     }
@@ -315,7 +315,6 @@ class MessageHandler {
     async prepareMessages() {
         await this.processImageUrls();
         this.applyTemplate();
-        this.messages = this.messages.slice(-this.config.max_history);
         this.messages = this.groupByRoles(this.messages);
         this.options.messages = this.messages;
     }
@@ -332,6 +331,7 @@ class MessageHandler {
                 try {
                     const result = await this.modelEntry.create({ options: this.options, config: this.config });
                     this.messages.push({ role: "assistant", content: result.message });
+                    this.messages = this.messages.slice(-this.config.max_history);
                     return result;
                 } catch (error) {
                     // If there are fallback models available, try the next one
@@ -727,7 +727,7 @@ class MixTogether extends MixCustom {
         };
     }
 
-    convertMessages(messages) {
+    static convertMessages(messages) {
         return messages.map(message => {
             if (message.content instanceof Array) {
                 message.content = message.content.map(content => content.text).join("\n\n");
@@ -742,7 +742,7 @@ class MixTogether extends MixCustom {
         }
 
         args.options.messages = [{ role: 'system', content: args.config.system }, ...args.options.messages || []];
-        args.options.messages = this.convertMessages(args.options.messages);
+        args.options.messages = MixTogether.convertMessages(args.options.messages);
 
         return super.create(args);
     }
@@ -760,6 +760,7 @@ class MixCerebras extends MixCustom {
 
     create(args = { config: {}, options: {} }) {
         args.options.messages = [{ role: 'system', content: args.config.system }, ...args.options.messages || []];
+        args.options.messages = MixTogether.convertMessages(args.options.messages);
         return super.create(args);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "modelmix",
-  "version": "3.0.4",
+  "version": "3.0.8",
   "description": "🧬 ModelMix - Unified API for Diverse AI LLM.",
   "main": "index.js",
   "repository": {