klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-06 21:29:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a model that specifies the user's search model configuration

Browse Source 
- Update all endpoints that generate embeddings to use the new model. Incl. generating text embeddings, creating embeddings for a search query
This commit is contained in:
sabaimran
2023-12-20 09:22:26 +05:30
parent 6dd2b05bf5
commit 0f6e4ff683
5 changed files with 44 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -18,7 +18,7 @@ from starlette.authentication import requires
# Internal Packages
from khoj.configure import configure_server
from khoj.database import adapters
from khoj.database.adapters import ConversationAdapters, EntryAdapters, get_default_search_model
from khoj.database.adapters import ConversationAdapters, EntryAdapters, get_user_search_model_or_default
from khoj.database.models import ChatModelOptions, SpeechToTextModelOptions
from khoj.database.models import Entry as DbEntry
from khoj.database.models import (
@@ -416,7 +416,7 @@ async def search(
]
if text_search_models:
with timer("Encoding query took", logger=logger):
search_model = await sync_to_async(get_default_search_model)()
search_model = await sync_to_async(get_user_search_model_or_default)(user)
encoded_asymmetric_query = state.embeddings_model[search_model.name].embed_query(defiltered_query)
with concurrent.futures.ThreadPoolExecutor() as executor: