@mozaik-ai/core 4.0.3 → 4.0.5

package/dist/index.mjs

@@ -11,7 +11,7 @@ var ModelContext = class _ModelContext {
     this.items.push(...contextItems);
     return this;
   }
-  addItem(item) {
+  addContextItem(item) {
     this.items.push(item);
     return this;
   }
@@ -401,7 +401,7 @@ var Agent = class _Agent extends Participant {
     const memory = Memory.create();
     const manifest = { id, name, capabilities, role: "agent" };
     const developerMessage = DeveloperMessageItem.create(instruction);
-    memory.getContext().addItem(developerMessage);
+    memory.getContext().addContextItem(developerMessage);
     return new _Agent(manifest, instruction, tools, memory, handlers);
   }
   getTools() {
@@ -560,7 +560,7 @@ var AgentLoop = class _AgentLoop {
   async run(message, loopVisitor) {
     var _a;
     let transition = {
-      nextStateId: "context_update",
+      nextStateId: "message_received",
       input: message
     };
     while (transition.nextStateId !== "idle") {
@@ -585,8 +585,8 @@ var AgentLoop = class _AgentLoop {
 
 // src/domain/agentic-environment/loop/state-executor.ts
 var LoopStateExecutor = class {
-  constructor(contextUpdateState, inferenceState, inferenceStreamingState, functionCallState, modelMessageState) {
-    this.contextUpdateState = contextUpdateState;
+  constructor(messageReceivedState, inferenceState, inferenceStreamingState, functionCallState, modelMessageState) {
+    this.messageReceivedState = messageReceivedState;
     this.inferenceState = inferenceState;
     this.inferenceStreamingState = inferenceStreamingState;
     this.functionCallState = functionCallState;
@@ -594,8 +594,8 @@ var LoopStateExecutor = class {
   }
   async execute(transition, loopVisitor) {
     switch (transition.nextStateId) {
-      case "context_update":
-        return await this.contextUpdateState.run(transition.input, loopVisitor);
+      case "message_received":
+        return await this.messageReceivedState.run(transition.input, loopVisitor);
       case "inference":
         return await this.inferenceState.run(transition.input, loopVisitor);
       case "inference_streaming":
@@ -640,19 +640,19 @@ var FunctionCallState = class {
   }
 };
 
-// src/application/states/context-update.ts
-var ContextUpdateState = class {
+// src/application/states/message-received.ts
+var MessageReceivedState = class {
   constructor() {
-    this.id = "context_update";
+    this.id = "message_received";
   }
   async run(input, loopVisitor) {
-    loopVisitor.visitContextUpdateStarted(input);
+    loopVisitor.visitMessageReceivedStarted(input);
     const userMessage = UserMessageItem.create(input.content);
     const output = {
       ...input.input,
       context: input.input.context.addContextItems([userMessage])
     };
-    loopVisitor.visitContextUpdateCompleted(output);
+    loopVisitor.visitMessageReceivedCompleted(output);
     return {
       stateId: this.id,
       input,
@@ -693,7 +693,7 @@ var TransitionResolver = class {
 // src/domain/agentic-environment/loop/transition-specification.ts
 var MessageReceivedSpecification = class {
   isSatisfiedBy(execution) {
-    return execution.stateId === "context_update";
+    return execution.stateId === "message_received";
   }
 };
 function isInferenceExecution(execution) {
@@ -726,7 +726,7 @@ var ContextPreparationToInferenceRule = class {
     this.specification = new MessageReceivedSpecification();
   }
   createTransition(execution) {
-    if (execution.stateId !== "context_update") {
+    if (execution.stateId !== "message_received") {
       throw new Error("Expected context_preparation execution");
     }
     const nextStateId = execution.output.streaming ? "inference_streaming" : "inference";
@@ -771,6 +771,7 @@ var InferenceToModelMessageRule = class {
     if (!modelMessageItem) {
       throw new Error("Expected inference model-message output");
     }
+    execution.input.context.addContextItem(modelMessageItem);
     return {
       nextStateId: "model_message",
       input: {
@@ -821,11 +822,11 @@ var EventPublisherLoopVisitor = class {
     this.runtime = runtime;
     this.cloud = createCloudClient();
   }
-  visitContextUpdateStarted(input) {
-    this.publish("context_update.started", input);
+  visitMessageReceivedStarted(input) {
+    this.publish("message_received.started", input);
   }
-  visitContextUpdateCompleted(output) {
-    this.publish("context_update.completed", output);
+  visitMessageReceivedCompleted(output) {
+    this.publish("message_received.completed", output);
   }
   visitInferenceStarted(input) {
     this.publish("inference.started", input);
@@ -921,7 +922,7 @@ function createRunLoop(resolveRuntime) {
       new ModelMessageToIdleRule()
     ]);
     const stateExecutor = new LoopStateExecutor(
-      new ContextUpdateState(),
+      new MessageReceivedState(),
       new InferenceState(inferenceRunner),
       new InferenceStreamingState(inferenceRunner),
       new FunctionCallState(functionCallRunner),