StanGirard/quivr
1
1
Fork 0
mirror of https://github.com/StanGirard/quivr.git synced 2024-09-11 21:17:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

feat(publicBrains): use join queries for better performance (#1730)

Browse Source 
Issue https://github.com/StanGirard/quivr/issues/1720
This commit is contained in:
Mamadou DICKO 2023-11-27 14:16:27 +01:00 committed by GitHub
parent 914e5b2468
commit f1ddaca7e2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
backend/models/databases/supabase/brains.py
View File

@ -96,23 +96,23 @@ class Brain(Repository):
def get_public_brains(self) -> list[PublicBrain]:
response = (
self.db.from_("brains")
.select("id:brain_id, name, description, last_update, brain_type")
.select(
"id:brain_id, name, description, last_update, brain_type, brain_definition: api_brain_definition(*), number_of_subscribers:brains_users(count)"
)
.filter("status", "eq", "public")
.execute()
)
public_brains: list[PublicBrain] = []
for item in response.data:
brain = PublicBrain(
id=item["id"],
name=item["name"],
description=item["description"],
last_update=item["last_update"],
brain_type=item["brain_type"],
brain_definition=self.get_api_brain_definition(item["id"]),
)
brain.number_of_subscribers = self.get_brain_subscribers_count(brain.id)
public_brains.append(brain)
item["number_of_subscribers"] = item["number_of_subscribers"][0]["count"]
if not item["brain_definition"]:
del item["brain_definition"]
else:
item["brain_definition"] = item["brain_definition"][0]
item["brain_definition"]["secrets"] = []
public_brains.append(PublicBrain(**item))
return public_brains
def update_brain_last_update_time(self, brain_id: UUID) -> None: