refactor(ai): enhance selection algorithm retry logic

This commit replaces the time-based for-loop in the selection
algorithm's retry logic with a more context-aware approach.

server/ai_process.go

@@ -22,7 +22,7 @@ import (
 	"github.com/livepeer/lpms/stream"
 )
 
-const maxProcessingRetryDuration = 500 * time.Millisecond
+const processingRetryTimeout = 2 * time.Second
 const defaultTextToImageModelID = "stabilityai/sdxl-turbo"
 const defaultImageToImageModelID = "stabilityai/sdxl-turbo"
 const defaultImageToVideoModelID = "stabilityai/stable-video-diffusion-img2vid-xt"
@@ -312,9 +312,11 @@ func processAIRequest(ctx context.Context, params aiRequestParams, req interface
 
 	var resp *worker.ImageResponse
 
+	cctx, cancel := context.WithTimeout(context.Background(), processingRetryTimeout)
+	defer cancel()
+
 	tries := 0
-	retryDuration := time.Now().Add(maxProcessingRetryDuration)
-	for time.Now().Before(retryDuration) {
+	for {
 		tries++
 
 		sess, err := params.sessManager.Select(ctx, cap, modelID)
@@ -336,6 +338,10 @@ func processAIRequest(ctx context.Context, params aiRequestParams, req interface
 		clog.Infof(ctx, "Error submitting request cap=%v modelID=%v try=%v orch=%v err=%v", cap, modelID, tries, sess.Transcoder(), err)
 
 		params.sessManager.Remove(ctx, sess)
+
+		if cctx.Err() == context.DeadlineExceeded {
+			break
+		}
 	}
 
 	if resp == nil {