Add/fastapi users addition

api.py

@@ -50,6 +50,8 @@
     batched_prompts,
     recipes,
     remote,
+    auth2,
+    teams,
 )
 import torch
 
@@ -82,6 +84,9 @@
 
 from dotenv import load_dotenv
 
+
+from transformerlab.shared.models.user_model import create_db_and_tables
+
 load_dotenv()
 
 # The following environment variable can be used by other scripts
@@ -110,6 +115,7 @@ async def lifespan(app: FastAPI):
     galleries.update_gallery_cache()
     spawn_fastchat_controller_subprocess()
     await db.init()
+    await create_db_and_tables()
     print("✅ SEED DATA")
     # Initialize experiments and cancel any running jobs
     seed_default_experiments()
@@ -230,6 +236,8 @@ async def validation_exception_handler(request, exc):
 app.include_router(batched_prompts.router)
 app.include_router(remote.router)
 app.include_router(fastchat_openai_api.router)
+app.include_router(teams.router)
+app.include_router(auth2.router)
 
 # Authentication and session management routes
 if os.getenv("TFL_MULTITENANT") == "true":

requirements-no-gpu-uv.txt

@@ -1,5 +1,5 @@
 # This file was autogenerated by uv via the following command:
-#    uv pip compile requirements.in -o requirements-no-gpu-uv.txt --index-strategy unsafe-best-match
+#    uv pip compile requirements.in -o requirements-no-gpu-uv.txt
 absl-py==2.1.0
     # via tensorboard
 accelerate==1.3.0
@@ -27,6 +27,10 @@ anyio==4.8.0
     #   sse-starlette
     #   starlette
     #   watchfiles
+argon2-cffi==23.1.0
+    # via pwdlib
+argon2-cffi-bindings==25.1.0
+    # via argon2-cffi
 attrs==25.1.0
     # via aiohttp
 audioread==3.0.1
@@ -39,6 +43,8 @@ azure-core==1.33.0
     #   azure-identity
 azure-identity==1.21.0
     # via markitdown
+bcrypt==4.3.0
+    # via pwdlib
 beautifulsoup4==4.13.3
     # via
     #   markdownify
@@ -51,6 +57,7 @@ certifi==2022.12.7
     #   sentry-sdk
 cffi==1.17.1
     # via
+    #   argon2-cffi-bindings
     #   cryptography
     #   soundfile
 charset-normalizer==2.1.1
@@ -77,6 +84,7 @@ cryptography==44.0.2
     #   msal
     #   pdfminer-six
     #   pyjwt
+    #   python-jose
     #   workos
 datasets==3.6.0
     # via
@@ -93,20 +101,33 @@ dill==0.3.8
     #   datasets
     #   evaluate
     #   multiprocess
+dnspython==2.8.0
+    # via email-validator
 docker-pycreds==0.4.0
     # via wandb
+ecdsa==0.19.1
+    # via python-jose
 einops==0.8.0
     # via
     #   -r requirements.in
     #   controlnet-aux
+email-validator==2.3.0
+    # via fastapi-users
 et-xmlfile==2.0.0
     # via openpyxl
 evaluate==0.4.3
     # via -r requirements.in
 fastapi==0.115.7
     # via
     #   -r requirements.in
+    #   fastapi-users
     #   transformerlab-inference
+fastapi-users==15.0.1
+    # via
+    #   -r requirements.in
+    #   fastapi-users-db-sqlalchemy
+fastapi-users-db-sqlalchemy==7.0.0
+    # via fastapi-users
 filelock==3.13.1
     # via
     #   controlnet-aux
@@ -169,6 +190,7 @@ humanfriendly==10.0
 idna==3.4
     # via
     #   anyio
+    #   email-validator
     #   httpx
     #   requests
     #   yarl
@@ -205,6 +227,8 @@ macmon-python==0.1.2
     # via -r requirements.in
 magika==0.6.1
     # via markitdown
+makefun==1.16.0
+    # via fastapi-users
 mammoth==1.9.0
     # via markitdown
 markdown==3.7
@@ -352,8 +376,14 @@ psutil==6.1.1
     #   peft
     #   transformerlab-inference
     #   wandb
+pwdlib==0.2.1
+    # via fastapi-users
 pyarrow==19.0.0
     # via datasets
+pyasn1==0.6.1
+    # via
+    #   python-jose
+    #   rsa
 pycparser==2.22
     # via cffi
 pydantic==2.11.7
@@ -381,6 +411,7 @@ pygments==2.19.1
     #   rich
 pyjwt==2.10.1
     # via
+    #   fastapi-users
     #   msal
     #   workos
 pytest==8.4.2
@@ -393,9 +424,12 @@ python-dotenv==1.0.1
     #   magika
     #   mcp
     #   pydantic-settings
+python-jose==3.5.0
+    # via -r requirements.in
 python-multipart==0.0.20
     # via
     #   -r requirements.in
+    #   fastapi-users
     #   mcp
 python-pptx==1.0.2
     # via markitdown
@@ -436,6 +470,8 @@ rich==13.9.4
     # via
     #   transformerlab-inference
     #   typer
+rsa==4.9.1
+    # via python-jose
 safetensors==0.5.3
     # via
     #   accelerate
@@ -477,6 +513,7 @@ six==1.17.0
     # via
     #   azure-core
     #   docker-pycreds
+    #   ecdsa
     #   markdownify
     #   python-dateutil
     #   tensorboard
