Update query.py

query.py

@@ -1,5 +1,11 @@
 import requests
 import json
+import re
+
+def extract_between_tags(text, start_tag, end_tag):
+    start_index = text.find(start_tag)
+    end_index = text.find(end_tag, start_index)
+    return text[start_index+len(start_tag):end_index-len(end_tag)]
 
 class VectaraQuery():
     def __init__(self, api_key: str, customer_id: str, corpus_id: str, prompt_name: str = None):