Merge pull request #38187 from code-dot-org/retrieve-user-model-ids

[AiLab] Retrieve user model ids and names from db

dashboard/app/controllers/api/v1/ml_models_controller.rb

@@ -17,9 +17,17 @@ def save
       name: params["ml_model"]["name"]
     )
     upload_to_s3(model_id, params["ml_model"].to_json)
+
     render json: "hooray!"
   end
 
+  # GET api/v1/ml_models/names
+  # Retrieve the names and ids of a user's trained ML models.
+  def user_ml_model_names
+    user_ml_model_data = UserMlModel.where(user_id: current_user.id).map {|user_ml_model| {"id": user_ml_model.model_id, "name": user_ml_model.name}}
+    render json: user_ml_model_data.to_json
+  end
+
   # GET api/v1/ml_models/:model_id
   # Retrieve a trained ML model from S3
   def get_trained_model

dashboard/config/routes.rb

@@ -693,6 +693,7 @@
       get 'peer_review_submissions/report_csv', to: 'peer_review_submissions#report_csv'
 
       post 'ml_models/save', to: 'ml_models#save'
+      get 'ml_models/names', to: 'ml_models#user_ml_model_names'
       get 'ml_models/:model_id', to: 'ml_models#get_trained_model'
 
       resources :teacher_feedbacks, only: [:index, :create] do