Increased timeouts to 5 minutes

RafalUlikowski · RafalUlikowski · commit b4af2ba18344 · 2025-05-08T22:50:36.000+02:00
diff --git a/src/main/java/com/APIprotector/apiprotector/service/OverviewService.java b/src/main/java/com/APIprotector/apiprotector/service/OverviewService.java
@@ -1,11 +1,14 @@
 package com.APIprotector.apiprotector.service;
 
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.HttpHeaders;
 import org.springframework.http.MediaType;
 import org.springframework.stereotype.Service;
 import org.springframework.web.reactive.function.BodyInserters;
 import org.springframework.web.reactive.function.client.WebClient;
 import reactor.core.publisher.Mono;
+
+import java.time.Duration;
 import java.util.Collections;
 import java.util.Map;
 
@@ -31,7 +34,10 @@ public OverviewService(
             throw new IllegalArgumentException("Prompt is empty or not configured");
         }
 
-        this.webClient = webClientBuilder.baseUrl(baseUrl).build();
+        this.webClient = webClientBuilder
+                .baseUrl(baseUrl)
+                .defaultHeaders(headers -> headers.add(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_VALUE))
+                .build();
         this.apiKey = apiKey;
         this.modelName = modelName;
         this.prompt = prompt;
@@ -64,6 +70,7 @@ public Mono<String> sendPrompt(String prompt) {
                 .contentType(MediaType.APPLICATION_JSON)
                 .body(BodyInserters.fromValue(requestBody))
                 .retrieve()
-                .bodyToMono(String.class);
+                .bodyToMono(String.class)
+                .timeout(Duration.ofMinutes(5));
     }
 }
