small fixups

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>

Author: mudler

core/agent/agent.go

@@ -180,6 +180,13 @@ func (a *Agent) SharedState() *types.AgentSharedState {
 	return a.sharedState
 }
 
+// StartConversationConsumer starts the goroutine that dispatches new conversation
+// messages to subscribers. This must be called when using AskDirect() without Run(),
+// otherwise the ConversationAction handler will deadlock on the newConversations channel.
+func (a *Agent) StartConversationConsumer() {
+	a.startNewConversationsConsumer()
+}
+
 func (a *Agent) startNewConversationsConsumer() {
 	go func() {
 		for {

core/state/pool.go

@@ -856,6 +856,10 @@ func (a *AgentPool) createAgentWithoutRun(name, pooldir string, config *AgentCon
 	a.agents[name] = agent
 	a.managers[name] = manager
 
+	// Start the conversation consumer so ConversationAction doesn't deadlock.
+	// This is normally started by Run(), but we skip Run() in distributed mode.
+	agent.StartConversationConsumer()
+
 	xlog.Info("Agent created (no Run)", "name", name)
 	return nil
 }