feat: wire the tool selection round to the FSM

[no ci]
valory-xyz · Sep 20, 2023 · 3eac849 · 3eac849
1 parent dcb3c0d
commit 3eac849
Showing 1 changed file with 10 additions and 1 deletion.
diff --git 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,