16 lines
516 B
Python
16 lines
516 B
Python
from collections.abc import Callable
|
|
from typing import Any, cast
|
|
|
|
from django.urls import path
|
|
from django.urls.resolvers import URLPattern, URLResolver
|
|
|
|
from .consumers import PromptConsumer, GuidanceConsumer
|
|
|
|
AsgiApplication = Callable[..., Any]
|
|
RoutePattern = URLPattern | URLResolver
|
|
|
|
websocket_urlpatterns: list[RoutePattern] = [
|
|
path("ws/prompt/<int:task_id>/", cast(AsgiApplication, PromptConsumer.as_asgi())),
|
|
path("ws/guidance/<int:task_id>/", cast(AsgiApplication, GuidanceConsumer.as_asgi())),
|
|
]
|