🔥 refactor(models): remove Token model and references to it to simplify the codebase and remove unused code

🔥 refactor(test_custom_component.py): remove unused fixture and imports to clean up the test file
🔧 chore(test_custom_component.py): update component fixture to include user_id parameter for testing purposes

src/backend/langflow/services/database/models/__init__.py

@@ -1,6 +1,5 @@
 from .flow import Flow
 from .user import User
-from .token import Token
 from .api_key import ApiKey
 
-__all__ = ["Flow", "User", "Token", "ApiKey"]
+__all__ = ["Flow", "User", "ApiKey"]

src/backend/langflow/services/database/models/token/__init__.py

@@ -1,5 +0,0 @@
-from .token import Token
-
-__all__ = [
-    "Token",
-]

src/backend/langflow/services/database/models/token/token.py

@@ -1,7 +0,0 @@
-from pydantic import BaseModel
-
-
-class Token(BaseModel):
-    access_token: str
-    refresh_token: str
-    token_type: str

tests/test_custom_component.py

@@ -473,15 +473,16 @@ def test_build_config_no_code():
 
 
 @pytest.fixture
-def component():
+def component(client, active_user):
     return CustomComponent(
+        user_id=active_user.id,
         field_config={
             "fields": {
                 "llm": {"type": "str"},
                 "url": {"type": "str"},
                 "year": {"type": "int"},
             }
-        }
+        },
     )