Update utils.py

utils.py

@@ -1,8 +1,22 @@
-def fill_prompt(prompt, json):
-    """
-        Takes in a prompt and fills in the {key} with the corresponding value from the json KV pairs
-    """
-
+def clean_pinecone(response):
+    #Dict for metadata that is removed from duplicate classes
+    additional_metadata = {}
+    classes = [] 
+    unique_codes = set()
+
+    for entry in response['matches']:
+        code = entry['metadata']['code']
+        if code not in unique_codes:
+            #Where class is unique
+            classes.append(entry)
+            unique_codes.add(code)
+        else:
+            #Class already added, but need to save alternate sessions
+            time = entry['metadata']['time']
+            days = entry['metadata']['days']
+            additional_metadata[code] = {'time': time, 'days': days}
+    
+    return filtered_matches, additional_metadata
 
 def build_filter(json_response):
     filter = {}