@mozaik-ai/core 4.0.0-beta.3 → 4.0.0-beta.5

package/dist/index.mjs

@@ -353,6 +353,9 @@ var Participant = class {
   getHandlers() {
     return this.handlers;
   }
+  setHandlers(handlers) {
+    this.handlers = handlers;
+  }
 };
 
 // src/domain/agentic-environment/participant/memory.ts
@@ -543,7 +546,7 @@ var AgentLoop = class {
   }
   async run(message) {
     let transition = {
-      nextStateId: "message_received",
+      nextStateId: "context_preparation",
       input: message
     };
     while (transition.nextStateId !== "idle") {
@@ -555,18 +558,21 @@ var AgentLoop = class {
 
 // src/domain/agentic-environment/loop/state-executor.ts
 var LoopStateExecutor = class {
-  constructor(messageReceivedState, inferenceState, functionCallState, modelMessageState) {
-    this.messageReceivedState = messageReceivedState;
+  constructor(contextPreparationState, inferenceState, inferenceStreamingState, functionCallState, modelMessageState) {
+    this.contextPreparationState = contextPreparationState;
     this.inferenceState = inferenceState;
+    this.inferenceStreamingState = inferenceStreamingState;
     this.functionCallState = functionCallState;
     this.modelMessageState = modelMessageState;
   }
   async execute(transition, loopVisitor) {
     switch (transition.nextStateId) {
-      case "message_received":
-        return await this.messageReceivedState.run(transition.input, loopVisitor);
+      case "context_preparation":
+        return await this.contextPreparationState.run(transition.input, loopVisitor);
       case "inference":
         return await this.inferenceState.run(transition.input, loopVisitor);
+      case "inference_streaming":
+        return await this.inferenceStreamingState.run(transition.input, loopVisitor);
       case "function_call":
         return await this.functionCallState.run(transition.input, loopVisitor);
       case "model_message":
@@ -607,19 +613,19 @@ var FunctionCallState = class {
   }
 };
 
-// src/application/states/message-received.ts
-var MessageReceivedState = class {
+// src/application/states/context-preparation.ts
+var ContextPreparationState = class {
   constructor() {
-    this.id = "message_received";
+    this.id = "context_preparation";
   }
   async run(input, loopVisitor) {
-    loopVisitor.visitMessageReceivedStarted(input);
+    loopVisitor.visitContextPreparationStarted(input);
     const userMessage = UserMessageItem.create(input.content);
     const output = {
       ...input.input,
       context: input.input.context.addContextItems([userMessage])
     };
-    loopVisitor.visitMessageReceivedCompleted(output);
+    loopVisitor.visitContextPreparationCompleted(output);
     return {
       stateId: this.id,
       input,
@@ -660,17 +666,20 @@ var TransitionResolver = class {
 // src/domain/agentic-environment/loop/transition-specification.ts
 var MessageReceivedSpecification = class {
   isSatisfiedBy(execution) {
-    return execution.stateId === "message_received";
+    return execution.stateId === "context_preparation";
   }
 };
+function isInferenceExecution(execution) {
+  return execution.stateId === "inference" || execution.stateId === "inference_streaming";
+}
 var InferenceProducedFunctionCallSpecification = class {
   isSatisfiedBy(execution) {
-    return execution.stateId === "inference" && execution.output.items.some((item) => item.type === "function_call");
+    return isInferenceExecution(execution) && execution.output.items.some((item) => item.type === "function_call");
   }
 };
 var InferenceProducedModelMessageSpecification = class {
   isSatisfiedBy(execution) {
-    return execution.stateId === "inference" && execution.output.items.some((item) => item.type === "message" && item.role === "assistant");
+    return isInferenceExecution(execution) && execution.output.items.some((item) => item.type === "message" && item.role === "assistant");
   }
 };
 var FunctionCallCompletedSpecification = class {
@@ -685,16 +694,17 @@ var ModelMessageCompletedSpecification = class {
 };
 
 // src/domain/agentic-environment/loop/transition-rule.ts
-var MessageReceivedToInferenceRule = class {
+var ContextPreparationToInferenceRule = class {
   constructor() {
     this.specification = new MessageReceivedSpecification();
   }
   createTransition(execution) {
-    if (execution.stateId !== "message_received") {
-      throw new Error("Expected message_received execution");
+    if (execution.stateId !== "context_preparation") {
+      throw new Error("Expected context_preparation execution");
     }
+    const nextStateId = execution.output.streaming ? "inference_streaming" : "inference";
     return {
-      nextStateId: "inference",
+      nextStateId,
       input: execution.output
     };
   }
@@ -704,7 +714,7 @@ var InferenceToFunctionCallRule = class {
     this.specification = new InferenceProducedFunctionCallSpecification();
   }
   createTransition(execution) {
-    if (execution.stateId !== "inference" || !execution.output.items.some((item) => item.type === "function_call")) {
+    if (execution.stateId !== "inference" && execution.stateId !== "inference_streaming" || !execution.output.items.some((item) => item.type === "function_call")) {
       throw new Error("Expected inference function-call output");
     }
     const functionCallItem = execution.output.items.find((item) => item.type === "function_call");
@@ -725,7 +735,7 @@ var InferenceToModelMessageRule = class {
     this.specification = new InferenceProducedModelMessageSpecification();
   }
   createTransition(execution) {
-    if (execution.stateId !== "inference" || !execution.output.items.some((item) => item.type === "message" && item.role === "assistant")) {
+    if (execution.stateId !== "inference" && execution.stateId !== "inference_streaming" || !execution.output.items.some((item) => item.type === "message" && item.role === "assistant")) {
       throw new Error("Expected inference model-message output");
     }
     const modelMessageItem = execution.output.items.find(
@@ -753,7 +763,7 @@ var FunctionCallToInferenceRule = class {
     const previousRequest = execution.input.inferenceInput;
     previousRequest.context.addContextItems([execution.input.call, execution.output.item]);
     return {
-      nextStateId: "inference",
+      nextStateId: previousRequest.streaming ? "inference_streaming" : "inference",
       input: {
         ...previousRequest
       }
@@ -781,20 +791,19 @@ var EventPublisherLoopVisitor = class {
     this.agentId = agentId;
     this.runtime = runtime;
   }
-  visitMessageReceivedStarted(input) {
+  visitContextPreparationStarted(input) {
     console.log("Visiting message received started: ", input);
-    this.publish("message_received.started", input);
+    this.publish("context_preparation.started", input);
   }
-  visitMessageReceivedCompleted(output) {
+  visitContextPreparationCompleted(output) {
     console.log("Visiting message received completed: ", output);
-    this.publish("message_received.completed", output);
+    this.publish("context_preparation.completed", output);
   }
   visitInferenceStarted(input) {
     console.log("Visiting inference started: ", input);
     this.publish("inference.started", input);
   }
   visitInferenceEvent(event) {
-    console.log("Visiting inference event: ", event);
     this.publish("inference.stream", event);
   }
   visitInferenceCompleted(output) {
@@ -823,6 +832,33 @@ var EventPublisherLoopVisitor = class {
   }
 };
 
+// src/application/states/inference-streaming.ts
+var InferenceStreamingState = class {
+  constructor(inferenceRunner) {
+    this.inferenceRunner = inferenceRunner;
+    this.id = "inference_streaming";
+  }
+  async run(input, loopVisitor) {
+    loopVisitor.visitInferenceStarted(input);
+    let output = void 0;
+    for await (const event of this.inferenceRunner.stream(input)) {
+      loopVisitor.visitInferenceEvent(event);
+      if (event.type === "inference.output") {
+        output = event.payload;
+      }
+    }
+    if (!output) {
+      throw new Error("Inference output not found");
+    }
+    loopVisitor.visitInferenceCompleted(output);
+    return {
+      stateId: "inference_streaming",
+      input,
+      output
+    };
+  }
+};
+
 // src/application/use-cases/run-loop.ts
 function createRunLoop(resolveRuntime) {
   return function runLoop(agentId, message, inferenceInput) {
@@ -831,15 +867,16 @@ function createRunLoop(resolveRuntime) {
     const functionCallRunner = runtime.getFunctionCallRunner();
     const transitionResolver = new TransitionResolver([
       // High-priority interception rules would go first.
-      new MessageReceivedToInferenceRule(),
+      new ContextPreparationToInferenceRule(),
       new InferenceToFunctionCallRule(),
       new InferenceToModelMessageRule(),
       new FunctionCallToInferenceRule(),
       new ModelMessageToIdleRule()
     ]);
     const stateExecutor = new LoopStateExecutor(
-      new MessageReceivedState(),
+      new ContextPreparationState(),
       new InferenceState(inferenceRunner),
+      new InferenceStreamingState(inferenceRunner),
       new FunctionCallState(functionCallRunner),
       new ModelMessageState()
     );
@@ -1010,7 +1047,14 @@ import Anthropic from "@anthropic-ai/sdk";
 var AnthropicMessages = class {
   constructor(endpointMapper = new AnthropicMessagesMapper(), config = {}) {
     this.endpointMapper = endpointMapper;
-    this.client = new Anthropic({ baseURL: config.baseURL, apiKey: config.apiKey });
+    this.clientConfig = config;
+  }
+  get client() {
+    var _a;
+    return (_a = this._client) != null ? _a : this._client = new Anthropic({
+      baseURL: this.clientConfig.baseURL,
+      apiKey: this.clientConfig.apiKey
+    });
   }
   async infer(inferenceInput) {
     const request = this.endpointMapper.toRequest(inferenceInput);
@@ -1019,10 +1063,17 @@ var AnthropicMessages = class {
   }
   async *stream(inferenceInput) {
     const request = this.endpointMapper.toRequest(inferenceInput);
-    const stream = await this.client.messages.create(request);
+    const stream = this.client.messages.stream(request);
     for await (const event of stream) {
       yield event;
     }
+    const output = this.endpointMapper.toResponse(await stream.finalMessage());
+    yield {
+      type: "inference.output",
+      producerId: request.model,
+      occurredAt: /* @__PURE__ */ new Date(),
+      payload: output
+    };
   }
 };
 
@@ -1184,7 +1235,10 @@ var OpenAIResponsesMapper = class {
 var OpenAIResponses = class {
   constructor(endpointMapper = new OpenAIResponsesMapper()) {
     this.endpointMapper = endpointMapper;
-    this.client = new OpenAI();
+  }
+  get client() {
+    var _a;
+    return (_a = this._client) != null ? _a : this._client = new OpenAI();
   }
   async infer(inferenceInput) {
     const request = this.endpointMapper.toRequest(inferenceInput);
@@ -1193,10 +1247,24 @@ var OpenAIResponses = class {
   }
   async *stream(inferenceInput) {
     const request = this.endpointMapper.toRequest(inferenceInput);
-    const response = await this.client.responses.create(request);
-    for await (const event of response) {
+    const stream = await this.client.responses.create({ ...request, stream: true });
+    let completedResponse = void 0;
+    for await (const event of stream) {
+      if (event.type === "response.completed") {
+        completedResponse = event.response;
+      }
       yield event;
     }
+    if (!completedResponse) {
+      throw new Error("Stream ended without a completed response");
+    }
+    const output = this.endpointMapper.toResponse(completedResponse);
+    yield {
+      type: "inference.output",
+      payload: output,
+      occurredAt: /* @__PURE__ */ new Date(),
+      producerId: inferenceInput.model
+    };
   }
 };
 
@@ -1353,11 +1421,14 @@ var GeminiGenerateContentMapper = class {
 // src/infrastructure/providers/gemini/endpoints/gemini-generate-content.ts
 var GeminiGenerateContent = class {
   constructor(endpointMapper = new GeminiGenerateContentMapper(), config = {}) {
-    var _a;
     this.endpointMapper = endpointMapper;
-    this.client = new GoogleGenAI({
-      apiKey: (_a = config.apiKey) != null ? _a : process.env.GEMINI_API_KEY,
-      ...config.baseURL ? { httpOptions: { baseUrl: config.baseURL } } : {}
+    this.clientConfig = config;
+  }
+  get client() {
+    var _a, _b;
+    return (_b = this._client) != null ? _b : this._client = new GoogleGenAI({
+      apiKey: (_a = this.clientConfig.apiKey) != null ? _a : process.env.GEMINI_API_KEY,
+      ...this.clientConfig.baseURL ? { httpOptions: { baseUrl: this.clientConfig.baseURL } } : {}
     });
   }
   async infer(inferenceInput) {
@@ -1368,9 +1439,21 @@ var GeminiGenerateContent = class {
   async *stream(inferenceInput) {
     const request = this.endpointMapper.toRequest(inferenceInput);
     const stream = await this.client.models.generateContentStream(request);
+    let lastEvent = void 0;
     for await (const chunk of stream) {
+      lastEvent = chunk;
       yield chunk;
     }
+    if (!lastEvent) {
+      throw new Error("Last event not found");
+    }
+    const output = this.endpointMapper.toResponse(lastEvent);
+    yield {
+      type: "inference.output",
+      producerId: request.model,
+      occurredAt: /* @__PURE__ */ new Date(),
+      payload: output
+    };
   }
 };
 
@@ -1487,12 +1570,16 @@ var OpenAIChatCompletions = class {
   constructor(endpointMapper = new OpenAIChatCompletionsMapper(), config = {}) {
     var _a;
     this.endpointMapper = endpointMapper;
-    this.client = new OpenAI2({
-      baseURL: config.baseURL,
-      apiKey: config.apiKey
-    });
+    this.clientConfig = config;
     this.extraBody = (_a = config.extraBody) != null ? _a : {};
   }
+  get client() {
+    var _a;
+    return (_a = this._client) != null ? _a : this._client = new OpenAI2({
+      baseURL: this.clientConfig.baseURL,
+      apiKey: this.clientConfig.apiKey
+    });
+  }
   buildRequest(inferenceInput) {
     return {
       ...this.extraBody,
@@ -1504,10 +1591,17 @@ var OpenAIChatCompletions = class {
     return this.endpointMapper.toResponse(response);
   }
   async *stream(inferenceInput) {
-    const stream = await this.client.chat.completions.create(this.buildRequest(inferenceInput));
-    for await (const event of stream) {
-      yield event;
+    const stream = this.client.chat.completions.stream(this.buildRequest(inferenceInput));
+    for await (const chunk of stream) {
+      yield chunk;
     }
+    const output = this.endpointMapper.toResponse(await stream.finalChatCompletion());
+    yield {
+      type: "inference.output",
+      payload: output,
+      occurredAt: /* @__PURE__ */ new Date(),
+      producerId: inferenceInput.model
+    };
   }
 };
 
@@ -1931,13 +2025,13 @@ var DefaultInferenceRunner = class {
     this.requestValidator.validate(input, generativeModel.specification);
     return await generativeModel.endpoint.infer(input);
   }
-  async streamInference(input) {
+  async *stream(input) {
     const generativeModel = this.supportedModels.find((model) => model.specification.name === input.model);
     if (!generativeModel) {
       throw new Error(`Unsupported model: ${input.model}`);
     }
     this.requestValidator.validate(input, generativeModel.specification);
-    return await generativeModel.endpoint.stream(input);
+    yield* generativeModel.endpoint.stream(input);
   }
 };
 
@@ -2014,13 +2108,57 @@ function defineRuntime() {
     runLoop
   };
 }
+
+// src/domain/agentic-environment/situation/situation-specification.ts
+var SituationSpecification = class {
+  and(other) {
+    return new AndSituationSpecification(this, other);
+  }
+  or(other) {
+    return new OrSituationSpecification(this, other);
+  }
+  not() {
+    return new NotSituationSpecification(this);
+  }
+};
+var AndSituationSpecification = class extends SituationSpecification {
+  constructor(left, right) {
+    super();
+    this.left = left;
+    this.right = right;
+  }
+  isSatisfiedBy(situationContext) {
+    return this.left.isSatisfiedBy(situationContext) && this.right.isSatisfiedBy(situationContext);
+  }
+};
+var OrSituationSpecification = class extends SituationSpecification {
+  constructor(left, right) {
+    super();
+    this.left = left;
+    this.right = right;
+  }
+  isSatisfiedBy(situationContext) {
+    return this.left.isSatisfiedBy(situationContext) || this.right.isSatisfiedBy(situationContext);
+  }
+};
+var NotSituationSpecification = class extends SituationSpecification {
+  constructor(rule) {
+    super();
+    this.rule = rule;
+  }
+  isSatisfiedBy(situationContext) {
+    return !this.rule.isSatisfiedBy(situationContext);
+  }
+};
 export {
+  Agent,
   AnthropicMessages,
   ContextItem,
   DeveloperMessageItem,
   FunctionCallItem,
   FunctionCallOutputItem,
   GeminiGenerateContent,
+  Human,
   InputTokenDetails,
   McpClient,
   McpToolRegistry,
@@ -2033,6 +2171,7 @@ export {
   ReasoningItem,
   RuntimeState,
   SemanticEvent,
+  SituationSpecification,
   SystemMessageItem,
   TokenUsage,
   UserMessageItem,