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
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.
|
| 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.
|
| 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 |
)
|