removed unused imports, minor fixes

src/trading/base.py

@@ -4,10 +4,9 @@ Base interfaces for trading operations.
 
 from abc import ABC, abstractmethod
 from dataclasses import dataclass
-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any
 
 from solders.pubkey import Pubkey
-from solders.signature import Signature
 
 
 @dataclass
@@ -23,7 +22,7 @@ class TokenInfo:
     user: Pubkey
 
     @classmethod
-    def from_dict(cls, data: Dict[str, Any]) -> "TokenInfo":
+    def from_dict(cls, data: dict[str, Any]) -> "TokenInfo":
         """Create TokenInfo from dictionary.
 
         Args:
@@ -42,7 +41,7 @@ class TokenInfo:
             user=Pubkey.from_string(data["user"]),
         )
 
-    def to_dict(self) -> Dict[str, str]:
+    def to_dict(self) -> dict[str, str]:
         """Convert to dictionary.
 
         Returns:

src/trading/buyer.py

@@ -2,20 +2,13 @@
 Buy operations for pump.fun tokens.
 """
 
-import asyncio
 import struct
-from typing import Final, List, Optional
+from typing import Final
 
-import spl.token.instructions as spl_token
-from solana.rpc.async_api import AsyncClient
-from solana.rpc.commitment import Confirmed
-from solana.rpc.types import TxOpts
 from solders.hash import Hash
 from solders.instruction import AccountMeta, Instruction
-from solders.keypair import Keypair
 from solders.message import Message
 from solders.pubkey import Pubkey
-from solders.system_program import TransferParams, transfer
 from solders.transaction import Transaction
 from spl.token.instructions import create_associated_token_account
 

src/trading/seller.py

@@ -2,9 +2,8 @@
 Sell operations for pump.fun tokens.
 """
 
-import asyncio
 import struct
-from typing import Final, Optional
+from typing import Final
 
 from solders.hash import Hash
 from solders.instruction import AccountMeta, Instruction

src/trading/trader.py

@@ -7,7 +7,6 @@ import asyncio
 import json
 import os
 from datetime import datetime
-from typing import Dict, Optional, Set
 
 import config
 from src.core.client import SolanaClient
@@ -74,13 +73,13 @@ class PumpTrader:
         self.buy_slippage = buy_slippage
         self.sell_slippage = sell_slippage
         self.max_retries = max_retries
-        self.max_token_age = max_token_age
+        self.max_token_age = 1  # seconds
 
         # Token processing state
         self.token_queue = asyncio.Queue()
         self.processing = False
-        self.processed_tokens: Set[str] = set()
-        self.token_timestamps: Dict[str, float] = {}
+        self.processed_tokens: set[str] = set()
+        self.token_timestamps: dict[str, float] = {}
 
     async def start(
         self,
@@ -139,7 +138,6 @@ class PumpTrader:
     async def _process_token_queue(self, marry_mode: bool, yolo_mode: bool) -> None:
         """Continuously process tokens from the queue, only if they're fresh."""
         while True:
-            # Get next token
             token_info = await self.token_queue.get()
             token_key = str(token_info.mint)
 
@@ -156,16 +154,13 @@ class PumpTrader:
                 self.token_queue.task_done()
                 continue
 
-            # Mark as processing
             self.processed_tokens.add(token_key)
 
-            # Process token
             logger.info(
                 f"Processing fresh token: {token_info.symbol} (age: {token_age:.1f}s)"
             )
             await self._handle_token(token_info, marry_mode, yolo_mode)
 
-            # Mark queue task as done
             self.token_queue.task_done()
 
     async def _handle_token(