Skip to content

Commit

Permalink
removed sorting by key
Browse files Browse the repository at this point in the history
Signed-off-by: AlexMikhalev <[email protected]>
  • Loading branch information
AlexMikhalev committed Jun 2, 2024
1 parent fc3ccef commit d5494a6
Show file tree
Hide file tree
Showing 7 changed files with 68 additions and 61 deletions.
1 change: 0 additions & 1 deletion crates/terraphim_rolegraph/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -197,7 +197,6 @@ impl RoleGraph {

let mut ranked_documents = results.into_iter().collect::<Vec<_>>();
ranked_documents.sort_by_key(|(_, doc)| std::cmp::Reverse(doc.rank));
ranked_documents.sort_by_key(|(_, doc)| std::cmp::Reverse(doc.id.clone()));

let documents: Vec<_> = ranked_documents
.into_iter()
Expand Down
3 changes: 2 additions & 1 deletion crates/terraphim_service/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -80,10 +80,11 @@ impl<'a> TerraphimService {
log::debug!("Sorting documents by relevance");
// Sort the documents by relevance
let documents = score::sort_documents(search_query, documents);
let total_length = documents.len();
let mut docs_ranked = Vec::new();
for (idx, doc) in documents.iter().enumerate() {
let document: &mut terraphim_types::Document = &mut doc.clone();
let rank = terraphim_types::Rank::new(idx.try_into().unwrap());
let rank = terraphim_types::Rank::new((total_length - idx).try_into().unwrap());
document.rank = Some(rank);
docs_ranked.push(document.clone());
}
Expand Down
1 change: 1 addition & 0 deletions crates/terraphim_service/src/score/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ pub fn sort_documents(search_query: &SearchQuery, documents: Vec<Document>) -> V
let mut results = scorer.score(&query, documents).unwrap();
//FIXME: results.score and rank solve the same purpose. Results score shall be normalized into rank and mapped to IndexedDocument
results.rescore(|doc| query.similarity.similarity(&query.name, &doc.title));
log::debug!("Rescore results {:#?}", results);
results
.into_vec()
.iter()
Expand Down
6 changes: 6 additions & 0 deletions desktop/src/lib/Search/ResultItem.svelte
Original file line number Diff line number Diff line change
Expand Up @@ -31,13 +31,19 @@
{#if document.tags}
<Taglist>
{#each document.tags as tag}
<!-- FIXME: link shall be config parameter for KG -->
<a
href="https://terraphim.github.io/terraphim-project/#/page/{tag}"
target="_blank"><Tag rounded>{tag}</Tag></a
>
{/each}
</Taglist>
{/if}
</div>
<div class="level-right">
<Taglist>
<Tag rounded>Rank {document.rank}</Tag>
</Taglist>
</div>
<div transition:fade>
<button on:click={onTitleClick}>
Expand Down
4 changes: 2 additions & 2 deletions desktop/src/lib/Search/Search.svelte
Original file line number Diff line number Diff line change
Expand Up @@ -88,8 +88,8 @@
{#if error}
<p class="error">{error}</p>
{:else if results.length}
{#each results as result_item}
<ResultItem document={result_item} />
{#each results as item}
<ResultItem document={item} />
{/each}
{:else}
<section class="section">
Expand Down

Large diffs are not rendered by default.

2 changes: 1 addition & 1 deletion terraphim_server/dist/index.html
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
<link rel="apple-touch-icon" type="image/png" href="/180x180.png">
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
<title>Terraphim AI</title>
<script type="module" crossorigin src="/assets/index-ed0ded8b.js"></script>
<script type="module" crossorigin src="/assets/index-4c117adb.js"></script>
<link rel="stylesheet" href="/assets/index-36fd5f08.css">
</head>
<body>
Expand Down

0 comments on commit d5494a6

Please sign in to comment.