Update appStore/rag.py

appStore/rag.py

@@ -10,27 +10,14 @@ from tenacity import retry, stop_after_attempt, wait_random_exponential
 
 # Get openai API key
 openai.api_key = os.environ["OPENAI_API_KEY"]
-model_select = "gpt-3.5-turbo-0125"
+model_select = "gpt-4-0125-preview"
 
 
 # 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. \
-#   Summarize only activities that address the vulnerability of "+label+" to climate change. \
-#   Formatting example: \
-#     - Bullet point 1 \
-#     - Bullet point 2 \
-# "
-
-#   # Add the meta data for references
-#   # context = ' - '.join([d.content for d in docs])
-#   prompt = base_prompt+"; Context: "+context+"; Answer:"
-  
-#   return prompt
-
 def get_prompt(context, label):
   base_prompt="Summarize the following context efficiently in bullet points, the less the better - but keep concrete goals. \
-  Summarize only elements of the context that address vulnerability to climate change. \
+  Summarize only elements of the context that address vulnerability of "+label+" to climate change. \
+  If there is no mention of "+label+" in the context, return nothing. \
   Formatting example: \
     - Bullet point 1 \
     - Bullet point 2 \
@@ -42,6 +29,20 @@ def get_prompt(context, label):
   
   return prompt
 
+# def get_prompt(context, label):
+#   base_prompt="Summarize the following context efficiently in bullet points, the less the better - but keep concrete goals. \
+#   Summarize only elements of the context that address vulnerability to climate change. \
+#   Formatting example: \
+#     - Bullet point 1 \
+#     - Bullet point 2 \
+# "
+
+#   # Add the meta data for references
+#   # context = ' - '.join([d.content for d in docs])
+#   prompt = base_prompt+"; Context: "+context+"; Answer:"
+  
+#   return prompt
+
 #   base_prompt="Summarize the following context efficiently in bullet points, the less the better- but keep concrete goals. \
 #   Summarize only activities that address the vulnerability of "+label+" to climate change. \
 #   Formatting example: \