mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-19 12:21:46 +03:00
575d9886c5
* feat: add notifications table * feat: add Notification model * feat: add notification repositories * feat: add upload and crawl notifications * feat: update notification message
39 lines
915 B
Python
39 lines
915 B
Python
from logger import get_logger
|
|
from models.databases.supabase import (
|
|
ApiKeyHandler,
|
|
Brain,
|
|
BrainSubscription,
|
|
Chats,
|
|
File,
|
|
Notifications,
|
|
Prompts,
|
|
UserUsage,
|
|
Vector,
|
|
)
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class SupabaseDB(
|
|
Brain,
|
|
UserUsage,
|
|
File,
|
|
BrainSubscription,
|
|
ApiKeyHandler,
|
|
Chats,
|
|
Vector,
|
|
Prompts,
|
|
Notifications,
|
|
):
|
|
def __init__(self, supabase_client):
|
|
self.db = supabase_client
|
|
Brain.__init__(self, supabase_client)
|
|
UserUsage.__init__(self, supabase_client)
|
|
File.__init__(self, supabase_client)
|
|
BrainSubscription.__init__(self, supabase_client)
|
|
ApiKeyHandler.__init__(self, supabase_client)
|
|
Chats.__init__(self, supabase_client)
|
|
Vector.__init__(self, supabase_client)
|
|
Prompts.__init__(self, supabase_client)
|
|
Notifications.__init__(self, supabase_client)
|