klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 13:23:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

FIx unit tests with new search model configurations

Browse Source
This commit is contained in:
sabaimran
2023-12-20 21:50:44 +05:30
parent aa23da60a3
commit 089e4bee12
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/khoj/database/adapters/__init__.py
View File

@@ -257,6 +257,9 @@ def get_user_search_model_or_default(user=None):
if SearchModelConfig.objects.filter(name="default").exists(): if SearchModelConfig.objects.filter(name="default").exists():
return SearchModelConfig.objects.filter(name="default").first() return SearchModelConfig.objects.filter(name="default").first()
else:
SearchModelConfig.objects.create()
return SearchModelConfig.objects.first() return SearchModelConfig.objects.first()

8
src/khoj/routers/api.py
View File

@@ -493,10 +493,10 @@ async def search(
# Collate results # Collate results
results += text_search.collate_results(hits, dedupe=dedupe) results += text_search.collate_results(hits, dedupe=dedupe)
# Sort results across all content types and take top results # Sort results across all content types and take top results
results = text_search.rerank_and_sort_results( results = text_search.rerank_and_sort_results(
results, query=defiltered_query, rank_results=r, search_model_name=search_model.name results, query=defiltered_query, rank_results=r, search_model_name=search_model.name
)[:results_count] )[:results_count]
# Cache results # Cache results
if user: if user: