llumo 0.1.9__py3-none-any.whl → 0.2.0__py3-none-any.whl

llumo/helpingFuntions.py

@@ -1,50 +1,137 @@
-import time
-import uuid
-import numpy as np
-
-def getProcessID():
-    return f"{int(time.time() * 1000)}{uuid.uuid4()}"
-
-
-def getInputPopulatedPrompt(promptTemplate, tempObj):
-    for key, value in tempObj.items():
-        promptTemplate = promptTemplate.replace(f"{{{{{key}}}}}", value)
-    return promptTemplate
-
-def costColumnMapping(costResults, allProcess):
-    # this dict will store cost column data for each row
-    cost_cols = {}
-
-    compressed_prompt = []
-    compressed_prompt_output = []
-    cost = []
-    cost_saving = []
-
-    for record in allProcess:
-        cost_cols[record] = []
-        for item in costResults:
-            if list(item.keys())[0].split("-")[0] == record.split("-")[0]:
-                cost_cols[record].append(list(item.values())[0])
-
-    for ky, val in cost_cols.items():
-        try:
-            compressed_prompt.append(val[0])
-        except IndexError:
-            compressed_prompt.append("error occured")
-
-        try:
-            compressed_prompt_output.append(val[1])
-        except IndexError:
-            compressed_prompt_output.append("error occured")
-
-        try:
-            cost.append(val[2])
-        except IndexError:
-            cost.append("error occured")
-
-        try:
-            cost_saving.append(val[3])
-        except IndexError:
-            cost_saving.append("error occured")
-
-    return compressed_prompt, compressed_prompt_output, cost, cost_saving
+import time
+import uuid
+import numpy as np
+from  datetime import datetime
+from dateutil import parser
+import requests
+import json
+import base64
+import os
+from dotenv import load_dotenv
+load_dotenv()
+
+subscriptionUrl = os.getenv("SUBSCRIPTION_URL")
+getStreamdataUrl = os.getenv("DATA_STREAM_URL")
+
+
+
+def getProcessID():
+    return f"{int(time.time() * 1000)}{uuid.uuid4()}"
+
+
+def getInputPopulatedPrompt(promptTemplate, tempObj):
+    for key, value in tempObj.items():
+        promptTemplate = promptTemplate.replace(f"{{{{{key}}}}}", value)
+    return promptTemplate
+
+def costColumnMapping(costResults, allProcess):
+    # this dict will store cost column data for each row
+    cost_cols = {}
+
+    compressed_prompt = []
+    compressed_prompt_output = []
+    cost = []
+    cost_saving = []
+
+    for record in allProcess:
+        cost_cols[record] = []
+        for item in costResults:
+            if list(item.keys())[0].split("-")[0] == record.split("-")[0]:
+                cost_cols[record].append(list(item.values())[0])
+
+    for ky, val in cost_cols.items():
+        try:
+            compressed_prompt.append(val[0])
+        except IndexError:
+            compressed_prompt.append("error occured")
+
+        try:
+            compressed_prompt_output.append(val[1])
+        except IndexError:
+            compressed_prompt_output.append("error occured")
+
+        try:
+            cost.append(val[2])
+        except IndexError:
+            cost.append("error occured")
+
+        try:
+            cost_saving.append(val[3])
+        except IndexError:
+            cost_saving.append("error occured")
+
+    return compressed_prompt, compressed_prompt_output, cost, cost_saving
+
+
+def checkUserHits(workspaceID, hasSubscribed, trialEndDate, subscriptionEndDate, remainingHits, datasetLength):
+    # Get the current date (only the date part)
+    current_date = datetime.now().date()
+
+    # Parse trialEndDate if provided
+    if trialEndDate is not None:
+        try:
+            trialEndDate = parser.parse(trialEndDate).date()
+        except Exception:
+            return {"success": False, "message": "Invalid trialEndDate format"}
+
+    # Parse subscriptionEndDate if provided
+    if subscriptionEndDate is not None:
+        try:
+            subscriptionEndDate = parser.parse(subscriptionEndDate).date()
+        except Exception:
+            return {"success": False, "message": "Invalid subscriptionEndDate format"}
+
+    # If user is on a free trial
+    if not hasSubscribed and trialEndDate is not None:
+        if current_date > trialEndDate:
+            return {"success": False, "message": "Trial expired. Access denied"}
+
+        if remainingHits < datasetLength or remainingHits <= 0:
+            return {"success": False, "message": "Trial Hits Exhausted"}
+
+    else:
+        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."}
+
+    return {"success": True, "message": "Access granted."}
+
+def getStreamId(workspaceID: str, token, dataStreamName):
+    headers = {
+        "Authorization": f"Bearer {token}",
+        "Content-Type": "application/json",
+    }
+    reqBody = {"workspaceID": workspaceID}
+    response = requests.post(url=getStreamdataUrl, json=reqBody, headers=headers)
+
+    if response.status_code == 200:
+        responseJson = response.json()
+        data = responseJson.get("data", [])
+
+        # Find stream by name
+        matchedStream = next((stream for stream in data if stream.get("name") == dataStreamName), None)
+
+        if matchedStream:
+            
+            return matchedStream.get("dataStreamID")
+            
+        else:
+            print(f"No stream found with name: {dataStreamName}")
+            return None
+    else:
+        print("Error:", response.status_code, response.text)
+        return None

