-
Notifications
You must be signed in to change notification settings - Fork 0
/
ModifyDictionary.py
24 lines (20 loc) · 1.02 KB
/
ModifyDictionary.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
class ModifyDictionary:
def __init__(self,running_dictionary):
self._running_dictionary = running_dictionary
def delete_items(self,dkey):
if self._running_dictionary[dkey]:
del self._running_dictionary[dkey]
return self._running_dictionary
def get_items(self,dkey):
'''
Purpose of this function is to combine multiple dictionary items into a single item for input to a prompt template.
searches the dictionary for the item_label and deletes the item_label
then searches the dictionary for all items that contain the item_label and adds them to the dictionary as a collapsed list.
'''
task_list = []
for key, value in self._running_dictionary.items():
if dkey in key.lower():
task_list.append(f"{value}")
return task_list
def filter_dict(self,keys):
return {k: self._running_dictionary[k.replace(" ","_")] for k in keys if k.replace(" ","_") in self._running_dictionary}