lib/bullet_train/api/engine.rb in bullet_train-api-1.0.4 vs lib/bullet_train/api/engine.rb in bullet_train-api-1.0.5

@@ -9,8 +9,14 @@
 require "rack/cors"
 
 module BulletTrain
   module Api
     class Engine < ::Rails::Engine
+      initializer "bullet_train.api.register_api_endpoints" do |app|
+        if BulletTrain::Api
+          BulletTrain::Api.endpoints << "Api::V1::MeEndpoint"
+          BulletTrain::Api.endpoints << "Api::V1::TeamsEndpoint"
+        end
+      end
     end
   end
 end