ai_flow.notification package¶

Submodules¶

ai_flow.notification.notification_client module¶

class ai_flow.notification.notification_client.AIFlowNotificationClient(server_uri: str)[source]¶

Bases: object

close()[source]¶

register_listener(listener_processor: notification_service.client.notification_client.ListenerProcessor, event_keys: Optional[List[str]] = None, begin_time: Optional[datetime.datetime] = None) → notification_service.client.notification_client.ListenerRegistrationId[source]¶

send_event(key: str, value: Optional[str] = None)[source]¶

Send event to current workflow execution. This function can only be used in AIFlow Operator runtime. It will retrieve the workflow execution info from runtime context and set to context of the event.

Parameters

key – the key of the event.
value – optional, the value of the event.

unregister_listener(registration_id: notification_service.client.notification_client.ListenerRegistrationId)[source]¶