Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
Commit ·
7872152
1
Parent(s): 3b6a2ca
Fall back to HF_TOKEN env var for local dev sessions
Browse files- backend/routes/agent.py +3 -5
backend/routes/agent.py
CHANGED
|
@@ -7,6 +7,7 @@ dependency. In dev mode (no OAUTH_CLIENT_ID), auth is bypassed automatically.
|
|
| 7 |
import asyncio
|
| 8 |
import json
|
| 9 |
import logging
|
|
|
|
| 10 |
from typing import Any
|
| 11 |
|
| 12 |
from dependencies import get_current_user
|
|
@@ -205,17 +206,14 @@ async def create_session(
|
|
| 205 |
|
| 206 |
Returns 503 if the server or user has reached the session limit.
|
| 207 |
"""
|
| 208 |
-
# Extract the user's HF token (Bearer header
|
| 209 |
-
# In dev mode, fall back to environment variable if no token in request
|
| 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 |
-
if not hf_token
|
| 217 |
-
# Dev mode: use HF_TOKEN from environment
|
| 218 |
-
import os
|
| 219 |
hf_token = os.environ.get("HF_TOKEN")
|
| 220 |
|
| 221 |
try:
|
|
|
|
| 7 |
import asyncio
|
| 8 |
import json
|
| 9 |
import logging
|
| 10 |
+
import os
|
| 11 |
from typing import Any
|
| 12 |
|
| 13 |
from dependencies import get_current_user
|
|
|
|
| 206 |
|
| 207 |
Returns 503 if the server or user has reached the session limit.
|
| 208 |
"""
|
| 209 |
+
# Extract the user's HF token (Bearer header, HttpOnly cookie, or env var)
|
|
|
|
| 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 |
+
if not hf_token:
|
|
|
|
|
|
|
| 217 |
hf_token = os.environ.get("HF_TOKEN")
|
| 218 |
|
| 219 |
try:
|