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

package/dist/index.d.mts

@@ -186,6 +186,24 @@ interface FunctionCallParams {
     inferenceInput: InferenceInput;
 }
 
+interface InterceptionHandler {
+    isSatisfiedBy(transition: ExecutableTransition): boolean;
+    handle(transition: ExecutableTransition): Promise<ExecutableTransition>;
+}
+type InterceptionOutput = {
+    type: "continue";
+    transition: LoopTransition;
+} | {
+    type: "pause";
+    pendingTransition: LoopTransition;
+} | {
+    type: "stop";
+    reason?: string;
+};
+interface InterceptionParams {
+    pendingTransition: LoopTransition;
+}
+
 type LoopStateId = "context_update" | "inference" | "inference_streaming" | "function_call" | "model_message" | "idle";
 type ReceivedMessage = {
     content: string;
@@ -227,19 +245,15 @@ interface LoopStateContract {
         output: void;
     };
 }
-type InterceptionOutput = {
-    type: "continue";
-    transition: LoopTransition;
-} | {
-    type: "pause";
-    pendingTransition: LoopTransition;
-} | {
-    type: "stop";
-    reason?: string;
-};
-interface InterceptionParams {
-    pendingTransition: LoopTransition;
-}
+type ExecutableLoopStateId = Exclude<LoopStateId, "idle">;
+type ExecutableTransition = LoopTransition<ExecutableLoopStateId>;
+type LoopStateExecution<TStateId extends ExecutableLoopStateId = ExecutableLoopStateId> = {
+    [K in TStateId]: {
+        stateId: K;
+        input: LoopStateContract[K]["input"];
+        output: LoopStateContract[K]["output"];
+    };
+}[TStateId];
 type LoopTransition<TStateId extends LoopStateId = LoopStateId> = {
     [K in TStateId]: {
         nextStateId: K;
@@ -247,6 +261,23 @@ type LoopTransition<TStateId extends LoopStateId = LoopStateId> = {
     };
 }[TStateId];
 
+declare class InputTokenDetails {
+    readonly cached_tokens: number;
+    constructor(cached_tokens: number);
+}
+declare class OutputTokenDetails {
+    readonly reasoning_tokens: number;
+    constructor(reasoning_tokens: number);
+}
+declare class TokenUsage {
+    readonly inputTokens: number;
+    readonly outputTokens: number;
+    readonly totalTokens: number;
+    readonly inputTokenDetails: InputTokenDetails;
+    readonly outputTokenDetails: OutputTokenDetails;
+    constructor(inputTokens: number, outputTokens: number, totalTokens: number, inputTokenDetails: InputTokenDetails, outputTokenDetails: OutputTokenDetails);
+}
+
 type InferenceInput = {
     model: string;
     maxOutputTokens?: number;
@@ -259,6 +290,7 @@ type InferenceInput = {
 type InferenceItem = FunctionCallItem | ReasoningItem | ModelMessageItem;
 type InferenceOutput = {
     items: InferenceItem[];
+    tokenUsage: TokenUsage | undefined;
     rowResponse: any;
 };
 interface InferenceRunner {
@@ -311,23 +343,6 @@ interface ModelContextRepository {
     getByProjectId(projectId: string): Promise<ModelContext[]>;
 }
 
-declare class InputTokenDetails {
-    readonly cached_tokens: number;
-    constructor(cached_tokens: number);
-}
-declare class OutputTokenDetails {
-    readonly reasoning_tokens: number;
-    constructor(reasoning_tokens: number);
-}
-declare class TokenUsage {
-    readonly inputTokens: number;
-    readonly outputTokens: number;
-    readonly totalTokens: number;
-    readonly inputTokenDetails: InputTokenDetails;
-    readonly outputTokenDetails: OutputTokenDetails;
-    constructor(inputTokens: number, outputTokens: number, totalTokens: number, inputTokenDetails: InputTokenDetails, outputTokenDetails: OutputTokenDetails);
-}
-
 interface McpServerConfig {
     /** The MCP server URL, e.g. https://api.githubcopilot.com/mcp/ */
     url: string;
@@ -459,11 +474,6 @@ declare abstract class RuntimeState {
     getParticipants(): readonly Participant[];
 }
 
-interface InterceptionHandler {
-    isSatisfiedBy(transition: LoopTransition): boolean;
-    handle(transition: LoopTransition): Promise<LoopTransition>;
-}
-
 declare class EventProcessor {
     process(event: SemanticEvent, consumer: Participant): void;
 }
@@ -497,7 +507,7 @@ declare function defineRuntime<TRuntimeState extends RuntimeState>(): {
     leave: (participant: Participant) => void;
     sendMessage: (message: string, senderId: string) => void;
     sendEvent: (event: SemanticEvent, senderId: string) => void;
-    runLoop: (agentId: string, message: string, inferenceInput: InferenceInput, interceptionHandler: InterceptionHandler) => void;
+    runLoop: (agentId: string, message: string, inferenceInput: InferenceInput, interceptionHandler?: InterceptionHandler) => void;
 };
 
 declare const supportedModels: GenerativeModel[];
@@ -615,4 +625,4 @@ declare class DefaultInferenceRunner implements InferenceRunner {
     stream(input: InferenceInput): AsyncGenerator<SemanticEvent>;
 }
 
-export { Agent, AnthropicMessages, ContextItem, DefaultInferenceRunner, DeveloperMessageItem, type Endpoint, FunctionCallItem, FunctionCallOutputItem, GeminiGenerateContent, Human, type InferenceInput, type InferenceOutput, type InferenceRunner, type InferenceRunnerConfig, InputTokenDetails, McpClient, type McpServerConfig, McpToolRegistry, type McpToolSpec, ModelContext, type ModelContextRepository, ModelMessageItem, OpenAIChatCompletions, OpenAIResponses, OutputTokenDetails, Participant, ReasoningItem, RuntimeState, SemanticEvent, type SituationContext, type SituationHandler, type SituationProcessor, SituationSpecification, type StructuredOutputFormat, SystemMessageItem, TokenUsage, type Tool, UserMessageItem, createAgent, createHuman, defineRuntime, supportedModels };
+export { Agent, AnthropicMessages, ContextItem, DefaultInferenceRunner, DeveloperMessageItem, type Endpoint, type ExecutableLoopStateId, type ExecutableTransition, FunctionCallItem, FunctionCallOutputItem, GeminiGenerateContent, Human, type InferenceInput, type InferenceOutput, type InferenceRunner, type InferenceRunnerConfig, InputTokenDetails, type InterceptionHandler, type LoopStateExecution, type LoopTransition, McpClient, type McpServerConfig, McpToolRegistry, type McpToolSpec, ModelContext, type ModelContextRepository, ModelMessageItem, OpenAIChatCompletions, OpenAIResponses, OutputTokenDetails, Participant, ReasoningItem, RuntimeState, SemanticEvent, type SituationContext, type SituationHandler, type SituationProcessor, SituationSpecification, type StructuredOutputFormat, SystemMessageItem, TokenUsage, type Tool, UserMessageItem, createAgent, createHuman, defineRuntime, supportedModels };

package/dist/index.d.ts

@@ -186,6 +186,24 @@ interface FunctionCallParams {
     inferenceInput: InferenceInput;
 }
 
+interface InterceptionHandler {
+    isSatisfiedBy(transition: ExecutableTransition): boolean;
+    handle(transition: ExecutableTransition): Promise<ExecutableTransition>;
+}
+type InterceptionOutput = {
+    type: "continue";
+    transition: LoopTransition;
+} | {
+    type: "pause";
+    pendingTransition: LoopTransition;
+} | {
+    type: "stop";
+    reason?: string;
+};
+interface InterceptionParams {
+    pendingTransition: LoopTransition;
+}
+
 type LoopStateId = "context_update" | "inference" | "inference_streaming" | "function_call" | "model_message" | "idle";
 type ReceivedMessage = {
     content: string;
@@ -227,19 +245,15 @@ interface LoopStateContract {
         output: void;
     };
 }
-type InterceptionOutput = {
-    type: "continue";
-    transition: LoopTransition;
-} | {
-    type: "pause";
-    pendingTransition: LoopTransition;
-} | {
-    type: "stop";
-    reason?: string;
-};
-interface InterceptionParams {
-    pendingTransition: LoopTransition;
-}
+type ExecutableLoopStateId = Exclude<LoopStateId, "idle">;
+type ExecutableTransition = LoopTransition<ExecutableLoopStateId>;
+type LoopStateExecution<TStateId extends ExecutableLoopStateId = ExecutableLoopStateId> = {
+    [K in TStateId]: {
+        stateId: K;
+        input: LoopStateContract[K]["input"];
+        output: LoopStateContract[K]["output"];
+    };
+}[TStateId];
 type LoopTransition<TStateId extends LoopStateId = LoopStateId> = {
     [K in TStateId]: {
         nextStateId: K;
@@ -247,6 +261,23 @@ type LoopTransition<TStateId extends LoopStateId = LoopStateId> = {
     };
 }[TStateId];
 
+declare class InputTokenDetails {
+    readonly cached_tokens: number;
+    constructor(cached_tokens: number);
+}
+declare class OutputTokenDetails {
+    readonly reasoning_tokens: number;
+    constructor(reasoning_tokens: number);
+}
+declare class TokenUsage {
+    readonly inputTokens: number;
+    readonly outputTokens: number;
+    readonly totalTokens: number;
+    readonly inputTokenDetails: InputTokenDetails;
+    readonly outputTokenDetails: OutputTokenDetails;
+    constructor(inputTokens: number, outputTokens: number, totalTokens: number, inputTokenDetails: InputTokenDetails, outputTokenDetails: OutputTokenDetails);
+}
+
 type InferenceInput = {
     model: string;
     maxOutputTokens?: number;
@@ -259,6 +290,7 @@ type InferenceInput = {
 type InferenceItem = FunctionCallItem | ReasoningItem | ModelMessageItem;
 type InferenceOutput = {
     items: InferenceItem[];
+    tokenUsage: TokenUsage | undefined;
     rowResponse: any;
 };
 interface InferenceRunner {
@@ -311,23 +343,6 @@ interface ModelContextRepository {
     getByProjectId(projectId: string): Promise<ModelContext[]>;
 }
 
-declare class InputTokenDetails {
-    readonly cached_tokens: number;
-    constructor(cached_tokens: number);
-}
-declare class OutputTokenDetails {
-    readonly reasoning_tokens: number;
-    constructor(reasoning_tokens: number);
-}
-declare class TokenUsage {
-    readonly inputTokens: number;
-    readonly outputTokens: number;
-    readonly totalTokens: number;
-    readonly inputTokenDetails: InputTokenDetails;
-    readonly outputTokenDetails: OutputTokenDetails;
-    constructor(inputTokens: number, outputTokens: number, totalTokens: number, inputTokenDetails: InputTokenDetails, outputTokenDetails: OutputTokenDetails);
-}
-
 interface McpServerConfig {
     /** The MCP server URL, e.g. https://api.githubcopilot.com/mcp/ */
     url: string;
@@ -459,11 +474,6 @@ declare abstract class RuntimeState {
     getParticipants(): readonly Participant[];
 }
 
-interface InterceptionHandler {
-    isSatisfiedBy(transition: LoopTransition): boolean;
-    handle(transition: LoopTransition): Promise<LoopTransition>;
-}
-
 declare class EventProcessor {
     process(event: SemanticEvent, consumer: Participant): void;
 }
@@ -497,7 +507,7 @@ declare function defineRuntime<TRuntimeState extends RuntimeState>(): {
     leave: (participant: Participant) => void;
     sendMessage: (message: string, senderId: string) => void;
     sendEvent: (event: SemanticEvent, senderId: string) => void;
-    runLoop: (agentId: string, message: string, inferenceInput: InferenceInput, interceptionHandler: InterceptionHandler) => void;
+    runLoop: (agentId: string, message: string, inferenceInput: InferenceInput, interceptionHandler?: InterceptionHandler) => void;
 };
 
 declare const supportedModels: GenerativeModel[];
@@ -615,4 +625,4 @@ declare class DefaultInferenceRunner implements InferenceRunner {
     stream(input: InferenceInput): AsyncGenerator<SemanticEvent>;
 }
 
-export { Agent, AnthropicMessages, ContextItem, DefaultInferenceRunner, DeveloperMessageItem, type Endpoint, FunctionCallItem, FunctionCallOutputItem, GeminiGenerateContent, Human, type InferenceInput, type InferenceOutput, type InferenceRunner, type InferenceRunnerConfig, InputTokenDetails, McpClient, type McpServerConfig, McpToolRegistry, type McpToolSpec, ModelContext, type ModelContextRepository, ModelMessageItem, OpenAIChatCompletions, OpenAIResponses, OutputTokenDetails, Participant, ReasoningItem, RuntimeState, SemanticEvent, type SituationContext, type SituationHandler, type SituationProcessor, SituationSpecification, type StructuredOutputFormat, SystemMessageItem, TokenUsage, type Tool, UserMessageItem, createAgent, createHuman, defineRuntime, supportedModels };
+export { Agent, AnthropicMessages, ContextItem, DefaultInferenceRunner, DeveloperMessageItem, type Endpoint, type ExecutableLoopStateId, type ExecutableTransition, FunctionCallItem, FunctionCallOutputItem, GeminiGenerateContent, Human, type InferenceInput, type InferenceOutput, type InferenceRunner, type InferenceRunnerConfig, InputTokenDetails, type InterceptionHandler, type LoopStateExecution, type LoopTransition, McpClient, type McpServerConfig, McpToolRegistry, type McpToolSpec, ModelContext, type ModelContextRepository, ModelMessageItem, OpenAIChatCompletions, OpenAIResponses, OutputTokenDetails, Participant, ReasoningItem, RuntimeState, SemanticEvent, type SituationContext, type SituationHandler, type SituationProcessor, SituationSpecification, type StructuredOutputFormat, SystemMessageItem, TokenUsage, type Tool, UserMessageItem, createAgent, createHuman, defineRuntime, supportedModels };

package/dist/index.js

@@ -622,20 +622,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() {
@@ -1130,7 +1130,7 @@ var AnthropicMessagesMapper = class {
         );
       }
     }
-    return { items, rowResponse: response };
+    return { items, tokenUsage: this.extractTokenUsage(response), rowResponse: response };
   }
 };
 
@@ -1318,6 +1318,7 @@ var OpenAIResponsesMapper = class {
     }
     return {
       items,
+      tokenUsage: this.extractTokenUsage(response),
       rowResponse: response
     };
   }
@@ -1506,7 +1507,7 @@ var GeminiGenerateContentMapper = class {
         );
       }
     }
-    return { items, rowResponse: response };
+    return { items, tokenUsage: this.extractTokenUsage(response), rowResponse: response };
   }
 };
 
@@ -1629,7 +1630,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(
@@ -1652,7 +1653,7 @@ var OpenAIChatCompletionsMapper = class {
         })
       );
     }
-    return { items, rowResponse: response };
+    return { items, tokenUsage: this.extractTokenUsage(response), rowResponse: response };
   }
 };