Skip to content

Commit eba8be5

Browse files
committed
feat: don't print relevance score by default
1 parent 6c1900f commit eba8be5

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

gptme_rag/cli.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -364,6 +364,11 @@ def index(
364364
multiple=True,
365365
help="Filter results by path pattern (glob). Can be specified multiple times.",
366366
)
367+
@click.option(
368+
"--print-relevance",
369+
is_flag=True,
370+
help="Print relevance scores",
371+
)
367372
def search(
368373
query: str,
369374
paths: list[Path],
@@ -378,6 +383,7 @@ def search(
378383
embedding_function: str | None,
379384
device: str | None,
380385
filter: tuple[str, ...],
386+
print_relevance: bool,
381387
):
382388
"""Search the index and assemble context."""
383389
paths = [path.resolve() for path in paths]
@@ -498,7 +504,7 @@ def get_expanded_content(doc: Document, expand: str, indexer: Indexer) -> str:
498504
if format == "full":
499505
for i, doc in enumerate(documents):
500506
# Show relevance info first
501-
if distances:
507+
if distances and print_relevance:
502508
formatter.print_relevance(1 - distances[i])
503509

504510
# Get and format content
@@ -550,7 +556,8 @@ def get_expanded_content(doc: Document, expand: str, indexer: Indexer) -> str:
550556
console.print(f"\n {'Total':15} [bold blue]{total:>7.3f}[/bold blue]")
551557
else:
552558
# Just show the base relevance score
553-
formatter.print_relevance(1 - distances[i])
559+
if distances and print_relevance:
560+
formatter.print_relevance(1 - distances[i])
554561

555562
# Display preview
556563
formatter.print_preview(doc)

0 commit comments

Comments
 (0)