Merge pull request #948 from transformerlab/add/remove-workos-routes-again

Remove WorkOS deps from the API

Author: aliasaria

api/api.py

@@ -59,11 +59,10 @@
     tools,
     batched_prompts,
     recipes,
-    remote,
-    auth2,
     teams,
+    auth,
 )
-from transformerlab.routers.auth2 import get_user_and_team  # noqa: E402
+from transformerlab.routers.auth import get_user_and_team  # noqa: E402
 import torch  # noqa: E402
 
 try:
@@ -221,9 +220,6 @@ async def validation_exception_handler(request, exc):
     return create_error_response(ErrorCode.VALIDATION_TYPE_ERROR, str(exc))
 
 
-### END GENERAL API - NOT OPENAI COMPATIBLE ###
-
-
 app.include_router(model.router, dependencies=[Depends(get_user_and_team)])
 app.include_router(serverinfo.router, dependencies=[Depends(get_user_and_team)])
 app.include_router(train.router, dependencies=[Depends(get_user_and_team)])
@@ -238,17 +234,9 @@ async def validation_exception_handler(request, exc):
 app.include_router(tools.router, dependencies=[Depends(get_user_and_team)])
 app.include_router(recipes.router, dependencies=[Depends(get_user_and_team)])
 app.include_router(batched_prompts.router, dependencies=[Depends(get_user_and_team)])
-app.include_router(remote.router, dependencies=[Depends(get_user_and_team)])
 app.include_router(fastchat_openai_api.router, dependencies=[Depends(get_user_and_team)])
 app.include_router(teams.router, dependencies=[Depends(get_user_and_team)])
-app.include_router(auth2.router)
-
-# Authentication and session management routes
-if os.getenv("TFL_MULTITENANT") == "true":
-    from transformerlab.routers import auth  # noqa: E402
-
-    app.include_router(auth.router)
-
+app.include_router(auth.router)
 
 controller_process = None
 worker_process = None

api/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
@@ -94,7 +94,6 @@ cryptography==44.0.2
     #   msal
     #   pdfminer-six
     #   pyjwt
-    #   workos
 datasets==3.6.0
     # via
     #   -r requirements.in
@@ -179,7 +178,6 @@ httpx==0.28.1
     # via
     #   mcp
     #   transformerlab-inference
-    #   workos
 httpx-sse==0.4.0
     # via mcp
 huggingface-hub==0.34.3
@@ -408,7 +406,6 @@ pydantic==2.11.7
     #   pydantic-settings
     #   transformerlab-inference
     #   wandb
-    #   workos
 pydantic-core==2.33.2
     # via pydantic
 pydantic-settings==2.8.1
@@ -426,7 +423,6 @@ pyjwt==2.10.1
     # via
     #   fastapi-users
     #   msal
-    #   workos
 pytest==8.4.2
     # via transformerlab
 python-dateutil==2.9.0.post0
@@ -518,6 +514,7 @@ setproctitle==1.3.5
 setuptools==70.2.0
     # via
     #   tensorboard
+    #   torch
     #   wandb
 shellingham==1.5.4
     # via typer
@@ -671,8 +668,6 @@ werkzeug==3.1.3
     #   -r requirements.in
     #   tensorboard
     #   transformerlab
-workos==5.28.0
-    # via -r requirements.in
 wrapt==1.17.3
     # via aiobotocore
 xlrd==2.0.1

api/requirements-rocm-uv.txt

@@ -521,6 +521,7 @@ setuptools==70.2.0
     # via
     #   pytorch-triton-rocm
     #   tensorboard
+    #   torch
     #   wandb
 shellingham==1.5.4
     # via typer

api/requirements-uv.txt

@@ -94,7 +94,6 @@ cryptography==44.0.2
     #   msal
     #   pdfminer-six
     #   pyjwt
-    #   workos
 datasets==3.6.0
     # via
     #   -r requirements.in
@@ -179,7 +178,6 @@ httpx==0.28.1
     # via
     #   mcp
     #   transformerlab-inference
-    #   workos
 httpx-sse==0.4.0
     # via mcp
 huggingface-hub==0.34.3
@@ -445,7 +443,6 @@ pydantic==2.11.7
     #   pydantic-settings
     #   transformerlab-inference
     #   wandb
-    #   workos
 pydantic-core==2.33.2
     # via pydantic
 pydantic-settings==2.8.1
@@ -463,7 +460,6 @@ pyjwt==2.10.1
     # via
     #   fastapi-users
     #   msal
-    #   workos
 pytest==8.4.2
     # via transformerlab
 python-dateutil==2.9.0.post0
@@ -555,6 +551,7 @@ setproctitle==1.3.5
 setuptools==70.2.0
     # via
     #   tensorboard
+    #   torch
     #   triton
     #   wandb
 shellingham==1.5.4
@@ -711,8 +708,6 @@ werkzeug==3.1.3
     #   -r requirements.in
     #   tensorboard
     #   transformerlab
-workos==5.28.0
-    # via -r requirements.in
 wrapt==1.17.3
     # via aiobotocore
 xlrd==2.0.1

api/requirements.in

@@ -24,7 +24,6 @@ watchfiles
 wandb==0.19.10
 tensorboardX
 werkzeug
-workos
 sqlalchemy[asyncio]
 sentence-transformers
 markitdown[all]