Extracted out "ignore lmstudio if not running" test

http4k · Jul 9, 2024 · 05a76f9 · 05a76f9
1 parent b7a8303
commit 05a76f9
Show file tree

Hide file tree

Showing 6 changed files with 40 additions and 2 deletions.
diff --git a/ai/langchain/src/test/kotlin/org/http4k/connect/langchain/LmStudioTestCase.kt b/ai/langchain/src/test/kotlin/org/http4k/connect/langchain/LmStudioTestCase.kt
@@ -0,0 +1,15 @@
+package org.http4k.connect.langchain
+
+import org.http4k.client.JavaHttpClient
+import org.http4k.core.Method
+import org.http4k.core.Request
+import org.junit.jupiter.api.Assumptions
+
+abstract class LmStudioTestCase {
+    init {
+        Assumptions.assumeTrue(
+            JavaHttpClient()(Request(Method.GET, "http://localhost:1234/")).status.successful,
+            "No LmStudio server running"
+        )
+    }
+}
diff --git a/...gchain/src/test/kotlin/org/http4k/connect/langchain/chat/LmStudioChatLanguageModelTest.kt b/...gchain/src/test/kotlin/org/http4k/connect/langchain/chat/LmStudioChatLanguageModelTest.kt
@@ -1,12 +1,13 @@
 package org.http4k.connect.langchain.chat
 
+import org.http4k.connect.langchain.LmStudioTestCase
 import org.http4k.connect.lmstudio.CHAT_MODEL
 import org.http4k.connect.lmstudio.FakeLmStudio
 import org.http4k.connect.lmstudio.Http
 import org.http4k.connect.lmstudio.LmStudio
 import org.http4k.connect.model.ModelName
 
-class LmStudioChatLanguageModelTest : ChatLanguageModelContract {
+class LmStudioChatLanguageModelTest : LmStudioTestCase(), ChatLanguageModelContract {
     override val model by lazy {
         LmStudioChatLanguageModel(
             LmStudio.Http(FakeLmStudio()),

diff --git a/...hain/src/test/kotlin/org/http4k/connect/langchain/embedding/LmStudioEmbeddingModelTest.kt b/...hain/src/test/kotlin/org/http4k/connect/langchain/embedding/LmStudioEmbeddingModelTest.kt
@@ -1,11 +1,12 @@
 package org.http4k.connect.langchain.embedding
 
+import org.http4k.connect.langchain.LmStudioTestCase
 import org.http4k.connect.lmstudio.CHAT_MODEL
 import org.http4k.connect.lmstudio.FakeLmStudio
 import org.http4k.connect.lmstudio.Http
 import org.http4k.connect.lmstudio.LmStudio
 import org.http4k.connect.model.ModelName
 
-class LmStudioEmbeddingModelTest : EmbeddingModelContract {
+class LmStudioEmbeddingModelTest : LmStudioTestCase(), EmbeddingModelContract {
     override val model = LmStudioEmbeddingModel(LmStudio.Http(FakeLmStudio()), ModelName.CHAT_MODEL)
 }
diff --git a/.../langchain/chat/LmStudioChatLanguageModelTest.can call through to language model.approved b/.../langchain/chat/LmStudioChatLanguageModelTest.can call through to language model.approved
@@ -0,0 +1,10 @@
+Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.
+
+Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.
+
+Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.
+Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.
+
+Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.
+
+Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.
diff --git a/...onnect/langchain/chat/LmStudioChatLanguageModelTest.can use the model in a chain.approved b/...onnect/langchain/chat/LmStudioChatLanguageModelTest.can use the model in a chain.approved
@@ -0,0 +1,10 @@
+Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.
+
+Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.
+
+Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.
+Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.
+
+Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.
+
+Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.
diff --git a/...angchain/embedding/LmStudioEmbeddingModelTest.can call through to language model.approved b/...angchain/embedding/LmStudioEmbeddingModelTest.can call through to language model.approved
@@ -0,0 +1 @@
+[0.9916232, 1.0206]