switch to the CPU version and use the @disabled annotation to skip un…

…supported test cases.
langchain4j · Nov 30, 2024 · e59ed1a · e59ed1a
1 parent ab518b2
commit e59ed1a
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 3 deletions.
diff --git a/...t/java/dev/langchain4j/community/model/xinference/XinferenceStreamingLanguageModelIT.java b/...t/java/dev/langchain4j/community/model/xinference/XinferenceStreamingLanguageModelIT.java
@@ -9,7 +9,6 @@
 import java.time.Duration;
 
 import static dev.langchain4j.model.output.FinishReason.LENGTH;
-import static dev.langchain4j.model.output.FinishReason.STOP;
 import static org.assertj.core.api.Assertions.assertThat;
 
 class XinferenceStreamingLanguageModelIT extends AbstractInferenceLanguageModelInfrastructure {

diff --git a/...e/src/test/java/dev/langchain4j/community/model/xinference/XinferenceToolChatModelIT.java b/...e/src/test/java/dev/langchain4j/community/model/xinference/XinferenceToolChatModelIT.java
@@ -127,4 +127,12 @@ void should_handle_tool_without_parameter() {
     protected void should_execute_tool_with_pojo_with_primitives() {
         super.should_execute_tool_with_pojo_with_primitives();
     }
+
+    @Test
+    @Disabled("The support isn't great, and there are cases where it fails.")
+    @Override
+    protected void should_execute_tool_with_map_parameter() {
+        super.should_execute_tool_with_map_parameter();
+    }
 }
+
diff --git a/...-xinference/src/test/java/dev/langchain4j/community/model/xinference/XinferenceUtils.java b/...-xinference/src/test/java/dev/langchain4j/community/model/xinference/XinferenceUtils.java
@@ -13,9 +13,9 @@ class XinferenceUtils {
     public static final String XINFERENCE_BASE_URL = System.getenv("XINFERENCE_BASE_URL");
     public static final String XINFERENCE_API_KEY = System.getenv("XINFERENCE_BASE_URL");
     // CPU
-//    public static final String XINFERENCE_IMAGE = "xprobe/xinference:latest-cpu";
+    public static final String XINFERENCE_IMAGE = "xprobe/xinference:latest-cpu";
     // GPU
-    public static final String XINFERENCE_IMAGE = "xprobe/xinference:latest";
+//    public static final String XINFERENCE_IMAGE = "xprobe/xinference:latest";
 
     public static final String CHAT_MODEL_NAME = "qwen2.5-instruct";
     public static final String GENERATE_MODEL_NAME = "qwen2.5";

diff --git a/...nce/src/test/java/dev/langchain4j/community/model/xinference/XinferenceVisionModelIT.java b/...nce/src/test/java/dev/langchain4j/community/model/xinference/XinferenceVisionModelIT.java
@@ -3,6 +3,7 @@
 import dev.langchain4j.data.message.*;
 import dev.langchain4j.model.chat.ChatLanguageModel;
 import dev.langchain4j.model.output.Response;
+import org.junit.jupiter.api.Disabled;
 import org.junit.jupiter.api.Test;
 
 import java.time.Duration;
@@ -11,6 +12,7 @@
 import static dev.langchain4j.internal.Utils.readBytes;
 import static org.assertj.core.api.Assertions.assertThat;
 
+@Disabled("Torch not compiled with CUDA enabled.")
 class XinferenceVisionModelIT extends AbstractXinferenceVisionModelInfrastructure {
 
     final String CAT_IMAGE_URL = "https://img0.baidu.com/it/u=317254799,1407991361&fm=253&fmt=auto&app=138&f=JPEG?w=889&h=500";