From c268b2be1728df1cfd1f9bcb62fc8aff5f3378cb Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Fri, 25 Aug 2023 15:19:13 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A5=20refactor(models):=20remove=20Tok?=
 =?UTF-8?q?en=20model=20and=20references=20to=20it=20to=20simplify=20the?=
 =?UTF-8?q?=20codebase=20and=20remove=20unused=20code=20=F0=9F=94=A5=20ref?=
 =?UTF-8?q?actor(test=5Fcustom=5Fcomponent.py):=20remove=20unused=20fixtur?=
 =?UTF-8?q?e=20and=20imports=20to=20clean=20up=20the=20test=20file=20?=
 =?UTF-8?q?=F0=9F=94=A7=20chore(test=5Fcustom=5Fcomponent.py):=20update=20?=
 =?UTF-8?q?component=20fixture=20to=20include=20user=5Fid=20parameter=20fo?=
 =?UTF-8?q?r=20testing=20purposes?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/services/database/models/__init__.py  | 3 +--
 .../langflow/services/database/models/token/__init__.py    | 5 -----
 .../langflow/services/database/models/token/token.py       | 7 -------
 tests/test_custom_component.py                             | 5 +++--
 4 files changed, 4 insertions(+), 16 deletions(-)
 delete mode 100644 src/backend/langflow/services/database/models/token/__init__.py
 delete mode 100644 src/backend/langflow/services/database/models/token/token.py

diff --git a/src/backend/langflow/services/database/models/__init__.py b/src/backend/langflow/services/database/models/__init__.py
index 01e81e277..3cc4231a3 100644
--- a/src/backend/langflow/services/database/models/__init__.py
+++ b/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"]
diff --git a/src/backend/langflow/services/database/models/token/__init__.py b/src/backend/langflow/services/database/models/token/__init__.py
deleted file mode 100644
index 9b9fa397d..000000000
--- a/src/backend/langflow/services/database/models/token/__init__.py
+++ /dev/null
@@ -1,5 +0,0 @@
-from .token import Token
-
-__all__ = [
-    "Token",
-]
diff --git a/src/backend/langflow/services/database/models/token/token.py b/src/backend/langflow/services/database/models/token/token.py
deleted file mode 100644
index 68c70f07f..000000000
--- a/src/backend/langflow/services/database/models/token/token.py
+++ /dev/null
@@ -1,7 +0,0 @@
-from pydantic import BaseModel
-
-
-class Token(BaseModel):
-    access_token: str
-    refresh_token: str
-    token_type: str
diff --git a/tests/test_custom_component.py b/tests/test_custom_component.py
index 4dc8c9f1a..e75dc0e5b 100644
--- a/tests/test_custom_component.py
+++ b/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"},
             }
-        }
+        },
     )