From 3eac849f00155af1f5947b217b4b3458acd3bfa8 Mon Sep 17 00:00:00 2001
From: Adamantios <adamantios.zaras@valory.xyz>
Date: Wed, 20 Sep 2023 17:34:49 +0300
Subject: [PATCH] feat: wire the tool selection round to the FSM

[no ci]
---
 packages/valory/skills/decision_maker_abci/rounds.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/packages/valory/skills/decision_maker_abci/rounds.py b/packages/valory/skills/decision_maker_abci/rounds.py
index ba22eace7..996634d0f 100644
--- a/packages/valory/skills/decision_maker_abci/rounds.py
+++ b/packages/valory/skills/decision_maker_abci/rounds.py
@@ -55,6 +55,9 @@
 )
 from packages.valory.skills.decision_maker_abci.states.redeem import RedeemRound
 from packages.valory.skills.decision_maker_abci.states.sampling import SamplingRound
+from packages.valory.skills.decision_maker_abci.states.tool_selection import (
+    ToolSelectionRound,
+)
 from packages.valory.skills.market_manager_abci.rounds import (
     Event as MarketManagerEvent,
 )
@@ -129,11 +132,17 @@ class DecisionMakerAbciApp(AbciApp[Event]):
     }
     transition_function: AbciAppTransitionFunction = {
         SamplingRound: {
-            Event.DONE: DecisionRequestRound,
+            Event.DONE: ToolSelectionRound,
             Event.NONE: FinishedWithoutDecisionRound,
             Event.NO_MAJORITY: SamplingRound,
             Event.ROUND_TIMEOUT: SamplingRound,
         },
+        ToolSelectionRound: {
+            Event.DONE: DecisionRequestRound,
+            Event.NONE: ToolSelectionRound,
+            Event.NO_MAJORITY: ToolSelectionRound,
+            Event.ROUND_TIMEOUT: ToolSelectionRound,
+        },
         DecisionRequestRound: {
             Event.DONE: FinishedDecisionMakerRound,
             Event.SLOTS_UNSUPPORTED_ERROR: BlacklistingRound,