File size: 568 Bytes
ad87194 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
class LimitChecker:
def __init__(self, supabase_client):
limits = self.fetch_limits(supabase_client)
self.url_limit = limits.get("total_website_url")
self.pdf_limit = limits.get("text_pdf_limit")
self.image_pdf_limit = limits.get("image_pdf_limit")
def fetch_limits(self, supabase_client):
table = supabase_client.table("ChatAI_limit_check")
results = table.select("*").execute()
return results.data[0]
def __iter__(self):
return iter((self.url_limit, self.pdf_limit, self.image_pdf_limit)) |