WIP: UX work and model id addition

Author: willschipp

embabel-database-core/src/main/kotlin/com/embabel/database/core/repository/AiModelRepository.kt

@@ -128,4 +128,12 @@ interface AiModelRepository {
      * @return Int of the list size
      */
     fun count(): Int
+
+
+    /**
+     * get a model by the model Id
+     * @param modelId
+     * @return the model
+     */
+    fun get(modelId: String): ModelMetadata?
 }

embabel-database-core/src/main/kotlin/com/embabel/database/core/repository/InMemoryAiModelRepository.kt

@@ -116,6 +116,10 @@ class InMemoryAiModelRepository(allModels: List<ModelMetadata> = emptyList()) :
         return models.filter { it.name.contains(name,ignoreCase = true) }
     }
 
+    override fun get(modelId: String): ModelMetadata? {
+        return models.find { it.modelId == modelId }
+    }
+
     //serialize the current list to json
     fun flushToFile() {
         // Create and configure the ObjectMapper

embabel-database-core/src/main/kotlin/com/embabel/database/core/repository/LLmModelMetadata.kt

@@ -20,8 +20,10 @@ import com.embabel.common.ai.model.ModelType;
 import com.embabel.common.ai.model.PricingModel;
 
 import java.time.LocalDate
+import java.util.UUID
 
 data class LlmModelMetadata (
+    val modelId: String = UUID.randomUUID().toString(),
     override val name: String,
     override val provider: String,
     override val knowledgeCutoffDate: LocalDate? = null,
@@ -72,8 +74,9 @@ data class LlmModelMetadata (
             pricingModel: PricingModel? = null,
             size: Long? = null,
             tags: List<String>? = null,
-            source: String? = null
-        ): LlmMetadata = LlmModelMetadata(name, provider, knowledgeCutoffDate, pricingModel, size, tags, source)
+            source: String? = null,
+            modelId: String = UUID.randomUUID().toString()
+        ): LlmMetadata = LlmModelMetadata(modelId, name, provider, knowledgeCutoffDate, pricingModel, size, tags, source)
 
         @JvmStatic
         @JvmOverloads
@@ -84,8 +87,9 @@ data class LlmModelMetadata (
             pricingModel: PricingModel? = null,
             size: Long? = null,
             tags: List<String>? = null,
-            source: String? = null
-        ): LlmMetadata = LlmModelMetadata(name, provider, knowledgeCutoffDate, pricingModel, size, tags, source)
+            source: String? = null,
+            modelId: String = UUID.randomUUID().toString()
+        ): LlmMetadata = LlmModelMetadata(modelId, name, provider, knowledgeCutoffDate, pricingModel, size, tags, source)
     }
 
 }

embabel-database-server/src/main/kotlin/com/embabel/database/server/controller/AiModelRepositoryController 2.kt

@@ -23,6 +23,7 @@ import org.springframework.web.bind.annotation.RestController
 import org.springframework.web.bind.annotation.GetMapping
 import org.springframework.web.bind.annotation.RequestParam
 import org.springframework.web.bind.annotation.RequestMapping
+import org.springframework.web.bind.annotation.PathVariable
 import org.springframework.web.bind.annotation.PostMapping
 import org.springframework.web.bind.annotation.RequestBody
 import org.springframework.beans.factory.annotation.Autowired
@@ -91,4 +92,14 @@ class AiModelRepositoryController {
         } //end if
         return aiModelRepository.findByTags(*tags.toTypedArray()) ?: throw ResponseStatusException(HttpStatus.NOT_FOUND,"No matching model found")
     }
+
+     @GetMapping("/models/{modelId}")
+    fun getById(@PathVariable modelId: String): ResponseEntity<ModelMetadata> {
+        val model = aiModelRepository.getById(modelId)
+        return if (model != null) {
+            ResponseEntity.ok(model)
+        } else {
+            ResponseEntity.notFound().build()
+        }
+    }
 }

embabel-database-server/src/main/kotlin/com/embabel/database/server/controller/AiModelRepositoryController.kt

@@ -0,0 +1,11 @@
+import { defineMock } from 'vite-plugin-mock-dev-server'
+
+export default defineMock({
+    url: '/api/v1/models/count',
+    method: 'GET',
+    body: () => {
+        return {
+            "count": 101
+        }
+    }
+})