llumo/models.py

@@ -1,43 +1,43 @@
-from enum import Enum
-
-class Provider(str, Enum):
-    OPENAI = "OPENAI"
-    GOOGLE = "GOOGLE"
-
-# Maps model aliases → (provider, actual model name for API)
-_MODEL_METADATA = {
-    "GPT_4": (Provider.OPENAI, "gpt-4"),
-    "GPT_4_32K": (Provider.OPENAI, "gpt-4-32k"),
-    "GPT_35T": (Provider.OPENAI, "gpt-3.5-turbo"),
-    "GPT_35T_INS": (Provider.OPENAI, "gpt-3.5-turbo-instruct"),
-    "GPT_35T_16K": (Provider.OPENAI, "gpt-3.5-turbo-16k"),
-    "GPT_35_TURBO": (Provider.OPENAI, "gpt-3.5-turbo"),
-
-    "GOOGLE_15_FLASH": (Provider.GOOGLE, "gemini-1.5-flash-latest"),
-    "GEMINI_PRO": (Provider.GOOGLE, "gemini-pro"),
-    "TEXT_BISON": (Provider.GOOGLE, "text-bison-001"),
-    "CHAT_BISON": (Provider.GOOGLE, "chat-bison-001"),
-    "TEXT_BISON_32K": (Provider.GOOGLE, "text-bison-32k"),
-    "TEXT_UNICORN": (Provider.GOOGLE, "text-unicorn-experimental"),
-}
-
-class AVAILABLEMODELS(str, Enum):
-    GPT_4 = "gpt-4"
-    GPT_4_32K = "gpt-4-32k"
-    GPT_35T = "gpt-3.5-turbo"
-    GPT_35T_INS = "gpt-3.5-turbo-instruct"
-    GPT_35T_16K = "gpt-3.5-turbo-16k"
-    GPT_35_TURBO = "gpt-3.5-turbo"
-
-    GOOGLE_15_FLASH = "gemini-1.5-flash-latest"
-    GEMINI_PRO = ""
-    TEXT_BISON = "text-bison-001"
-    CHAT_BISON = "chat-bison-001"
-    TEXT_BISON_32K = "text-bison-32k"
-    TEXT_UNICORN = "text-unicorn-experimental"
-
-def getProviderFromModel(model: AVAILABLEMODELS) -> Provider:
-    for alias, (provider, apiName) in _MODEL_METADATA.items():
-        if model.value == apiName:
-            return provider
+from enum import Enum
+
+class Provider(str, Enum):
+    OPENAI = "OPENAI"
+    GOOGLE = "GOOGLE"
+
+# Maps model aliases → (provider, actual model name for API)
+_MODEL_METADATA = {
+    "GPT_4": (Provider.OPENAI, "gpt-4"),
+    "GPT_4_32K": (Provider.OPENAI, "gpt-4-32k"),
+    "GPT_35T": (Provider.OPENAI, "gpt-3.5-turbo"),
+    "GPT_35T_INS": (Provider.OPENAI, "gpt-3.5-turbo-instruct"),
+    "GPT_35T_16K": (Provider.OPENAI, "gpt-3.5-turbo-16k"),
+    "GPT_35_TURBO": (Provider.OPENAI, "gpt-3.5-turbo"),
+
+    "GOOGLE_15_FLASH": (Provider.GOOGLE, "gemini-1.5-flash-latest"),
+    "GEMINI_PRO": (Provider.GOOGLE, "gemini-pro"),
+    "TEXT_BISON": (Provider.GOOGLE, "text-bison-001"),
+    "CHAT_BISON": (Provider.GOOGLE, "chat-bison-001"),
+    "TEXT_BISON_32K": (Provider.GOOGLE, "text-bison-32k"),
+    "TEXT_UNICORN": (Provider.GOOGLE, "text-unicorn-experimental"),
+}
+
+class AVAILABLEMODELS(str, Enum):
+    GPT_4 = "gpt-4"
+    GPT_4_32K = "gpt-4-32k"
+    GPT_35T = "gpt-3.5-turbo"
+    GPT_35T_INS = "gpt-3.5-turbo-instruct"
+    GPT_35T_16K = "gpt-3.5-turbo-16k"
+    GPT_35_TURBO = "gpt-3.5-turbo"
+
+    GOOGLE_15_FLASH = "gemini-1.5-flash-latest"
+    GEMINI_PRO = ""
+    TEXT_BISON = "text-bison-001"
+    CHAT_BISON = "chat-bison-001"
+    TEXT_BISON_32K = "text-bison-32k"
+    TEXT_UNICORN = "text-unicorn-experimental"
+
+def getProviderFromModel(model: AVAILABLEMODELS) -> Provider:
+    for alias, (provider, apiName) in _MODEL_METADATA.items():
+        if model.value == apiName:
+            return provider
     raise ValueError(f"Provider not found for model: {model}")

