Update Classes/Owiki_Class.py

Classes/Owiki_Class.py

@@ -10,8 +10,8 @@ import re
 from Classes.Helper_Class import DB_Retriever
 from typing import Optional
 
-os.environ["GOOGLE_API_KEY"] = "AIzaSyD5BAPI-PH4nbw4gY7BkcFFp1BTf2yR4bY"
-genai.configure(api_key="AIzaSyD5BAPI-PH4nbw4gY7BkcFFp1BTf2yR4bY")
+
+
 
 class OWiki:
     def __init__(self,**kwargs):
@@ -19,6 +19,9 @@ class OWiki:
         self.summary = kwargs['summary_length']
         model = kwargs["model"]
         self.db_loc = kwargs["db_loc"]
+        self.api_key = kwargs["api_key"]
+        os.environ["GOOGLE_API_KEY"]  = self.api_key
+        genai.configure(api_key=self.api_key)
         self.llm = ChatGoogleGenerativeAI(model=model,
                             temperature=temperature)
         self.model_embedding = kwargs['model_embeddings']