sort and fix backend imports

This commit is contained in:
Pascal Lim
2024-08-30 22:26:22 +02:00
parent 08efabc696
commit c386d0b1a5
63 changed files with 598 additions and 1023 deletions
@@ -1,5 +1,3 @@
import pytest
from test.util.abstract_integration_test import AbstractPostgresTest
from test.util.mock_user import mock_webui_user
@@ -9,8 +7,8 @@ class TestAuths(AbstractPostgresTest):
def setup_class(cls):
super().setup_class()
from apps.webui.models.users import Users
from apps.webui.models.auths import Auths
from apps.webui.models.users import Users
cls.users = Users
cls.auths = Auths
@@ -5,7 +5,6 @@ from test.util.mock_user import mock_webui_user
class TestChats(AbstractPostgresTest):
BASE_PATH = "/api/v1/chats"
def setup_class(cls):
@@ -13,8 +12,7 @@ class TestChats(AbstractPostgresTest):
def setup_method(self):
super().setup_method()
from apps.webui.models.chats import ChatForm
from apps.webui.models.chats import Chats
from apps.webui.models.chats import ChatForm, Chats
self.chats = Chats
self.chats.insert_new_chat(
@@ -3,7 +3,6 @@ from test.util.mock_user import mock_webui_user
class TestDocuments(AbstractPostgresTest):
BASE_PATH = "/api/v1/documents"
def setup_class(cls):
@@ -3,7 +3,6 @@ from test.util.mock_user import mock_webui_user
class TestModels(AbstractPostgresTest):
BASE_PATH = "/api/v1/models"
def setup_class(cls):
@@ -3,7 +3,6 @@ from test.util.mock_user import mock_webui_user
class TestPrompts(AbstractPostgresTest):
BASE_PATH = "/api/v1/prompts"
def test_prompts(self):
@@ -21,7 +21,6 @@ def _assert_user(data, id, **kwargs):
class TestUsers(AbstractPostgresTest):
BASE_PATH = "/api/v1/users"
def setup_class(cls):