Spaces:
Sleeping
Sleeping
Commit
·
d786593
1
Parent(s):
45d9469
test some cookie stuff
Browse files- evolutiontransformer/api.py +12 -2
- pyproject.toml +1 -0
- uv.lock +2 -0
evolutiontransformer/api.py
CHANGED
|
@@ -22,7 +22,7 @@ app = FastAPI()
|
|
| 22 |
|
| 23 |
app.add_middleware(
|
| 24 |
CORSMiddleware,
|
| 25 |
-
allow_origins=["
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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" },
|