feat(chat): add prompt method to inline chat service #157

Introduce a new `prompt` method in `ShireInlineChatService` to handle input processing. This method is now used in `ShireInlineChatPanel` to generate the prompt for the LLM provider.
phodal · Jan 2, 2025 · ab09b82 · ab09b82
1 parent 7485a88
commit ab09b82
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/main/kotlin/com/phodal/shire/inline/ShireInlineChatPanel.kt b/src/main/kotlin/com/phodal/shire/inline/ShireInlineChatPanel.kt
@@ -37,7 +37,8 @@ class ShireInlineChatPanel(val editor: Editor) : JPanel(GridBagLayout()), Editor
         this.centerPanel.isVisible = true
         val project = editor.project!!
 
-        val flow: Flow<String>? = LlmProvider.provider(project)?.stream(input, "", false)
+        val prompt = ShireInlineChatService.getInstance().prompt(input)
+        val flow: Flow<String>? = LlmProvider.provider(project)?.stream(prompt, "", false)
 
         val panelView = ShirePanelView(project, showInput = false)
         panelView.minimumSize = Dimension(800, 40)

diff --git a/src/main/kotlin/com/phodal/shire/inline/ShireInlineChatService.kt b/src/main/kotlin/com/phodal/shire/inline/ShireInlineChatService.kt
@@ -59,6 +59,11 @@ class ShireInlineChatService : Disposable {
         allChats.remove(editor.virtualFile.url)
     }
 
+    fun prompt(input: String): String {
+        // todo: call our service to get action prompt
+        return input
+    }
+
     companion object {
         fun getInstance(): ShireInlineChatService {
             return ApplicationManager.getApplication().getService(ShireInlineChatService::class.java)