llumo/sockets.py

@@ -1,148 +1,148 @@
-import socketio
-import threading
-import time
-
-
-class LlumoSocketClient:
-    def __init__(self, socket_url):
-        self.socket_url = socket_url
-        self._received_data = []
-        self._last_update_time = None
-        self._listening_done = threading.Event()
-        self._connection_established = threading.Event()
-        self._lock = threading.Lock()
-        self._connected = False
-        self.server_socket_id = None  # Store the server-assigned socket ID
-        self._expected_results = None  # NEW
-
-        # Initialize client
-        self.sio = socketio.Client(
-            # logger=True,
-            # engineio_logger=True,
-            reconnection=True,
-            reconnection_attempts=10,
-            reconnection_delay=1,
-        )
-
-        @self.sio.on("connect")
-        def on_connect():
-            # print("Socket connection established")
-            self._connected = True
-            # Don't set connection_established yet - wait for server confirmation
-
-        # Listen for the connection-established event from the server
-        @self.sio.on("connection-established")
-        def on_connection_established(data):
-            # print(
-            #     f"Server acknowledged connection with 'connection-established' event: {data}"
-            # )
-            if isinstance(data, dict) and "socketId" in data:
-                self.server_socket_id = data["socketId"]
-                # print(f"Received server socket ID: {self.server_socket_id}")
-            self._connection_established.set()
-
-        @self.sio.on("result-update")
-        def on_result_update(data):
-            with self._lock:
-                # print(f"Received result-update event: {data}")
-                self._received_data.append(data)
-                self._last_update_time = time.time()
-
-                # ✅ Stop if all expected results are received
-                if self._expected_results and len(self._received_data) >= self._expected_results:
-                    # print("✅ All expected results received.")
-                    self._listening_done.set()
-
-        @self.sio.on("disconnect")
-        def on_disconnect():
-            # print("Socket disconnected")
-            self._connected = False
-
-        @self.sio.on("connect_error")
-        def on_connect_error(error):
-            print(f"Socket connection error: {error}")
-
-        @self.sio.on("error")
-        def on_error(error):
-            print(f"Socket error event: {error}")
-
-    def connect(self, timeout=20):
-        self._received_data = []
-        self._connection_established.clear()
-        self._listening_done.clear()
-        self.server_socket_id = None
-        self._connected = False
-
-        try:
-            # print("[DEBUG] Connecting to socket...")
-            self.sio.connect(self.socket_url, transports=["websocket"], wait=True)
-
-            # Wait for socket connection
-            start = time.time()
-            while not self.sio.connected:
-                if time.time() - start > timeout:
-                    raise RuntimeError("Timed out waiting for low-level socket connection.")
-                time.sleep(0.1)
-            # print("[DEBUG] SocketIO low-level connection established.")
-
-            # Wait for server "connection-established" event
-            if not self._connection_established.wait(timeout):
-                raise RuntimeError("Timed out waiting for connection-established event.")
-
-            self._connected = True
-            self._last_update_time = time.time()
-            # print(f"[DEBUG] Full connection established. Connected: {self._connected}")
-
-            return self.server_socket_id or self.sio.sid
-
-        except Exception as e:
-            # print(f"[DEBUG] Connection failed with error: {e}")
-            self._connected = False
-            raise RuntimeError(f"WebSocket connection failed: {e}")
-
-    def listenForResults(self, min_wait=30, max_wait=300, inactivity_timeout=50, expected_results=None):
-        if not self._connected:
-            raise RuntimeError("WebSocket is not connected. Call connect() first.")
-
-        self._expected_results = expected_results  # NEW
-        start_time = time.time()
-        self._last_update_time = time.time()
-
-        def timeout_watcher():
-            while not self._listening_done.is_set():
-                current_time = time.time()
-                time_since_last_update = current_time - self._last_update_time
-                total_elapsed = current_time - start_time
-
-                if total_elapsed < min_wait:
-                    time.sleep(0.5)
-                    continue
-
-                if total_elapsed > max_wait:
-                    # print(f"⚠️ Max wait time {max_wait}s exceeded.")
-                    self._listening_done.set()
-                    break
-
-                if time_since_last_update > inactivity_timeout:
-                    # print(f"⚠️ Inactivity timeout {inactivity_timeout}s exceeded.")
-                    self._listening_done.set()
-                    break
-
-                time.sleep(1)
-
-        timeout_thread = threading.Thread(target=timeout_watcher, daemon=True)
-        timeout_thread.start()
-        self._listening_done.wait()
-
-    def getReceivedData(self):
-        with self._lock:
-            return self._received_data.copy()
-
-    def disconnect(self):
-        try:
-            if self._connected:
-                self.sio.disconnect()
-                self._connected = False
-                # print("WebSocket client disconnected")
-        except Exception as e:
-            print(f"Error during WebSocket disconnect: {e}")
+import socketio
+import threading
+import time
+
+
+class LlumoSocketClient:
+    def __init__(self, socket_url):
+        self.socket_url = socket_url
+        self._received_data = []
+        self._last_update_time = None
+        self._listening_done = threading.Event()
+        self._connection_established = threading.Event()
+        self._lock = threading.Lock()
+        self._connected = False
+        self.server_socket_id = None  # Store the server-assigned socket ID
+        self._expected_results = None  # NEW
+
+        # Initialize client
+        self.sio = socketio.Client(
+            # logger=True,
+            # engineio_logger=True,
+            reconnection=True,
+            reconnection_attempts=10,
+            reconnection_delay=1,
+        )
+
+        @self.sio.on("connect")
+        def on_connect():
+            # print("Socket connection established")
+            self._connected = True
+            # Don't set connection_established yet - wait for server confirmation
+
+        # Listen for the connection-established event from the server
+        @self.sio.on("connection-established")
+        def on_connection_established(data):
+            # print(
+            #     f"Server acknowledged connection with 'connection-established' event: {data}"
+            # )
+            if isinstance(data, dict) and "socketId" in data:
+                self.server_socket_id = data["socketId"]
+                # print(f"Received server socket ID: {self.server_socket_id}")
+            self._connection_established.set()
+
+        @self.sio.on("result-update")
+        def on_result_update(data):
+            with self._lock:
+                # print(f"Received result-update event: {data}")
+                self._received_data.append(data)
+                self._last_update_time = time.time()
+
+                # ✅ Stop if all expected results are received
+                if self._expected_results and len(self._received_data) >= self._expected_results:
+                    # print("✅ All expected results received.")
+                    self._listening_done.set()
+
+        @self.sio.on("disconnect")
+        def on_disconnect():
+            # print("Socket disconnected")
+            self._connected = False
+
+        @self.sio.on("connect_error")
+        def on_connect_error(error):
+            print(f"Socket connection error: {error}")
+
+        @self.sio.on("error")
+        def on_error(error):
+            print(f"Socket error event: {error}")
+
+    def connect(self, timeout=20):
+        self._received_data = []
+        self._connection_established.clear()
+        self._listening_done.clear()
+        self.server_socket_id = None
+        self._connected = False
+
+        try:
+            # print("[DEBUG] Connecting to socket...")
+            self.sio.connect(self.socket_url, transports=["websocket"], wait=True)
+
+            # Wait for socket connection
+            start = time.time()
+            while not self.sio.connected:
+                if time.time() - start > timeout:
+                    raise RuntimeError("Timed out waiting for low-level socket connection.")
+                time.sleep(0.1)
+            # print("[DEBUG] SocketIO low-level connection established.")
+
+            # Wait for server "connection-established" event
+            if not self._connection_established.wait(timeout):
+                raise RuntimeError("Timed out waiting for connection-established event.")
+
+            self._connected = True
+            self._last_update_time = time.time()
+            # print(f"[DEBUG] Full connection established. Connected: {self._connected}")
+
+            return self.server_socket_id or self.sio.sid
+
+        except Exception as e:
+            # print(f"[DEBUG] Connection failed with error: {e}")
+            self._connected = False
+            raise RuntimeError(f"WebSocket connection failed: {e}")
+
+    def listenForResults(self, min_wait=30, max_wait=300, inactivity_timeout=50, expected_results=None):
+        if not self._connected:
+            raise RuntimeError("WebSocket is not connected. Call connect() first.")
+
+        self._expected_results = expected_results  # NEW
+        start_time = time.time()
+        self._last_update_time = time.time()
+
+        def timeout_watcher():
+            while not self._listening_done.is_set():
+                current_time = time.time()
+                time_since_last_update = current_time - self._last_update_time
+                total_elapsed = current_time - start_time
+
+                if total_elapsed < min_wait:
+                    time.sleep(0.5)
+                    continue
+
+                if total_elapsed > max_wait:
+                    # print(f"⚠️ Max wait time {max_wait}s exceeded.")
+                    self._listening_done.set()
+                    break
+
+                if time_since_last_update > inactivity_timeout:
+                    # print(f"⚠️ Inactivity timeout {inactivity_timeout}s exceeded.")
+                    self._listening_done.set()
+                    break
+
+                time.sleep(1)
+
+        timeout_thread = threading.Thread(target=timeout_watcher, daemon=True)
+        timeout_thread.start()
+        self._listening_done.wait()
+
+    def getReceivedData(self):
+        with self._lock:
+            return self._received_data.copy()
+
+    def disconnect(self):
+        try:
+            if self._connected:
+                self.sio.disconnect()
+                self._connected = False
+                # print("WebSocket client disconnected")
+        except Exception as e:
+            print(f"Error during WebSocket disconnect: {e}")

