fix: listen to incoming messages on agent initialize not constructor

Signed-off-by: Niall Shaw <niall.shaw@absa.africa>

packages/core/src/agent/Agent.ts

@@ -110,26 +110,6 @@ export class Agent<AgentModules extends AgentModulesInput = any> extends BaseAge
     }
 
     super(agentConfig, dependencyManager)
-
-    const stop$ = this.dependencyManager.resolve<Subject<boolean>>(InjectionSymbols.Stop$)
-
-    // Listen for new messages (either from transports or somewhere else in the framework / extensions)
-    this.messageSubscription = this.eventEmitter
-      .observable<AgentMessageReceivedEvent>(AgentEventTypes.AgentMessageReceived)
-      .pipe(
-        takeUntil(stop$),
-        concatMap((e) =>
-          this.messageReceiver
-            .receiveMessage(e.payload.message, {
-              connection: e.payload.connection,
-              contextCorrelationId: e.payload.contextCorrelationId,
-            })
-            .catch((error) => {
-              this.logger.error('Failed to process message', { error })
-            })
-        )
-      )
-      .subscribe()
   }
 
   public registerInboundTransport(inboundTransport: InboundTransport) {
@@ -199,6 +179,26 @@ export class Agent<AgentModules extends AgentModulesInput = any> extends BaseAge
     await this.mediator.initialize()
     await this.mediationRecipient.initialize()
 
+    const stop$ = this.dependencyManager.resolve<Subject<boolean>>(InjectionSymbols.Stop$)
+
+    // Listen for new messages (either from transports or somewhere else in the framework / extensions)
+    this.messageSubscription = this.eventEmitter
+      .observable<AgentMessageReceivedEvent>(AgentEventTypes.AgentMessageReceived)
+      .pipe(
+        takeUntil(stop$),
+        concatMap((e) =>
+          this.messageReceiver
+            .receiveMessage(e.payload.message, {
+              connection: e.payload.connection,
+              contextCorrelationId: e.payload.contextCorrelationId,
+            })
+            .catch((error) => {
+              this.logger.error('Failed to process message', { error })
+            })
+        )
+      )
+      .subscribe()
+
     this._isInitialized = true
   }