Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.simiacryptus.skyenet.apps.plan.PlanSettings.kt Maven / Gradle / Ivy
package com.simiacryptus.skyenet.apps.plan
import com.simiacryptus.jopenai.describe.AbbrevWhitelistYamlDescriber
import com.simiacryptus.jopenai.models.OpenAITextModel
import com.simiacryptus.skyenet.apps.plan.CommandAutoFixTask.CommandAutoFixTaskData
import com.simiacryptus.skyenet.apps.plan.file.FileModificationTask.FileModificationTaskData
import com.simiacryptus.skyenet.apps.plan.PlanUtil.isWindows
import com.simiacryptus.skyenet.apps.plan.PlanningTask.PlanningTaskData
import com.simiacryptus.skyenet.apps.plan.PlanningTask.TaskBreakdownResult
import com.simiacryptus.skyenet.apps.plan.TaskType.Companion.getAvailableTaskTypes
import com.simiacryptus.skyenet.apps.plan.TaskType.Companion.getImpl
import com.simiacryptus.skyenet.core.actors.ParsedActor
data class TaskSettings(
var enabled: Boolean = false,
var model: OpenAITextModel? = null
)
open class PlanSettings(
var defaultModel: OpenAITextModel,
var parsingModel: OpenAITextModel,
val command: List = listOf(if (isWindows) "powershell" else "bash"),
var temperature: Double = 0.2,
val budget: Double = 2.0,
val taskSettings: MutableMap = TaskType.values().associateWith { taskType ->
TaskSettings(
when (taskType) {
TaskType.FileModification, TaskType.Inquiry -> true
else -> false
}
)
}.mapKeys { it.key.name }.toMutableMap(),
var autoFix: Boolean = false,
var allowBlocking: Boolean = true,
var commandAutoFixCommands: List? = listOf(),
val env: Map? = mapOf(),
val workingDir: String? = ".",
val language: String? = if (isWindows) "powershell" else "bash",
) {
fun getTaskSettings(taskType: TaskType<*>): TaskSettings =
taskSettings[taskType.name] ?: TaskSettings()
fun setTaskSettings(taskType: TaskType<*>, settings: TaskSettings) {
taskSettings[taskType.name] = settings
}
fun copy(
model: OpenAITextModel = this.defaultModel,
parsingModel: OpenAITextModel = this.parsingModel,
command: List = this.command,
temperature: Double = this.temperature,
budget: Double = this.budget,
taskSettings: MutableMap = this.taskSettings,
autoFix: Boolean = this.autoFix,
allowBlocking: Boolean = this.allowBlocking,
commandAutoFixCommands: List? = this.commandAutoFixCommands,
env: Map? = this.env,
workingDir: String? = this.workingDir,
language: String? = this.language,
) = PlanSettings(
defaultModel = model,
parsingModel = parsingModel,
command = command,
temperature = temperature,
budget = budget,
taskSettings = taskSettings,
autoFix = autoFix,
allowBlocking = allowBlocking,
commandAutoFixCommands = commandAutoFixCommands,
env = env,
workingDir = workingDir,
language = language,
)
fun planningActor(): ParsedActor {
val planTaskSettings = this.getTaskSettings(TaskType.TaskPlanning)
val prompt = """
|Given a user request, identify and list smaller, actionable tasks that can be directly implemented in code.
|
|For each task:
|* Detail files input and output file
|* Describe task execution dependencies and order
|* Provide a brief description of the task
|* Specify important interface and integration details (each task will run independently off of a copy of this plan)
|
|Tasks can be of the following types:
|${
getAvailableTaskTypes(this).joinToString("\n") { taskType ->
"* ${getImpl(this, taskType).promptSegment()}"
}
}
|
|Creating directories and initializing source control are out of scope.
|${if (planTaskSettings.enabled) "Do not start your plan with a plan to plan!\n" else ""}
""".trimMargin()
val describer = describer()
val parserPrompt = """
Task Subtype Schema:
${getAvailableTaskTypes(this).joinToString("\n\n") { taskType -> """
${taskType.name}:
${describer.describe(taskType.taskDataClass).replace("\n", "\n ")}
""".trim()
}}
""".trimIndent()
return ParsedActor(
name = "TaskBreakdown",
resultClass = TaskBreakdownResult::class.java,
exampleInstance = exampleInstance,
prompt = prompt,
model = planTaskSettings.model ?: this.defaultModel,
parsingModel = this.parsingModel,
temperature = this.temperature,
describer = describer,
parserPrompt = parserPrompt
)
}
open fun describer() = object : AbbrevWhitelistYamlDescriber(
"com.simiacryptus", "com.github.simiacryptus"
) {
override val includeMethods: Boolean get() = false
override fun getEnumValues(clazz: Class<*>): List {
return if (clazz == TaskType::class.java) {
taskSettings.filter { it.value.enabled }.map { it.key.toString() }
} else {
super.getEnumValues(clazz)
}
}
}
companion object {
var exampleInstance = TaskBreakdownResult(
tasksByID = mapOf(
"1" to CommandAutoFixTaskData(
task_description = "Task 1",
task_dependencies = listOf(),
command = listOf("npx", "create-react-app", ".", "--template", "typescript"),
workingDir = "."
),
"2" to FileModificationTaskData(
task_description = "Task 2",
task_dependencies = listOf("1"),
input_files = listOf("input2.txt"),
output_files = listOf("output2.txt"),
),
"3" to PlanningTaskData(
task_description = "Task 3",
task_dependencies = listOf("2"),
)
),
)
}
}