llumo-0.2.0.dist-info/METADATA

@@ -0,0 +1,17 @@
+Metadata-Version: 2.4
+Name: llumo
+Version: 0.2.0
+Summary: Python SDK for interacting with the Llumo ai API.
+Home-page: https://www.llumo.ai/
+Author: Llumo
+Author-email: product@llumo.ai
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Dynamic: author
+Dynamic: author-email
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-python
+Dynamic: summary

llumo-0.2.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+llumo/__init__.py,sha256=O04b4yW1BnOvcHzxWFddAKhtdBEhBNhLdb6xgnpHH_Q,205
+llumo/client.py,sha256=F4zabvAjLnu0N61qSw5DqerNlV5ybC2DbxaI55olldg,31916
+llumo/exceptions.py,sha256=iCj7HhtO_ckC2EaVBdXbAudNpuMDsYmmMEV5lwynZ-E,1854
+llumo/execution.py,sha256=x88wQV8eL99wNN5YtjFaAMCIfN1PdfQVlAZQb4vzgQ0,1413
+llumo/functionCalling.py,sha256=QtuTtyoz5rnfNUrNT1kzegNPOrMFjrlgxZfwTqRMdiA,7190
+llumo/helpingFuntions.py,sha256=9w1J4wMnJV1v_5yFMyxIHcvc16sEq5MZzOFBPagij5Q,4467
+llumo/models.py,sha256=YH-qAMnShmUpmKE2LQAzQdpRsaXkFSlOqMxHwU4zBUI,1560
+llumo/sockets.py,sha256=Qxxqtx3Hg07HLhA4QfcipK1ChiOYhHZBu02iA6MfYlQ,5579
+llumo-0.2.0.dist-info/licenses/LICENSE,sha256=tF9yAcfPV9xGT3ViWmC8hPvOo8BEk4ZICbUfcEo8Dlk,182
+llumo-0.2.0.dist-info/METADATA,sha256=z8Ec84QAe1khloO38UkicxqVGukFid_3aeptYpHra0I,426
+llumo-0.2.0.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+llumo-0.2.0.dist-info/top_level.txt,sha256=d5zUTMI99llPtLRB8rtSrqELm_bOqX-bNC5IcwlDk88,6
+llumo-0.2.0.dist-info/RECORD,,

