schwab_mcp 0.1.0 → 0.3.0

data/lib/schwab_mcp/schwab_client_factory.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require_relative "loggable"
+
+module SchwabMCP
+  module SchwabClientFactory
+    extend Loggable
+
+    def self.create_client
+      begin
+        log_debug("Initializing Schwab client")
+        client = SchwabRb::Auth.init_client_easy(
+          ENV['SCHWAB_API_KEY'],
+          ENV['SCHWAB_APP_SECRET'],
+          ENV['SCHWAB_CALLBACK_URI'],
+          ENV['SCHWAB_TOKEN_PATH']
+        )
+
+        unless client
+          log_error("Failed to initialize Schwab client - check credentials")
+          return nil
+        end
+
+        log_debug("Schwab client initialized successfully")
+        client
+      rescue => e
+        log_error("Error initializing Schwab client: #{e.message}")
+        log_debug("Backtrace: #{e.backtrace.first(3).join('\n')}")
+        nil
+      end
+    end
+
+    def self.cached_client
+      @client ||= create_client
+    end
+
+    def self.client_error_response
+      MCP::Tool::Response.new([{
+        type: "text",
+        text: "**Error**: Failed to initialize Schwab client. Check your credentials."
+      }])
+    end
+  end
+end

data/lib/schwab_mcp/tools/cancel_order_tool.rb

@@ -2,6 +2,7 @@ require "mcp"
 require "schwab_rb"
 require "json"
 require_relative "../loggable"
+require_relative "../schwab_client_factory"
 
 module SchwabMCP
   module Tools
@@ -52,69 +53,23 @@ module SchwabMCP
         end
 
         begin
-          client = SchwabRb::Auth.init_client_easy(
-            ENV['SCHWAB_API_KEY'],
-            ENV['SCHWAB_APP_SECRET'],
-            ENV['SCHWAB_CALLBACK_URI'],
-            ENV['TOKEN_PATH']
-          )
-
-          unless client
-            log_error("Failed to initialize Schwab client")
-            return MCP::Tool::Response.new([{
-              type: "text",
-              text: "**Error**: Failed to initialize Schwab client. Check your credentials."
-            }])
-          end
+          client = SchwabClientFactory.create_client
+          return SchwabClientFactory.client_error_response unless client
 
-          account_id = ENV[account_name]
-          unless account_id
-            available_accounts = ENV.keys.select { |key| key.end_with?('_ACCOUNT') }
-            log_error("Account name '#{account_name}' not found in environment variables")
+          available_accounts = client.available_account_names
+          unless available_accounts.include?(account_name)
+            log_error("Account name '#{account_name}' not found in configured accounts")
             return MCP::Tool::Response.new([{
               type: "text",
-              text: "**Error**: Account name '#{account_name}' not found in environment variables.\n\nAvailable accounts: #{available_accounts.join(', ')}\n\nTo configure: Set ENV['#{account_name}'] to your account ID."
+              text: "**Error**: Account name '#{account_name}' not found in configured accounts.\n\nAvailable accounts: #{available_accounts.join(', ')}\n\nTo configure: Add the account to your schwab_rb configuration file."
             }])
           end
 
-          log_debug("Found account ID: [REDACTED] for account name: #{account_name}")
-          log_debug("Fetching account numbers mapping")
-
-          account_numbers_response = client.get_account_numbers
-
-          unless account_numbers_response&.body
-            log_error("Failed to retrieve account numbers")
-            return MCP::Tool::Response.new([{
-              type: "text",
-              text: "**Error**: Failed to retrieve account numbers from Schwab API"
-            }])
-          end
-
-          account_mappings = JSON.parse(account_numbers_response.body, symbolize_names: true)
-          log_debug("Account mappings retrieved (#{account_mappings.length} accounts found)")
-
-          account_hash = nil
-          account_mappings.each do |mapping|
-            if mapping[:accountNumber] == account_id
-              account_hash = mapping[:hashValue]
-              break
-            end
-          end
-
-          unless account_hash
-            log_error("Account ID not found in available accounts")
-            return MCP::Tool::Response.new([{
-              type: "text",
-              text: "**Error**: Account ID not found in available accounts. #{account_mappings.length} accounts available."
-            }])
-          end
-
-          log_debug("Found account hash for account ID: #{account_name}")
+          log_debug("Using account name: #{account_name}")
           log_debug("Verifying order exists before attempting cancellation")
 
