@mozaik-ai/core 4.0.0-beta.11 → 4.0.0-beta.12

package/dist/index.d.mts

@@ -240,6 +240,14 @@ type InterceptionOutput = {
 interface InterceptionParams {
     pendingTransition: LoopTransition;
 }
+type ExecutableLoopStateId = Exclude<LoopStateId, "idle">;
+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 +255,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 +284,7 @@ type InferenceInput = {
 type InferenceItem = FunctionCallItem | ReasoningItem | ModelMessageItem;
 type InferenceOutput = {
     items: InferenceItem[];
+    tokenUsage: TokenUsage | undefined;
     rowResponse: any;
 };
 interface InferenceRunner {
@@ -311,23 +337,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;
@@ -611,8 +620,8 @@ declare class DefaultInferenceRunner implements InferenceRunner {
 }
 
 interface InterceptionHandler {
-    isSatisfiedBy(transition: LoopTransition): boolean;
-    handle(transition: LoopTransition): Promise<LoopTransition>;
+    isSatisfiedBy(transition: LoopTransition<ExecutableLoopStateId>): boolean;
+    handle(transition: LoopTransition<ExecutableLoopStateId>): Promise<LoopTransition<ExecutableLoopStateId>>;
 }
 
-export { Agent, AnthropicMessages, ContextItem, DefaultInferenceRunner, DeveloperMessageItem, type Endpoint, FunctionCallItem, FunctionCallOutputItem, GeminiGenerateContent, Human, type InferenceInput, type InferenceOutput, type InferenceRunner, type InferenceRunnerConfig, InputTokenDetails, type InterceptionHandler, 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, 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

@@ -240,6 +240,14 @@ type InterceptionOutput = {
 interface InterceptionParams {
     pendingTransition: LoopTransition;
 }
+type ExecutableLoopStateId = Exclude<LoopStateId, "idle">;
+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 +255,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 +284,7 @@ type InferenceInput = {
 type InferenceItem = FunctionCallItem | ReasoningItem | ModelMessageItem;
 type InferenceOutput = {
     items: InferenceItem[];
+    tokenUsage: TokenUsage | undefined;
     rowResponse: any;
 };
 interface InferenceRunner {
@@ -311,23 +337,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;
@@ -611,8 +620,8 @@ declare class DefaultInferenceRunner implements InferenceRunner {
 }
 
 interface InterceptionHandler {
-    isSatisfiedBy(transition: LoopTransition): boolean;
-    handle(transition: LoopTransition): Promise<LoopTransition>;
+    isSatisfiedBy(transition: LoopTransition<ExecutableLoopStateId>): boolean;
+    handle(transition: LoopTransition<ExecutableLoopStateId>): Promise<LoopTransition<ExecutableLoopStateId>>;
 }
 
-export { Agent, AnthropicMessages, ContextItem, DefaultInferenceRunner, DeveloperMessageItem, type Endpoint, FunctionCallItem, FunctionCallOutputItem, GeminiGenerateContent, Human, type InferenceInput, type InferenceOutput, type InferenceRunner, type InferenceRunnerConfig, InputTokenDetails, type InterceptionHandler, 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, 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

@@ -633,10 +633,9 @@ var AgentLoop = class _AgentLoop {
         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() {
@@ -1131,7 +1130,7 @@ var AnthropicMessagesMapper = class {
         );
       }
     }
-    return { items, rowResponse: response };
+    return { items, tokenUsage: this.extractTokenUsage(response), rowResponse: response };
   }
 };
 
@@ -1319,6 +1318,7 @@ var OpenAIResponsesMapper = class {
     }
     return {
       items,
+      tokenUsage: this.extractTokenUsage(response),
       rowResponse: response
     };
   }
@@ -1507,7 +1507,7 @@ var GeminiGenerateContentMapper = class {
         );
       }
     }
-    return { items, rowResponse: response };
+    return { items, tokenUsage: this.extractTokenUsage(response), rowResponse: response };
   }
 };
 
@@ -1630,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(
@@ -1653,7 +1653,7 @@ var OpenAIChatCompletionsMapper = class {
         })
       );
     }
-    return { items, rowResponse: response };
+    return { items, tokenUsage: this.extractTokenUsage(response), rowResponse: response };
   }
 };