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.PlanUtil.isWindows
import com.simiacryptus.skyenet.apps.plan.PlanningTask.TaskBreakdownResult
import com.simiacryptus.skyenet.apps.plan.TaskType.Companion.getAvailableTaskTypes
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
}
)
}.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] ?: TaskSettings()
fun setTaskSettings(taskType: TaskType, settings: TaskSettings) {
taskSettings[taskType] = 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,
)
var exampleInstance = TaskBreakdownResult(
tasksByID = mapOf(
"1" to PlanningTask.PlanTask(
task_description = "Task 1",
task_type = TaskType.CommandAutoFix,
task_dependencies = listOf(),
execution_task = PlanningTask.ExecutionTask(
command = listOf("npx", "create-react-app", ".", "--template", "typescript"),
workingDir = ".",
)
),
"2" to PlanningTask.PlanTask(
task_description = "Task 2",
task_type = TaskType.FileModification,
task_dependencies = listOf("1"),
input_files = listOf("input2.txt"),
output_files = listOf("output2.txt"),
),
"3" to PlanningTask.PlanTask(
task_description = "Task 3",
task_type = TaskType.TaskPlanning,
task_dependencies = listOf("2"),
input_files = listOf("input3.txt"),
)
),
)
open fun planningActor(): ParsedActor {
val planTaskSettings = this.getTaskSettings(TaskType.TaskPlanning)
return ParsedActor(
name = "TaskBreakdown",
resultClass = TaskBreakdownResult::class.java,
exampleInstance = exampleInstance,
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") { "* ${it.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(),
model = planTaskSettings.model ?: this.defaultModel,
parsingModel = this.parsingModel,
temperature = this.temperature,
describer = describer(),
)
}
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)
}
}
}
}