-          order_response = client.get_order(order_id, account_hash)
-
-          unless order_response&.body
+          order = client.get_order(order_id, account_name: account_name) # returns SchwabRb::DataObjects::Order
+          unless order
             log_warn("Order not found or empty response for order ID: #{order_id}")
             return MCP::Tool::Response.new([{
               type: "text",
@@ -122,9 +77,8 @@ module SchwabMCP
             }])
           end
 
-          order_data = JSON.parse(order_response.body)
-          order_status = order_data['status']
-          cancelable = order_data['cancelable']
+          order_status = order.status
+          cancelable = order.respond_to?(:cancelable) ? order.cancelable : true # fallback if attribute not present
 
           log_debug("Order found - Status: #{order_status}, Cancelable: #{cancelable}")
           if cancelable == false
@@ -136,11 +90,11 @@ module SchwabMCP
           end
 
           log_info("Attempting to cancel order ID: #{order_id} (Status: #{order_status})")
-          cancel_response = client.cancel_order(order_id, account_hash)
+          cancel_response = client.cancel_order(order_id, account_name: account_name)
 
           if cancel_response.respond_to?(:status) && cancel_response.status == 200
             log_info("Successfully cancelled order ID: #{order_id}")
-            formatted_response = format_cancellation_success(order_id, account_name, order_data)
+            formatted_response = format_cancellation_success(order_id, account_name, order)
           elsif cancel_response.respond_to?(:status) && cancel_response.status == 404
             log_warn("Order not found during cancellation: #{order_id}")
             return MCP::Tool::Response.new([{
@@ -149,7 +103,7 @@ module SchwabMCP
             }])
           else
             log_info("Order cancellation request submitted for order ID: #{order_id}")
-            formatted_response = format_cancellation_success(order_id, account_name, order_data)
+            formatted_response = format_cancellation_success(order_id, account_name, order)
           end
 
           MCP::Tool::Response.new([{
@@ -157,12 +111,7 @@ module SchwabMCP
             text: formatted_response
           }])
 
-        rescue JSON::ParserError => e
-          log_error("JSON parsing error: #{e.message}")
-          MCP::Tool::Response.new([{
-            type: "text",
-            text: "**Error**: Failed to parse API response: #{e.message}"
-          }])
+        # No JSON::ParserError rescue needed with data objects
         rescue => e
           log_error("Error cancelling order ID #{order_id}: #{e.message}")
           log_debug("Backtrace: #{e.backtrace.first(3).join('\n')}")
@@ -186,27 +135,26 @@ module SchwabMCP
 
       private
 
-      def self.format_cancellation_success(order_id, account_name, order_data)
+      def self.format_cancellation_success(order_id, account_name, order)
         friendly_name = account_name.gsub('_ACCOUNT', '').split('_').map(&:capitalize).join(' ')
 
         formatted = "**✅ Order Cancellation Successful**\n\n"
         formatted += "**Order ID**: #{order_id}\n"
         formatted += "**Account**: #{friendly_name} (#{account_name})\n\n"
         formatted += "**Order Details:**\n"
-        formatted += "- Original Status: #{order_data['status']}\n" if order_data['status']
-        formatted += "- Order Type: #{order_data['orderType']}\n" if order_data['orderType']
-        formatted += "- Session: #{order_data['session']}\n" if order_data['session']
-        formatted += "- Duration: #{order_data['duration']}\n" if order_data['duration']
-        formatted += "- Quantity: #{order_data['quantity']}\n" if order_data['quantity']
-        formatted += "- Price: $#{format_currency(order_data['price'])}\n" if order_data['price']
-
-        if order_data['orderLegCollection'] && order_data['orderLegCollection'].any?
+        formatted += "- Original Status: #{order.status}\n" if order.status
+        formatted += "- Order Type: #{order.order_type}\n" if order.order_type
+        formatted += "- Duration: #{order.duration}\n" if order.duration
+        formatted += "- Quantity: #{order.quantity}\n" if order.quantity
+        formatted += "- Price: $#{format_currency(order.price)}\n" if order.price
+
+        if order.order_leg_collection && order.order_leg_collection.any?
           formatted += "\n**Instruments:**\n"
