diff --git a/examples/llama.android/app/src/main/java/com/example/llama/revamp/data/repository/ModelRepository.kt b/examples/llama.android/app/src/main/java/com/example/llama/revamp/data/repository/ModelRepository.kt index 8690486c62..40d6d33247 100644 --- a/examples/llama.android/app/src/main/java/com/example/llama/revamp/data/repository/ModelRepository.kt +++ b/examples/llama.android/app/src/main/java/com/example/llama/revamp/data/repository/ModelRepository.kt @@ -36,6 +36,7 @@ import javax.inject.Singleton interface ModelRepository { fun getStorageMetrics(): Flow fun getModels(): Flow> + suspend fun getModelById(id: String): ModelInfo? suspend fun importModel( uri: Uri, @@ -89,6 +90,9 @@ class ModelRepositoryImpl @Inject constructor( } } + override suspend fun getModelById(id: String) = + modelDao.getModelById(id)?.toModelInfo() + override suspend fun importModel( uri: Uri, name: String?,