Update QnA.py

QnA.py

@@ -64,11 +64,16 @@ def get_hugging_face_model(model_id='mistralai/Mistral-7B-Instruct-v0.2',tempera
         )
     return llm
 
+def get_groq_model(api_key):
+    os.environ["GROQ_API_KEY"] = API_KEY
+    llm = ChatGroq(model="llama3-8b-8192") # (model="gemma2-9b-it")
+    return llm
+    
+
 def Q_A(vectorstore,question,API_KEY,compressor=False):
     
     if API_KEY.startswith('gsk'):
-        os.environ["GROQ_API_KEY"] = API_KEY
-        chat_llm = ChatGroq(model="llama3-8b-8192") # (model="gemma2-9b-it")
+       chat_llm = get_groq_model(api_key=API_KEY)
     elif API_KEY.startswith('hf'):
         chat_llm = get_hugging_face_model(api_key=API_KEY)