generated from JetBrains/intellij-platform-plugin-template
-
Notifications
You must be signed in to change notification settings - Fork 18
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
5f1c62d
commit c54db6d
Showing
8 changed files
with
657 additions
and
159 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
121 changes: 121 additions & 0 deletions
121
src/main/kotlin/com/github/simiacryptus/aicoder/actions/generic/LineFilterChatAction.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,121 @@ | ||
| package com.github.simiacryptus.aicoder.actions.generic | ||
|
|
||
| import com.github.simiacryptus.aicoder.ApplicationEvents | ||
| import com.github.simiacryptus.aicoder.actions.BaseAction | ||
| import com.github.simiacryptus.aicoder.actions.dev.AppServer | ||
| import com.github.simiacryptus.aicoder.config.AppSettingsState | ||
| import com.github.simiacryptus.aicoder.util.ComputerLanguage | ||
| import com.intellij.openapi.actionSystem.AnActionEvent | ||
| import com.intellij.openapi.actionSystem.CommonDataKeys | ||
| import com.intellij.openapi.fileEditor.FileDocumentManager | ||
| import com.simiacryptus.skyenet.core.platform.ApplicationServices | ||
| import com.simiacryptus.skyenet.core.platform.Session | ||
| import com.simiacryptus.skyenet.core.platform.StorageInterface | ||
| import com.simiacryptus.skyenet.core.platform.User | ||
| import com.simiacryptus.skyenet.webui.application.ApplicationServer | ||
| import com.simiacryptus.skyenet.webui.chat.ChatServer | ||
| import com.simiacryptus.skyenet.webui.chat.ChatSocketManager | ||
| import com.simiacryptus.skyenet.webui.session.SocketManager | ||
| import com.simiacryptus.skyenet.webui.util.MarkdownUtil.renderMarkdown | ||
| import org.slf4j.LoggerFactory | ||
| import java.awt.Desktop | ||
| import java.io.File | ||
|
|
||
| class LineFilterChatAction : BaseAction() { | ||
|
|
||
| val path = "/codeChat" | ||
|
|
||
| override fun handle(e: AnActionEvent) { | ||
| val editor = e.getData(CommonDataKeys.EDITOR) ?: return | ||
| val session = StorageInterface.newGlobalID() | ||
| val language = ComputerLanguage.getComputerLanguage(e)?.name ?: return | ||
| val filename = FileDocumentManager.getInstance().getFile(editor.document)?.name ?: return | ||
| val code = editor.caretModel.primaryCaret.selectedText ?: editor.document.text | ||
| val lines = code.split("\n").toTypedArray() | ||
| val codelines = lines.withIndex().joinToString("\n") { (i, line) -> | ||
| "${i.toString().padStart(3, '0')} $line" | ||
| } | ||
| agents[session] = object : ChatSocketManager( | ||
| session = session, | ||
| model = AppSettingsState.instance.defaultChatModel(), | ||
| userInterfacePrompt = """ | ||
| |# `$filename` | ||
| | | ||
| |```$language | ||
| |$code | ||
| |``` | ||
| """.trimMargin().trim(), | ||
| systemPrompt = """ | ||
| |You are a helpful AI that helps people with coding. | ||
| | | ||
| |You will be answering questions about the following code located in `$filename`: | ||
| | | ||
| |```$language | ||
| |$codelines | ||
| |``` | ||
| | | ||
| |Responses may use markdown formatting. Lines from the prompt can be included by using the line number in a response line (e.g. `\nLINE\n`). | ||
| | | ||
| |For example: | ||
| | | ||
| |```text | ||
| |001 | ||
| |## Injected subtitle | ||
| | | ||
| |025 | ||
| |026 | ||
| | | ||
| |013 | ||
| |014 | ||
| |``` | ||
| """.trimMargin(), | ||
| api = api, | ||
| applicationClass = ApplicationServer::class.java, | ||
| storage = ApplicationServices.dataStorageFactory(root), | ||
| ) { | ||
| override fun canWrite(user: User?): Boolean = true | ||
| override fun renderResponse(response: String): String { | ||
| return renderMarkdown(response.split("\n").joinToString("\n") { | ||
| when { | ||
| // Is numeric, use line if in range | ||
| it.toIntOrNull()?.let { i -> lines.indices.contains(i) } == true -> lines[it.toInt()] | ||
| // Otherwise, use response | ||
| else -> it | ||
| } | ||
| } | ||
| ) } | ||
| } | ||
|
|
||
| val server = AppServer.getServer(e.project) | ||
| val app = initApp(server, path) | ||
| app.sessions[session] = app.newSession(null, session) | ||
|
|
||
| Thread { | ||
| Thread.sleep(500) | ||
| try { | ||
| Desktop.getDesktop().browse(server.server.uri.resolve("$path/#$session")) | ||
| } catch (e: Throwable) { | ||
| log.warn("Error opening browser", e) | ||
| } | ||
| }.start() | ||
| } | ||
|
|
||
| override fun isEnabled(event: AnActionEvent) = true | ||
|
|
||
| companion object { | ||
| private val log = LoggerFactory.getLogger(LineFilterChatAction::class.java) | ||
| private val agents = mutableMapOf<Session, SocketManager>() | ||
| val root: File get() = File(ApplicationEvents.pluginHome, "code_chat") | ||
| private fun initApp(server: AppServer, path: String): ChatServer { | ||
| server.appRegistry[path]?.let { return it } | ||
| val socketServer = object : ApplicationServer(applicationName = "Code Chat", path = path) { | ||
| override val singleInput = false | ||
| override val stickyInput = true | ||
| override fun newSession(user: User?, session: Session) = agents[session]!! | ||
| } | ||
| server.addApp(path, socketServer) | ||
| return socketServer | ||
| } | ||
|
|
||
| } | ||
| } |
Oops, something went wrong.