@agentscope-ai/agentscope 0.0.5 → 0.0.6

package/dist/agent/index.mjs

@@ -2,6 +2,21 @@
 import { z as z9 } from "zod";
 
 // src/message/message.ts
+function assertContentBlocksForRole(role, content) {
+  if (role === "user") {
+    for (const block of content) {
+      if (block.type !== "text" && block.type !== "data") {
+        throw new Error("User message can only contain text blocks or data blocks.");
+      }
+    }
+  } else if (role === "system") {
+    for (const block of content) {
+      if (block.type !== "text") {
+        throw new Error("System message can only contain text blocks.");
+      }
+    }
+  }
+}
 function createMsg({
   name,
   content,
@@ -13,6 +28,7 @@ function createMsg({
   usage
 }) {
   const contentBlocks = typeof content === "string" ? [{ id: crypto.randomUUID(), type: "text", text: content }] : content;
+  assertContentBlocksForRole(role, contentBlocks);
   return { id, name, role, content: contentBlocks, metadata, created_at, finished_at, usage };
 }
 function getContentBlocks(msg, blockType) {
@@ -732,26 +748,37 @@ var Agent = class {
    * @param usage
    */
   _saveToContext(blocks, usage) {
+    const msgUsage = usage ? { input_tokens: usage.inputTokens, output_tokens: usage.outputTokens } : void 0;
     const lastMsg = this.context.at(-1);
     if (this.context.length === 0) {
       this.context.push(
-        createMsg({ name: this.name, content: blocks, role: "assistant", usage })
+        createMsg({
+          name: this.name,
+          content: blocks,
+          role: "assistant",
+          usage: msgUsage
+        })
       );
     } else if (lastMsg && lastMsg.role === "assistant" && lastMsg.name === this.name) {
       lastMsg.content.push(...blocks);
-      if (usage) {
+      if (msgUsage) {
         if (!lastMsg.usage) {
           lastMsg.usage = {
-            inputTokens: 0,
-            outputTokens: 0
+            input_tokens: 0,
+            output_tokens: 0
           };
         }
-        lastMsg.usage.inputTokens = lastMsg.usage.inputTokens + usage.inputTokens;
-        lastMsg.usage.outputTokens = lastMsg.usage.outputTokens + usage.outputTokens;
+        lastMsg.usage.input_tokens = lastMsg.usage.input_tokens + msgUsage.input_tokens;
+        lastMsg.usage.output_tokens = lastMsg.usage.output_tokens + msgUsage.output_tokens;
       }
     } else {
       this.context.push(
-        createMsg({ name: this.name, content: blocks, role: "assistant", usage })
+        createMsg({
+          name: this.name,
+          content: blocks,
+          role: "assistant",
+          usage: msgUsage
+        })
       );
     }
   }