harshpatel080503 commited on
Commit
38114c8
·
verified ·
1 Parent(s): 9cd5da9

Update rag_chain.py

Browse files
Files changed (1) hide show
  1. rag_chain.py +6 -2
rag_chain.py CHANGED
@@ -14,8 +14,12 @@ from langchain.prompts import PromptTemplate
14
 
15
  load_dotenv()
16
 
17
- os.environ["HUGGINGFACEHUB_API_TOKEN"] = os.getenv("HUGGINGFACEHUB_ACCESS_TOKEN")
18
- os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
 
 
 
 
19
 
20
  # Hugging Face Embeddings
21
  os.environ['HF_HOME'] = 'E:/Generative AI/AI Models/Embedding Models'
 
14
 
15
  load_dotenv()
16
 
17
+ # Already handled by Hugging Face via secret env vars
18
+ hf_token = os.getenv("HUGGINGFACEHUB_ACCESS_TOKEN")
19
+ openai_token = os.getenv("OPENAI_API_KEY")
20
+
21
+ os.environ["HUGGINGFACEHUB_API_TOKEN"] = hf_token
22
+ os.environ["OPENAI_API_KEY"] = openai_token
23
 
24
  # Hugging Face Embeddings
25
  os.environ['HF_HOME'] = 'E:/Generative AI/AI Models/Embedding Models'