Skip to content

Commit

Permalink
feat(ui): maintain score order when curating sources (#1643)
Browse files Browse the repository at this point in the history
* Update ui.py

Changed 'curated_sources' from a list, in order to maintain score order when returning the curated sources.

* Maintain score order after curating sources
  • Loading branch information
Hoffelhas authored Mar 11, 2024
1 parent 290b9fb commit 410bf7a
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions private_gpt/ui/ui.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,8 +44,8 @@ class Config:
frozen = True

@staticmethod
def curate_sources(sources: list[Chunk]) -> set["Source"]:
curated_sources = set()
def curate_sources(sources: list[Chunk]) -> list["Source"]:
curated_sources = []

for chunk in sources:
doc_metadata = chunk.document.doc_metadata
Expand All @@ -54,7 +54,10 @@ def curate_sources(sources: list[Chunk]) -> set["Source"]:
page_label = doc_metadata.get("page_label", "-") if doc_metadata else "-"

source = Source(file=file_name, page=page_label, text=chunk.text)
curated_sources.add(source)
curated_sources.append(source)
curated_sources = list(
dict.fromkeys(curated_sources).keys()
) # Unique sources only

return curated_sources

Expand Down

0 comments on commit 410bf7a

Please sign in to comment.