Eoin McGrath commited on
Commit
91dc605
·
1 Parent(s): b796ec0

ensures api_key can be loaded

Browse files
Files changed (1) hide show
  1. utils.py +4 -5
utils.py CHANGED
@@ -1,8 +1,6 @@
1
  import os
2
- import logging
3
  import csv
4
  import hashlib
5
- from dotenv import load_dotenv
6
 
7
  import chromadb
8
  from llama_index.core import Settings
@@ -18,13 +16,14 @@ from llama_index.llms.openai import OpenAI
18
 
19
  from config import CHROMA_PATH, CHROMA_COLLECTION, FILES, CHUNK_SIZE, CHUNK_OVERLAP
20
 
21
- Settings.llm = OpenAI(temperature=0, model="gpt-4o-mini")
 
 
 
22
  Settings.embed_model = OpenAIEmbedding(
23
  model="text-embedding-3-small"
24
  )
25
 
26
- load_dotenv()
27
-
28
  def deterministic_id_func(i: int, doc: BaseNode) -> str:
29
  """Deterministic ID function for the text splitter.
30
  This will be used to generate a unique repeatable identifier for each node."""
 
1
  import os
 
2
  import csv
3
  import hashlib
 
4
 
5
  import chromadb
6
  from llama_index.core import Settings
 
16
 
17
  from config import CHROMA_PATH, CHROMA_COLLECTION, FILES, CHUNK_SIZE, CHUNK_OVERLAP
18
 
19
+
20
+ api_key = os.getenv("OPENAI_API_KEY")
21
+
22
+ Settings.llm = OpenAI(temperature=0, model="gpt-4o-mini", api_key=api_key)
23
  Settings.embed_model = OpenAIEmbedding(
24
  model="text-embedding-3-small"
25
  )
26
 
 
 
27
  def deterministic_id_func(i: int, doc: BaseNode) -> str:
28
  """Deterministic ID function for the text splitter.
29
  This will be used to generate a unique repeatable identifier for each node."""