refactor(task_pipeline): Remove unused EventHandler

Signed-off-by: -LAN- <laipz8200@outlook.com>
pull/22600/head
-LAN- 10 months ago
parent 4a3307e9e8
commit 705eafc132
No known key found for this signature in database
GPG Key ID: 6BA0D108DED011FF

@ -3,7 +3,7 @@ import time
from collections.abc import Callable, Generator, Mapping from collections.abc import Callable, Generator, Mapping
from contextlib import contextmanager from contextlib import contextmanager
from threading import Thread from threading import Thread
from typing import Any, Optional, Protocol, Union from typing import Any, Optional, Union
from sqlalchemy import select from sqlalchemy import select
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
@ -277,18 +277,6 @@ class AdvancedChatAppGenerateTaskPipeline:
if tts_publisher: if tts_publisher:
yield MessageAudioEndStreamResponse(audio="", task_id=task_id) yield MessageAudioEndStreamResponse(audio="", task_id=task_id)
class EventHandler(Protocol):
"""Protocol for event handlers."""
def __call__(
self,
event: Any,
*,
graph_runtime_state: Optional[Any] = None,
tts_publisher: Optional[Any] = None,
trace_manager: Optional[Any] = None,
) -> Generator[Any, None, None]: ...
@contextmanager @contextmanager
def _database_session(self): def _database_session(self):
"""Context manager for database sessions.""" """Context manager for database sessions."""

@ -2,7 +2,7 @@ import logging
import time import time
from collections.abc import Callable, Generator from collections.abc import Callable, Generator
from contextlib import contextmanager from contextlib import contextmanager
from typing import Any, Optional, Protocol, Union from typing import Any, Optional, Union
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
@ -247,19 +247,6 @@ class WorkflowAppGenerateTaskPipeline:
if tts_publisher: if tts_publisher:
yield MessageAudioEndStreamResponse(audio="", task_id=task_id) yield MessageAudioEndStreamResponse(audio="", task_id=task_id)
class EventHandler(Protocol):
"""Protocol for event handlers."""
def __call__(
self,
event: Any,
*,
graph_runtime_state: Optional[Any] = None,
tts_publisher: Optional[Any] = None,
trace_manager: Optional[Any] = None,
queue_message: Optional[Any] = None,
) -> Generator[Any, None, None]: ...
@contextmanager @contextmanager
def _database_session(self): def _database_session(self):
"""Context manager for database sessions.""" """Context manager for database sessions."""

Loading…
Cancel
Save