-          order_data['orderLegCollection'].each do |leg|
-            if leg['instrument']
-              symbol = leg['instrument']['symbol']
-              instruction = leg['instruction']
-              quantity = leg['quantity']
+          order.order_leg_collection.each do |leg|
+            if leg.instrument
+              symbol = leg.instrument.symbol
+              instruction = leg.instruction
+              quantity = leg.quantity
               formatted += "- #{symbol}: #{instruction} #{quantity}\n"
             end
           end

data/lib/schwab_mcp/tools/get_account_names_tool.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+require "mcp"
+require_relative "../loggable"
+require "schwab_rb"
+
+module SchwabMCP
+  module Tools
+    class GetAccountNamesTool < MCP::Tool
+      extend Loggable
+      description "Get a list of configured Schwab account names"
+
+      input_schema(
+        properties: {
+          topic: {
+            type: "string",
+            description: "Asking about a specific topic related to account names (optional)"
+          }
+        },
+        required: []
+      )
+
+      annotations(
+        title: "Get Account Names",
+        read_only_hint: true,
+        destructive_hint: false,
+        idempotent_hint: true
+      )
+
+      def self.call(topic: nil, server_context:)
+        account_names = SchwabRb::AccountHashManager.new.available_account_names
+        acct_names_content = if account_names && !account_names.empty?
+          formatted_names = account_names.map { |name| "- #{name}" }.join("\n")
+          "Configured Schwab Account Names:\n\n#{formatted_names}"
+        else
+          <<~NO_ACCOUNTS
+            No Schwab Account Names Configured
+
+            You need to configure your Schwab account names in the account_names.json file.
+
+            This file should be located in your schwab_home directory (typically ~/.schwab_rb/).
+
+            For detailed setup instructions, please refer to:
+            https://github.com/jwplatta/schwab_rb/blob/main/doc/ACCOUNT_MANAGEMENT.md
+
+            The account_names.json file should map friendly names to your Schwab account hashes.
+          NO_ACCOUNTS
+        end
+
+        MCP::Tool::Response.new([{
+          type: "text",
+          text: acct_names_content
+        }])
+      end
+
+    end
+  end
+end

data/lib/schwab_mcp/tools/get_market_hours_tool.rb

@@ -1,8 +1,10 @@
+# frozen_string_literal: true
+
 require "mcp"
 require "schwab_rb"
-require "json"
 require "date"
 require_relative "../loggable"
+require_relative "../schwab_client_factory"
 
 module SchwabMCP
   module Tools
@@ -42,22 +44,9 @@ module SchwabMCP
         log_debug("Date parameter: #{date || 'today'}")
 
         begin
-          client = SchwabRb::Auth.init_client_easy(
-            ENV['SCHWAB_API_KEY'],
-            ENV['SCHWAB_APP_SECRET'],
-            ENV['SCHWAB_CALLBACK_URI'],
-            ENV['TOKEN_PATH']
-          )
-
-          unless client
-            log_error("Failed to initialize Schwab client")
-            return MCP::Tool::Response.new([{
-              type: "text",
-              text: "**Error**: Failed to initialize Schwab client. Check your credentials."
-            }])
-          end
+          client = SchwabClientFactory.create_client
+          return SchwabClientFactory.client_error_response unless client
 
-          # Parse date if provided
           parsed_date = nil
           if date
             begin
@@ -73,23 +62,24 @@ module SchwabMCP
           end
 
           log_debug("Making API request for markets: #{markets.join(', ')}")
-          response = client.get_market_hours(markets, date: parsed_date)
+          market_hours_obj = client.get_market_hours(markets, date: parsed_date, return_data_objects: true)
 
