feat(ui): maintain score order when curating sources (#1643)
* 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
This commit is contained in:
parent
290b9fb084
commit
410bf7a71f
|
@ -44,8 +44,8 @@ class Source(BaseModel):
|
||||||
frozen = True
|
frozen = True
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def curate_sources(sources: list[Chunk]) -> set["Source"]:
|
def curate_sources(sources: list[Chunk]) -> list["Source"]:
|
||||||
curated_sources = set()
|
curated_sources = []
|
||||||
|
|
||||||
for chunk in sources:
|
for chunk in sources:
|
||||||
doc_metadata = chunk.document.doc_metadata
|
doc_metadata = chunk.document.doc_metadata
|
||||||
|
@ -54,7 +54,10 @@ class Source(BaseModel):
|
||||||
page_label = doc_metadata.get("page_label", "-") if doc_metadata else "-"
|
page_label = doc_metadata.get("page_label", "-") if doc_metadata else "-"
|
||||||
|
|
||||||
source = Source(file=file_name, page=page_label, text=chunk.text)
|
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
|
return curated_sources
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue