feat(tools): enhance AgentTool and add new tool examples #308

- Refactor `AgentTool` toString method for better formatting.
- Implement `toolInfo` in `DatabaseFunctionProvider` to return `AgentTool` instance.
- Add new tool examples for component view, styling view, container view, history, dependencies, and SCC.
- Update `BridgeToolProvider` to include function tools from `ToolchainFunctionProvider`.
- Add test case for loading all bridge command examples.

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/agenttool/AgentTool.kt

@@ -1,10 +1,14 @@
 package cc.unitmesh.devti.agenttool
 
 data class AgentTool(val commandName: String, val description: String, val example: String) {
-    override fun toString(): String =
-        """<tool>name: ${commandName}, desc: $description, example:
+    override fun toString(): String {
+//        val string = """desc: $description"""
+        val string = if (description.isEmpty()) "" else """desc: $description"""
+        return """<tool>name: ${commandName}, $string
+example:
 <devin>
 $example
 </devin>
 </tool>"""
+    }
 }

core/src/main/kotlin/cc/unitmesh/devti/bridge/BridgeToolProvider.kt

@@ -3,6 +3,7 @@ package cc.unitmesh.devti.bridge
 import cc.unitmesh.devti.agenttool.AgentTool
 import cc.unitmesh.devti.devin.dataprovider.BuiltinCommand
 import cc.unitmesh.devti.devin.dataprovider.BuiltinCommand.*
+import cc.unitmesh.devti.provider.toolchain.ToolchainFunctionProvider
 import com.intellij.openapi.project.Project
 
 object BridgeToolProvider {
@@ -14,9 +15,21 @@ object BridgeToolProvider {
             .map {
                 val example = BuiltinCommand.example(it)
                 AgentTool(it.commandName, it.description, example)
+            }.toMutableList()
+
+        val functions = ToolchainFunctionProvider.all()
+
+        commonTools += functions.map {
+            if (it.toolInfo() != null) {
+                return@map listOf(it.toolInfo()!!)
             }
 
-        /// collect function tools in Bridge.kt
+            val funcNames = it.funcNames()
+            funcNames.map { name ->
+                val example = BuiltinCommand.example(name)
+                AgentTool(name, "", example)
+            }
+        }.flatten()
 
         return commonTools
     }

core/src/main/kotlin/cc/unitmesh/devti/bridge/BridgeToolWindow.kt

@@ -9,18 +9,18 @@ import com.intellij.openapi.project.Project
 
 class BridgeToolWindow(val myProject: Project, val myEditor: Editor?, private val showInput: Boolean = false) :
     SketchToolWindow(myProject, myEditor, showInput, ChatActionType.BRIDGE) {
-    override val inputListener = object : SketchInputListener(project, chatCodingService, this) {
+    override val inputListener
+        get() = object : SketchInputListener(project, chatCodingService, this) {
+            init {
+                // no super
+                val template = templateRender.getTemplate("bridge.vm")
+                var systemPrompt = ""
+                val customContext = BridgeRunContext.create(project, null, "")
 
-        init {
-            // no super
-            val template = templateRender.getTemplate("bridge.vm")
-            var systemPrompt = ""
-            val customContext = BridgeRunContext.create(project, null, "")
-
-            systemPrompt = templateRender.renderTemplate(template, customContext)
-            invokeLater {
-                toolWindow.addSystemPrompt(systemPrompt)
+                systemPrompt = templateRender.renderTemplate(template, customContext)
+                invokeLater {
+                    toolWindow.addSystemPrompt(systemPrompt)
+                }
             }
         }
-    }
 }

core/src/main/kotlin/cc/unitmesh/devti/provider/toolchain/ToolchainFunctionProvider.kt

@@ -1,9 +1,12 @@
 package cc.unitmesh.devti.provider.toolchain
 
+import cc.unitmesh.devti.agenttool.AgentTool
 import com.intellij.openapi.extensions.ExtensionPointName
 import com.intellij.openapi.project.Project
 
 interface ToolchainFunctionProvider {
+    fun toolInfo(): AgentTool? = null
+
     fun funcNames(): List<String>
 
     fun isApplicable(project: Project, funcName: String): Boolean

core/src/main/kotlin/cc/unitmesh/devti/sketch/SketchToolWindow.kt

@@ -61,7 +61,7 @@ open class SketchToolWindow(
 ) :
     SimpleToolWindowPanel(true, true), NullableComponent, Disposable {
     open val chatCodingService = ChatCodingService(chatActionType, project)
-    open val inputListener = SketchInputListener(project, chatCodingService, this)
+    open val inputListener get() = SketchInputListener(project, chatCodingService, this)
     private var progressBar: CustomProgressBar = CustomProgressBar(this)
     private var inputSection: AutoDevInputSection = AutoDevInputSection(project, this, showAgent = false)
 

exts/devins-lang/src/main/resources/agent/toolExamples/componentView.devin

@@ -0,0 +1,2 @@
+List all UI Component List of current project, like React Vue components
+/componentView

exts/devins-lang/src/main/resources/agent/toolExamples/containerView.devin

@@ -0,0 +1,2 @@
+List all modules of current project
+/moduleView

exts/devins-lang/src/main/resources/agent/toolExamples/dependencies.devin

@@ -0,0 +1,2 @@
+Get all dependencies (Gradle, Maven, package.json) of current project
+/dependencies

exts/devins-lang/src/main/resources/agent/toolExamples/history.devin

@@ -0,0 +1,2 @@
+Get history commit message of current file
+/history:package.json

exts/devins-lang/src/main/resources/agent/toolExamples/scc.devin

@@ -0,0 +1,2 @@
+Scc is a very fast accurate code counter with complexity calculations and COCOMO estimates
+/scc

exts/devins-lang/src/main/resources/agent/toolExamples/stylingView.devin

@@ -0,0 +1,2 @@
+List all CSS, SCSS classes of current project
+/stylingView

exts/devins-lang/src/test/kotlin/cc/unitmesh/devti/language/completion/dataprovider/FunctionToolsCommandTest.kt

@@ -0,0 +1,19 @@
+package cc.unitmesh.devti.language.completion.dataprovider
+
+import cc.unitmesh.devti.devin.dataprovider.BuiltinCommand
+import cc.unitmesh.devti.provider.toolchain.ToolchainFunctionProvider
+import com.intellij.testFramework.LightPlatformTestCase
+
+class FunctionToolsCommandTest : LightPlatformTestCase() {
+    fun testShouldLoadAllBridgeCommandExamples() {
+        val functions = ToolchainFunctionProvider.all()
+
+        val map = functions.map {
+            it.funcNames().map { name ->
+                BuiltinCommand.example(name)
+            }
+        }
+
+        println(map)
+    }
+}

exts/ext-database/src/main/kotlin/cc/unitmesh/database/provider/DatabaseFunctionProvider.kt

@@ -2,7 +2,9 @@ package cc.unitmesh.database.provider
 
 import cc.unitmesh.database.util.DatabaseSchemaAssistant
 import cc.unitmesh.database.util.DatabaseSchemaAssistant.getTableColumn
+import cc.unitmesh.devti.agenttool.AgentTool
 import cc.unitmesh.devti.bridge.provider.DatabaseFunction
+import cc.unitmesh.devti.devin.dataprovider.BuiltinCommand
 import cc.unitmesh.devti.provider.toolchain.ToolchainFunctionProvider
 import com.intellij.database.model.DasTable
 import com.intellij.database.model.RawDataSource
@@ -12,10 +14,13 @@ import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.project.Project
 
 class DatabaseFunctionProvider : ToolchainFunctionProvider {
-    override fun isApplicable(project: Project, funcName: String): Boolean {
-        return DatabaseFunction.values().any { it.funName == funcName }
+    override fun toolInfo(): AgentTool? {
+        val example = BuiltinCommand.example("database")
+        return AgentTool("database", "Database schema and query tool", example)
     }
 
+    override fun isApplicable(project: Project, funcName: String): Boolean = DatabaseFunction.entries.any { it.funName == funcName }
+
     override fun funcNames(): List<String> = DatabaseFunction.allFuncNames()
 
     override fun execute(