parent
fb7a77202f
commit
d92035c242
@ -0,0 +1,47 @@
|
||||
package kr.re.etri.autoflow.controllers;
|
||||
|
||||
import io.swagger.v3.oas.annotations.Operation;
|
||||
import io.swagger.v3.oas.annotations.tags.Tag;
|
||||
import org.springframework.http.MediaType;
|
||||
import org.springframework.http.ResponseEntity;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
import org.springframework.web.reactive.function.client.WebClient;
|
||||
import reactor.core.publisher.Mono;
|
||||
|
||||
import java.util.Collections;
|
||||
import java.util.Map;
|
||||
|
||||
@Tag(name = "MLflow API", description = "MLflow Experiment 및 Run 조회 API")
|
||||
@RestController
|
||||
@RequestMapping("/api/mlflow")
|
||||
public class MlflowController {
|
||||
|
||||
private final WebClient webClient;
|
||||
|
||||
public MlflowController() {
|
||||
this.webClient = WebClient.builder()
|
||||
.baseUrl("http://192.168.10.135:30128/api/2.0/mlflow")
|
||||
.defaultHeaders(headers -> headers.setBasicAuth("user", "LImQa2Me37nu"))
|
||||
.build();
|
||||
}
|
||||
|
||||
@Operation(summary = "Experiment의 Run 목록 조회", description = "주어진 Experiment ID의 Run을 최대 1000개까지 조회합니다.")
|
||||
@GetMapping(value = "/runs", produces = MediaType.APPLICATION_JSON_VALUE)
|
||||
public Mono<ResponseEntity<String>> getRuns(@RequestParam String experimentId) {
|
||||
Map<String, Object> body = Map.of(
|
||||
"experiment_ids", Collections.singletonList(experimentId),
|
||||
"order_by", Collections.singletonList("attribute.start_time DESC"),
|
||||
"max_results", 1000
|
||||
);
|
||||
|
||||
return webClient.post()
|
||||
.uri("/runs/search")
|
||||
.contentType(MediaType.APPLICATION_JSON)
|
||||
.bodyValue(body)
|
||||
.retrieve()
|
||||
.bodyToMono(String.class)
|
||||
.map(ResponseEntity::ok)
|
||||
.onErrorResume(e -> Mono.just(ResponseEntity.internalServerError().body(e.getMessage())));
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in new issue