cache

app.py

@@ -11,7 +11,6 @@ from cachetools import TTLCache, cached
 client = rg.Argilla(
     api_url=os.getenv("ARGILLA_API_URL"), api_key=os.getenv("ARGILLA_API_KEY")
 )
-
 cache = TTLCache(maxsize=100, ttl=timedelta(minutes=10), timer=datetime.now)
 
 
@@ -20,7 +19,8 @@ def fetch_data(dataset_name: str, workspace: str):
 
 
 @cached(cache)
-def get_progress(dataset) -> dict:
+def get_progress(dataset_name: str, workspace: str) -> dict:
+    dataset = fetch_data(dataset_name, workspace)
     records = list(dataset.records)
     total_records = len(records)
     annotated_records = len(
@@ -35,7 +35,8 @@ def get_progress(dataset) -> dict:
 
 
 @cached(cache)
-def get_leaderboard(dataset) -> dict:
+def get_leaderboard(dataset_name: str, workspace: str) -> dict:
+    dataset = fetch_data(dataset_name, workspace)
     user_annotations = {}
     for record in dataset.records:
         for response in record.responses:
@@ -155,9 +156,10 @@ def create_treemap(user_annotations, total_records):
 
 
 def update_dashboard():
-    dataset = fetch_data(os.getenv("DATASET_NAME"), os.getenv("WORKSPACE"))
-    progress = get_progress(dataset)
-    user_annotations = get_leaderboard(dataset)
+    dataset_name = os.getenv("DATASET_NAME")
+    workspace = os.getenv("WORKSPACE")
+    progress = get_progress(dataset_name, workspace)
+    user_annotations = get_leaderboard(dataset_name, workspace)
 
     gauge_chart = create_gauge_chart(progress)
     treemap = create_treemap(user_annotations, progress["total"])