-          if response&.body
-            log_info("Successfully retrieved market hours for #{markets.join(', ')}")
-            date_info = date ? " for #{date}" : " for today"
-            MCP::Tool::Response.new([{
-              type: "text",
-              text: "**Market Hours#{date_info}:**\n\n```json\n#{response.body}\n```"
-            }])
-          else
-            log_warn("Empty response from Schwab API for markets: #{markets.join(', ')}")
-            MCP::Tool::Response.new([{
+          unless market_hours_obj
+            log_warn("No market hours data object returned for markets: #{markets.join(', ')}")
+            return MCP::Tool::Response.new([{
               type: "text",
-              text: "**No Data**: Empty response from Schwab API for markets: #{markets.join(', ')}"
+              text: "**No Data**: No market hours data returned for markets: #{markets.join(', ')}"
             }])
           end
 
+          formatted = format_market_hours_object(market_hours_obj)
+          log_info("Successfully retrieved market hours for #{markets.join(', ')}")
+          date_info = date ? " for #{date}" : " for today"
+          MCP::Tool::Response.new([{
+            type: "text",
+            text: "**Market Hours#{date_info}:**\n\n#{formatted}"
+          }])
+
         rescue => e
           log_error("Error retrieving market hours for #{markets.join(', ')}: #{e.message}")
           log_debug("Backtrace: #{e.backtrace.first(3).join('\n')}")
@@ -99,6 +89,15 @@ module SchwabMCP
           }])
         end
       end
+
+      # Format the market hours object for display
+      def self.format_market_hours_object(obj)
+        if obj.respond_to?(:to_h)
+          obj.to_h.inspect
+        else
+          obj.inspect
+        end
+      end
     end
   end
 end

data/lib/schwab_mcp/tools/get_order_tool.rb

@@ -3,6 +3,7 @@ require "schwab_rb"
 require "json"
 require_relative "../loggable"
 require_relative "../redactor"
+require_relative "../schwab_client_factory"
 
 module SchwabMCP
   module Tools
@@ -53,74 +54,25 @@ module SchwabMCP
         end
 
         begin
-          client = SchwabRb::Auth.init_client_easy(
-            ENV['SCHWAB_API_KEY'],
-            ENV['SCHWAB_APP_SECRET'],
-            ENV['SCHWAB_CALLBACK_URI'],
-            ENV['TOKEN_PATH']
-          )
+          client = SchwabClientFactory.create_client
+          return SchwabClientFactory.client_error_response unless client
 
-          unless client
-            log_error("Failed to initialize Schwab client")
+          available_accounts = client.available_account_names
+          unless available_accounts.include?(account_name)
+            log_error("Account name '#{account_name}' not found in configured accounts")
             return MCP::Tool::Response.new([{
               type: "text",
-              text: "**Error**: Failed to initialize Schwab client. Check your credentials."
+              text: "**Error**: Account name '#{account_name}' not found in configured accounts.\n\nAvailable accounts: #{available_accounts.join(', ')}\n\nTo configure: Add the account to your schwab_rb configuration file."
             }])
           end
 
-          account_id = ENV[account_name]
-          unless account_id
-            available_accounts = ENV.keys.select { |key| key.end_with?('_ACCOUNT') }
-            log_error("Account name '#{account_name}' not found in environment variables")
-            return MCP::Tool::Response.new([{
-              type: "text",
-              text: "**Error**: Account name '#{account_name}' not found in environment variables.\n\nAvailable accounts: #{available_accounts.join(', ')}\n\nTo configure: Set ENV['#{account_name}'] to your account ID."
-            }])
-          end
-
-          log_debug("Found account ID: [REDACTED] for account name: #{account_name}")
-          log_debug("Fetching account numbers mapping")
-
-          account_numbers_response = client.get_account_numbers
-
-          unless account_numbers_response&.body
-            log_error("Failed to retrieve account numbers")
-            return MCP::Tool::Response.new([{
-              type: "text",
-              text: "**Error**: Failed to retrieve account numbers from Schwab API"
-            }])
-          end
-
-          account_mappings = JSON.parse(account_numbers_response.body, symbolize_names: true)
-          log_debug("Account mappings retrieved (#{account_mappings.length} accounts found)")
-
-          account_hash = nil
-          account_mappings.each do |mapping|
-            if mapping[:accountNumber] == account_id
-              account_hash = mapping[:hashValue]
-              break
-            end
-          end
-
-          unless account_hash
-            log_error("Account ID not found in available accounts")
-            return MCP::Tool::Response.new([{
-              type: "text",
-              text: "**Error**: Account ID not found in available accounts. #{account_mappings.length} accounts available."
-            }])
-          end
-
-          log_debug("Found account hash for account ID: #{account_name}")
+          log_debug("Using account name: #{account_name}")
           log_debug("Fetching order details for order ID: #{order_id}")
 
