diff --git a/langfuse/api/resources/annotation_queues/types/annotation_queue_object_type.py b/langfuse/api/resources/annotation_queues/types/annotation_queue_object_type.py index 1bef33b55..6e63a7015 100644 --- a/langfuse/api/resources/annotation_queues/types/annotation_queue_object_type.py +++ b/langfuse/api/resources/annotation_queues/types/annotation_queue_object_type.py @@ -9,13 +9,17 @@ class AnnotationQueueObjectType(str, enum.Enum): TRACE = "TRACE" OBSERVATION = "OBSERVATION" + SESSION = "SESSION" def visit( self, trace: typing.Callable[[], T_Result], observation: typing.Callable[[], T_Result], + session: typing.Callable[[], T_Result], ) -> T_Result: if self is AnnotationQueueObjectType.TRACE: return trace() if self is AnnotationQueueObjectType.OBSERVATION: return observation() + if self is AnnotationQueueObjectType.SESSION: + return session()