@mozaik-ai/core 4.0.0-beta.9 → 4.0.1

package/dist/index.mjs

@@ -558,20 +558,20 @@ var AgentLoop = class _AgentLoop {
     this.interceptionHandler = interceptionHandler;
   }
   async run(message, loopVisitor) {
+    var _a;
     let transition = {
       nextStateId: "context_update",
       input: message
     };
     while (transition.nextStateId !== "idle") {
-      const isInterceptionSatisfied = this.interceptionHandler.isSatisfiedBy(transition);
-      if (isInterceptionSatisfied) {
+      const isInterceptionSatisfied = (_a = this.interceptionHandler) == null ? void 0 : _a.isSatisfiedBy(transition);
+      if (isInterceptionSatisfied && this.interceptionHandler) {
         loopVisitor.visitInterceptionStarted(transition);
         transition = await this.interceptionHandler.handle(transition);
         loopVisitor.visitInterceptionFinished(transition);
-      } else {
-        const execution = await this.stateExecutor.execute(transition, loopVisitor);
-        transition = this.transitionResolver.resolve(execution);
       }
+      const execution = await this.stateExecutor.execute(transition, loopVisitor);
+      transition = this.transitionResolver.resolve(execution);
     }
   }
   getLoopId() {
@@ -943,7 +943,7 @@ var AnthropicMessagesMapper = class {
     if (inferenceInput.structuredOutput) {
       outputConfig.format = {
         type: "json_schema",
-        json_schema: inferenceInput.structuredOutput.schema
+        schema: inferenceInput.structuredOutput.schema
       };
     }
     if (inferenceInput.reasoningEffort) {
@@ -1066,7 +1066,7 @@ var AnthropicMessagesMapper = class {
         );
       }
     }
-    return { items, rowResponse: response };
+    return { items, tokenUsage: this.extractTokenUsage(response), rowResponse: response };
   }
 };
 
@@ -1254,6 +1254,7 @@ var OpenAIResponsesMapper = class {
     }
     return {
       items,
+      tokenUsage: this.extractTokenUsage(response),
       rowResponse: response
     };
   }
@@ -1442,7 +1443,7 @@ var GeminiGenerateContentMapper = class {
         );
       }
     }
-    return { items, rowResponse: response };
+    return { items, tokenUsage: this.extractTokenUsage(response), rowResponse: response };
   }
 };
 
@@ -1565,7 +1566,7 @@ var OpenAIChatCompletionsMapper = class {
     const items = [];
     const message = (_b = (_a = response.choices) == null ? void 0 : _a[0]) == null ? void 0 : _b.message;
     if (!message) {
-      return { items: [], rowResponse: response };
+      return { items: [], tokenUsage: this.extractTokenUsage(response), rowResponse: response };
     }
     if (message.reasoning_content) {
       items.push(
@@ -1588,7 +1589,7 @@ var OpenAIChatCompletionsMapper = class {
         })
       );
     }
-    return { items, rowResponse: response };
+    return { items, tokenUsage: this.extractTokenUsage(response), rowResponse: response };
   }
 };