|
|
|
@ -750,10 +750,20 @@ class PromptQueue:
|
|
|
|
|
return True |
|
|
|
|
return False |
|
|
|
|
|
|
|
|
|
def get_history(self, prompt_id=None): |
|
|
|
|
def get_history(self, prompt_id=None, max_items=None, offset=-1): |
|
|
|
|
with self.mutex: |
|
|
|
|
if prompt_id is None: |
|
|
|
|
return copy.deepcopy(self.history) |
|
|
|
|
out = {} |
|
|
|
|
i = 0 |
|
|
|
|
if offset < 0 and max_items is not None: |
|
|
|
|
offset = len(self.history) - max_items |
|
|
|
|
for k in self.history: |
|
|
|
|
if i >= offset: |
|
|
|
|
out[k] = self.history[k] |
|
|
|
|
if max_items is not None and len(out) >= max_items: |
|
|
|
|
break |
|
|
|
|
i += 1 |
|
|
|
|
return out |
|
|
|
|
elif prompt_id in self.history: |
|
|
|
|
return {prompt_id: copy.deepcopy(self.history[prompt_id])} |
|
|
|
|
else: |
|
|
|
|