amanm10000 commited on
Commit
29ec62a
·
1 Parent(s): 944f540

Refactor environment variable access in main.py to use os.environ for GROQ and Cohere API keys, and comment out debug print statements.

Browse files
Files changed (1) hide show
  1. main.py +4 -4
main.py CHANGED
@@ -28,11 +28,11 @@ if not os.environ.get("GROQ_API_KEY"):
28
 
29
  # load_dotenv()
30
 
31
- print(f"GROQ_API_KEY: {os.getenv('GROQ_API_KEY')}")
32
- print(f"HUGGING_FACE_API_KEY: {os.getenv('HUGGING_FACE_API_KEY')}")
33
 
34
 
35
- llm = init_chat_model("deepseek-r1-distill-llama-70b", model_provider="groq", api_key=os.getenv("GROQ_API_KEY"))
36
  '''
37
  embeddings = HuggingFaceInferenceAPIEmbeddings(
38
  api_key = os.getenv('HUGGING_FACE_API_KEY'),
@@ -44,7 +44,7 @@ embeddings = HuggingFaceInferenceAPIEmbeddings(
44
  )'''
45
 
46
  embeddings = CohereEmbeddings(
47
- cohere_api_key=os.getenv('COHERE'),
48
  model="embed-english-v3.0", # Added this line
49
  user_agent="langchain-cohere-embeddings"
50
  )
 
28
 
29
  # load_dotenv()
30
 
31
+ # print(f"GROQ_API_KEY: {os.getenv('GROQ_API_KEY')}")
32
+ # print(f"HUGGING_FACE_API_KEY: {os.getenv('HUGGING_FACE_API_KEY')}")
33
 
34
 
35
+ llm = init_chat_model("deepseek-r1-distill-llama-70b", model_provider="groq", api_key=os.environ["GROQ_API_KEY"])
36
  '''
37
  embeddings = HuggingFaceInferenceAPIEmbeddings(
38
  api_key = os.getenv('HUGGING_FACE_API_KEY'),
 
44
  )'''
45
 
46
  embeddings = CohereEmbeddings(
47
+ cohere_api_key=os.environ['COHERE'],
48
  model="embed-english-v3.0", # Added this line
49
  user_agent="langchain-cohere-embeddings"
50
  )