@mozaik-ai/core 3.3.4 → 3.4.0

package/dist/index.d.mts

@@ -122,6 +122,26 @@ declare class ReasoningItem extends ContextItem {
     toJSON(): any;
 }
 
+interface ReasoningEffort<Effort extends string> {
+    setReasoningEffort(effort: Effort): void;
+    getReasoningEffort(): Effort;
+}
+
+interface FunctionTool {
+    type: "function";
+    name: string;
+    description: string;
+    parameters: Record<string, any>;
+    strict: boolean;
+    invoke: (args: any) => Promise<any>;
+}
+type Tool = FunctionTool;
+
+interface ToolCallingCapability {
+    setTools(tools: Tool[]): void;
+    getTools(): Tool[];
+}
+
 type ModelSpecification = {
     name: string;
     supportReasoningEffort: boolean;
@@ -131,7 +151,7 @@ type ModelSpecification = {
     maxOutputTokens: number;
     supportFunctionCalling: boolean;
 };
-interface GenerativeModel {
+interface GenerativeModel extends ReasoningEffort<string>, ToolCallingCapability {
     readonly specification: ModelSpecification;
 }
 
@@ -154,26 +174,6 @@ interface ModelContextRepository {
     getByProjectId(projectId: string): Promise<ModelContext[]>;
 }
 
-interface ReasoningEffort<Effort extends string> {
-    setReasoningEffort(effort: Effort): void;
-    getReasoningEffort(): Effort;
-}
-
-interface FunctionTool {
-    type: "function";
-    name: string;
-    description: string;
-    parameters: Record<string, any>;
-    strict: boolean;
-    invoke: (args: any) => Promise<any>;
-}
-type Tool = FunctionTool;
-
-interface ToolCallingCapability {
-    setTools(tools: Tool[]): void;
-    getTools(): Tool[];
-}
-
 declare class InferenceRequest {
     readonly model: GenerativeModel & ReasoningEffort<string> & ToolCallingCapability;
     readonly context: ModelContext;
@@ -324,6 +324,18 @@ declare class InMemoryModelContextRepository implements ModelContextRepository {
     private clone;
 }
 
+declare enum HookId {
+    BEFORE_MESSAGE_RECEIVED = "BEFORE_MESSAGE_RECEIVED",
+    AFTER_MESSAGE_RECEIVED = "AFTER_MESSAGE_RECEIVED",
+    BEFORE_INFERENCE = "BEFORE_INFERENCE",
+    AFTER_INFERENCE = "AFTER_INFERENCE",
+    BEFORE_FUNCTION_CALL = "BEFORE_FUNCTION_CALL",
+    AFTER_FUNCTION_CALL = "AFTER_FUNCTION_CALL",
+    BEFORE_MODEL_RESPONDED = "BEFORE_MODEL_RESPONDED",
+    AFTER_MODEL_RESPONDED = "AFTER_MODEL_RESPONDED",
+    ON_ERROR = "ON_ERROR"
+}
+
 interface TransitionRecord {
     fromState: StateId;
     toState: StateId;
@@ -367,6 +379,19 @@ interface RuntimeContext {
     error?: Error;
 }
 
+declare class AgentRuntime {
+    private hooksRegistry;
+    private stateHandlerRegistry;
+    constructor();
+    onMessageReceived(runtimeContext: RuntimeContext): Promise<void>;
+    onInferencePending(runtimeContext: RuntimeContext): Promise<void>;
+    onFunctionCallPending(runtimeContext: RuntimeContext): Promise<void>;
+    onError(context: RuntimeContext): Promise<void>;
+    onModelResponded(context: RuntimeContext): Promise<void>;
+    on(hookId: HookId, callback: (data: any) => Promise<void>): Promise<void>;
+    start(userMessage: UserMessageItem, model: GenerativeModel & ReasoningEffort<string> & ToolCallingCapability, context: ModelContext): Promise<void>;
+}
+
 interface InferenceVisitor {
     onStart(context: RuntimeContext): Promise<void>;
     afterInference(inferenceResponse: InferenceResponse): Promise<void>;
@@ -380,7 +405,7 @@ interface FunctionCallVisitor {
 declare class Agent {
     private visitor;
     private functionCallVisitor;
-    private runtime;
+    protected runtime: AgentRuntime;
     constructor();
     beforeMessageReceived(context: RuntimeContext): Promise<void>;
     afterMessageReceived(context: RuntimeContext): Promise<void>;
@@ -420,4 +445,49 @@ declare class SystemMessageItem extends ContextItem {
     }): SystemMessageItem;
 }
 
-export { type Action, Agent, AgentSociety, type AsyncAction, type AsyncRule, BaseSpecification, ContextItem, DeveloperMessageItem, FunctionCallItem, FunctionCallOutputItem, type GenerativeModel, Gpt54, Gpt54Mini, Gpt54Nano, If, InMemoryModelContextRepository, InferenceRequest, InferenceResponse, type InferenceVisitor, InputTokenDetails, Loop, ModelContext, type ModelContextRepository, ModelMessageItem, OpenAIResponses, OutputTokenDetails, ReasoningItem, type RuntimeContext, type Specification, SystemMessageItem, TokenUsage, type Tool, UserMessageItem };
+interface ToolExecutor {
+    produce(functionCallItem: FunctionCallItem, signal?: AbortSignal): AsyncIterable<FunctionCallOutputItem>;
+}
+
+interface InferenceHandler {
+    produce(context: ModelContext, model: GenerativeModel, signal?: AbortSignal): AsyncIterable<ReasoningItem | FunctionCallItem | ModelMessageItem>;
+}
+
+interface MessageSender {
+    stream(message: string, signal?: AbortSignal): AsyncIterable<string>;
+}
+
+declare abstract class Participant {
+    abstract onFunctionCallOutput(participant: Participant, item: FunctionCallOutputItem): Promise<void>;
+    abstract onMessage(participant: Participant, message: string): Promise<void>;
+    abstract onFunctionCall(participant: Participant, item: FunctionCallItem): Promise<void>;
+    abstract onReasoning(participant: Participant, item: ReasoningItem): Promise<void>;
+    abstract onOutputMessage(participant: Participant, item: ModelMessageItem): Promise<void>;
+}
+
+declare abstract class AgenticEnvironment {
+    protected toolExecutor: ToolExecutor;
+    protected messageSender: MessageSender;
+    protected inferenceHandler: InferenceHandler;
+    protected participants: Participant[];
+    constructor(toolExecutor: ToolExecutor, messageSender: MessageSender, inferenceHandler: InferenceHandler);
+    join(user: Participant): void;
+    leave(user: Participant): void;
+    notifyFunctionCallOutput(user: Participant, item: FunctionCallOutputItem): void;
+    notifyMessage(user: Participant, message: string): void;
+    abstract executeFunctionCall(user: Participant, functionCallItem: FunctionCallItem): Promise<void>;
+    abstract sendMessage(user: Participant, message: string): Promise<void>;
+    abstract runInference(user: Participant, context: ModelContext, model: GenerativeModel, signal?: AbortSignal): Promise<void>;
+    notifyFunctionCall(user: Participant, item: FunctionCallItem): void;
+    notifyReasoning(user: Participant, item: ReasoningItem): void;
+    notifyOutputMessage(user: Participant, item: ModelMessageItem): void;
+}
+
+declare class Environment extends AgenticEnvironment {
+    constructor(toolExecutor: ToolExecutor, messageSender: MessageSender, inferenceHandler: InferenceHandler);
+    executeFunctionCall(user: Participant, functionCallItem: FunctionCallItem): Promise<void>;
+    sendMessage(user: Participant, message: string): Promise<void>;
+    runInference(user: Participant, context: ModelContext, model: GenerativeModel, signal?: AbortSignal): Promise<void>;
+}
+
+export { type Action, Agent, AgentSociety, type AsyncAction, type AsyncRule, BaseSpecification, ContextItem, DeveloperMessageItem, Environment, FunctionCallItem, FunctionCallOutputItem, type GenerativeModel, Gpt54, Gpt54Mini, Gpt54Nano, If, InMemoryModelContextRepository, type InferenceHandler, InferenceRequest, InferenceResponse, type InferenceVisitor, InputTokenDetails, Loop, type MessageSender, ModelContext, type ModelContextRepository, ModelMessageItem, OpenAIResponses, OutputTokenDetails, ReasoningItem, type RuntimeContext, type Specification, SystemMessageItem, TokenUsage, type Tool, type ToolExecutor, UserMessageItem };

package/dist/index.d.ts

@@ -122,6 +122,26 @@ declare class ReasoningItem extends ContextItem {
     toJSON(): any;
 }
 
+interface ReasoningEffort<Effort extends string> {
+    setReasoningEffort(effort: Effort): void;
+    getReasoningEffort(): Effort;
+}
+
+interface FunctionTool {
+    type: "function";
+    name: string;
+    description: string;
+    parameters: Record<string, any>;
+    strict: boolean;
+    invoke: (args: any) => Promise<any>;
+}
+type Tool = FunctionTool;
+
+interface ToolCallingCapability {
+    setTools(tools: Tool[]): void;
+    getTools(): Tool[];
+}
+
 type ModelSpecification = {
     name: string;
     supportReasoningEffort: boolean;
@@ -131,7 +151,7 @@ type ModelSpecification = {
     maxOutputTokens: number;
     supportFunctionCalling: boolean;
 };
-interface GenerativeModel {
+interface GenerativeModel extends ReasoningEffort<string>, ToolCallingCapability {
     readonly specification: ModelSpecification;
 }
 
@@ -154,26 +174,6 @@ interface ModelContextRepository {
     getByProjectId(projectId: string): Promise<ModelContext[]>;
 }
 
-interface ReasoningEffort<Effort extends string> {
-    setReasoningEffort(effort: Effort): void;
-    getReasoningEffort(): Effort;
-}
-
-interface FunctionTool {
-    type: "function";
-    name: string;
-    description: string;
-    parameters: Record<string, any>;
-    strict: boolean;
-    invoke: (args: any) => Promise<any>;
-}
-type Tool = FunctionTool;
-
-interface ToolCallingCapability {
-    setTools(tools: Tool[]): void;
-    getTools(): Tool[];
-}
-
 declare class InferenceRequest {
     readonly model: GenerativeModel & ReasoningEffort<string> & ToolCallingCapability;
     readonly context: ModelContext;
@@ -324,6 +324,18 @@ declare class InMemoryModelContextRepository implements ModelContextRepository {
     private clone;
 }
 
+declare enum HookId {
+    BEFORE_MESSAGE_RECEIVED = "BEFORE_MESSAGE_RECEIVED",
+    AFTER_MESSAGE_RECEIVED = "AFTER_MESSAGE_RECEIVED",
+    BEFORE_INFERENCE = "BEFORE_INFERENCE",
+    AFTER_INFERENCE = "AFTER_INFERENCE",
+    BEFORE_FUNCTION_CALL = "BEFORE_FUNCTION_CALL",
+    AFTER_FUNCTION_CALL = "AFTER_FUNCTION_CALL",
+    BEFORE_MODEL_RESPONDED = "BEFORE_MODEL_RESPONDED",
+    AFTER_MODEL_RESPONDED = "AFTER_MODEL_RESPONDED",
+    ON_ERROR = "ON_ERROR"
+}
+
 interface TransitionRecord {
     fromState: StateId;
     toState: StateId;
@@ -367,6 +379,19 @@ interface RuntimeContext {
     error?: Error;
 }
 
+declare class AgentRuntime {
+    private hooksRegistry;
+    private stateHandlerRegistry;
+    constructor();
+    onMessageReceived(runtimeContext: RuntimeContext): Promise<void>;
+    onInferencePending(runtimeContext: RuntimeContext): Promise<void>;
+    onFunctionCallPending(runtimeContext: RuntimeContext): Promise<void>;
+    onError(context: RuntimeContext): Promise<void>;
+    onModelResponded(context: RuntimeContext): Promise<void>;
+    on(hookId: HookId, callback: (data: any) => Promise<void>): Promise<void>;
+    start(userMessage: UserMessageItem, model: GenerativeModel & ReasoningEffort<string> & ToolCallingCapability, context: ModelContext): Promise<void>;
+}
+
 interface InferenceVisitor {
     onStart(context: RuntimeContext): Promise<void>;
     afterInference(inferenceResponse: InferenceResponse): Promise<void>;
@@ -380,7 +405,7 @@ interface FunctionCallVisitor {
 declare class Agent {
     private visitor;
     private functionCallVisitor;
-    private runtime;
+    protected runtime: AgentRuntime;
     constructor();
     beforeMessageReceived(context: RuntimeContext): Promise<void>;
     afterMessageReceived(context: RuntimeContext): Promise<void>;
@@ -420,4 +445,49 @@ declare class SystemMessageItem extends ContextItem {
     }): SystemMessageItem;
 }
 
-export { type Action, Agent, AgentSociety, type AsyncAction, type AsyncRule, BaseSpecification, ContextItem, DeveloperMessageItem, FunctionCallItem, FunctionCallOutputItem, type GenerativeModel, Gpt54, Gpt54Mini, Gpt54Nano, If, InMemoryModelContextRepository, InferenceRequest, InferenceResponse, type InferenceVisitor, InputTokenDetails, Loop, ModelContext, type ModelContextRepository, ModelMessageItem, OpenAIResponses, OutputTokenDetails, ReasoningItem, type RuntimeContext, type Specification, SystemMessageItem, TokenUsage, type Tool, UserMessageItem };
+interface ToolExecutor {
+    produce(functionCallItem: FunctionCallItem, signal?: AbortSignal): AsyncIterable<FunctionCallOutputItem>;
+}
+
+interface InferenceHandler {
+    produce(context: ModelContext, model: GenerativeModel, signal?: AbortSignal): AsyncIterable<ReasoningItem | FunctionCallItem | ModelMessageItem>;
+}
+
+interface MessageSender {
+    stream(message: string, signal?: AbortSignal): AsyncIterable<string>;
+}
+
+declare abstract class Participant {
+    abstract onFunctionCallOutput(participant: Participant, item: FunctionCallOutputItem): Promise<void>;
+    abstract onMessage(participant: Participant, message: string): Promise<void>;
+    abstract onFunctionCall(participant: Participant, item: FunctionCallItem): Promise<void>;
+    abstract onReasoning(participant: Participant, item: ReasoningItem): Promise<void>;
+    abstract onOutputMessage(participant: Participant, item: ModelMessageItem): Promise<void>;
+}
+
+declare abstract class AgenticEnvironment {
+    protected toolExecutor: ToolExecutor;
+    protected messageSender: MessageSender;
+    protected inferenceHandler: InferenceHandler;
+    protected participants: Participant[];
+    constructor(toolExecutor: ToolExecutor, messageSender: MessageSender, inferenceHandler: InferenceHandler);
+    join(user: Participant): void;
+    leave(user: Participant): void;
+    notifyFunctionCallOutput(user: Participant, item: FunctionCallOutputItem): void;
+    notifyMessage(user: Participant, message: string): void;
+    abstract executeFunctionCall(user: Participant, functionCallItem: FunctionCallItem): Promise<void>;
+    abstract sendMessage(user: Participant, message: string): Promise<void>;
+    abstract runInference(user: Participant, context: ModelContext, model: GenerativeModel, signal?: AbortSignal): Promise<void>;
+    notifyFunctionCall(user: Participant, item: FunctionCallItem): void;
+    notifyReasoning(user: Participant, item: ReasoningItem): void;
+    notifyOutputMessage(user: Participant, item: ModelMessageItem): void;
+}
+
+declare class Environment extends AgenticEnvironment {
+    constructor(toolExecutor: ToolExecutor, messageSender: MessageSender, inferenceHandler: InferenceHandler);
+    executeFunctionCall(user: Participant, functionCallItem: FunctionCallItem): Promise<void>;
+    sendMessage(user: Participant, message: string): Promise<void>;
+    runInference(user: Participant, context: ModelContext, model: GenerativeModel, signal?: AbortSignal): Promise<void>;
+}
+
+export { type Action, Agent, AgentSociety, type AsyncAction, type AsyncRule, BaseSpecification, ContextItem, DeveloperMessageItem, Environment, FunctionCallItem, FunctionCallOutputItem, type GenerativeModel, Gpt54, Gpt54Mini, Gpt54Nano, If, InMemoryModelContextRepository, type InferenceHandler, InferenceRequest, InferenceResponse, type InferenceVisitor, InputTokenDetails, Loop, type MessageSender, ModelContext, type ModelContextRepository, ModelMessageItem, OpenAIResponses, OutputTokenDetails, ReasoningItem, type RuntimeContext, type Specification, SystemMessageItem, TokenUsage, type Tool, type ToolExecutor, UserMessageItem };

package/dist/index.js

@@ -35,6 +35,7 @@ __export(index_exports, {
   BaseSpecification: () => BaseSpecification,
   ContextItem: () => ContextItem,
   DeveloperMessageItem: () => DeveloperMessageItem,
+  Environment: () => Environment,
   FunctionCallItem: () => FunctionCallItem,
   FunctionCallOutputItem: () => FunctionCallOutputItem,
   Gpt54: () => Gpt54,
@@ -1086,6 +1087,78 @@ var SystemMessageItem = class _SystemMessageItem extends ContextItem {
     return new _SystemMessageItem(content);
   }
 };
+
+// src/domain/agentic-environment/agentic-environment.ts
+var AgenticEnvironment = class {
+  constructor(toolExecutor, messageSender, inferenceHandler) {
+    this.participants = [];
+    this.toolExecutor = toolExecutor;
+    this.messageSender = messageSender;
+    this.inferenceHandler = inferenceHandler;
+  }
+  join(user) {
+    this.participants.push(user);
+  }
+  leave(user) {
+    this.participants = this.participants.filter((p) => p !== user);
+  }
+  notifyFunctionCallOutput(user, item) {
+    for (const participant of this.participants) {
+      participant.onFunctionCallOutput(user, item);
+    }
+  }
+  notifyMessage(user, message) {
+    for (const participant of this.participants) {
+      if (participant === user) {
+        continue;
+      }
+      participant.onMessage(user, message);
+    }
+  }
+  notifyFunctionCall(user, item) {
+    for (const participant of this.participants) {
+      participant.onFunctionCall(user, item);
+    }
+  }
+  notifyReasoning(user, item) {
+    for (const participant of this.participants) {
+      participant.onReasoning(user, item);
+    }
+  }
+  notifyOutputMessage(user, item) {
+    for (const participant of this.participants) {
+      participant.onOutputMessage(user, item);
+    }
+  }
+};
+
+// src/application/environment.ts
+var Environment = class extends AgenticEnvironment {
+  constructor(toolExecutor, messageSender, inferenceHandler) {
+    super(toolExecutor, messageSender, inferenceHandler);
+  }
+  async executeFunctionCall(user, functionCallItem) {
+    for await (const item of this.toolExecutor.produce(functionCallItem)) {
+      this.notifyFunctionCallOutput(user, item);
+    }
+  }
+  async sendMessage(user, message) {
+    for await (const messagePart of this.messageSender.stream(message)) {
+      this.notifyMessage(user, messagePart);
+    }
+  }
+  async runInference(user, context, model, signal) {
+    for await (const item of this.inferenceHandler.produce(context, model, signal)) {
+      if (item instanceof FunctionCallItem) {
+        this.notifyFunctionCall(user, item);
+      } else if (item instanceof ReasoningItem) {
+        this.notifyReasoning(user, item);
+      } else if (item instanceof ModelMessageItem) {
+        this.notifyOutputMessage(user, item);
+      }
+    }
+  }
+};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   Agent,
@@ -1093,6 +1166,7 @@ var SystemMessageItem = class _SystemMessageItem extends ContextItem {
   BaseSpecification,
   ContextItem,
   DeveloperMessageItem,
+  Environment,
   FunctionCallItem,
   FunctionCallOutputItem,
   Gpt54,