✨ Feature: index 메타데이터 CRUD 기능 추가

src/main/kotlin/simplerag/ragback/domain/chat/entity/Model.kt

@@ -26,4 +26,4 @@ class Model(
     @Id @GeneratedValue(strategy = GenerationType.IDENTITY)
     @Column(name = "models_id")
     val id: Long? = null,
-): BaseEntity()
+) : BaseEntity()

src/main/kotlin/simplerag/ragback/domain/document/entity/DataFile.kt

@@ -2,7 +2,6 @@ package simplerag.ragback.domain.document.entity
 
 import jakarta.persistence.*
 import simplerag.ragback.global.entity.BaseEntity
-import java.time.LocalDateTime
 
 @Entity
 @Table(

src/main/kotlin/simplerag/ragback/domain/document/entity/DataFileTag.kt

@@ -21,4 +21,4 @@ class DataFileTag(
     @Id @GeneratedValue(strategy = GenerationType.IDENTITY)
     @Column(name = "data_files_tags_id")
     val id: Long? = null,
-): BaseEntity()
+) : BaseEntity()

src/main/kotlin/simplerag/ragback/domain/document/entity/Tag.kt

@@ -16,4 +16,4 @@ class Tag(
     @Id @GeneratedValue(strategy = GenerationType.IDENTITY)
     @Column(name = "tags_id")
     val id: Long? = null,
-): BaseEntity()
+) : BaseEntity()

src/main/kotlin/simplerag/ragback/domain/document/repository/DataFileTagRepository.kt

