akseljoonas HF Staff commited on
Commit
01cc20c
·
1 Parent(s): 154b509

fixing wrong token loading

Browse files
Files changed (1) hide show
  1. backend/routes/agent.py +3 -1
backend/routes/agent.py CHANGED
@@ -206,11 +206,13 @@ async def create_session(
206
 
207
  Returns 503 if the server or user has reached the session limit.
208
  """
209
- # Extract the user's HF token from the Bearer header
210
  hf_token = None
211
  auth_header = request.headers.get("Authorization", "")
212
  if auth_header.startswith("Bearer "):
213
  hf_token = auth_header[7:]
 
 
214
 
215
  try:
216
  session_id = await session_manager.create_session(
 
206
 
207
  Returns 503 if the server or user has reached the session limit.
208
  """
209
+ # Extract the user's HF token (Bearer header or HttpOnly cookie)
210
  hf_token = None
211
  auth_header = request.headers.get("Authorization", "")
212
  if auth_header.startswith("Bearer "):
213
  hf_token = auth_header[7:]
214
+ if not hf_token:
215
+ hf_token = request.cookies.get("hf_access_token")
216
 
217
  try:
218
  session_id = await session_manager.create_session(