github/MemGPT
7
0
Fork 0
mirror of https://github.com/cpacker/MemGPT.git synced 2025-06-03 04:30:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: allow setting pass for server --secure (#2132)

Browse Source
This commit is contained in:
Charles Packer 2024-12-01 21:35:31 -08:00 committed by GitHub
parent 7765cad817
commit aa36ee2142
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
letta/server/rest_api/app.py
View File

@ -1,5 +1,6 @@
import json import json
import logging import logging
import os
import sys import sys
from pathlib import Path from pathlib import Path
from typing import Optional from typing import Optional
@ -103,7 +104,7 @@ def generate_password():
return secrets.token_urlsafe(16) return secrets.token_urlsafe(16)
random_password = generate_password() random_password = os.getenv("LETTA_SERVER_PASSWORD") or generate_password()
class CheckPasswordMiddleware(BaseHTTPMiddleware): class CheckPasswordMiddleware(BaseHTTPMiddleware):
@ -132,11 +133,11 @@ def create_application() -> "FastAPI":
debug=True, debug=True,
) )
if "--ade" in sys.argv: if (os.getenv("LETTA_SERVER_ADE") == "true") or "--ade" in sys.argv:
settings.cors_origins.append("https://app.letta.com") settings.cors_origins.append("https://app.letta.com")
print(f"▶ View using ADE at: https://app.letta.com/development-servers/local/dashboard") print(f"▶ View using ADE at: https://app.letta.com/development-servers/local/dashboard")
if "--secure" in sys.argv: if (os.getenv("LETTA_SERVER_SECURE") == "true") or "--secure" in sys.argv:
print(f"▶ Using secure mode with password: {random_password}") print(f"▶ Using secure mode with password: {random_password}")
app.add_middleware(CheckPasswordMiddleware) app.add_middleware(CheckPasswordMiddleware)