sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 23:48:56 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 16

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
Show stats 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: