Skip to content

Commit 1d7d681

Browse files
authored
Merge pull request #4 from SteveDala/development
Development
2 parents c5a217f + a8fa2ef commit 1d7d681

2 files changed

Lines changed: 5 additions & 1 deletion

File tree

‎app/api/sidekiq_api.rb‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ class SidekiqApi < Grape::API
1717
job_data = Sidekiq::Status.get_all(job_id)
1818
initiator = Sidekiq::Status.get(job_id, :initiator)
1919

20-
if initiator.nil?
20+
if job_data.blank? || initiator.nil?
2121
error!({ error: 'Job not found or has no owner' }, 404)
2222
end
2323

‎app/sidekiq/predict_effort_job.rb‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,16 @@ def perform(task_def_id, user_id)
1515
raise StandardError, "ML_SERVICE_URL is not configured"
1616
end
1717
Rails.logger.info("ML payload: #{payload.to_json}")
18+
1819
response = Net::HTTP.post(
1920
URI("#{ml_url}predict"),
2021
payload.to_json,
2122
"Content-Type" => "application/json"
2223
)
2324

25+
unless response.is_a?(Net::HTTPSuccess)
26+
raise StandardError, "ML service returned #{response.code}: #{response.body}"
27+
end
2428
result = JSON.parse(response.body)
2529
Rails.logger.info("FastAPI response: #{response.body}")
2630
td.update(predicted_effort: result["predicted_effort"])

0 commit comments

Comments
 (0)