Merge branch #Piloterra/ia-subvention' into 'mlcl-trading/ia-subvention'

apis/LangSmith.py

@@ -40,6 +40,10 @@ class LangSmithAPI:
 
 
     def getLastTraces(self, id_request: str = ''):
+
+        if not hasattr(self, 'session_id') or not self.session_id:
+            return 0
+        
         payload = {
             "session": [self.session_id],
             "run_type":"chain",
@@ -66,6 +70,9 @@ class LangSmithAPI:
     def getDailyMessages(self, id_request: str = ""):
         trace = self.getLastTraces(id_request)
         
+        if trace == 0:
+            return 0
+        
         current_date = datetime.today().date()
         # current_date = current_date - timedelta(days=1) # Yesterday
         

app.py

@@ -5,7 +5,7 @@ from dotenv import load_dotenv
 from rag import Rag
 from vectore_store.PineconeConnector import PineconeConnector
 from vectore_store.VectoreStoreManager import VectoreStoreManager
-from subscription.index import get_usages, is_usage_limit_reached
+from subscription.index import get_usages, is_usage_limit_reached, daily_limit_enabled
 
 from util import getYamlConfig
 
@@ -51,14 +51,16 @@ def main():
     st.logo(LOGO)
     st.title(GROUP_NAME)
     
-    usages = get_usages()
-    reached = is_usage_limit_reached()
-
-    st.sidebar.html(f"<p style='margin:0;text-align:center;font-weight:bold;font-size:24px;'>{usages['count']} / {usages['daily_limit']}</p>")
-    if reached:
-        st.sidebar.html(f"<p style='text-align:center;font-style:italic;'>Vous avez atteint la limite d'utilisation<p>")
-    else:
-        st.sidebar.html(f"<p style='text-align:center;'>Nombre de messages envoyés aujourd'hui<p>")
+    if daily_limit_enabled():
+        usages = get_usages()
+        reached = is_usage_limit_reached()
+
+        st.sidebar.html(f"<p style='margin:0;text-align:center;font-weight:bold;font-size:24px;'>{usages['count']} / {usages['daily_limit']}</p>")
+        if reached:
+            st.sidebar.html(f"<p style='text-align:center;font-style:italic;'>Vous avez atteint la limite d'utilisation<p>")
+        else:
+            st.sidebar.html(f"<p style='text-align:center;'>Nombre de messages envoyés aujourd'hui<p>")
+
     
 
     saved_documents = st.Page("pages/persistent_documents.py", title="Communs", icon="🗃️")

subscription/index.py

@@ -4,11 +4,15 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-langsmith = LangSmithAPI()
-count = langsmith.getDailyMessages()
+# Convert DAILY_LIMIT to int with a default value of 0
+DAILY_LIMIT = int(os.getenv("DAILY_MESSAGES_LIMIT", "0"))
 
-# Convert DAILY_LIMIT to int with a default value of 5
-DAILY_LIMIT = int(os.getenv("DAILY_MESSAGES_LIMIT", "5"))
+if(DAILY_LIMIT > 0):
+    langsmith = LangSmithAPI()
+    count = langsmith.getDailyMessages()
+
+def daily_limit_enabled():
+    return DAILY_LIMIT > 0
 
 def is_usage_limit_reached():
     return count >= DAILY_LIMIT