diff --git a/cozepy/websockets/chat/__init__.py b/cozepy/websockets/chat/__init__.py index a76508a..6507de8 100644 --- a/cozepy/websockets/chat/__init__.py +++ b/cozepy/websockets/chat/__init__.py @@ -181,8 +181,8 @@ def __init__( auth: Auth, requester: Requester, bot_id: str, - workflow_id: str, on_event: Union[WebsocketsChatEventHandler, Dict[WebsocketsEventType, Callable]], + workflow_id: Optional[str] = None, **kwargs, ): if isinstance(on_event, WebsocketsChatEventHandler): @@ -352,8 +352,8 @@ def create( self, *, bot_id: str, - workflow_id: str, on_event: Union[WebsocketsChatEventHandler, Dict[WebsocketsEventType, Callable]], + workflow_id: Optional[str] = None, **kwargs, ) -> WebsocketsChatClient: return WebsocketsChatClient( @@ -361,8 +361,8 @@ def create( auth=self._auth, requester=self._requester, bot_id=bot_id, - workflow_id=workflow_id, on_event=on_event, # type: ignore + workflow_id=workflow_id, **kwargs, ) @@ -433,8 +433,8 @@ def __init__( auth: Auth, requester: Requester, bot_id: str, - workflow_id: str, on_event: Union[AsyncWebsocketsChatEventHandler, Dict[WebsocketsEventType, Callable]], + workflow_id: Optional[str] = None, **kwargs, ): if isinstance(on_event, AsyncWebsocketsChatEventHandler): @@ -604,8 +604,8 @@ def create( self, *, bot_id: str, - workflow_id: str, on_event: Union[AsyncWebsocketsChatEventHandler, Dict[WebsocketsEventType, Callable]], + workflow_id: Optional[str] = None, **kwargs, ) -> AsyncWebsocketsChatClient: return AsyncWebsocketsChatClient( @@ -613,7 +613,7 @@ def create( auth=self._auth, requester=self._requester, bot_id=bot_id, - workflow_id=workflow_id, on_event=on_event, # type: ignore + workflow_id=workflow_id, **kwargs, )