Fix history retrieval bug when accessing a distributed frontend

This commit is contained in:
doctorpangloss 2024-03-29 10:49:51 -07:00
parent c6f4301e88
commit 1f705ba9d9

View File

@ -17,8 +17,10 @@ class History:
outputs=outputs, outputs=outputs,
status=ExecutionStatus(*status)._asdict()) status=ExecutionStatus(*status)._asdict())
def copy(self, prompt_id: Optional[str | int] = None, max_items: int = MAXIMUM_HISTORY_SIZE, def copy(self, prompt_id: Optional[str | int] = None, max_items: Optional[int] = None,
offset: int = 0) -> Dict[str, HistoryEntry]: offset: Optional[int] = None) -> Dict[str, HistoryEntry]:
offset = offset or 0
max_items = max_items or MAXIMUM_HISTORY_SIZE
if prompt_id in self.history: if prompt_id in self.history:
return {prompt_id: copy.deepcopy(self.history[prompt_id])} return {prompt_id: copy.deepcopy(self.history[prompt_id])}
else: else: