llumo 0.1.6b3__py3-none-any.whl → 0.1.8__py3-none-any.whl

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,146 +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
-
-        # Initialize client
-        self.sio = socketio.Client(
-            # logger=True,
-            # engineio_logger=True,
-            reconnection=True,
-            reconnection_attempts=5,
-            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()
-
-        @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
-
-        try:
-            # print("Attempting direct WebSocket connection...")
-            # Connect with websocket transport
-            self.sio.connect(self.socket_url, transports=["websocket"], wait=True)
-
-            # print(f"Engine.IO connection established with SID: {self.sio.sid}")
-            # print( "Waiting for server to acknowledge connection with connection-established event...")
-
-            # Wait for the connection-established event
-            if not self._connection_established.wait(timeout):
-                raise RuntimeError("Timed out waiting for connection-established event")
-
-            self._last_update_time = time.time()
-            # print(f"Connection fully established. Server socket ID: {self.server_socket_id}")
-
-            # Return the server-assigned socket ID if available, otherwise fall back to the client's SID
-            return self.server_socket_id or self.sio.sid
-        except Exception as e:
-            self._connected = False
-            raise RuntimeError(f"WebSocket connection failed: {e}")
-
-    def listenForResults(self, min_wait=30, max_wait=300, inactivity_timeout=50):
-        """
-        Listen for results with improved timeout handling:
-        - min_wait: Minimum time to wait even if no data is received
-        - max_wait: Maximum total time to wait for results
-        - inactivity_timeout: Time to wait after last data received
-        """
-        if not self._connected:
-            raise RuntimeError("WebSocket is not connected. Call connect() first.")
-
-        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
-
-                # Always wait for minimum time
-                if total_elapsed < min_wait:
-                    time.sleep(0.5)
-                    continue
-
-                # Stop if maximum time exceeded
-                if total_elapsed > max_wait:
-                    # print(f"⚠️ Maximum wait time of {max_wait}s reached, stopping listener.")
-                    self._listening_done.set()
-                    break
-
-                # Stop if no activity for inactivity_timeout
-                if time_since_last_update > inactivity_timeout:
-                    # print(f"⚠️ No data received for {inactivity_timeout}s, stopping listener.")
-                    self._listening_done.set()
-                    break
-
-                # Check every second
-                time.sleep(1)
-
-        timeout_thread = threading.Thread(target=timeout_watcher, daemon=True)
-        timeout_thread.start()
-        # print("Started listening for WebSocket events...")
-        self._listening_done.wait()
-        # print(f"Finished listening. Received {len(self._received_data)} data updates.")
-
-    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.1.6b3.dist-info → llumo-0.1.8.dist-info}/METADATA

@@ -1,26 +1,26 @@
-Metadata-Version: 2.4
-Name: llumo
-Version: 0.1.6b3
-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
+Metadata-Version: 2.4
+Name: llumo
+Version: 0.1.8
+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

llumo-0.1.8.dist-info/RECORD

@@ -0,0 +1,13 @@
+llumo/__init__.py,sha256=O04b4yW1BnOvcHzxWFddAKhtdBEhBNhLdb6xgnpHH_Q,205
+llumo/client.py,sha256=Qc-LTgAiW4D8Q18oZlkNTDP8Csd_fPDI41zYaXkFD1M,22437
+llumo/exceptions.py,sha256=KGmztP61dkkzCTBEiRv5Xe9HrLNv1s_fYioRCG64GUU,1656
+llumo/execution.py,sha256=x88wQV8eL99wNN5YtjFaAMCIfN1PdfQVlAZQb4vzgQ0,1413
+llumo/functionCalling.py,sha256=QtuTtyoz5rnfNUrNT1kzegNPOrMFjrlgxZfwTqRMdiA,7190
+llumo/helpingFuntions.py,sha256=BWRoAAXG3Dsovc9bk-pDD2feQM_3ERXz_MwNgux0e7Q,1418
+llumo/models.py,sha256=YH-qAMnShmUpmKE2LQAzQdpRsaXkFSlOqMxHwU4zBUI,1560
+llumo/sockets.py,sha256=Qxxqtx3Hg07HLhA4QfcipK1ChiOYhHZBu02iA6MfYlQ,5579
+llumo-0.1.8.dist-info/licenses/LICENSE,sha256=tF9yAcfPV9xGT3ViWmC8hPvOo8BEk4ZICbUfcEo8Dlk,182
+llumo-0.1.8.dist-info/METADATA,sha256=nRQD4QA9Bf5OPisfsG4tjJwXN4fg5eADW1YuDxvZ-dg,695
+llumo-0.1.8.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
+llumo-0.1.8.dist-info/top_level.txt,sha256=d5zUTMI99llPtLRB8rtSrqELm_bOqX-bNC5IcwlDk88,6
+llumo-0.1.8.dist-info/RECORD,,

{llumo-0.1.6b3.dist-info → llumo-0.1.8.dist-info}/WHEEL

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

{llumo-0.1.6b3.dist-info → llumo-0.1.8.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.6b3.dist-info/RECORD

@@ -1,14 +0,0 @@
-llumo/.env,sha256=Vx5FkuywpYHXH2N8epJ7PlNOPiwx9UP9DUz4vWd0urs,373
-llumo/__init__.py,sha256=Ro08YEqv03qSyn54Gj8x2LUSLSZeXBQjvuODmMpqjHA,212
-llumo/client.py,sha256=hrKH97PZ6i3KMfT7a6g3KzN_pzvpN8W5oMWMn2gO4tI,23986
-llumo/exceptions.py,sha256=bmYQYdTz2ooM0CYs3M4iHYxQrVmc46dNey_fCkiaUxo,1700
-llumo/execution.py,sha256=jSkyI2l6bowum82RDI60Tkmoe2bVnrvkJnqr8mNWjBE,1451
-llumo/functionCalling.py,sha256=JwCxK_TJcFZMPP7ujqsoYfrX45XLxQfGqmyOdlwAtcM,7315
-llumo/helpingFuntions.py,sha256=_eGra7Ci79eI2e37hSIq3CMN2bgdQUTv55PydITs5NI,1468
-llumo/models.py,sha256=WBtnu7ckOy9TGRiwswz04xOGYF6EslTUOxHUz4QWzUA,1602
-llumo/sockets.py,sha256=rxeUHaxwqg1vSlplohS-aLycbW125ocqk04RiV20Ldg,5835
-llumo-0.1.6b3.dist-info/licenses/LICENSE,sha256=vMiqSi3KpDHq3RFxKiqdh10ZUF3PjE3nnntANU-HEu4,186
-llumo-0.1.6b3.dist-info/METADATA,sha256=zQ3RyaCVX1j11HraIjFvevqfJYyyR_iIG9gUu_No83A,723
-llumo-0.1.6b3.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-llumo-0.1.6b3.dist-info/top_level.txt,sha256=d5zUTMI99llPtLRB8rtSrqELm_bOqX-bNC5IcwlDk88,6
-llumo-0.1.6b3.dist-info/RECORD,,