Push works

cccp-education · Sep 25, 2024 · a220577 · a220577
1 parent ca19a4a
commit a220577
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 6 deletions.
diff --git a/buildSrc/src/main/kotlin/school/ai/AssistantManager.kt b/buildSrc/src/main/kotlin/school/ai/AssistantManager.kt
@@ -20,8 +20,20 @@ import org.slf4j.LoggerFactory
 
 
 object AssistantManager {
+
     val logger = LoggerFactory.getLogger(AssistantPlugin::class.java)
 
+    @JvmStatic
+    val Project.localModels
+        get() = mapOf(
+            "llama3.2:3b" to "LlamaTiny",
+            "llama3.1:8b" to "LlamaSmall",
+            "mistral:7b" to "Mistral",
+            "phi3.5:3.8b" to "Phi",
+            "smollm:135m" to "SmollM",
+        )
+
+    @JvmStatic
     val Project.apiKey: String
         get() = Properties().apply {
             "$projectDir/private.properties"

diff --git a/buildSrc/src/main/kotlin/school/ai/AssistantPlugin.kt b/buildSrc/src/main/kotlin/school/ai/AssistantPlugin.kt
@@ -10,6 +10,7 @@ import school.ai.AssistantManager.apiKey
 import school.ai.AssistantManager.createOllamaChatModel
 import school.ai.AssistantManager.createOllamaStreamingChatModel
 import school.ai.AssistantManager.generateStreamingResponse
+import school.ai.AssistantManager.localModels
 import school.ai.AssistantManager.userMessage
 
 class AssistantPlugin : Plugin<Project> {
@@ -74,12 +75,7 @@ class AssistantPlugin : Plugin<Project> {
             }
 
             // Creating tasks for each model
-            mapOf(
-                "llama3.1:latest" to "Llama3",
-                "mistral" to "Mistral",
-                "phi3.5:latest" to "Phi",
-                "smollm:135m" to "SmollM",
-            ).forEach { model ->
+            localModels.forEach { model ->
                 createChatTask("helloOllama${model.value}", model.key)
                 createStreamingChatTask("helloOllamaStream${model.value}", model.key)
             }

diff --git a/buildSrc/src/main/kotlin/school/ai/datas.kt b/buildSrc/src/main/kotlin/school/ai/datas.kt
@@ -0,0 +1,2 @@
+package school.ai
+