-          order_response = client.get_order(order_id, account_hash)
-
-          if order_response&.body
+          order = client.get_order(order_id, account_name: account_name) # returns SchwabRb::DataObjects::Order
+          if order
             log_info("Successfully retrieved order details for order ID: #{order_id}")
-            order_data = JSON.parse(order_response.body)
-
-            formatted_response = format_order_data(order_data, order_id, account_name)
-
+            formatted_response = format_order_object(order, order_id, account_name)
             MCP::Tool::Response.new([{
               type: "text",
               text: formatted_response
@@ -132,13 +84,6 @@ module SchwabMCP
               text: "**No Data**: Empty response from Schwab API for order ID: #{order_id}. Order may not exist or may be in a different account."
             }])
           end
-
-        rescue JSON::ParserError => e
-          log_error("JSON parsing error: #{e.message}")
-          MCP::Tool::Response.new([{
-            type: "text",
-            text: "**Error**: Failed to parse API response: #{e.message}"
-          }])
         rescue => e
           log_error("Error retrieving order details for order ID #{order_id}: #{e.message}")
           log_debug("Backtrace: #{e.backtrace.first(3).join('\n')}")
@@ -151,104 +96,72 @@ module SchwabMCP
 
       private
 
-      def self.format_order_data(order_data, order_id, account_name)
-        friendly_name = account_name.gsub('_ACCOUNT', '').split('_').map(&:capitalize).join(' ')
 
+      def self.format_order_object(order, order_id, account_name)
+        friendly_name = account_name.gsub('_ACCOUNT', '').split('_').map(&:capitalize).join(' ')
         formatted = "**Order Details for Order ID #{order_id}:**\n\n"
         formatted += "**Account:** #{friendly_name} (#{account_name})\n\n"
 
         formatted += "**Order Information:**\n"
-        formatted += "- Order ID: #{order_data['orderId']}\n" if order_data['orderId']
-        formatted += "- Status: #{order_data['status']}\n" if order_data['status']
-        formatted += "- Order Type: #{order_data['orderType']}\n" if order_data['orderType']
-        formatted += "- Session: #{order_data['session']}\n" if order_data['session']
-        formatted += "- Duration: #{order_data['duration']}\n" if order_data['duration']
-        formatted += "- Complex Order Strategy Type: #{order_data['complexOrderStrategyType']}\n" if order_data['complexOrderStrategyType']
-        formatted += "- Cancelable: #{order_data['cancelable']}\n" if order_data.key?('cancelable')
-        formatted += "- Editable: #{order_data['editable']}\n" if order_data.key?('editable')
+        formatted += "- Order ID: #{order.order_id}\n" if order.order_id
+        formatted += "- Status: #{order.status}\n" if order.status
+        formatted += "- Order Type: #{order.order_type}\n" if order.order_type
+        formatted += "- Duration: #{order.duration}\n" if order.duration
+        formatted += "- Complex Order Strategy Type: #{order.complex_order_strategy_type}\n" if order.complex_order_strategy_type
 
         formatted += "\n**Timing:**\n"
-        formatted += "- Entered Time: #{order_data['enteredTime']}\n" if order_data['enteredTime']
-        formatted += "- Close Time: #{order_data['closeTime']}\n" if order_data['closeTime']
+        formatted += "- Entered Time: #{order.entered_time}\n" if order.entered_time
+        formatted += "- Close Time: #{order.close_time}\n" if order.close_time
 
         formatted += "\n**Quantity & Pricing:**\n"