@@ -9,12 +9,14 @@ import simplerag.ragback.domain.document.entity.DataFileTag
 interface DataFileTagRepository : JpaRepository<DataFileTag, Long> {
     fun existsByDataFileIdAndTagId(dataFileId: Long, tagId: Long): Boolean
 
-    @Query("""
+    @Query(
+        """
         SELECT dft
         FROM DataFileTag dft
         JOIN FETCH dft.tag t
         WHERE dft.dataFile = :dataFile
-    """)
+    """
+    )
     fun findTagsByDataFile(@Param("dataFile") dataFile: DataFile): List<DataFileTag>
 
     fun deleteAllByDataFile(dataFile: DataFile)

src/main/kotlin/simplerag/ragback/domain/document/service/DataFileService.kt

@@ -2,7 +2,6 @@ package simplerag.ragback.domain.document.service
 
 import org.springframework.dao.DataIntegrityViolationException
 import org.springframework.data.domain.PageRequest
-import org.springframework.data.domain.Pageable
 import org.springframework.stereotype.Service
 import org.springframework.transaction.annotation.Transactional
 import org.springframework.transaction.support.TransactionSynchronization
@@ -22,9 +21,7 @@ import simplerag.ragback.global.util.S3Type
 import simplerag.ragback.global.util.S3Util
 import simplerag.ragback.global.util.computeMetricsStreaming
 import simplerag.ragback.global.util.resolveContentType
-import java.time.LocalDateTime
 import java.util.*
-import kotlin.collections.ArrayList
 
 @Service
 class DataFileService(
@@ -83,11 +80,10 @@ class DataFileService(
         val dataSlice = dataFileRepository.findByIdGreaterThanOrderById(cursor, PageRequest.of(0, take))
 
         val dataFileList: MutableList<DataFileDetailResponse> = ArrayList()
-        dataSlice.forEach{ dataFile ->
+        dataSlice.forEach { dataFile ->
             val dataFileTags: List<DataFileTag> = dataFileTagRepository.findTagsByDataFile(dataFile)
 
-            val tagDtos: List<TagDTO> = dataFileTags.map{
-                dataFileTag ->
+            val tagDtos: List<TagDTO> = dataFileTags.map { dataFileTag ->
                 val tag = dataFileTag.tag
                 TagDTO(tag.id, tag.name)
             }

src/main/kotlin/simplerag/ragback/domain/index/controller/IndexController.kt

@@ -0,0 +1,59 @@
+package simplerag.ragback.domain.index.controller
+
+import jakarta.validation.Valid
+import org.springframework.http.HttpStatus
+import org.springframework.validation.annotation.Validated
+import org.springframework.web.bind.annotation.*
+import simplerag.ragback.domain.index.dto.*
+import simplerag.ragback.domain.index.service.IndexService
+import simplerag.ragback.global.response.ApiResponse
+
+@RestController
+@RequestMapping("/api/v1/indexes")
+@Validated
+class IndexController(
+    private val indexService: IndexService
+) {
+
+    @PostMapping
+    @ResponseStatus(HttpStatus.CREATED)
+    fun createIndex(
+        @RequestBody @Valid indexCreateRequest: IndexCreateRequest
+    ): ApiResponse<IndexPreviewResponse> {
+        val createdIndex = indexService.createIndex(indexCreateRequest)
+        return ApiResponse.ok(createdIndex)
+    }
+
+    @GetMapping
+    fun getIndexes(): ApiResponse<IndexPreviewResponseList> {
+        val indexPreviewResponseList = indexService.getIndexes()
+        return ApiResponse.ok(indexPreviewResponseList)
+    }
+
+    @GetMapping("/{indexId}")
+    fun getIndex(
+        @PathVariable indexId: Long
+    ): ApiResponse<IndexDetailResponse> {
+        val indexDetailResponse = indexService.getIndex(indexId)
+        return ApiResponse.ok(indexDetailResponse)
+    }
+
+    @PutMapping("/{indexId}")
+    fun updateIndexes(
+        @PathVariable indexId: Long,
+        @RequestBody @Valid indexUpdateRequest: IndexUpdateRequest,
+    ): ApiResponse<IndexPreviewResponse> {
+        val indexPreviewResponse = indexService.updateIndex(indexId, indexUpdateRequest)
+        return ApiResponse.ok(indexPreviewResponse)
+    }
+
+    @DeleteMapping("/{indexId}")
+    fun deleteIndex(
+        @PathVariable indexId: Long
+    ): ApiResponse<Unit> {
+        indexService.deleteIndex(indexId)
+        return ApiResponse.ok(null, "인덱스가 삭제 되었습니다.")
+    }
+
+
+}

src/main/kotlin/simplerag/ragback/domain/index/converter/IndexConverter.kt

@@ -0,0 +1,47 @@
+package simplerag.ragback.domain.index.converter
+
+import simplerag.ragback.domain.index.dto.IndexCreateRequest
+import simplerag.ragback.domain.index.dto.IndexDetailResponse
+import simplerag.ragback.domain.index.dto.IndexPreviewResponse
+import simplerag.ragback.domain.index.dto.IndexPreviewResponseList
+import simplerag.ragback.domain.index.entity.Index
+
+
+fun toIndex(createRequest: IndexCreateRequest): Index {
+    return Index(
+        snapshotName = createRequest.snapshotName.trim(),
+        overlapSize = createRequest.overlapSize,
+        chunkingSize = createRequest.chunkingSize,
+        similarityMetric = createRequest.similarityMetric,
+        topK = createRequest.topK,
+        embeddingModel = createRequest.embeddingModel,
+        reranker = createRequest.reranker
+    )
+}
+
+fun toIndexPreviewResponseList(
+    indexes: List<Index>
+): IndexPreviewResponseList {
+    val indexList = indexes.map { toIndexPreviewResponse(it) }
+    return IndexPreviewResponseList(indexList)
+}
+
+fun toIndexPreviewResponse(index: Index): IndexPreviewResponse {
+    return IndexPreviewResponse(
+        indexId = index.id,
+        snapshotName = index.snapshotName,
+    )
+}
+
+fun toIndexDetailResponse(index: Index): IndexDetailResponse {
+    return IndexDetailResponse(
+        indexId = index.id,
+        chunkingSize = index.chunkingSize,
+        overlapSize = index.overlapSize,
+        similarityMetric = index.similarityMetric,
+        topK = index.topK,
+        embeddingModel = index.embeddingModel,
+        reranker = index.reranker,
+        snapshotName = index.snapshotName,
+    )
+}

src/main/kotlin/simplerag/ragback/domain/index/dto/IndexRequestDTO.kt

@@ -0,0 +1,48 @@
+package simplerag.ragback.domain.index.dto
+
+import jakarta.validation.constraints.NotBlank
+import jakarta.validation.constraints.Positive
+import jakarta.validation.constraints.PositiveOrZero
+import org.hibernate.validator.constraints.Length
+import simplerag.ragback.domain.index.entity.enums.EmbeddingModel
+import simplerag.ragback.domain.index.entity.enums.SimilarityMetric
+
+data class IndexCreateRequest(
+    @field:Length(max = 255)
+    @field:NotBlank
+    val snapshotName: String,
+
+    @field:Positive
+    val chunkingSize: Int,
+
+    @field:PositiveOrZero
+    val overlapSize: Int,
+
+    val similarityMetric: SimilarityMetric,
+
+    @field:Positive
+    val topK: Int,
+
+    val embeddingModel: EmbeddingModel,
+
+    val reranker: Boolean,
+)
+
+data class IndexUpdateRequest(
+    @field:Length(max = 255)
+    @field:NotBlank
+    val snapshotName: String,
+
+    @field:Positive
+    val chunkingSize: Int,
+
+    @field:PositiveOrZero
+    val overlapSize: Int,
+
+    val similarityMetric: SimilarityMetric,
+
+    @field:Positive
+    val topK: Int,
+
+    val reranker: Boolean,
+)

src/main/kotlin/simplerag/ragback/domain/index/dto/IndexResponseDTO.kt

@@ -0,0 +1,24 @@
+package simplerag.ragback.domain.index.dto
+
+import simplerag.ragback.domain.index.entity.enums.EmbeddingModel
+import simplerag.ragback.domain.index.entity.enums.SimilarityMetric
+
+data class IndexPreviewResponseList(
+    val indexDetailResponse: List<IndexPreviewResponse>,
+)
+
+data class IndexPreviewResponse(
+    var indexId: Long?,
+    val snapshotName: String
+)
+
+data class IndexDetailResponse(
+    var indexId: Long?,
+    val snapshotName: String,
+    val chunkingSize: Int,
+    val overlapSize: Int,
+    val similarityMetric: SimilarityMetric,
+    val topK: Int,
+    val embeddingModel: EmbeddingModel,
+    val reranker: Boolean,
+)

src/main/kotlin/simplerag/ragback/domain/index/entity/ChunkEmbedding.kt

@@ -27,7 +27,7 @@ class ChunkEmbedding(
     @Id @GeneratedValue(strategy = GenerationType.IDENTITY)
     @Column(name = "chunk_embeddings_id")
     val id: Long? = null,
-): BaseEntity() {
+) : BaseEntity() {
 
     @get:Transient
     val embedding: FloatArray get() = _embedding.copyOf()

src/main/kotlin/simplerag/ragback/domain/index/entity/DataFileIndex.kt

@@ -19,4 +19,4 @@ class DataFileIndex(
     @Id @GeneratedValue(strategy = GenerationType.IDENTITY)
     @Column(name = "data_files_indexes_id")
     val id: Long? = null,
-): BaseEntity()
+) : BaseEntity()

src/main/kotlin/simplerag/ragback/domain/index/entity/Index.kt

@@ -2,6 +2,7 @@ package simplerag.ragback.domain.index.entity
 
 import jakarta.persistence.*
 import jakarta.validation.constraints.Min
+import simplerag.ragback.domain.index.dto.IndexUpdateRequest
 import simplerag.ragback.domain.index.entity.enums.EmbeddingModel
 import simplerag.ragback.domain.index.entity.enums.SimilarityMetric
 import simplerag.ragback.global.entity.BaseEntity
@@ -11,32 +12,46 @@ import simplerag.ragback.global.entity.BaseEntity
 class Index(
 
     @Column(name = "snapshot_name", length = 255, nullable = false)
-    val snapshotName: String,
+    var snapshotName: String,
 
     @Column(name = "chunking_size", nullable = false)
     @Min(1)
-    val chunkingSize: Int,
+    var chunkingSize: Int,
 
     @Column(name = "overlap_size", nullable = false)
     @Min(0)
-    val overlapSize: Int,
+    var overlapSize: Int,
 
     @Column(name = "similarity_metric", nullable = false)
     @Enumerated(EnumType.STRING)
-    val similarityMetric: SimilarityMetric,
+    var similarityMetric: SimilarityMetric,
 
     @Column(name = "top_k", nullable = false)
     @Min(1)
-    val topK: Int,
+    var topK: Int,
 
     @Column(name = "embedding_model", nullable = false, length = 255)
     @Enumerated(EnumType.STRING)
     val embeddingModel: EmbeddingModel,
 
     @Column(name = "reranker", nullable = false)
-    val reranker: Boolean,
+    var reranker: Boolean,
+
+    @OneToMany(cascade = [CascadeType.ALL], orphanRemoval = true, mappedBy = "index")
+    val chunkEmbeddings: MutableList<ChunkEmbedding> = mutableListOf(),
 
     @Id @GeneratedValue(strategy = GenerationType.IDENTITY)
     @Column(name = "indexes_id")
-    val id: Long? = null,
-): BaseEntity()
+    var id: Long? = null,
+) : BaseEntity() {
+
+    fun update(req: IndexUpdateRequest) {
+        snapshotName = req.snapshotName.trim()
+        chunkingSize = req.chunkingSize
+        overlapSize = req.overlapSize
+        similarityMetric = req.similarityMetric
+        topK = req.topK
+        reranker = req.reranker
+    }
+
+}

src/main/kotlin/simplerag/ragback/domain/index/repository/ChunkEmbeddingRepository.kt

@@ -0,0 +1,6 @@
+package simplerag.ragback.domain.index.repository
+
+import org.springframework.data.jpa.repository.JpaRepository
+import simplerag.ragback.domain.index.entity.ChunkEmbedding
+
+interface ChunkEmbeddingRepository : JpaRepository<ChunkEmbedding, Long>

src/main/kotlin/simplerag/ragback/domain/index/repository/IndexRepository.kt

@@ -0,0 +1,9 @@
+package simplerag.ragback.domain.index.repository
+
+import org.springframework.data.jpa.repository.JpaRepository
+import simplerag.ragback.domain.index.entity.Index
+
+interface IndexRepository : JpaRepository<Index, Long> {
+
+    fun findAllByOrderByCreatedAtDesc(): List<Index>
+}