Spaces:
Running
Running
Commit
·
01cc20c
1
Parent(s):
154b509
fixing wrong token loading
Browse files- 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
|
| 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(
|