llumo 0.1.8__py3-none-any.whl → 0.1.9__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,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.1.8.dist-info → llumo-0.1.9.dist-info}/METADATA

@@ -1,26 +1,26 @@
-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
+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

llumo-0.1.9.dist-info/RECORD

@@ -0,0 +1,14 @@
+llumo/.env,sha256=Vx5FkuywpYHXH2N8epJ7PlNOPiwx9UP9DUz4vWd0urs,373
+llumo/__init__.py,sha256=Ro08YEqv03qSyn54Gj8x2LUSLSZeXBQjvuODmMpqjHA,212
+llumo/client.py,sha256=9QtL8xLH0doqkJIQqhNIaR3Mlma2hkiQsylFLfaboKs,22998
+llumo/exceptions.py,sha256=bmYQYdTz2ooM0CYs3M4iHYxQrVmc46dNey_fCkiaUxo,1700
+llumo/execution.py,sha256=jSkyI2l6bowum82RDI60Tkmoe2bVnrvkJnqr8mNWjBE,1451
+llumo/functionCalling.py,sha256=_qVC987oR7nWIFmRSKDkysMgXC7tJycN6nUUniBusAc,7380
+llumo/helpingFuntions.py,sha256=_eGra7Ci79eI2e37hSIq3CMN2bgdQUTv55PydITs5NI,1468
+llumo/models.py,sha256=WBtnu7ckOy9TGRiwswz04xOGYF6EslTUOxHUz4QWzUA,1602
+llumo/sockets.py,sha256=uehnOHKqZYKek4XN5ya348OmdED-hGTXRcO8onL1XUM,5727
+llumo-0.1.9.dist-info/licenses/LICENSE,sha256=vMiqSi3KpDHq3RFxKiqdh10ZUF3PjE3nnntANU-HEu4,186
+llumo-0.1.9.dist-info/METADATA,sha256=NpdYEwl-E-gmFVSLtfEyxafcZF8ykErKEvVha2VDIdY,721
+llumo-0.1.9.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
+llumo-0.1.9.dist-info/top_level.txt,sha256=d5zUTMI99llPtLRB8rtSrqELm_bOqX-bNC5IcwlDk88,6
+llumo-0.1.9.dist-info/RECORD,,

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

@@ -1,13 +0,0 @@
-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,,