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.
package com.simiacryptus.skyenet.apps.plan
import com.simiacryptus.diff.addApplyFileDiffLinks
import com.simiacryptus.jopenai.API
import com.simiacryptus.jopenai.util.JsonUtil
import com.simiacryptus.skyenet.Retryable
import com.simiacryptus.skyenet.TabbedDisplay
import com.simiacryptus.skyenet.apps.plan.PlanningTask.TaskBreakdownInterface
import com.simiacryptus.skyenet.core.actors.SimpleActor
import com.simiacryptus.skyenet.webui.session.SessionTask
import com.simiacryptus.skyenet.webui.util.MarkdownUtil
import org.slf4j.LoggerFactory
import java.util.concurrent.Semaphore
class FileModificationTask(
planSettings: PlanSettings,
planTask: PlanningTask.PlanTask
) : AbstractTask(planSettings, planTask) {
val fileModificationActor by lazy {
SimpleActor(
name = "FileModification",
prompt = """
|Generate patches for existing files or create new files based on the given requirements and context.
|For existing files:
|- Ensure modifications are efficient, maintain readability, and adhere to coding standards.
|- Carefully review the existing code and project structure to ensure changes are consistent and do not introduce bugs.
|- Consider the impact of modifications on other parts of the codebase.
|For new files:
|- Provide a clear relative file path based on the content and purpose of the file.
|- Ensure the code is well-structured, follows best practices, and meets the specified functionality.
|- Carefully consider how the new file fits into the existing project structure and architecture.
|- Avoid creating files that duplicate functionality or introduce inconsistencies.
|Provide a summary of the changes made.
|Response format:
|- For existing files: Use ${TRIPLE_TILDE}diff code blocks with a header specifying the file path.
|- For new files: Use ${TRIPLE_TILDE} code blocks with a header specifying the new file path.
|- The diff format should use + for line additions, - for line deletions.
|- Include 2 lines of context before and after every change in diffs.
|- Separate code blocks with a single blank line.
|- For new files, specify the language for syntax highlighting after the opening triple backticks.
|Example:
|Here are the modifications:
|### src/utils/existingFile.js
|${TRIPLE_TILDE}diff
| // Existing utility functions
| function existingFunction() {
|- return 'old result';
|+ return 'new result';
| }
|${TRIPLE_TILDE}
|### src/utils/newFile.js
|${TRIPLE_TILDE}js
|// New utility functions
|function newFunction() {
| return 'new functionality';
|}
|${TRIPLE_TILDE}
""".trimMargin(),
model = planSettings.model,
temperature = planSettings.temperature,
)
}
override fun promptSegment(): String {
return """
|FileModification - Modify existing files or create new files
| ** For each file, specify the relative file path and the goal of the modification or creation
| ** List input files/tasks to be examined when designing the modifications or new files
""".trimMargin()
}
override fun run(
agent: PlanCoordinator,
taskId: String,
userMessage: String,
plan: TaskBreakdownInterface,
planProcessingState: PlanProcessingState,
task: SessionTask,
taskTabs: TabbedDisplay,
api: API
) {
if(((planTask.input_files ?: listOf()) + (planTask.output_files ?: listOf())).isEmpty()) {
task.complete("No input files specified")
return
}
val semaphore = Semaphore(0)
val onComplete = { semaphore.release() }
val process = { sb: StringBuilder ->
val codeResult = fileModificationActor.answer(
listOf(
userMessage,
JsonUtil.toJson(plan),
getPriorCode(planProcessingState),
getInputFileCode(),
this.planTask.description ?: "",
).filter { it.isNotBlank() }, api
)
planProcessingState.taskResult[taskId] = codeResult
if (agent.planSettings.autoFix) {
val diffLinks = agent.ui.socketManager!!.addApplyFileDiffLinks(
root = agent.root,
response = codeResult,
handle = { newCodeMap ->
newCodeMap.forEach { (path, newCode) ->
task.complete("$path Updated")
}
},
ui = agent.ui,
api = api,
shouldAutoApply = { true }
)
taskTabs.selectedTab += 1
taskTabs.update()
task.complete()
onComplete()
MarkdownUtil.renderMarkdown(diffLinks + "\n\n## Auto-applied changes", ui = agent.ui)
} else {
MarkdownUtil.renderMarkdown(
agent.ui.socketManager!!.addApplyFileDiffLinks(
root = agent.root,
response = codeResult,
handle = { newCodeMap ->
newCodeMap.forEach { (path, newCode) ->
task.complete("$path Updated")
}
},
ui = agent.ui,
api = api
) + acceptButtonFooter(agent.ui) {
taskTabs.selectedTab += 1
taskTabs.update()
task.complete()
onComplete()
}, ui = agent.ui
)
}
}
Retryable(agent.ui, task = task, process = process)
try {
semaphore.acquire()
} catch (e: Throwable) {
log.warn("Error", e)
}
}
companion object {
private val log = LoggerFactory.getLogger(FileModificationTask::class.java)
}
}
