@mozaik-ai/core 4.0.0-beta.8 → 4.0.0

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() {
@@ -813,11 +813,13 @@ var ModelMessageToIdleRule = class {
 };
 
 // src/domain/agentic-environment/loop/event-publisher-visitor.ts
+import { createCloudClient } from "@mozaik-ai/cloud-sdk";
 var EventPublisherLoopVisitor = class {
   constructor(agentId, loopId, runtime) {
     this.agentId = agentId;
     this.loopId = loopId;
     this.runtime = runtime;
+    this.cloud = createCloudClient();
   }
   visitContextUpdateStarted(input) {
     this.publish("context_update.started", { ...input, loopId: this.loopId });
@@ -850,12 +852,11 @@ var EventPublisherLoopVisitor = class {
     this.publish("interception.finished", transition);
   }
   publish(type, payload) {
-    this.runtime.publish({
-      type,
-      producerId: this.agentId,
-      occurredAt: /* @__PURE__ */ new Date(),
-      payload: { ...payload, loopId: this.loopId }
-    });
+    const event = new SemanticEvent(type, this.agentId, /* @__PURE__ */ new Date(), payload);
+    this.runtime.publish(event);
+    if (this.cloud.enabled) {
+      this.cloud.send(event);
+    }
   }
 };
 
@@ -1065,7 +1066,7 @@ var AnthropicMessagesMapper = class {
         );
       }
     }
-    return { items, rowResponse: response };
+    return { items, tokenUsage: this.extractTokenUsage(response), rowResponse: response };
   }
 };
 
@@ -1253,6 +1254,7 @@ var OpenAIResponsesMapper = class {
     }
     return {
       items,
+      tokenUsage: this.extractTokenUsage(response),
       rowResponse: response
     };
   }
@@ -1441,7 +1443,7 @@ var GeminiGenerateContentMapper = class {
         );
       }
     }
-    return { items, rowResponse: response };
+    return { items, tokenUsage: this.extractTokenUsage(response), rowResponse: response };
   }
 };
 
@@ -1564,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(
@@ -1587,7 +1589,7 @@ var OpenAIChatCompletionsMapper = class {
         })
       );
     }
-    return { items, rowResponse: response };
+    return { items, tokenUsage: this.extractTokenUsage(response), rowResponse: response };
   }
 };