Use Plugin ClassLoader for fallback

Author: jGauravGupta

src/main/java/io/github/jeddict/ai/hints/AssistantChatManager.java

@@ -1,5 +1,5 @@
 /**
- * Copyright 2025 the original author or authors from the Jeddict project (https://jeddict.github.io/).
+ * Copyright 2025-26 the original author or authors from the Jeddict project (https://jeddict.github.io/).
  *
  * Licensed under the Apache License, Version 2.0 (the "License"); you may not
  * use this file except in compliance with the License. You may obtain a copy of
@@ -61,6 +61,7 @@
 import static io.github.jeddict.ai.review.ReviewUtil.parseReviewsFromYaml;
 import io.github.jeddict.ai.scanner.ProjectMetadataInfo;
 import io.github.jeddict.ai.settings.PreferencesManager;
+import io.github.jeddict.ai.util.ClassLoaderUtil;
 import io.github.jeddict.ai.util.ColorUtil;
 import static io.github.jeddict.ai.util.ContextHelper.getFilesContextList;
 import static io.github.jeddict.ai.util.ContextHelper.getImageFilesContext;
@@ -477,13 +478,9 @@ public void displayHtmlContent(String filename, String title) {
         }));
     }
 
-    private void usePluginClassLoader() {
-        Thread.currentThread().setContextClassLoader(getClass().getClassLoader());
-    }
-
     public void openChat(String type, final String query, String fileName, String title, Consumer<String> action) {
         SwingUtilities.invokeLater(() -> {
-            usePluginClassLoader();
+            ClassLoaderUtil.usePluginClassLoaderIfNeeded(getClass());
             new JeddictUpdateManager().checkForJeddictUpdate();
             ac = createChatInstance(title, type, getProject());
             ac.setLayout(new BorderLayout());

src/main/java/io/github/jeddict/ai/lang/JeddictBrain.java

@@ -1,5 +1,5 @@
 /**
- * Copyright 2025 the original author or authors from the Jeddict project (https://jeddict.github.io/).
+ * Copyright 2025-26 the original author or authors from the Jeddict project (https://jeddict.github.io/).
  *
  * Licensed under the Apache License, Version 2.0 (the "License"); you may not
  * use this file except in compliance with the License. You may obtain a copy of
@@ -49,6 +49,7 @@
 import io.github.jeddict.ai.agent.ToolsProbingTool;
 import io.github.jeddict.ai.agent.pair.PairProgrammer;
 import static io.github.jeddict.ai.lang.InteractionMode.INTERACTIVE;
+import io.github.jeddict.ai.util.ClassLoaderUtil;
 import io.github.jeddict.ai.util.PropertyChangeEmitter;
 import java.lang.reflect.InvocationTargetException;
 import org.openide.DialogDisplayer;
@@ -309,18 +310,23 @@ protected boolean probeToolSupport() {
             );
 
             return toolsSupport;
-        } catch (final NoClassDefFoundError e) {
-            LOG.severe(() ->
-                "error probing tool support (conflicting classpath), returning false %s\n%s".formatted(
-                    e.toString(),
-                    Arrays.toString(e.getStackTrace())
-                )
+        } catch (final NoClassDefFoundError | java.util.ServiceConfigurationError e) {
+
+            LOG.severe(()
+                    -> "Error probing tool support (likely classloader conflict), returning false %s\n%s".formatted(
+                            e.toString(),
+                            Arrays.toString(e.getStackTrace())
+                    )
             );
+
+            ClassLoaderUtil.enablePluginClassLoaderFallback();
+
             final NotifyDescriptor nd = new NotifyDescriptor.Message(
-                "AI tool support probe failed due to a conflicting classpath issue: " + e.toString(),
-                NotifyDescriptor.ERROR_MESSAGE
+                    "AI tool support probe failed due to a classpath conflict: " + e.toString(),
+                    NotifyDescriptor.ERROR_MESSAGE
             );
             DialogDisplayer.getDefault().notify(nd);
+
         } catch (final Throwable t) {
             LOG.severe(() ->
                 "error probing tool support, returning false %s\n%s".formatted(

src/main/java/io/github/jeddict/ai/util/ClassLoaderUtil.java

@@ -0,0 +1,38 @@
+/**
+ * Copyright 2026 the original author or authors from the Jeddict project (https://jeddict.github.io/).
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not
+ * use this file except in compliance with the License. You may obtain a copy of
+ * the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+ * License for the specific language governing permissions and limitations under
+ * the License.
+ */
+package io.github.jeddict.ai.util;
+
+/**
+ *
+ * @author Gaurav Gupta
+ */
+public final class ClassLoaderUtil {
+
+    private static volatile boolean USE_PLUGIN_CLASSLOADER_FALLBACK = false;
+
+    private ClassLoaderUtil() {
+    }
+
+    public static void usePluginClassLoaderIfNeeded(Class<?> clazz) {
+        if (USE_PLUGIN_CLASSLOADER_FALLBACK) {
+            Thread.currentThread().setContextClassLoader(clazz.getClassLoader());
+        }
+    }
+
+    public static void enablePluginClassLoaderFallback() {
+        USE_PLUGIN_CLASSLOADER_FALLBACK = true;
+    }
+}