From 23b347b1c5777c00c638afcebc471291d2357669 Mon Sep 17 00:00:00 2001
From: Matt Upham <30577966+mattupham@users.noreply.github.com>
Date: Tue, 17 Sep 2024 23:01:14 -0700
Subject: [PATCH] Update 1ct session allowed functions

---
 packages/types/src/one-click-trading-types.ts                 | 4 +++-
 .../use-create-one-click-trading-session.tsx                  | 2 ++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/packages/types/src/one-click-trading-types.ts b/packages/types/src/one-click-trading-types.ts
index 88ef0040a4..2485410aa7 100644
--- a/packages/types/src/one-click-trading-types.ts
+++ b/packages/types/src/one-click-trading-types.ts
@@ -4,7 +4,9 @@ export type AvailableOneClickTradingMessages =
   | "/osmosis.poolmanager.v1beta1.MsgSwapExactAmountIn"
   | "/osmosis.poolmanager.v1beta1.MsgSplitRouteSwapExactAmountIn"
   | "/osmosis.poolmanager.v1beta1.MsgSwapExactAmountOut"
-  | "/osmosis.poolmanager.v1beta1.MsgSplitRouteSwapExactAmountOut";
+  | "/osmosis.poolmanager.v1beta1.MsgSplitRouteSwapExactAmountOut"
+  | "/osmosis.concentratedliquidity.v1beta1.MsgWithdrawPosition"
+  | "/osmosis.valsetpref.v1beta1.MsgSetValidatorSetPreference";
 
 export type OneClickTradingResetPeriods = "day" | "week" | "month" | "year";
 export interface OneClickTradingTimeLimit {
diff --git a/packages/web/hooks/mutations/one-click-trading/use-create-one-click-trading-session.tsx b/packages/web/hooks/mutations/one-click-trading/use-create-one-click-trading-session.tsx
index 8ecaff7a1c..e8d1719c56 100644
--- a/packages/web/hooks/mutations/one-click-trading/use-create-one-click-trading-session.tsx
+++ b/packages/web/hooks/mutations/one-click-trading/use-create-one-click-trading-session.tsx
@@ -249,6 +249,8 @@ export const useCreateOneClickTradingSession = ({
         "/osmosis.poolmanager.v1beta1.MsgSplitRouteSwapExactAmountIn",
         "/osmosis.poolmanager.v1beta1.MsgSwapExactAmountOut",
         "/osmosis.poolmanager.v1beta1.MsgSplitRouteSwapExactAmountOut",
+        "/osmosis.concentratedliquidity.v1beta1.MsgWithdrawPosition",
+        "/osmosis.valsetpref.v1beta1.MsgSetValidatorSetPreference",
       ];
 
       let sessionPeriod: OneClickTradingTimeLimit;