ionosphere commited on
Commit
cc4bc88
1 Parent(s): c8d7bc8

Update Mistral api key

Browse files
Files changed (1) hide show
  1. rag_module.py +1 -1
rag_module.py CHANGED
@@ -27,7 +27,7 @@ from collections import defaultdict
27
 
28
  class RagModule():
29
  def __init__(self):
30
- self.mistral_api_key = "Yb2kAF0DR4Mva5AEmoYFV3kYRAKdXB7i"
31
  self.model_name_embedding = "mistral-embed"
32
  self.embedding_model = MistralAIEmbeddings(model=self.model_name_embedding, mistral_api_key=self.mistral_api_key)
33
  self.chunk_size = 1000
 
27
 
28
  class RagModule():
29
  def __init__(self):
30
+ self.mistral_api_key = os.environ.get("MISTRAL_API_KEY")
31
  self.model_name_embedding = "mistral-embed"
32
  self.embedding_model = MistralAIEmbeddings(model=self.model_name_embedding, mistral_api_key=self.mistral_api_key)
33
  self.chunk_size = 1000