Skip to content

Commit 1860286

Browse files
committed
filter out new hidden context item type
1 parent 4a4b81b commit 1860286

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

chatkit/server.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -441,11 +441,11 @@ async def _process_non_streaming(
441441
after=items_list_params.after,
442442
context=context,
443443
)
444-
# filter out HiddenContextItems
444+
# filter out hidden context items
445445
items.data = [
446446
item
447447
for item in items.data
448-
if not isinstance(item, HiddenContextItem)
448+
if not isinstance(item, (HiddenContextItem, SDKHiddenContextItem))
449449
]
450450
return self._serialize(items)
451451
case ThreadsUpdateReq():
@@ -711,7 +711,9 @@ async def _process_events(
711711
# special case - don't send hidden context items back to the client
712712
should_swallow_event = isinstance(
713713
event, ThreadItemDoneEvent
714-
) and isinstance(event.item, HiddenContextItem)
714+
) and isinstance(
715+
event.item, (HiddenContextItem, SDKHiddenContextItem)
716+
)
715717

716718
if not should_swallow_event:
717719
yield event
@@ -867,7 +869,7 @@ def _serialize(self, obj: BaseModel) -> bytes:
867869

868870
def _to_thread_response(self, thread: ThreadMetadata | Thread) -> Thread:
869871
def is_hidden(item: ThreadItem) -> bool:
870-
return isinstance(item, HiddenContextItem)
872+
return isinstance(item, (HiddenContextItem, SDKHiddenContextItem))
871873

872874
items = thread.items if isinstance(thread, Thread) else Page()
873875
items.data = [item for item in items.data if not is_hidden(item)]

0 commit comments

Comments
 (0)