From b484953bb331ff3a8dd4c2963bb82d3ef64703f9 Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Sun, 28 May 2023 10:11:38 +0530 Subject: [PATCH] Import app state correctly to generate embeddings with OpenAI model Resolves #216 --- src/khoj/utils/models.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/khoj/utils/models.py b/src/khoj/utils/models.py index 12935818..b5850851 100644 --- a/src/khoj/utils/models.py +++ b/src/khoj/utils/models.py @@ -8,7 +8,7 @@ import torch from tqdm import trange # Internal Packages -from khoj.utils.state import processor_config, config_file +from khoj.utils import state class BaseEncoder(ABC): @@ -25,14 +25,14 @@ class OpenAI(BaseEncoder): def __init__(self, model_name, device=None): self.model_name = model_name if ( - not processor_config - or not processor_config.conversation - or not processor_config.conversation.openai_api_key + not state.processor_config + or not state.processor_config.conversation + or not state.processor_config.conversation.openai_api_key ): raise Exception( - f"Set OpenAI API key under processor-config > conversation > openai-api-key in config file: {config_file}" + f"Set OpenAI API key under processor-config > conversation > openai-api-key in config file: {state.config_file}" ) - openai.api_key = processor_config.conversation.openai_api_key + openai.api_key = state.processor_config.conversation.openai_api_key self.embedding_dimensions = None def encode(self, entries, device=None, **kwargs):