rlmgrep 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

rlmgrep/__init__.py

@@ -1,2 +1,2 @@
 __all__ = ["__version__"]
-__version__ = "0.1.0"
+__version__ = "0.1.4"

rlmgrep/cli.py

@@ -6,6 +6,7 @@ import sys
 from pathlib import Path
 
 import dspy
+from . import __version__
 from .config import ensure_default_config, load_config
 from .file_map import build_file_map
 from .ingest import FileRecord, collect_candidates, load_files, resolve_type_exts
@@ -67,14 +68,16 @@ def _parse_args(argv: list[str]) -> argparse.Namespace:
         prog="rlmgrep",
         description="Grep-shaped CLI search powered by DSPy RLM.",
     )
+    parser.add_argument("--version", action="version", version=f"rlmgrep {__version__}")
     parser.add_argument("pattern", nargs="?", help="Query string (interpreted by RLM)")
     parser.add_argument("paths", nargs="*", help="Files or directories")
 
-    parser.add_argument("-n", dest="line_numbers", action="store_true", help="Show line numbers")
-    parser.add_argument("-H", dest="with_filename", action="store_true", help="Always show filenames")
+    parser.add_argument("-n", dest="line_numbers", action="store_true", help="Show line numbers (default)")
+    parser.add_argument("--no-line-number", dest="line_numbers", action="store_false", help="Hide line numbers")
+    parser.add_argument("-H", dest="with_filename", action="store_true", help="Always show filenames (default)")
     parser.add_argument("-r", dest="recursive", action="store_true", help="Recursive (directories are searched recursively by default)")
     parser.add_argument("--no-recursive", dest="recursive", action="store_false", help="Do not recurse directories")
-    parser.set_defaults(recursive=True)
+    parser.set_defaults(recursive=True, line_numbers=True)
 
     parser.add_argument("-C", dest="context", type=int, default=0, help="Context lines before/after")
     parser.add_argument("-A", dest="after", type=int, default=None, help="Context lines after")
@@ -511,13 +514,16 @@ def main(argv: list[str] | None = None) -> int:
     before = args.before if args.before is not None else args.context
     after = args.after if args.after is not None else args.context
 
+    use_color = sys.stdout.isatty() and not os.getenv("NO_COLOR")
+
     output_lines = render_matches(
         files=files,
         matches=verified,
         show_line_numbers=args.line_numbers,
-        show_filename=args.with_filename,
         before=before,
         after=after,
+        use_color=use_color,
+        heading=True,
     )
 
     if args.answer:

rlmgrep/render.py

@@ -2,22 +2,36 @@ from __future__ import annotations
 
 from .ingest import FileRecord
 
+COLOR_RESET = "\x1b[0m"
+COLOR_PATH = "\x1b[35m"
+COLOR_LINE_NO = "\x1b[32m"
+
+
+def _colorize(text: str, color: str, use_color: bool) -> str:
+    if not use_color:
+        return text
+    return f"{color}{text}{COLOR_RESET}"
+
+
+def _format_heading(path: str, use_color: bool) -> str:
+    if not path.startswith((".", "/")):
+        path = f"./{path}"
+    return _colorize(path, COLOR_PATH, use_color)
+
 
 def _format_line(
-    path: str,
     line_no: int,
     text: str,
     is_match: bool,
-    show_filename: bool,
     show_line_numbers: bool,
+    use_color: bool,
+    heading: bool,
 ) -> str:
     delim = ":" if is_match else "-"
-    if show_filename and show_line_numbers:
-        return f"{path}{delim}{line_no}{delim}{text}"
-    if show_filename:
-        return f"{path}{delim}{text}"
     if show_line_numbers:
-        return f"{line_no}{delim}{text}"
+        prefix = _colorize(str(line_no), COLOR_LINE_NO, use_color)
+        sep = "\t" if heading else ""
+        return f"{prefix}{delim}{sep}{text}"
     return text
 
 
@@ -39,18 +53,22 @@ def render_matches(
     files: dict[str, FileRecord],
     matches: dict[str, list[int]],
     show_line_numbers: bool,
-    show_filename: bool,
     before: int,
     after: int,
+    use_color: bool = False,
+    heading: bool = True,
 ) -> list[str]:
     output: list[str] = []
-    multiple_files = len(files) > 1
-    show_filename = show_filename or multiple_files
 