-        formatted += "- Quantity: #{order_data['quantity']}\n" if order_data['quantity']
-        formatted += "- Filled Quantity: #{order_data['filledQuantity']}\n" if order_data['filledQuantity']
-        formatted += "- Remaining Quantity: #{order_data['remainingQuantity']}\n" if order_data['remainingQuantity']
-        formatted += "- Requested Destination: #{order_data['requestedDestination']}\n" if order_data['requestedDestination']
-        formatted += "- Destination Link Name: #{order_data['destinationLinkName']}\n" if order_data['destinationLinkName']
-        formatted += "- Price: $#{format_currency(order_data['price'])}\n" if order_data['price']
-        formatted += "- Stop Price: $#{format_currency(order_data['stopPrice'])}\n" if order_data['stopPrice']
-        formatted += "- Stop Price Link Basis: #{order_data['stopPriceLinkBasis']}\n" if order_data['stopPriceLinkBasis']
-        formatted += "- Stop Price Link Type: #{order_data['stopPriceLinkType']}\n" if order_data['stopPriceLinkType']
-        formatted += "- Stop Price Offset: $#{format_currency(order_data['stopPriceOffset'])}\n" if order_data['stopPriceOffset']
-        formatted += "- Stop Type: #{order_data['stopType']}\n" if order_data['stopType']
+        formatted += "- Quantity: #{order.quantity}\n" if order.quantity
+        formatted += "- Filled Quantity: #{order.filled_quantity}\n" if order.filled_quantity
+        formatted += "- Remaining Quantity: #{order.remaining_quantity}\n" if order.remaining_quantity
+        formatted += "- Price: $#{format_currency(order.price)}\n" if order.price
 
-        if order_data['orderLegCollection'] && order_data['orderLegCollection'].any?
+        if order.order_leg_collection && order.order_leg_collection.any?
           formatted += "\n**Order Legs:**\n"
-          order_data['orderLegCollection'].each_with_index do |leg, index|
+          order.order_leg_collection.each_with_index do |leg, index|
             formatted += "**Leg #{index + 1}:**\n"
-            formatted += "- Instruction: #{leg['instruction']}\n" if leg['instruction']
-            formatted += "- Quantity: #{leg['quantity']}\n" if leg['quantity']
-            formatted += "- Position Effect: #{leg['positionEffect']}\n" if leg['positionEffect']
-            formatted += "- Quantity Type: #{leg['quantityType']}\n" if leg['quantityType']
+            formatted += "- Instruction: #{leg.instruction}\n" if leg.instruction
+            formatted += "- Quantity: #{leg.quantity}\n" if leg.quantity
+            formatted += "- Position Effect: #{leg.position_effect}\n" if leg.position_effect
 
-            if leg['instrument']
-              instrument = leg['instrument']
+            if leg.instrument
+              instrument = leg.instrument
               formatted += "- **Instrument:**\n"
-              formatted += "  * Asset Type: #{instrument['assetType']}\n" if instrument['assetType']
-              formatted += "  * Symbol: #{instrument['symbol']}\n" if instrument['symbol']
-              formatted += "  * Description: #{instrument['description']}\n" if instrument['description']
-              formatted += "  * CUSIP: #{instrument['cusip']}\n" if instrument['cusip']
-              formatted += "  * Net Change: #{instrument['netChange']}\n" if instrument['netChange']
-
-              if instrument['putCall']
-                formatted += "  * Option Type: #{instrument['putCall']}\n"
-                formatted += "  * Strike Price: $#{format_currency(instrument['strikePrice'])}\n" if instrument['strikePrice']
-                formatted += "  * Expiration Date: #{instrument['expirationDate']}\n" if instrument['expirationDate']
-                formatted += "  * Days to Expiration: #{instrument['daysToExpiration']}\n" if instrument['daysToExpiration']
-                formatted += "  * Expiration Type: #{instrument['expirationType']}\n" if instrument['expirationType']
-                formatted += "  * Exercise Type: #{instrument['exerciseType']}\n" if instrument['exerciseType']
-                formatted += "  * Settlement Type: #{instrument['settlementType']}\n" if instrument['settlementType']
-                formatted += "  * Deliverables: #{instrument['deliverables']}\n" if instrument['deliverables']
-              end
+              formatted += "  * Asset Type: #{instrument.asset_type}\n" if instrument.asset_type
+              formatted += "  * Symbol: #{instrument.symbol}\n" if instrument.symbol
+              formatted += "  * Description: #{instrument.description}\n" if instrument.description
             end
