Update appStore/rag.py

appStore/rag.py

@@ -10,7 +10,6 @@ from huggingface_hub import InferenceClient
 
 # Get openai API key
 hf_token = os.environ["HF_API_KEY"]
-
 # define a special function for putting the prompt together (as we can't use haystack)
 def get_prompt(context, label):
   base_prompt="Summarize the following context efficiently in bullet points, the less the better - but keep concrete goals. \
@@ -62,6 +61,8 @@ def run_query(context, label):
             # res_box.success(result) # output to response text box
             res_box.success(result)
 
+            if chunk.choices[0].finish_reason != None:
+                break