", options).response.let {
+ }.generate(model, " $prefix $suffix ", null).response.let {
if (it.endsWith(END)) it.substring(0, it.length - END.length).trim(' ', '\t', '\n') else it
}
} catch (e: HttpTimeoutException) {
diff --git a/src/main/kotlin/com/plexworlds/l3/llm/LLM.kt b/src/main/kotlin/com/plexworlds/l3/llm/provider/Provider.kt
similarity index 76%
rename from src/main/kotlin/com/plexworlds/l3/llm/LLM.kt
rename to src/main/kotlin/com/plexworlds/l3/llm/provider/Provider.kt
index d2ded8f..c26b07f 100644
--- a/src/main/kotlin/com/plexworlds/l3/llm/LLM.kt
+++ b/src/main/kotlin/com/plexworlds/l3/llm/provider/Provider.kt
@@ -1,10 +1,10 @@
-package com.plexworlds.l3.llm
+package com.plexworlds.l3.llm.provider
/**
- * This is an interface for a Large Language Model (LLM).
+ * This is an interface for a Large Language Model (Provider).
* It provides a method to generate a completion suggestion based on a given prefix and suffix.
*/
-interface LLM {
+interface Provider {
/**
* This method generates a completion suggestion based on a given prefix and suffix.
*
@@ -15,7 +15,7 @@ interface LLM {
fun call(prefix: String, suffix: String): String
/**
- * This method changes the current model used by the LLM.
+ * This method changes the current model used by the Provider.
*
* @param model The name of the model to be used.
*/
diff --git a/src/main/kotlin/com/plexworlds/l3/llm/provider/Providers.kt b/src/main/kotlin/com/plexworlds/l3/llm/provider/Providers.kt
new file mode 100644
index 0000000..65da985
--- /dev/null
+++ b/src/main/kotlin/com/plexworlds/l3/llm/provider/Providers.kt
@@ -0,0 +1,7 @@
+package com.plexworlds.l3.llm.provider
+
+enum class Providers(val provider: Provider) {
+
+ OLLAMA(Ollama),
+ DUMMY(Dummy)
+}