mirror of
https://github.com/open-webui/open-webui
synced 2025-06-26 18:26:48 +00:00
wip
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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"):
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user