From 372d23a8ff39441225b41d9b45ea163813dce44d Mon Sep 17 00:00:00 2001
From: Oliver Kopp <kopp.dev@gmail.com>
Date: Sun, 15 Sep 2024 11:27:00 +0200
Subject: [PATCH 1/2] Add uses directive

---
 src/main/java/module-info.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/main/java/module-info.java b/src/main/java/module-info.java
index 0ef0ae1ea6e..dc1fce79618 100644
--- a/src/main/java/module-info.java
+++ b/src/main/java/module-info.java
@@ -142,8 +142,9 @@
 
     requires org.jooq.jool;
 
-    // region: AI
+    // region AI
     requires ai.djl.api;
+    uses ai.djl.repository.zoo.ZooProvider;
     requires ai.djl.tokenizers;
     requires jvm.openai;
     requires langchain4j;

From 486da99e02f7813af40b55dab7e258c947da87d1 Mon Sep 17 00:00:00 2001
From: Siedlerchr <siedlerkiller@gmail.com>
Date: Sun, 15 Sep 2024 14:26:39 +0200
Subject: [PATCH 2/2] add zoo factory

---
 build.gradle | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/build.gradle b/build.gradle
index abcd0eec189..ac32717a6be 100644
--- a/build.gradle
+++ b/build.gradle
@@ -779,6 +779,7 @@ jlink {
         uses 'org.eclipse.jgit.transport.SshSessionFactory'
         uses 'org.eclipse.jgit.lib.GpgSigner'
         uses 'kong.unirest.core.json.JsonEngine';
+        uses 'ai.djl.repository.zoo.ZooProvider';
         provides 'org.mariadb.jdbc.tls.TlsSocketPlugin' with 'org.mariadb.jdbc.internal.protocol.tls.DefaultTlsSocketPlugin'
         provides 'java.sql.Driver' with 'org.postgresql.Driver'
         provides 'org.mariadb.jdbc.authentication.AuthenticationPlugin' with 'org.mariadb.jdbc.internal.com.send.authentication.CachingSha2PasswordPlugin',
@@ -795,7 +796,10 @@ jlink {
         provides 'java.security.Provider' with 'org.bouncycastle.jce.provider.BouncyCastleProvider',
                 'org.bouncycastle.pqc.jcajce.provider.BouncyCastlePQCProvider'
         provides 'kong.unirest.core.json.JsonEngine' with 'kong.unirest.modules.gson.GsonEngine';
-
+        provides 'ai.djl.repository.zoo.ZooProvider' with 'ai.djl.engine.rust.zoo.RsZooProvider',
+                'ai.djl.huggingface.zoo.HfZooProvider',
+                'ai.djl.pytorch.zoo.PtZooProvider',
+                'ai.djl.repository.zoo.DefaultZooProvider';
     }
 
     jpackage {