Reset daily message count at the start of a new day

apis/LangSmith.py

@@ -1,9 +1,8 @@
 import requests
 import json
 import os 
-from time import sleep
 from dotenv import load_dotenv
-from datetime import datetime
+from datetime import datetime, timedelta
 
 
 load_dotenv()
@@ -11,6 +10,8 @@ load_dotenv()
 API_BASE_URL = "https://api.smith.langchain.com/api/v1"
 
 class LangSmithAPI:
+    _daily_count = 0
+    _last_count_date = None
 
     def __init__(self):
         self.base_url = API_BASE_URL
@@ -20,6 +21,10 @@ class LangSmithAPI:
 
         self.getSessionId()
 
+        if self._last_count_date != datetime.today().date():
+            self._daily_count = 0
+            self._last_count_date = datetime.today().date()
+
     def getSessionId(self):
 
         payload = {}
@@ -59,14 +64,20 @@ class LangSmithAPI:
         
     
     def getDailyMessages(self, id_request: str = ""):
-
         trace = self.getLastTraces(id_request)
         
-        daily_counter = 0
+        current_date = datetime.today().date()
+        # current_date = current_date - timedelta(days=1) # Yesterday
+        
+        # Reset counter if it's a new day
+        if self._last_count_date != current_date:
+            self._daily_count = 0
+            self._last_count_date = current_date
         
+        # Count only new messages from the current day
         for t in trace:
             start_time = datetime.fromisoformat(t['start_time'])
-            if start_time.date() == datetime.today().date():
-                daily_counter += 1
+            if start_time.date() == current_date:
+                self._daily_count += 1
 
-        return daily_counter
+        return self._daily_count