-    for path in sorted(matches.keys()):
+    paths = sorted(matches.keys())
+    for idx, path in enumerate(paths):
         record = files.get(path)
         if record is None:
             continue
+        if heading:
+            if idx > 0:
+                output.append("")
+            output.append(_format_heading(path, use_color))
         lines = record.lines
         page_map = record.page_map
         n_lines = len(lines)
@@ -65,12 +83,12 @@ def render_matches(
                         text = f"{text}\tpage={page_map[line_no - 1]}"
                     output.append(
                         _format_line(
-                            path,
                             line_no,
                             text,
                             True,
-                            show_filename,
                             show_line_numbers,
+                            use_color,
+                            heading,
                         )
                     )
             continue
@@ -90,12 +108,12 @@ def render_matches(
                 is_match = line_no in match_set
                 output.append(
                     _format_line(
-                        path,
                         line_no,
                         text,
                         is_match,
-                        show_filename,
                         show_line_numbers,
+                        use_color,
+                        heading,
                     )
                 )
             if idx < len(merged) - 1:

rlmgrep/rlm.py

@@ -24,6 +24,10 @@ class RLMGrepSignature(dspy.Signature):
     find relevant lines. Return all relevant matches you can find, avoid duplicates,
     and only use exact paths from the directory keys.
     Always read the ASCII file map first to orient yourself to the available paths.
+    Do not wrap code in backticks; only raw Python.
+    Do not import pandas or numpy; use built-ins only.
+
+    Files like "photo.jpg.md" or "audio.mp3.md" are LLM descriptions/transcriptions of images/audio that were originally in the directory but have been converted to md to make them searchable by you. 
     """
 
     directory: dict = dspy.InputField(
@@ -52,7 +56,21 @@ class RLMGrepSignature(dspy.Signature):
 
 class RLMGrepAnswerSignature(dspy.Signature):
     """
-    Same as RLMGrepSignature, but also return a narrative answer to the query.
+    You are the search engine for rlmgrep, a grep-shaped CLI for coding agents.
+    Inputs include a directory mapping of files (path -> full text), an ASCII file
+    map, and a user query string. Your output must be grep-printable matches as
+    (path, line) pairs that point to real lines in the provided texts.
+    The query may be natural language or a short pattern; interpret it freely to
+    find relevant lines. Return all relevant matches you can find, avoid duplicates,
+    and only use exact paths from the directory keys.
+    Always read the ASCII file map first to orient yourself to the available paths.
+    Do not wrap code in backticks; only raw Python.
+    Do not import pandas or numpy; use built-ins only.
+
+    In this mode you are also responsible for generating a narrative answer to the query based on the provided files.
+
+    Files like "photo.jpg.md" or "audio.mp3.md" are LLM descriptions/transcriptions of images/audio that were originally in the directory but have been converted to md to make them searchable by you.
+
     """
 
     directory: dict = dspy.InputField(

{rlmgrep-0.1.2.dist-info → rlmgrep-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rlmgrep
-Version: 0.1.2
+Version: 0.1.4
 Summary: Grep-shaped CLI search powered by DSPy RLM
 Author: rlmgrep
 License: MIT
@@ -61,8 +61,9 @@ rlmgrep [options] "query" [paths...]
 
 Common options:
 
-- `-n` show line numbers
-- `-H` always show filenames
+- `-n` show line numbers (default)
+- `--no-line-number` hide line numbers
+- `-H` always show filenames (default)
 - `-C N` context lines before/after (grep-style)
 - `-A N` context lines after
 - `-B N` context lines before
@@ -109,10 +110,9 @@ rg -l "token" . | rlmgrep --stdin-files --answer "what does this token control?"
 ## Output contract (stable for agents)
 
 - Matches are written to stdout; warnings go to stderr.
-- Output uses grep-like prefixes:
-  - `path:line:text` for match lines when both `-H` and `-n` are enabled.
-  - `path-line-text` for context lines (note the `-` separator).
-  - If `-H` or `-n` are omitted, their parts are omitted.
+- Output uses rg-style headings by default:
+  - A file header line like `./path/to/file`
+  - Then `line:\ttext` for matches, `line-\ttext` for context lines
 - Line numbers are 1-based.
 - When context ranges are disjoint, a `--` line separates groups.
 - Exit codes:

rlmgrep-0.1.4.dist-info/RECORD

@@ -0,0 +1,14 @@
+rlmgrep/__init__.py,sha256=V2k5h8cZ-5Dt0onxqKP5wn7Rn2cmXmsSVcJMQZ9X1sY,48
+rlmgrep/__main__.py,sha256=MHKZ_ae3fSLGTLUUMOx15fWdeOnJSHhq-zslRP5F5Lc,79
+rlmgrep/cli.py,sha256=4qwR5cdQV0H40XZhLXdAO9tzlubeQCgA2sepWrXzKWI,18996
+rlmgrep/config.py,sha256=A6VLuuXSgQ1vM207CP0G92Mg3et93dGSmkkLQ0IOfwk,2388
+rlmgrep/file_map.py,sha256=x2Ri1wzK8_87GUorsAV01K_nYLZcv30yIquDeTCcdEw,876
+rlmgrep/ingest.py,sha256=uCz2el9B-RIT9umFo-gFEdAsmWPP1IJOArFFQY0D_1A,9127
+rlmgrep/interpreter.py,sha256=s_nMRxLlAU9C0JmUzUBW5NbVbuH67doVWF54K54STlA,2478
+rlmgrep/render.py,sha256=mxU7ByarJ6FU8ERt7biPVIOmpZQdKRrdyoZqlngploQ,3558
+rlmgrep/rlm.py,sha256=i3rCTp8OABByF60Un5gO7265gaW4spwU0OFKIz4surg,5750
+rlmgrep-0.1.4.dist-info/METADATA,sha256=s8iJbqKgLZSiUv7q4dkun4f40uDTQOYom3qjE1OMaks,6600
+rlmgrep-0.1.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+rlmgrep-0.1.4.dist-info/entry_points.txt,sha256=UV6QkEbkwBO1JJ53mm84_n35tVyOczPvOQ14ga7vrCI,45
+rlmgrep-0.1.4.dist-info/top_level.txt,sha256=gTujSRsO58c80eN7aRH2cfe51FHxx8LJ1w1Y2YlHti0,8
+rlmgrep-0.1.4.dist-info/RECORD,,

rlmgrep-0.1.2.dist-info/RECORD

@@ -1,14 +0,0 @@
-rlmgrep/__init__.py,sha256=tXbRXsO0NE_UV1kIHiZTTQQH0fj0U2KoxxNusu_gzrM,48
-rlmgrep/__main__.py,sha256=MHKZ_ae3fSLGTLUUMOx15fWdeOnJSHhq-zslRP5F5Lc,79
-rlmgrep/cli.py,sha256=-R8wosqWIibuWVYNFNQ2NpTh-uRpeQms3OqqBX9vuwk,18653
-rlmgrep/config.py,sha256=A6VLuuXSgQ1vM207CP0G92Mg3et93dGSmkkLQ0IOfwk,2388
-rlmgrep/file_map.py,sha256=x2Ri1wzK8_87GUorsAV01K_nYLZcv30yIquDeTCcdEw,876
-rlmgrep/ingest.py,sha256=uCz2el9B-RIT9umFo-gFEdAsmWPP1IJOArFFQY0D_1A,9127
-rlmgrep/interpreter.py,sha256=s_nMRxLlAU9C0JmUzUBW5NbVbuH67doVWF54K54STlA,2478
-rlmgrep/render.py,sha256=w6KOfont2M7pQz_EEngTFMY5xJEE11N_ko8P9x5FdH8,3097
-rlmgrep/rlm.py,sha256=LZfkyWxjvtf8dwo5JxetKvvpBYeGKhajwHEVpCb2eo4,4474
-rlmgrep-0.1.2.dist-info/METADATA,sha256=9_EA5Gy94JFgSLy9nFuoR1B0auVlhNhIyEBTtUyBTaY,6615
-rlmgrep-0.1.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-rlmgrep-0.1.2.dist-info/entry_points.txt,sha256=UV6QkEbkwBO1JJ53mm84_n35tVyOczPvOQ14ga7vrCI,45
-rlmgrep-0.1.2.dist-info/top_level.txt,sha256=gTujSRsO58c80eN7aRH2cfe51FHxx8LJ1w1Y2YlHti0,8
-rlmgrep-0.1.2.dist-info/RECORD,,