@mozaik-ai/core 2.13.8 → 3.0.1

package/README.md

@@ -1,6 +1,6 @@
 # Mozaik
 
-**Mozaik** is a TypeScript library for building, managing, and evolving **LLM context**.
+**Mozaik** is a TypeScript framework for building controllable multi-step AI agents.
 
 Instead of focusing on agents themselves, Mozaik provides a structured way to **model, manipulate, persist, and restore the context** that drives language model behavior. It implements a clean object model aligned with the **OpenResponses specification**, enabling developers to work with LLM inputs and outputs as composable, typed entities.
 

package/dist/index.d.mts

@@ -23,11 +23,11 @@ declare class Context {
     toJSON(): any;
 }
 
-declare abstract class Content {
+declare abstract class ItemContent {
     abstract readonly type: string;
 }
 
-declare class InputText extends Content {
+declare class InputText extends ItemContent {
     readonly text: string;
     readonly type = "input_text";
     private constructor();
@@ -62,7 +62,7 @@ declare class DeveloperMessage extends ContextItem {
     }): DeveloperMessage;
 }
 
-declare class OutputText extends Content {
+declare class OutputText extends ItemContent {
     readonly text: string;
     readonly type = "output_text";
     private constructor();
@@ -97,7 +97,7 @@ declare class FunctionCall extends ContextItem {
     toJSON(): any;
 }
 
-declare class SummaryText extends Content {
+declare class SummaryText extends ItemContent {
     readonly text: string;
     readonly type = "summary_text";
     private constructor();
@@ -290,11 +290,6 @@ interface Action<T> {
 interface AsyncAction<T> {
     apply(candidate: T): Promise<T>;
 }
-interface Rule<T> {
-    condition: Condition<T>;
-    action: AsyncAction<T>;
-    apply(candidate: T): T;
-}
 interface AsyncRule<T> {
     condition: Condition<T>;
     action: AsyncAction<T>;
@@ -328,4 +323,87 @@ declare class InMemoryContextRepository implements ContextRepository {
     private clone;
 }
 
-export { type Action, type AsyncAction, type AsyncRule, BaseCondition, type Condition, Context, ContextItem, type ContextRepository, DeveloperMessage, FunctionCall, FunctionCallOutput, type GenerativeModel, Gpt54, Gpt54Mini, Gpt54Nano, If, InMemoryContextRepository, InferenceRequest, InferenceResponse, InputTokenDetails, Loop, ModelMessage, OpenAIResponses, OutputTokenDetails, Reasoning, type Rule, TokenUsage, type Tool, UserMessage };
+declare enum HookId {
+    BEFORE_INFERENCE = "BEFORE_INFERENCE",
+    AFTER_INFERENCE = "AFTER_INFERENCE",
+    BEFORE_FUNCTION_CALL = "BEFORE_FUNCTION_CALL",
+    AFTER_FUNCTION_CALL = "AFTER_FUNCTION_CALL",
+    BEFORE_MODEL_MESSAGE = "BEFORE_MODEL_MESSAGE",
+    AFTER_MODEL_MESSAGE = "AFTER_MODEL_MESSAGE"
+}
+
+interface TransitionRecord {
+    fromState: StateId;
+    toState: StateId;
+    timestamp: Date;
+    duration: number;
+    error: Error | null;
+}
+
+declare enum StateId {
+    INFERENCE_PENDING = "INFERENCE_PENDING",
+    FUNCTION_CALL_PENDING = "FUNCTION_CALL_PENDING",
+    MODEL_MESSAGE_RECEIVED = "MODEL_MESSAGE_RECEIVED"
+}
+
+declare enum ExecutionStatus {
+    RUNNING = 0,
+    COMPLETED = 1,
+    FAILED = 2
+}
+declare class Execution {
+    executionId: string;
+    currentStateId: StateId;
+    previousStateId: StateId | null;
+    status: ExecutionStatus;
+    stepCount: number;
+    retryCounts: Map<StateId, number>;
+    history: TransitionRecord[];
+    constructor(executionId: string, currentStateId: StateId, previousStateId: StateId | null, status: ExecutionStatus, stepCount: number, retryCounts: Map<StateId, number>, history: TransitionRecord[]);
+    isTerminal(): boolean;
+    static create(): Execution;
+}
+
+interface RuntimeContext {
+    execution: Execution;
+    userMessage: UserMessage;
+    inferenceRequest?: InferenceRequest;
+    model: GenerativeModel & ReasoningEffort<string> & ToolCallingCapability;
+    context: Context;
+    inferenceResponse?: InferenceResponse;
+    functionCallOutput?: FunctionCallOutput;
+    modelMessage?: ModelMessage;
+}
+
+declare class AgentRuntime {
+    private hooksRegistry;
+    private stateHandlerRegistry;
+    constructor();
+    onInferencePending(context: RuntimeContext): Promise<void>;
+    onFunctionCallPending(context: RuntimeContext): Promise<void>;
+    onModelMessageReceived(context: RuntimeContext): Promise<void>;
+    on(hookId: HookId, callback: (data: any) => Promise<void>): Promise<void>;
+    start(userMessage: UserMessage, model: GenerativeModel & ReasoningEffort<string> & ToolCallingCapability, context: Context): Promise<void>;
+}
+
+interface InferenceVisitor {
+    onStart(context: RuntimeContext): Promise<void>;
+    afterInference(inferenceResponse: InferenceResponse): Promise<void>;
+}
+
+declare class Agent {
+    private readonly runtime;
+    private visitor;
+    constructor(runtime: AgentRuntime);
+    beforeInference(context: RuntimeContext): Promise<void>;
+    afterInference(context: RuntimeContext): Promise<void>;
+    beforeFunctionCall(context: RuntimeContext): Promise<void>;
+    afterFunctionCall(context: RuntimeContext): Promise<void>;
+    beforeModelMessage(context: RuntimeContext): Promise<void>;
+    afterModelMessage(context: RuntimeContext): Promise<void>;
+    onStart(context: RuntimeContext): Promise<void>;
+    setInferenceVisitor(visitor: InferenceVisitor): void;
+    run(userMessage: string, model: GenerativeModel & ReasoningEffort<string> & ToolCallingCapability, context: Context): Promise<void>;
+}
+
+export { type Action, Agent, AgentRuntime, type AsyncAction, type AsyncRule, BaseCondition, type Condition, Context, ContextItem, type ContextRepository, DeveloperMessage, FunctionCall, FunctionCallOutput, type GenerativeModel, Gpt54, Gpt54Mini, Gpt54Nano, If, InMemoryContextRepository, InferenceRequest, InferenceResponse, type InferenceVisitor, InputTokenDetails, Loop, ModelMessage, OpenAIResponses, OutputTokenDetails, Reasoning, TokenUsage, type Tool, UserMessage };

package/dist/index.d.ts

@@ -23,11 +23,11 @@ declare class Context {
     toJSON(): any;
 }
 
-declare abstract class Content {
+declare abstract class ItemContent {
     abstract readonly type: string;
 }
 
-declare class InputText extends Content {
+declare class InputText extends ItemContent {
     readonly text: string;
     readonly type = "input_text";
     private constructor();
@@ -62,7 +62,7 @@ declare class DeveloperMessage extends ContextItem {
     }): DeveloperMessage;
 }
 
-declare class OutputText extends Content {
+declare class OutputText extends ItemContent {
     readonly text: string;
     readonly type = "output_text";
     private constructor();
@@ -97,7 +97,7 @@ declare class FunctionCall extends ContextItem {
     toJSON(): any;
 }
 
-declare class SummaryText extends Content {
+declare class SummaryText extends ItemContent {
     readonly text: string;
     readonly type = "summary_text";
     private constructor();
@@ -290,11 +290,6 @@ interface Action<T> {
 interface AsyncAction<T> {
     apply(candidate: T): Promise<T>;
 }
-interface Rule<T> {
-    condition: Condition<T>;
-    action: AsyncAction<T>;
-    apply(candidate: T): T;
-}
 interface AsyncRule<T> {
     condition: Condition<T>;
     action: AsyncAction<T>;
@@ -328,4 +323,87 @@ declare class InMemoryContextRepository implements ContextRepository {
     private clone;
 }
 
-export { type Action, type AsyncAction, type AsyncRule, BaseCondition, type Condition, Context, ContextItem, type ContextRepository, DeveloperMessage, FunctionCall, FunctionCallOutput, type GenerativeModel, Gpt54, Gpt54Mini, Gpt54Nano, If, InMemoryContextRepository, InferenceRequest, InferenceResponse, InputTokenDetails, Loop, ModelMessage, OpenAIResponses, OutputTokenDetails, Reasoning, type Rule, TokenUsage, type Tool, UserMessage };
+declare enum HookId {
+    BEFORE_INFERENCE = "BEFORE_INFERENCE",
+    AFTER_INFERENCE = "AFTER_INFERENCE",
+    BEFORE_FUNCTION_CALL = "BEFORE_FUNCTION_CALL",
+    AFTER_FUNCTION_CALL = "AFTER_FUNCTION_CALL",
+    BEFORE_MODEL_MESSAGE = "BEFORE_MODEL_MESSAGE",
+    AFTER_MODEL_MESSAGE = "AFTER_MODEL_MESSAGE"
+}
+
+interface TransitionRecord {
+    fromState: StateId;
+    toState: StateId;
+    timestamp: Date;
+    duration: number;
+    error: Error | null;
+}
+
+declare enum StateId {
+    INFERENCE_PENDING = "INFERENCE_PENDING",
+    FUNCTION_CALL_PENDING = "FUNCTION_CALL_PENDING",
+    MODEL_MESSAGE_RECEIVED = "MODEL_MESSAGE_RECEIVED"
+}
+
+declare enum ExecutionStatus {
+    RUNNING = 0,
+    COMPLETED = 1,
+    FAILED = 2
+}
+declare class Execution {
+    executionId: string;
+    currentStateId: StateId;
+    previousStateId: StateId | null;
+    status: ExecutionStatus;
+    stepCount: number;
+    retryCounts: Map<StateId, number>;
+    history: TransitionRecord[];
+    constructor(executionId: string, currentStateId: StateId, previousStateId: StateId | null, status: ExecutionStatus, stepCount: number, retryCounts: Map<StateId, number>, history: TransitionRecord[]);
+    isTerminal(): boolean;
+    static create(): Execution;
+}
+
+interface RuntimeContext {
+    execution: Execution;
+    userMessage: UserMessage;
+    inferenceRequest?: InferenceRequest;
+    model: GenerativeModel & ReasoningEffort<string> & ToolCallingCapability;
+    context: Context;
+    inferenceResponse?: InferenceResponse;
+    functionCallOutput?: FunctionCallOutput;
+    modelMessage?: ModelMessage;
+}
+
+declare class AgentRuntime {
+    private hooksRegistry;
+    private stateHandlerRegistry;
+    constructor();
+    onInferencePending(context: RuntimeContext): Promise<void>;
+    onFunctionCallPending(context: RuntimeContext): Promise<void>;
+    onModelMessageReceived(context: RuntimeContext): Promise<void>;
+    on(hookId: HookId, callback: (data: any) => Promise<void>): Promise<void>;
+    start(userMessage: UserMessage, model: GenerativeModel & ReasoningEffort<string> & ToolCallingCapability, context: Context): Promise<void>;
+}
+
+interface InferenceVisitor {
+    onStart(context: RuntimeContext): Promise<void>;
+    afterInference(inferenceResponse: InferenceResponse): Promise<void>;
+}
+
+declare class Agent {
+    private readonly runtime;
+    private visitor;
+    constructor(runtime: AgentRuntime);
+    beforeInference(context: RuntimeContext): Promise<void>;
+    afterInference(context: RuntimeContext): Promise<void>;
+    beforeFunctionCall(context: RuntimeContext): Promise<void>;
+    afterFunctionCall(context: RuntimeContext): Promise<void>;
+    beforeModelMessage(context: RuntimeContext): Promise<void>;
+    afterModelMessage(context: RuntimeContext): Promise<void>;
+    onStart(context: RuntimeContext): Promise<void>;
+    setInferenceVisitor(visitor: InferenceVisitor): void;
+    run(userMessage: string, model: GenerativeModel & ReasoningEffort<string> & ToolCallingCapability, context: Context): Promise<void>;
+}
+
+export { type Action, Agent, AgentRuntime, type AsyncAction, type AsyncRule, BaseCondition, type Condition, Context, ContextItem, type ContextRepository, DeveloperMessage, FunctionCall, FunctionCallOutput, type GenerativeModel, Gpt54, Gpt54Mini, Gpt54Nano, If, InMemoryContextRepository, InferenceRequest, InferenceResponse, type InferenceVisitor, InputTokenDetails, Loop, ModelMessage, OpenAIResponses, OutputTokenDetails, Reasoning, TokenUsage, type Tool, UserMessage };