Update modules.py

modules.py

@@ -1,3 +1,4 @@
+import sys
 import time
 from typing import Any, List, Optional, Tuple, Union
 import numpy as np
@@ -103,6 +104,7 @@ def get_prompt(context:str, query:str)->str:
 def retrieve_query(query:str, df:str, column:str, api_key:str, model:str, temperature:float=0.5, chunk_size:int=100)->str:
     if not api_key:
         st.write("Please enter your key")
+        sys.exit(0)
     context = semantic_chunk(query=query, df=df, column=column, model=model, api_key=api_key, chunk_size=chunk_size)
     llm = select_llm(model=model, api_key=api_key, temperature=temperature)
     prompt = get_prompt(context, query)