@ax-llm/ax 12.0.0 → 12.0.1

package/index.cjs

@@ -1270,6 +1270,7 @@ var AxBaseAI = class {
     if (chatReq.functions && chatReq.functions.length > 0) {
       functions = chatReq.functions.map((fn2) => this.cleanupFunctionSchema(fn2));
     }
+    validateChatPrompt(chatReq.chatPrompt);
     const req = {
       ...chatReq,
       model,
@@ -1611,6 +1612,31 @@ function setChatResponseEvents(res, span, excludeContentFromTrace) {
     });
   }
 }
+function validateAxMessageArray(values) {
+  for (let i = 0; i < values.length; i++) {
+    const message = values[i];
+    if (!message || typeof message !== "object") {
+      throw new Error(
+        `AxMessage array validation failed: Item at index ${i} is not a valid message object`
+      );
+    }
+    if ("content" in message && typeof message.content === "string" && message.content.trim() === "") {
+      throw new Error(
+        `AxMessage array validation failed: Item at index ${i} has empty content`
+      );
+    }
+  }
+}
+function validateChatPrompt(chatPrompt) {
+  for (let i = 0; i < chatPrompt.length; i++) {
+    const message = chatPrompt[i];
+    if (message && "content" in message && typeof message.content === "string" && message.content.trim() === "") {
+      throw new Error(
+        `Chat prompt validation failed: Message at index ${i} has empty content`
+      );
+    }
+  }
+}
 function validateModels(models) {
   const keys = /* @__PURE__ */ new Set();
   for (const model of models) {
@@ -3387,7 +3413,9 @@ var AxAIGoogleGeminiImpl = class {
     const contents = req.chatPrompt.filter((p) => p.role !== "system").map((msg, i) => {
       switch (msg.role) {
         case "user": {
-          const parts = Array.isArray(msg.content) ? msg.content.map((c, i2) => {
+          const parts = Array.isArray(
+            msg.content
+          ) ? msg.content.map((c, i2) => {
             switch (c.type) {
               case "text":
                 return { text: c.text };
@@ -3448,12 +3476,14 @@ var AxAIGoogleGeminiImpl = class {
             }
           ];
           return {
-            role: "function",
+            role: "model",
             parts
           };
         }
         default:
-          throw new Error("Invalid role");
+          throw new Error(
+            `Invalid role: ${JSON.stringify(msg)} (index: ${i})`
+          );
       }
     });
     let tools = [];
@@ -9185,6 +9215,7 @@ Content: ${result.content}`
     }
     let prompt;
     if (Array.isArray(values)) {
+      validateAxMessageArray(values);
       prompt = this.promptTemplate.render(values, {
         examples: this.examples,
         demos: this.demos