{llumo-0.1.9.dist-info → llumo-0.2.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.1.0)
+Generator: setuptools (80.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

{llumo-0.1.9.dist-info → llumo-0.2.0.dist-info}/licenses/LICENSE

@@ -1,4 +1,4 @@
-Copyright (c) 2025 Llumo AI
-
-All rights reserved. This software is proprietary and confidential. 
-Unauthorized copying, distribution, or use of this software is strictly prohibited.
+Copyright (c) 2025 Llumo AI
+
+All rights reserved. This software is proprietary and confidential. 
+Unauthorized copying, distribution, or use of this software is strictly prohibited.

llumo/.env

@@ -1,6 +0,0 @@
-
-BASE_URL="https://app.llumo.ai/api"
-postUrl = "https://red-skull-service-392377961931.us-central1.run.app/api/process-playground"
-fetchUrl = "https://red-skull-service-392377961931.us-central1.run.app/api/get-cells-data"
-validateUrl = "https://backend-api.llumo.ai/api/v1/workspace-key-details"
-SOCKET_URL="https://red-skull-service-392377961931.us-central1.run.app/"

llumo-0.1.9.dist-info/METADATA

@@ -1,26 +0,0 @@
-Metadata-Version: 2.4
-Name: llumo
-Version: 0.1.9
-Summary: Python SDK for interacting with the Llumo ai API.
-Home-page: https://www.llumo.ai/
-Author: Llumo
-Author-email: product@llumo.ai
-License: Proprietary
-Requires-Python: >=3.7
-License-File: LICENSE
-Requires-Dist: requests>=2.0.0
-Requires-Dist: websocket-client>=1.0.0
-Requires-Dist: pandas>=1.0.0
-Requires-Dist: numpy>=1.0.0
-Requires-Dist: python-socketio[client]==5.13.0
-Requires-Dist: python-dotenv==1.1.0
-Requires-Dist: openai==1.75.0
-Requires-Dist: google-generativeai==0.8.5
-Dynamic: author
-Dynamic: author-email
-Dynamic: home-page
-Dynamic: license
-Dynamic: license-file
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary