tcmmichaelb139 commited on
Commit
d786593
·
1 Parent(s): 45d9469

test some cookie stuff

Browse files
Files changed (3) hide show
  1. evolutiontransformer/api.py +12 -2
  2. pyproject.toml +1 -0
  3. uv.lock +2 -0
evolutiontransformer/api.py CHANGED
@@ -22,7 +22,7 @@ app = FastAPI()
22
 
23
  app.add_middleware(
24
  CORSMiddleware,
25
- allow_origins=["http://localhost:5173"], # todo add website url
26
  allow_credentials=True,
27
  allow_methods=["*"],
28
  allow_headers=["*"],
@@ -47,10 +47,20 @@ class MergeRequest(BaseModel):
47
 
48
  def get_session_id(request: Request, response: Response):
49
  session_id = request.cookies.get("session_id")
 
 
50
 
51
  if not session_id:
52
  session_id = str(uuid.uuid4())
53
- response.set_cookie(key="session_id", value=session_id)
 
 
 
 
 
 
 
 
54
 
55
  return session_id
56
 
 
22
 
23
  app.add_middleware(
24
  CORSMiddleware,
25
+ allow_origins=["*"], # Allow all origins for now to debug
26
  allow_credentials=True,
27
  allow_methods=["*"],
28
  allow_headers=["*"],
 
47
 
48
  def get_session_id(request: Request, response: Response):
49
  session_id = request.cookies.get("session_id")
50
+ print(f"Received cookies: {request.cookies}")
51
+ print(f"Current session_id: {session_id}")
52
 
53
  if not session_id:
54
  session_id = str(uuid.uuid4())
55
+ print(f"Generated new session_id: {session_id}")
56
+ # Set cookie with appropriate settings
57
+ response.set_cookie(
58
+ key="session_id",
59
+ value=session_id,
60
+ httponly=True, # Prevent XSS attacks
61
+ secure=True, # Only send over HTTPS in production
62
+ samesite="lax", # Allow same-site requests
63
+ )
64
 
65
  return session_id
66
 
pyproject.toml CHANGED
@@ -15,6 +15,7 @@ dependencies = [
15
  "matplotlib>=3.10.6",
16
  "numpy>=2.3.3",
17
  "pytest>=8.4.2",
 
18
  "redis>=6.4.0",
19
  "torch>=2.8.0",
20
  "transformers>=4.56.1",
 
15
  "matplotlib>=3.10.6",
16
  "numpy>=2.3.3",
17
  "pytest>=8.4.2",
18
+ "python-multipart>=0.0.20",
19
  "redis>=6.4.0",
20
  "torch>=2.8.0",
21
  "transformers>=4.56.1",
uv.lock CHANGED
@@ -526,6 +526,7 @@ dependencies = [
526
  { name = "matplotlib" },
527
  { name = "numpy" },
528
  { name = "pytest" },
 
529
  { name = "redis" },
530
  { name = "torch" },
531
  { name = "transformers" },
@@ -544,6 +545,7 @@ requires-dist = [
544
  { name = "matplotlib", specifier = ">=3.10.6" },
545
  { name = "numpy", specifier = ">=2.3.3" },
546
  { name = "pytest", specifier = ">=8.4.2" },
 
547
  { name = "redis", specifier = ">=6.4.0" },
548
  { name = "torch", specifier = ">=2.8.0" },
549
  { name = "transformers", specifier = ">=4.56.1" },
 
526
  { name = "matplotlib" },
527
  { name = "numpy" },
528
  { name = "pytest" },
529
+ { name = "python-multipart" },
530
  { name = "redis" },
531
  { name = "torch" },
532
  { name = "transformers" },
 
545
  { name = "matplotlib", specifier = ">=3.10.6" },
546
  { name = "numpy", specifier = ">=2.3.3" },
547
  { name = "pytest", specifier = ">=8.4.2" },
548
+ { name = "python-multipart", specifier = ">=0.0.20" },
549
  { name = "redis", specifier = ">=6.4.0" },
550
  { name = "torch", specifier = ">=2.8.0" },
551
  { name = "transformers", specifier = ">=4.56.1" },