Ibraaheem commited on
Commit
afe3376
·
1 Parent(s): 7344079

Update private_gpt/components/embedding/embedding_component.py

Browse files
private_gpt/components/embedding/embedding_component.py CHANGED
@@ -7,6 +7,8 @@ from llama_index.embeddings.base import BaseEmbedding
7
  from private_gpt.paths import models_cache_path
8
  from private_gpt.settings.settings import Settings
9
 
 
 
10
  logger = logging.getLogger(__name__)
11
 
12
 
@@ -38,7 +40,7 @@ class EmbeddingComponent:
38
  case "openai":
39
  from llama_index import OpenAIEmbedding
40
 
41
- openai_settings = settings.openai.api_key
42
  self.embedding_model = OpenAIEmbedding(api_key=openai_settings)
43
  case "mock":
44
  # Not a random number, is the dimensionality used by
 
7
  from private_gpt.paths import models_cache_path
8
  from private_gpt.settings.settings import Settings
9
 
10
+ import os
11
+
12
  logger = logging.getLogger(__name__)
13
 
14
 
 
40
  case "openai":
41
  from llama_index import OpenAIEmbedding
42
 
43
+ openai_settings = os.environ.get("OPENAI_API_KEY")
44
  self.embedding_model = OpenAIEmbedding(api_key=openai_settings)
45
  case "mock":
46
  # Not a random number, is the dimensionality used by