include featues without intervals in search results (this was a bug)

AbanteAI · Apr 18, 2024 · c81776c · c81776c
1 parent 83662ca
commit c81776c
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/mentat/code_context.py b/mentat/code_context.py
@@ -423,10 +423,14 @@ async def search(
                 continue
             distance = node["distance"]
             path, interval = split_intervals_from_path(Path(node["ref"]))
-            intervals = parse_intervals(interval)
-            for _interval in intervals:
-                feature = CodeFeature(cwd / path, _interval)
+            if not interval:
+                feature = CodeFeature(cwd / path)
                 all_features_sorted.append((feature, distance))
+            else:
+                intervals = parse_intervals(interval)
+                for _interval in intervals:
+                    feature = CodeFeature(cwd / path, _interval)
+                    all_features_sorted.append((feature, distance))
         if max_results is None:
             return all_features_sorted
         else: