From e2fff5dc475afa937dee5e1b5c589b29267dce6a Mon Sep 17 00:00:00 2001 From: sabaimran Date: Sat, 18 Nov 2023 14:01:01 -0800 Subject: [PATCH] Don't explicitly use value to get the model type value --- src/khoj/routers/api.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/khoj/routers/api.py b/src/khoj/routers/api.py index 790c5041..6d67fcbe 100644 --- a/src/khoj/routers/api.py +++ b/src/khoj/routers/api.py @@ -678,7 +678,7 @@ async def extract_references_and_questions( if ( offline_chat_config 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 offline_chat = await ConversationAdapters.get_offline_chat() @@ -691,7 +691,7 @@ async def extract_references_and_questions( inferred_queries = extract_questions_offline( 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 = await ConversationAdapters.get_openai_chat() api_key = openai_chat_config.api_key