Initial commit
Co-Authored-By: Quentin Torroba <quentin.torroba@mistral.ai> Co-Authored-By: Laure Hugo <laure.hugo@mistral.ai> Co-Authored-By: Benjamin Trom <benjamin.trom@mistral.ai> Co-Authored-By: Mathias Gesbert <mathias.gesbert@ext.mistral.ai> Co-Authored-By: Michel Thomazo <michel.thomazo@mistral.ai> Co-Authored-By: Clément Drouin <clement.drouin@mistral.ai> Co-Authored-By: Vincent Guilloux <vincent.guilloux@mistral.ai> Co-Authored-By: Valentin Berard <val@mistral.ai> Co-Authored-By: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
Quentin Torroba
Quentin Torroba
commit
fa15fc977b
200 changed files with 30484 additions and 0 deletions
247
vibe/core/autocompletion/completers.py
Normal file
247
vibe/core/autocompletion/completers.py
Normal file
|
|
@ -0,0 +1,247 @@
|
|||
from __future__ import annotations
|
||||
|
||||
from pathlib import Path
|
||||
from typing import NamedTuple
|
||||
|
||||
from vibe.core.autocompletion.file_indexer import FileIndexer, IndexEntry
|
||||
from vibe.core.autocompletion.fuzzy import fuzzy_match
|
||||
|
||||
DEFAULT_MAX_ENTRIES_TO_PROCESS = 32000
|
||||
DEFAULT_TARGET_MATCHES = 100
|
||||
|
||||
|
||||
class Completer:
|
||||
def get_completions(self, text: str, cursor_pos: int) -> list[str]:
|
||||
return []
|
||||
|
||||
def get_completion_items(self, text: str, cursor_pos: int) -> list[tuple[str, str]]:
|
||||
return [
|
||||
(completion, "") for completion in self.get_completions(text, cursor_pos)
|
||||
]
|
||||
|
||||
def get_replacement_range(
|
||||
self, text: str, cursor_pos: int
|
||||
) -> tuple[int, int] | None:
|
||||
return None
|
||||
|
||||
|
||||
class CommandCompleter(Completer):
|
||||
def __init__(self, commands: list[tuple[str, str]]) -> None:
|
||||
aliases_with_descriptions: dict[str, str] = {}
|
||||
for alias, description in commands:
|
||||
aliases_with_descriptions[alias] = description
|
||||
|
||||
self._descriptions = aliases_with_descriptions
|
||||
self._aliases: list[str] = list(aliases_with_descriptions.keys())
|
||||
|
||||
def get_completions(self, text: str, cursor_pos: int) -> list[str]:
|
||||
if not text.startswith("/"):
|
||||
return []
|
||||
|
||||
word = text[1:cursor_pos].lower()
|
||||
search_str = "/" + word
|
||||
return [
|
||||
alias for alias in self._aliases if alias.lower().startswith(search_str)
|
||||
]
|
||||
|
||||
def get_completion_items(self, text: str, cursor_pos: int) -> list[tuple[str, str]]:
|
||||
completions = self.get_completions(text, cursor_pos)
|
||||
return [(alias, self._descriptions.get(alias, "")) for alias in completions]
|
||||
|
||||
def get_replacement_range(
|
||||
self, text: str, cursor_pos: int
|
||||
) -> tuple[int, int] | None:
|
||||
if text.startswith("/"):
|
||||
return (0, cursor_pos)
|
||||
return None
|
||||
|
||||
|
||||
class PathCompleter(Completer):
|
||||
def __init__(
|
||||
self,
|
||||
max_entries_to_process: int = DEFAULT_MAX_ENTRIES_TO_PROCESS,
|
||||
target_matches: int = DEFAULT_TARGET_MATCHES,
|
||||
) -> None:
|
||||
self._indexer = FileIndexer()
|
||||
self._max_entries_to_process = max_entries_to_process
|
||||
self._target_matches = target_matches
|
||||
|
||||
class _SearchContext(NamedTuple):
|
||||
suffix: str
|
||||
search_pattern: str
|
||||
path_prefix: str
|
||||
immediate_only: bool
|
||||
|
||||
def _extract_partial(self, before_cursor: str) -> str | None:
|
||||
if "@" not in before_cursor:
|
||||
return None
|
||||
|
||||
at_index = before_cursor.rfind("@")
|
||||
fragment = before_cursor[at_index + 1 :]
|
||||
|
||||
if " " in fragment:
|
||||
return None
|
||||
|
||||
return fragment
|
||||
|
||||
def _build_search_context(self, partial_path: str) -> _SearchContext:
|
||||
suffix = partial_path.split("/")[-1]
|
||||
|
||||
if not partial_path:
|
||||
# "@" => show top-level dir and files
|
||||
return self._SearchContext(
|
||||
search_pattern="", path_prefix="", suffix=suffix, immediate_only=True
|
||||
)
|
||||
|
||||
if partial_path.endswith("/"):
|
||||
# "@something/" => list immediate children
|
||||
return self._SearchContext(
|
||||
search_pattern="",
|
||||
path_prefix=partial_path,
|
||||
suffix=suffix,
|
||||
immediate_only=True,
|
||||
)
|
||||
|
||||
return self._SearchContext(
|
||||
# => run fuzzy search across the index
|
||||
search_pattern=partial_path,
|
||||
path_prefix="",
|
||||
suffix=suffix,
|
||||
immediate_only=False,
|
||||
)
|
||||
|
||||
def _matches_prefix(self, entry: IndexEntry, context: _SearchContext) -> bool:
|
||||
path_str = entry.rel
|
||||
|
||||
if context.path_prefix:
|
||||
prefix_without_slash = context.path_prefix.rstrip("/")
|
||||
prefix_with_slash = f"{prefix_without_slash}/"
|
||||
|
||||
if path_str == prefix_without_slash and entry.is_dir:
|
||||
# do not suggest the dir itself (e.g. "@src/" => don't suggest "@src/")
|
||||
return False
|
||||
|
||||
if path_str.startswith(prefix_with_slash):
|
||||
after_prefix = path_str[len(prefix_with_slash) :]
|
||||
else:
|
||||
idx = path_str.find(prefix_with_slash)
|
||||
if idx == -1 or (idx > 0 and path_str[idx - 1] != "/"):
|
||||
return False
|
||||
after_prefix = path_str[idx + len(prefix_with_slash) :]
|
||||
|
||||
# only suggest files/dirs that are immediate children of the prefix
|
||||
return bool(after_prefix) and "/" not in after_prefix
|
||||
|
||||
if context.immediate_only and "/" in path_str:
|
||||
# when user just typed "@", only show top-level entries
|
||||
return False
|
||||
|
||||
# entry matches the prefix: let the fuzzy matcher decide if it's a good match
|
||||
return True
|
||||
|
||||
def _is_visible(self, entry: IndexEntry, context: _SearchContext) -> bool:
|
||||
return not (entry.name.startswith(".") and not context.suffix.startswith("."))
|
||||
|
||||
def _format_label(self, entry: IndexEntry) -> str:
|
||||
suffix = "/" if entry.is_dir else ""
|
||||
return f"@{entry.rel}{suffix}"
|
||||
|
||||
def _score_matches(
|
||||
self, entries: list[IndexEntry], context: _SearchContext
|
||||
) -> list[tuple[str, float]]:
|
||||
scored_matches: list[tuple[str, float]] = []
|
||||
MAX_MATCHES = 50
|
||||
|
||||
for i, entry in enumerate(entries):
|
||||
if i >= self._max_entries_to_process:
|
||||
break
|
||||
|
||||
if not self._matches_prefix(entry, context):
|
||||
continue
|
||||
|
||||
if not self._is_visible(entry, context):
|
||||
continue
|
||||
|
||||
label = self._format_label(entry)
|
||||
|
||||
if not context.search_pattern:
|
||||
scored_matches.append((label, 0.0))
|
||||
if len(scored_matches) >= self._target_matches:
|
||||
break
|
||||
continue
|
||||
|
||||
match_result = fuzzy_match(
|
||||
context.search_pattern, entry.rel, entry.rel_lower
|
||||
)
|
||||
if match_result.matched:
|
||||
scored_matches.append((label, match_result.score))
|
||||
if (
|
||||
len(scored_matches) >= self._target_matches
|
||||
and match_result.score > MAX_MATCHES
|
||||
):
|
||||
break
|
||||
|
||||
scored_matches.sort(key=lambda x: (-x[1], x[0]))
|
||||
return scored_matches
|
||||
|
||||
def _collect_matches(self, text: str, cursor_pos: int) -> list[str]:
|
||||
before_cursor = text[:cursor_pos]
|
||||
partial_path = self._extract_partial(before_cursor)
|
||||
if partial_path is None:
|
||||
return []
|
||||
|
||||
context = self._build_search_context(partial_path)
|
||||
|
||||
try:
|
||||
# TODO (Vince): doing the assumption that "." is the root directory... Reliable?
|
||||
file_index = self._indexer.get_index(Path("."))
|
||||
except (OSError, RuntimeError):
|
||||
return []
|
||||
|
||||
scored_matches = self._score_matches(file_index, context)
|
||||
return [path for path, _ in scored_matches]
|
||||
|
||||
def get_completions(self, text: str, cursor_pos: int) -> list[str]:
|
||||
return self._collect_matches(text, cursor_pos)
|
||||
|
||||
def get_completion_items(self, text: str, cursor_pos: int) -> list[tuple[str, str]]:
|
||||
matches = self._collect_matches(text, cursor_pos)
|
||||
return [(completion, "") for completion in matches]
|
||||
|
||||
def get_replacement_range(
|
||||
self, text: str, cursor_pos: int
|
||||
) -> tuple[int, int] | None:
|
||||
before_cursor = text[:cursor_pos]
|
||||
if "@" in before_cursor:
|
||||
at_index = before_cursor.rfind("@")
|
||||
return (at_index, cursor_pos)
|
||||
return None
|
||||
|
||||
|
||||
class MultiCompleter(Completer):
|
||||
def __init__(self, completers: list[Completer]) -> None:
|
||||
self.completers = completers
|
||||
|
||||
def get_completions(self, text: str, cursor_pos: int) -> list[str]:
|
||||
all_completions = []
|
||||
for completer in self.completers:
|
||||
completions = completer.get_completions(text, cursor_pos)
|
||||
all_completions.extend(completions)
|
||||
|
||||
seen = set()
|
||||
unique = []
|
||||
for comp in all_completions:
|
||||
if comp not in seen:
|
||||
seen.add(comp)
|
||||
unique.append(comp)
|
||||
|
||||
return unique
|
||||
|
||||
def get_replacement_range(
|
||||
self, text: str, cursor_pos: int
|
||||
) -> tuple[int, int] | None:
|
||||
for completer in self.completers:
|
||||
range_result = completer.get_replacement_range(text, cursor_pos)
|
||||
if range_result is not None:
|
||||
return range_result
|
||||
return None
|
||||
Loading…
Add table
Add a link
Reference in a new issue