llumo 0.2.5__py3-none-any.whl → 0.2.7__py3-none-any.whl

llumo/client.py

@@ -1,16 +1,18 @@
 import requests
 
-from .exceptions import LlumoAIError
+
 import time
 import re
 import json
 import uuid
+
 import os
 import itertools
 import pandas as pd
 from typing import List, Dict
 from .models import AVAILABLEMODELS, getProviderFromModel
 from .execution import ModelExecutor
+from .exceptions import LlumoAIError
 from .helpingFuntions import *
 from .sockets import LlumoSocketClient
 from .functionCalling import LlumoAgentExecutor
@@ -29,7 +31,7 @@ class LlumoClient:
         self.socket = LlumoSocketClient(socketUrl)
         self.processMapping = {}
 
-    def validateApiKey(self, evalName=""):
+    def validateApiKey(self, evalName=" "):
         headers = {
             "Authorization": f"Bearer {self.apiKey}",
             "Content-Type": "application/json",

llumo/helpingFuntions.py

@@ -106,23 +106,28 @@ def checkUserHits(
         if subscriptionEndDate and current_date > subscriptionEndDate:
             return {"success": False, "message": "Subscription expired. Access denied."}
 
-        if remainingHits <= 0 or remainingHits < datasetLength:
-            headers = {
-                "Authorization": f"Bearer {base64.b64encode(workspaceID.encode()).decode()}",
-                "Content-Type": "application/json",
-            }
-            reqBody = {"unitsToSet": 1}
-            responseBody = requests.post(
-                url=subscriptionUrl, json=reqBody, headers=headers
-            )
-            response = json.loads(responseBody.text)
-
-            proceed = response.get("execution", "")
-            print(proceed)
-
-            if proceed:
-                return {"success": True, "message": "Hits added and access granted."}
 
+        
+        if remainingHits <= 0 or remainingHits < datasetLength:
+            if workspaceID:
+                workspaceID=str(workspaceID)
+                headers = {
+                    "Authorization": f"Bearer {base64.b64encode(workspaceID.encode()).decode()}",
+                    "Content-Type": "application/json",
+                }
+                reqBody = {"unitsToSet": 1}
+                responseBody = requests.post(
+                    url=subscriptionUrl, json=reqBody, headers=headers
+                )
+                response = json.loads(responseBody.text)
+
+                proceed = response.get("execution", "")
+                print(proceed)
+
+                if proceed:
+                    return {"success": True, "message": "Hits added and access granted."}
+            else:
+                return {"success": False, "message": "Workspace ID is required for subscription."}
     return {"success": True, "message": "Access granted."}
 
 

{llumo-0.2.5.dist-info → llumo-0.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: llumo
-Version: 0.2.5
+Version: 0.2.7
 Summary: Python SDK for interacting with the Llumo ai API.
 Home-page: https://www.llumo.ai/
 Author: Llumo

{llumo-0.2.5.dist-info → llumo-0.2.7.dist-info}/RECORD

@@ -1,13 +1,13 @@
 llumo/__init__.py,sha256=O04b4yW1BnOvcHzxWFddAKhtdBEhBNhLdb6xgnpHH_Q,205
-llumo/client.py,sha256=aCYeUrmEssSqydA_p3nokJ9MkspgitJ0HTmzL7SpkQw,35540
+llumo/client.py,sha256=8Ne-Dkr0ftT9f1KfWp72bea4ae2RrVJ3FvM4Xs7AFrM,35543
 llumo/exceptions.py,sha256=iCj7HhtO_ckC2EaVBdXbAudNpuMDsYmmMEV5lwynZ-E,1854
 llumo/execution.py,sha256=x88wQV8eL99wNN5YtjFaAMCIfN1PdfQVlAZQb4vzgQ0,1413
 llumo/functionCalling.py,sha256=D5jYapu1rIvdIJNUYPYMTyhQ1H-6nkwoOLMi6eekfUE,7241
-llumo/helpingFuntions.py,sha256=G_pqLhYNH3bZ47gA--w6YFNvB443xkrbh8CeQfYxazk,8839
+llumo/helpingFuntions.py,sha256=lG_d3lQgJj6pI7v1YdLqdPojrLCNwybKz29zXrGaL5k,9090
 llumo/models.py,sha256=YH-qAMnShmUpmKE2LQAzQdpRsaXkFSlOqMxHwU4zBUI,1560
 llumo/sockets.py,sha256=Qxxqtx3Hg07HLhA4QfcipK1ChiOYhHZBu02iA6MfYlQ,5579
-llumo-0.2.5.dist-info/licenses/LICENSE,sha256=tF9yAcfPV9xGT3ViWmC8hPvOo8BEk4ZICbUfcEo8Dlk,182
-llumo-0.2.5.dist-info/METADATA,sha256=luOkLN_md2iGVCEBTGs2jZybFrko_XKvmF_Ny04txjg,1490
-llumo-0.2.5.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-llumo-0.2.5.dist-info/top_level.txt,sha256=d5zUTMI99llPtLRB8rtSrqELm_bOqX-bNC5IcwlDk88,6
-llumo-0.2.5.dist-info/RECORD,,
+llumo-0.2.7.dist-info/licenses/LICENSE,sha256=tF9yAcfPV9xGT3ViWmC8hPvOo8BEk4ZICbUfcEo8Dlk,182
+llumo-0.2.7.dist-info/METADATA,sha256=VDeZQ64GPuvyozUR0sui4syLE98RiRm3gUzeLmF8cZg,1490
+llumo-0.2.7.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+llumo-0.2.7.dist-info/top_level.txt,sha256=d5zUTMI99llPtLRB8rtSrqELm_bOqX-bNC5IcwlDk88,6
+llumo-0.2.7.dist-info/RECORD,,