QuivrHQ/quivr
1
1
Fork 0
mirror of https://github.com/QuivrHQ/quivr.git synced 2024-12-14 17:03:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

feat(db): Add Supabase client and database instances caching (#2513)

Browse Source 
This pull request adds caching for the Supabase client and database
instances in order to improve performance and reduce unnecessary API
calls. The `get_supabase_client()` and `get_supabase_db()` functions now
check if the instances have already been created and return the cached
instances if available. This avoids creating new instances for every
function call, resulting in faster execution times.
This commit is contained in:
Stan Girard 2024-04-28 06:46:16 -07:00 committed by GitHub
parent bdb115ad0a
commit 30b9e057ff
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
backend/models/settings.py
View File

@ -1,3 +1,4 @@
from typing import Optional
from uuid import UUID from uuid import UUID
from langchain.embeddings.ollama import OllamaEmbeddings from langchain.embeddings.ollama import OllamaEmbeddings
@ -120,17 +121,26 @@ class ResendSettings(BaseSettings):
resend_api_key: str = "null" resend_api_key: str = "null"
# Global variables to store the Supabase client and database instances
_supabase_client: Optional[Client] = None
_supabase_db: Optional[SupabaseDB] = None
def get_supabase_client() -> Client: def get_supabase_client() -> Client:
settings = BrainSettings() # pyright: ignore reportPrivateUsage=none global _supabase_client
supabase_client: Client = create_client( if _supabase_client is None:
settings.supabase_url, settings.supabase_service_key settings = BrainSettings() # pyright: ignore reportPrivateUsage=none
) _supabase_client = create_client(
return supabase_client settings.supabase_url, settings.supabase_service_key
)
return _supabase_client
def get_supabase_db() -> SupabaseDB: def get_supabase_db() -> SupabaseDB:
supabase_client = get_supabase_client() global _supabase_db
return SupabaseDB(supabase_client) if _supabase_db is None:
_supabase_db = SupabaseDB(get_supabase_client())
return _supabase_db
def get_embeddings(): def get_embeddings():