@@ -496,7 +533,9 @@ soxr==0.5.0.post1
 speechrecognition==3.14.2
     # via markitdown
 sqlalchemy==2.0.38
-    # via -r requirements.in
+    # via
+    #   -r requirements.in
+    #   fastapi-users-db-sqlalchemy
 sse-starlette==2.3.5
     # via mcp
 starlette==0.45.3

requirements-rocm-uv.txt

@@ -1,5 +1,5 @@
 # This file was autogenerated by uv via the following command:
-#    uv pip compile requirements-rocm.in -o requirements-rocm-uv.txt --index-strategy unsafe-best-match
+#    uv pip compile requirements-rocm.in -o requirements-rocm-uv.txt --index-strategy unsafe-best-match --python-platform linux
 absl-py==2.2.2
     # via tensorboard
 accelerate==1.6.0
@@ -27,6 +27,10 @@ anyio==4.9.0
     #   sse-starlette
     #   starlette
     #   watchfiles
+argon2-cffi==23.1.0
+    # via pwdlib
+argon2-cffi-bindings==25.1.0
+    # via argon2-cffi
 attrs==25.3.0
     # via aiohttp
 audioread==3.0.1
@@ -39,6 +43,8 @@ azure-core==1.33.0
     #   azure-identity
 azure-identity==1.21.0
     # via markitdown
+bcrypt==4.3.0
+    # via pwdlib
 beautifulsoup4==4.13.4
     # via
     #   markdownify
@@ -51,6 +57,7 @@ certifi==2022.12.7
     #   sentry-sdk
 cffi==1.17.1
     # via
+    #   argon2-cffi-bindings
     #   cryptography
     #   soundfile
 charset-normalizer==2.1.1
@@ -77,6 +84,7 @@ cryptography==44.0.2
     #   msal
     #   pdfminer-six
     #   pyjwt
+    #   python-jose
     #   workos
 datasets==3.6.0
     # via
@@ -93,20 +101,33 @@ dill==0.3.8
     #   datasets
     #   evaluate
     #   multiprocess
+dnspython==2.8.0
+    # via email-validator
 docker-pycreds==0.4.0
     # via wandb
+ecdsa==0.19.1
+    # via python-jose
 einops==0.8.1
     # via
     #   -r requirements-rocm.in
     #   controlnet-aux
+email-validator==2.3.0
+    # via fastapi-users
 et-xmlfile==2.0.0
     # via openpyxl
 evaluate==0.4.3
     # via -r requirements-rocm.in
 fastapi==0.115.12
     # via
     #   -r requirements-rocm.in
+    #   fastapi-users
     #   transformerlab-inference
+fastapi-users==15.0.1
+    # via
+    #   -r requirements-rocm.in
+    #   fastapi-users-db-sqlalchemy
+fastapi-users-db-sqlalchemy==7.0.0
+    # via fastapi-users
 filelock==3.13.1
     # via
     #   controlnet-aux
@@ -169,6 +190,7 @@ humanfriendly==10.0
 idna==3.4
     # via
     #   anyio
+    #   email-validator
     #   httpx
     #   requests
     #   yarl
@@ -205,6 +227,8 @@ macmon-python==0.1.2
     # via -r requirements-rocm.in
 magika==0.6.1
     # via markitdown
+makefun==1.16.0
+    # via fastapi-users
 mammoth==1.9.0
     # via markitdown
 markdown==3.8
@@ -350,8 +374,14 @@ psutil==7.0.0
     #   peft
     #   transformerlab-inference
     #   wandb
+pwdlib==0.2.1
+    # via fastapi-users
 pyarrow==20.0.0
     # via datasets
+pyasn1==0.6.1
+    # via
+    #   python-jose
+    #   rsa
 pycparser==2.22
     # via cffi
 pydantic==2.11.7
@@ -379,6 +409,7 @@ pygments==2.19.1
     #   rich
 pyjwt==2.10.1
     # via
+    #   fastapi-users
     #   msal
     #   workos
 pyrsmi==0.2.0
@@ -393,9 +424,12 @@ python-dotenv==1.1.0
     #   magika
     #   mcp
     #   pydantic-settings
+python-jose==3.5.0
+    # via -r requirements-rocm.in
 python-multipart==0.0.20
     # via
     #   -r requirements-rocm.in
+    #   fastapi-users
     #   mcp
 python-pptx==1.0.2
     # via markitdown
@@ -438,6 +472,8 @@ rich==14.0.0
     # via
     #   transformerlab-inference
     #   typer
+rsa==4.9.1
+    # via python-jose
 safetensors==0.5.3
     # via
     #   accelerate
@@ -480,6 +516,7 @@ six==1.17.0
     # via
     #   azure-core
     #   docker-pycreds
+    #   ecdsa
     #   markdownify
     #   python-dateutil
     #   tensorboard
@@ -499,7 +536,9 @@ soxr==0.5.0.post1
 speechrecognition==3.14.2
     # via markitdown
 sqlalchemy==2.0.40
-    # via -r requirements-rocm.in
+    # via
+    #   -r requirements-rocm.in
+    #   fastapi-users-db-sqlalchemy
 sse-starlette==2.3.5
     # via mcp
 starlette==0.46.2

requirements-rocm.in

@@ -5,10 +5,12 @@ datasets==3.6.0
 einops
 evaluate
 fastapi
+fastapi-users[sqlalchemy]
 packaging
 psutil
 python-multipart
 python-dotenv
+python-jose[cryptography]
 pydantic>= 2.0
 nltk==3.9.1
 scipy