From 7f8b9c4a1475b868936bad36c98d50d79adac253 Mon Sep 17 00:00:00 2001
From: Alessandro Benedetti <a.benedetti@sease.io>
Date: Mon, 16 Dec 2024 18:32:54 +0000
Subject: [PATCH] gradle dependencies fix

---
 solr/modules/llm/build.gradle                                | 5 -----
 .../solr/llm/texttovector/store/rest/TestModelManager.java   | 3 +--
 2 files changed, 1 insertion(+), 7 deletions(-)

diff --git a/solr/modules/llm/build.gradle b/solr/modules/llm/build.gradle
index 45bdb9f874a..b6c3d57eb15 100644
--- a/solr/modules/llm/build.gradle
+++ b/solr/modules/llm/build.gradle
@@ -36,11 +36,6 @@ dependencies {
 
   testImplementation project(':solr:test-framework')
   testImplementation libs.junit.junit
-  testImplementation libs.hamcrest.hamcrest
   testImplementation libs.commonsio.commonsio
-  testImplementation libs.langchain4j.cohere
-  testImplementation libs.langchain4j.hugging.face
-  testImplementation libs.langchain4j.mistral.ai
-  testImplementation libs.langchain4j.open.ai
   
 }
diff --git a/solr/modules/llm/src/test/org/apache/solr/llm/texttovector/store/rest/TestModelManager.java b/solr/modules/llm/src/test/org/apache/solr/llm/texttovector/store/rest/TestModelManager.java
index f8605c02c1c..37d40b3f6c4 100644
--- a/solr/modules/llm/src/test/org/apache/solr/llm/texttovector/store/rest/TestModelManager.java
+++ b/solr/modules/llm/src/test/org/apache/solr/llm/texttovector/store/rest/TestModelManager.java
@@ -16,7 +16,6 @@
  */
 package org.apache.solr.llm.texttovector.store.rest;
 
-import dev.langchain4j.model.cohere.CohereEmbeddingModel;
 import org.apache.solr.common.util.NamedList;
 import org.apache.solr.core.SolrResourceLoader;
 import org.apache.solr.llm.TestLlmBase;
@@ -60,7 +59,7 @@ public void test() throws Exception {
   public void testRestManagerEndpoints() throws Exception {
     assertJQ("/schema/managed", "/responseHeader/status==0");
 
-    final String cohereModelClassName = CohereEmbeddingModel.class.getName();
+    final String cohereModelClassName = "dev.langchain4j.model.cohere.CohereEmbeddingModel";
 
     // Add models
     String model = "{ \"name\":\"testModel1\", \"class\":\"" + cohereModelClassName + "\"}";