-            formatted += "\n" unless index == order_data['orderLegCollection'].length - 1
-          end
-        end
-
-        if order_data['childOrderStrategies'] && order_data['childOrderStrategies'].any?
-          formatted += "\n**Child Order Strategies:**\n"
-          formatted += "- Number of Child Orders: #{order_data['childOrderStrategies'].length}\n"
-          order_data['childOrderStrategies'].each_with_index do |child, index|
-            formatted += "- Child Order #{index + 1}: #{child['orderId']} (Status: #{child['status']})\n" if child['orderId'] && child['status']
+            formatted += "\n" unless index == order.order_leg_collection.length - 1
           end
         end
 
-        if order_data['orderActivityCollection'] && order_data['orderActivityCollection'].any?
+        if order.order_activity_collection && order.order_activity_collection.any?
           formatted += "\n**Order Activities:**\n"
-          order_data['orderActivityCollection'].each_with_index do |activity, index|
+          order.order_activity_collection.each_with_index do |activity, index|
             formatted += "**Activity #{index + 1}:**\n"
-            formatted += "- Activity Type: #{activity['activityType']}\n" if activity['activityType']
-            formatted += "- Execution Type: #{activity['executionType']}\n" if activity['executionType']
-            formatted += "- Quantity: #{activity['quantity']}\n" if activity['quantity']
-            formatted += "- Order Remaining Quantity: #{activity['orderRemainingQuantity']}\n" if activity['orderRemainingQuantity']
+            formatted += "- Activity Type: #{activity.activity_type}\n" if activity.activity_type
+            formatted += "- Execution Type: #{activity.execution_type}\n" if activity.execution_type
+            formatted += "- Quantity: #{activity.quantity}\n" if activity.quantity
+            formatted += "- Order Remaining Quantity: #{activity.order_remaining_quantity}\n" if activity.order_remaining_quantity
 
-            if activity['executionLegs'] && activity['executionLegs'].any?
-              activity['executionLegs'].each_with_index do |exec_leg, leg_index|
+            if activity.execution_legs && activity.execution_legs.any?
+              activity.execution_legs.each_with_index do |exec_leg, leg_index|
                 formatted += "- **Execution Leg #{leg_index + 1}:**\n"
-                formatted += "  * Leg ID: #{exec_leg['legId']}\n" if exec_leg['legId']
-                formatted += "  * Price: $#{format_currency(exec_leg['price'])}\n" if exec_leg['price']
-                formatted += "  * Quantity: #{exec_leg['quantity']}\n" if exec_leg['quantity']
-                formatted += "  * Mismarked Quantity: #{exec_leg['mismarkedQuantity']}\n" if exec_leg['mismarkedQuantity']
-                formatted += "  * Time: #{exec_leg['time']}\n" if exec_leg['time']
+                formatted += "  * Leg ID: #{exec_leg.leg_id}\n" if exec_leg.leg_id
+                formatted += "  * Price: $#{format_currency(exec_leg.price)}\n" if exec_leg.price
+                formatted += "  * Quantity: #{exec_leg.quantity}\n" if exec_leg.quantity
+                formatted += "  * Mismarked Quantity: #{exec_leg.mismarked_quantity}\n" if exec_leg.mismarked_quantity
+                formatted += "  * Time: #{exec_leg.time}\n" if exec_leg.time
               end
             end
-            formatted += "\n" unless index == order_data['orderActivityCollection'].length - 1
+            formatted += "\n" unless index == order.order_activity_collection.length - 1
           end
         end
 
-        redacted_data = Redactor.redact(order_data)
+        redacted_data = Redactor.redact(order.to_h)
         formatted += "\n**Full Response (Redacted):**\n"
         formatted += "```json\n#{JSON.pretty_generate(redacted_data)}\n```"
         formatted