mirror of
https://github.com/suitenumerique/docs.git
synced 2026-09-17 23:28:01 +02:00
Manuel RaynaudandAnthony LC
8ce216f6e8
The frontend application is using Vercel AI SDK and it's data stream protocol. We decided to use the pydantic AI library to use it's vercel ai adapter. It will make the payload validation, use AsyncIterator and deal with vercel specification.
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
"""Force session creation for all requests."""
|
|
|
|
|
|
class ForceSessionMiddleware:
|
|
"""
|
|
Force session creation for unauthenticated users.
|
|
Must be used after Authentication middleware.
|
|
"""
|
|
|
|
def __init__(self, get_response):
|
|
"""Initialize the middleware."""
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request):
|
|
"""Force session creation for unauthenticated users."""
|
|
|
|
if not request.user.is_authenticated and request.session.session_key is None:
|
|
request.session.create()
|
|
|
|
response = self.get_response(request)
|
|
return response
|
|
|
|
|
|
class SaveRawBodyMiddleware:
|
|
"""
|
|
Save the raw request body to use it later.
|
|
"""
|
|
|
|
def __init__(self, get_response):
|
|
"""Initialize the middleware."""
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request):
|
|
"""Save the raw request body in the request to use it later."""
|
|
if request.path.endswith(("/ai-proxy/", "/ai-proxy")):
|
|
request.raw_body = request.body
|
|
|
|
response = self.get_response(request)
|
|
return response
|