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.general.PlanChatApp.kt Maven / Gradle / Ivy
package com.simiacryptus.skyenet.apps.general
import com.simiacryptus.jopenai.API
import com.simiacryptus.jopenai.ChatClient
import com.simiacryptus.jopenai.models.OpenAITextModel
import com.simiacryptus.skyenet.apps.plan.*
import com.simiacryptus.skyenet.apps.plan.file.InquiryTask.InquiryTaskData
import com.simiacryptus.skyenet.core.platform.Session
import com.simiacryptus.skyenet.core.platform.User
import com.simiacryptus.skyenet.util.MarkdownUtil
import com.simiacryptus.skyenet.webui.application.ApplicationInterface
import org.slf4j.LoggerFactory
import java.io.File
import java.util.*
open class PlanChatApp(
applicationName: String = "Task Planning Chat v1.0",
path: String = "/taskChat",
planSettings: PlanSettings,
model: OpenAITextModel,
parsingModel: OpenAITextModel,
domainName: String = "localhost",
showMenubar: Boolean = true,
initialPlan: PlanUtil.TaskBreakdownWithPrompt? = null,
api: API? = null,
) : PlanAheadApp(
applicationName = applicationName,
path = path,
planSettings = planSettings,
model = model,
parsingModel = parsingModel,
domainName = domainName,
showMenubar = showMenubar,
initialPlan = initialPlan,
api = api,
) {
override val stickyInput = true
override val singleInput = false
private val sessionHandlers: MutableMap = mutableMapOf()
override fun userMessage(
session: Session,
user: User?,
userMessage: String,
ui: ApplicationInterface,
api: API
) {
ui.socketManager?.pool!!.submit {
val sessionHandler = sessionHandlers.getOrPut(session.sessionId) {
ChatSessionHandler(
ui = ui,
session = session,
user = user,
api = api
)
}
sessionHandler.handleUserMessage(
userMessage = userMessage,
)
}
}
private inner class ChatSessionHandler(
val ui: ApplicationInterface,
val session: Session,
val user: User?,
val api: API
) {
val messageHistory: MutableList = mutableListOf()
fun handleUserMessage(userMessage: String) {
try {
messageHistory.add(userMessage)
val planSettings = (getSettings(session, user, PlanSettings::class.java) ?: PlanSettings(
defaultModel = model,
parsingModel = parsingModel,
command = planSettings.command,
temperature = planSettings.temperature,
workingDir = planSettings.workingDir,
env = planSettings.env
)).copy(
allowBlocking = false,
)
if (api is ChatClient) api.budget = planSettings.budget
val coordinator = PlanCoordinator(
user = user,
session = session,
dataStorage = dataStorage,
ui = ui,
root = planSettings?.workingDir?.let { File(it).toPath() } ?: dataStorage.getDataDir(user, session).toPath(),
planSettings = planSettings
)
val mainTask = coordinator.ui.newTask()
val sessionTask = ui.newTask(false).apply { mainTask.verbose(placeholder) }
val api = (api as ChatClient).getChildClient().apply {
val createFile = sessionTask.createFile(".logs/api-${UUID.randomUUID()}.log")
createFile.second?.apply {
logStreams += this.outputStream().buffered()
sessionTask.verbose("API log: $this ")
}
}
val plan = PlanCoordinator.initialPlan(
codeFiles = coordinator.codeFiles,
files = coordinator.files,
root = dataStorage.getDataDir(user, session).toPath(),
task = sessionTask,
userMessage = userMessage,
ui = coordinator.ui,
planSettings = coordinator.planSettings,
api = api
)
val modifiedPlan = addRespondToChatTask(plan.plan)
val planProcessingState = coordinator.executePlan(
plan = modifiedPlan,
task = sessionTask,
userMessage = userMessage,
api = api
)
val response = planProcessingState.taskResult["respond_to_chat"] as? String
if (response != null) {
mainTask.add(MarkdownUtil.renderMarkdown(response, ui = ui))
messageHistory.add(response)
} else {
mainTask.add("Sorry, I couldn't generate a response.")
messageHistory.add("Sorry, I couldn't generate a response.")
}
mainTask.complete()
} catch (e: Throwable) {
ui.newTask().error(ui, e)
log.warn("Error", e)
}
}
}
protected open fun addRespondToChatTask(plan: Map): Map {
val tasksByID = plan?.toMutableMap() ?: mutableMapOf()
val respondTaskId = "respond_to_chat"
tasksByID[respondTaskId] = InquiryTaskData(
task_description = "Respond to the user's chat message based on the executed plan",
task_dependencies = tasksByID.keys.toList()
)
return tasksByID
}
companion object {
private val log = LoggerFactory.getLogger(PlanChatApp::class.java)
}
}
