Merge pull request #193 from tradingstrategy-ai/guard-tokensniffer-tax

Add guard for tokensniffer data

tradingstrategy/top.py

@@ -220,8 +220,12 @@ def get_buy_tax(self, epsilon=0.0001, rounding=4) -> float | None:
 
         if not self.has_tax_data():
             return None
+
+        raw_buy_fee = self.token_sniffer_data["swap_simulation"].get("buy_fee")
+        if raw_buy_fee is None:
+            return None
 
-        fee = float(self.token_sniffer_data["swap_simulation"]["buy_fee"]) / 100
+        fee = float(raw_buy_fee) / 100
         if fee < epsilon:
             return 0
         return round(fee, rounding)
@@ -246,8 +250,12 @@ def get_sell_tax(self, epsilon=0.0001, rounding=4) -> float | None:
 
         if not self.has_tax_data():
             return None
+
+        raw_sell_fee = self.token_sniffer_data["swap_simulation"].get("sell_fee")
+        if raw_sell_fee is None:
+            return None
 
-        fee = float(self.token_sniffer_data["swap_simulation"]["sell_fee"]) / 100
+        fee = float(raw_sell_fee) / 100
         if fee < epsilon:
             return 0
         return round(fee, rounding)