This commit is contained in:
Timothy Jaeryang Baek
2024-12-10 00:54:13 -08:00
parent f6bec8d9f3
commit d3d161f723
112 changed files with 1217 additions and 1165 deletions

View File

@@ -7,8 +7,8 @@ class TestAuths(AbstractPostgresTest):
def setup_class(cls):
super().setup_class()
from open_webui.apps.webui.models.auths import Auths
from open_webui.apps.webui.models.users import Users
from open_webui.models.auths import Auths
from open_webui.models.users import Users
cls.users = Users
cls.auths = Auths

View File

@@ -12,7 +12,7 @@ class TestChats(AbstractPostgresTest):
def setup_method(self):
super().setup_method()
from open_webui.apps.webui.models.chats import ChatForm, Chats
from open_webui.models.chats import ChatForm, Chats
self.chats = Chats
self.chats.insert_new_chat(
@@ -88,7 +88,7 @@ class TestChats(AbstractPostgresTest):
def test_get_user_archived_chats(self):
self.chats.archive_all_chats_by_user_id("2")
from open_webui.apps.webui.internal.db import Session
from open_webui.internal.db import Session
Session.commit()
with mock_webui_user(id="2"):

View File

@@ -7,7 +7,7 @@ class TestModels(AbstractPostgresTest):
def setup_class(cls):
super().setup_class()
from open_webui.apps.webui.models.models import Model
from open_webui.models.models import Model
cls.models = Model

View File

@@ -25,7 +25,7 @@ class TestUsers(AbstractPostgresTest):
def setup_class(cls):
super().setup_class()
from open_webui.apps.webui.models.users import Users
from open_webui.models.users import Users
cls.users = Users