- get or create working directory for jlama

Author: jknack

modules/jooby-langchain4j/src/main/java/io/jooby/internal/langchain4j/BuiltInModel.java

@@ -5,7 +5,6 @@
  */
 package io.jooby.internal.langchain4j;
 
-import java.nio.file.Paths;
 import java.time.Duration;
 
 import com.typesafe.config.Config;
@@ -106,21 +105,15 @@ public ChatModel createChatModel(@NonNull Config config) {
       check("dev.langchain4j.model.jlama.JlamaChatModel", "langchain4j-jlama");
       return JlamaChatModel.builder()
           .modelName(config.getString("model-name"))
-          .workingDirectory(
-              config.hasPath("working-dir")
-                  ? Paths.get(config.getString("working-dir"))
-                  : Paths.get(System.getProperty("user.dir"), "./models"))
+          .workingDirectory(getOrCreateWorkingDir(config))
           .build();
     }
 
     @Override
     public StreamingChatModel createStreamingModel(@NonNull Config config) {
       return JlamaStreamingChatModel.builder()
           .modelName(config.getString("model-name"))
-          .workingDirectory(
-              config.hasPath("working-dir")
-                  ? Paths.get(config.getString("working-dir"))
-                  : Paths.get(System.getProperty("user.dir"), "./models"))
+          .workingDirectory(getOrCreateWorkingDir(config))
           .build();
     }
   };
@@ -164,4 +157,21 @@ protected Duration getStreamTimeout(Config config) {
   protected double getTemp(Config config) {
     return config.hasPath("temperature") ? config.getDouble("temperature") : 0.7;
   }
+
+  protected java.nio.file.Path getOrCreateWorkingDir(Config config) {
+    java.nio.file.Path path =
+        config.hasPath("working-dir")
+            ? java.nio.file.Paths.get(config.getString("working-dir"))
+            : java.nio.file.Paths.get(System.getProperty("user.dir"), "models");
+
+    try {
+      // Jlama explicitly requires the directory to exist before booting
+      if (!java.nio.file.Files.exists(path)) {
+        java.nio.file.Files.createDirectories(path);
+      }
+      return path;
+    } catch (java.io.IOException e) {
+      throw new IllegalStateException("Failed to create a working directory at: " + path, e);
+    }
+  }
 }