akseljoonas HF Staff commited on
Commit
e17cb5d
·
1 Parent(s): 0736c2f

Fall back to HF_TOKEN env var for local dev sessions

Browse files
Files changed (1) hide show
  1. 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 or HttpOnly cookie)
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 and user["user_id"] == "dev":
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: