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

Don't explicitly use value to get the model type value

Browse Source
This commit is contained in:
sabaimran
2023-11-18 14:01:01 -08:00
parent a8a25ceac2
commit e2fff5dc47
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -678,7 +678,7 @@ async def extract_references_and_questions(
if ( if (
offline_chat_config offline_chat_config
and offline_chat_config.enabled and offline_chat_config.enabled
and conversation_config.model_type == ChatModelOptions.ModelType.OFFLINE.value and conversation_config.model_type == ChatModelOptions.ModelType.OFFLINE
): ):
using_offline_chat = True using_offline_chat = True
offline_chat = await ConversationAdapters.get_offline_chat() offline_chat = await ConversationAdapters.get_offline_chat()
@@ -691,7 +691,7 @@ async def extract_references_and_questions(
inferred_queries = extract_questions_offline( inferred_queries = extract_questions_offline(
defiltered_query, loaded_model=loaded_model, conversation_log=meta_log, should_extract_questions=False defiltered_query, loaded_model=loaded_model, conversation_log=meta_log, should_extract_questions=False
) )
elif openai_chat_config and conversation_config.model_type == ChatModelOptions.ModelType.OPENAI.value: elif openai_chat_config and conversation_config.model_type == ChatModelOptions.ModelType.OPENAI:
openai_chat_config = await ConversationAdapters.get_openai_chat_config() openai_chat_config = await ConversationAdapters.get_openai_chat_config()
openai_chat = await ConversationAdapters.get_openai_chat() openai_chat = await ConversationAdapters.get_openai_chat()
api_key = openai_chat_config.api_key api_key = openai_chat_config.api_key