feat: avante repo map rust crate (#628)

2024-09-26 03:45:49 +08:00
parent 5461342fce
commit 0d90c047ef
23 changed files with 1399 additions and 737 deletions
--- a/crates/avante-repo-map/queries/tree-sitter-python-defs.scm
+++ b/crates/avante-repo-map/queries/tree-sitter-python-defs.scm
@@ -0,0 +1,25 @@
+;; Capture top-level functions, class, and method definitions
+(module
+  (expression_statement
+    (assignment) @assignment
+  )
+)
+(module
+  (function_definition) @function
+)
+(module
+  (class_definition
+    body: (block
+      (expression_statement
+        (assignment) @class_assignment
+      )
+    )
+  )
+)
+(module
+  (class_definition
+    body: (block
+      (function_definition) @method
+    )
+  )
+)