llumo 0.2.34__py3-none-any.whl → 0.2.36__py3-none-any.whl

llumo/callback.py

@@ -4,6 +4,7 @@ from langchain_core.messages import BaseMessage
 from langchain_core.outputs import LLMResult
 from langchain_core.agents import AgentAction, AgentFinish
 import json
+
 from llumo.llumoLogger import LlumoLogger
 from llumo.llumoSessionContext import LlumoSessionContext
 import time
@@ -93,7 +94,7 @@ class LlumoCallbackHandler(BaseCallbackHandler):
 
                 self.agentStartTime = time.time()
                 self.isAgentExecution = True
-                print(f"[DEBUG] Agent execution started: {self.currentAgentName} - Reset counters for new query")
+                # print(f"[DEBUG] Agent execution started: {self.currentAgentName} - Reset counters for new query")
             else:
                 self.isAgentExecution = False
 
@@ -168,6 +169,10 @@ class LlumoCallbackHandler(BaseCallbackHandler):
 
     def on_llm_end(self, response: Any, **kwargs: Any) -> None:
         """Called when LLM completes"""
+        # print("ON LLM END kwargs: ",kwargs)
+        # print("ON LLM END response: ",response)
+
+
         duration_ms = int((time.time() - self.llmStartTime) * 1000) if self.llmStartTime else 0
 
         # Initialize default values
@@ -233,7 +238,7 @@ class LlumoCallbackHandler(BaseCallbackHandler):
 
         except Exception as e:
             error_message = f"Response processing error: {str(e)}"
-            status = "ERROR"
+            # status = "ERROR"
 
         # Ensure we have string values
         output = str(output) if output is not None else ""
@@ -347,8 +352,8 @@ class LlumoCallbackHandler(BaseCallbackHandler):
 
     def on_tool_start(self, serialized: Dict[str, Any], input_str: str, **kwargs: Any) -> None:
         """Called when a tool starts executing"""
-        # print("ON TOOL START: ",serialized)
-        # print("ON TOOL START: ",kwargs)
+        # print("ON TOOL START serialized: ",serialized)
+        # print("ON TOOL START kwargs: ",kwargs)
 
         self.toolStartTime = time.time()
         self.stepTime = time.time()
@@ -376,7 +381,7 @@ class LlumoCallbackHandler(BaseCallbackHandler):
         if self.currentToolName not in self.toolsUsed:
             self.toolsUsed.append(self.currentToolName)
 
-        print(f"[DEBUG] Tool started: {self.currentToolName} with input: {input_str}")
+        # print(f"[DEBUG] Tool started: {self.currentToolName} with input: {input_str}")
 
     def on_tool_end(self, output: Any, **kwargs: Any) -> None:
         """Called when a tool completes execution"""
@@ -409,7 +414,7 @@ class LlumoCallbackHandler(BaseCallbackHandler):
                 status="SUCCESS",
                 # message="",
             )
-            print(f"[DEBUG] Tool completed: {self.currentToolName} -> {output_str}")
+            # print(f"[DEBUG] Tool completed: {self.currentToolName} -> {output_str}")
 
         except Exception as e:
             print(f"[ERROR] Failed to log tool end: {e}")
@@ -500,7 +505,7 @@ class LlumoCallbackHandler(BaseCallbackHandler):
                 toolName=self.currentToolName or "unknown",
                 description=self.currentToolDescription,
                 input=self.currentToolInput or {"input": ""},
-                output="",
+                output=f'{error}' if error else "",
                 latencyMs=0,
                 status="FAILURE",
                 # message=str(error),
@@ -557,7 +562,7 @@ class LlumoCallbackHandler(BaseCallbackHandler):
         """Called when arbitrary text is logged"""
         # Only log significant text events during agent execution
         if self.isAgentExecution and text.strip():
-            print(f"[DEBUG] Additional text: {text}")
+            # print(f"[DEBUG] Additional text: {text}")
 
             # Check if this text contains important ReAct information like "Observation:"
             if any(keyword in text.lower() for keyword in ['observation:']):

llumo/client.py

@@ -204,10 +204,10 @@ class LlumoClient:
             "Content-Type": "application/json",
         }
         try:
-            print(postUrl)
+            # print(postUrl)
             response = requests.post(postUrl, json=payload, headers=headers)
-            print(f"Post API Status Code: {response.status_code}")
-            print(response.text)
+            # print(f"Post API Status Code: {response.status_code}")
+            # print(response.text)
             # print(response.status_code)
 
         except Exception as e:
@@ -795,10 +795,10 @@ class LlumoClient:
         email = None
         try:
             socketID = self.socket.connect(timeout=250)
-            print("Socket connected with ID:", socketID)
+            # print("Socket connected with ID:", socketID)
         except Exception as e:
             socketID = "DummySocketID"
-            print(f"Socket connection failed, using dummy ID. Error: {str(e)}")
+            # print(f"Socket connection failed, using dummy ID. Error: {str(e)}")
 
         self.evalData = []
         self.evals = evals
@@ -936,10 +936,10 @@ class LlumoClient:
             
             compoundKey = f"{rowID}-{columnID}-{columnID}"
             rowIdMapping[compoundKey] = {"index": index}
-            print("rowIdMapping:", rowIdMapping)
+            # print("rowIdMapping:", rowIdMapping)
 
             # Create evaluation payload
-            print("socketID in before templateData: ", socketID)
+            # print("socketID in before templateData: ", socketID)
             templateData = {
                 "processID": getProcessID(),
                 "socketID": socketID,
@@ -997,13 +997,13 @@ class LlumoClient:
 
         rawResults = self.socket.getReceivedData()
 
-        print(f"Total results received: {len(rawResults)}")
+        # print(f"Total results received: {len(rawResults)}")
         # print("Raw results:", rawResults)
 
         # print("data from db #####################",dataFromDb)
         # Fix here: keep full keys, do not split keys
         receivedRowIDs = {key for item in rawResults for key in item.keys()}
-        print("Received Row IDs:", receivedRowIDs)
+        # print("Received Row IDs:", receivedRowIDs)
         expectedRowIDs = set(rowIdMapping.keys())
         missingRowIDs = expectedRowIDs - receivedRowIDs
         # print("All expected keys:", expectedRowIDs)
@@ -1011,13 +1011,13 @@ class LlumoClient:
         # print("Missing keys:", len(missingRowIDs))
         missingRowIDs = list(missingRowIDs)
 
-        print("Missing Row IDs:", missingRowIDs)
-        print(f"Total results before fetching missing data: {len(rawResults)}")
+        # print("Missing Row IDs:", missingRowIDs)
+        # print(f"Total results before fetching missing data: {len(rawResults)}")
         if len(missingRowIDs) > 0:
             print('''It's taking longer than expected to get results for some rows. You can close this now. 
-                  Please wait for 15 mins while we create the flow graph for you. You can check the graph at app.llumo.ai/debugger''')
+                  Please wait for 15 mins while we create the flow graph for you. You can check the graph at app.llumo.ai/debugging''')
         else:
-            print('''All results received successfully. You can check flowgraph in 5 mins at app.llumo.ai/debugger''')
+            print('''All results received successfully. You can check flowgraph in 5 mins at app.llumo.ai/debugging''')
         # if len(missingRowIDs) > 0:
         #     dataFromDb = self.fetchDataForMissingKeys(workspaceID, missingRowIDs)
         #     # print("Fetched missing data from DB:", dataFromDb)

llumo/llumoLogger.py

@@ -23,6 +23,12 @@ class LlumoLogger:
                 timeout=10,
             )
 
+            if response.status_code == 401:
+                # Wrong API key
+                print("❌ SDK integration failed! ")
+                raise Exception("Your Llumo API key is Invalid. Try again.")
+
+
             response.raise_for_status()
             res_json = response.json()
 
@@ -33,19 +39,19 @@ class LlumoLogger:
             self.playgroundID = inner_data.get("playgroundID")
             self.userEmailID = inner_data.get("createdBy")
 
-            if not self.workspaceID or not self.playgroundID:
-                raise RuntimeError(
-                    f"Invalid response: workspaceID or playgroundID missing. Full response: {res_json}"
-                )
-
+            # if not self.workspaceID or not self.playgroundID:
+            #     raise RuntimeError(
+            #         f"Invalid response: workspaceID or playgroundID missing. Full response: {res_json}"
+            #     )
+            print("✅ SDK integration successful! ")
         except requests.exceptions.RequestException as req_err:
             raise RuntimeError(
                 f"Network or HTTP error during authentication: {req_err}"
             )
-        except ValueError as json_err:
-            raise RuntimeError(f"Invalid JSON in authentication response: {json_err}")
-        except Exception as e:
-            raise RuntimeError(f"Authentication failed: {e}")
+        # except ValueError as json_err:
+        #     raise RuntimeError(f"Invalid JSON in authentication response: {json_err}")
+        # except Exception as e:
+        #     raise RuntimeError(f"Authentication failed: {e}")
 
     def getWorkspaceID(self):
         return self.workspaceID

llumo/llumoSessionContext.py

@@ -114,7 +114,7 @@ class LlumoSessionContext(LlumoClient):
         }
 
         try:
-            print(run)
+            # print(run)
             response = requests.post(url, headers=headers, json=run, timeout=20)
             response.raise_for_status()
             # print(response.json())

llumo/sockets.py

@@ -110,7 +110,8 @@ class LlumoSocketClient:
         except Exception as e:
             # print(f"[DEBUG] Connection failed with error: {e}")
             self._connected = False
-            # raise RuntimeError(f"WebSocket connection failed: {e}")
+            # raise RuntimeError(f"WebSocket
+            # connection failed: {e}")
             print("It seems your internet connection is a bit unstable. This might take a little longer than usual—thanks for your patience!")
 
     def listenForResults(
@@ -150,7 +151,7 @@ class LlumoSocketClient:
 
     def getReceivedData(self):
         with self._lock:
-            print("Total received:", len(self._received_data))  # DEBUG
+            # print("Total received:", len(self._received_data))  # DEBUG
             return self._received_data.copy()
 
     def disconnect(self):

{llumo-0.2.34.dist-info → llumo-0.2.36.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: llumo
-Version: 0.2.34
+Version: 0.2.36
 Summary: Python SDK for interacting with the Llumo ai API.
 Home-page: https://www.llumo.ai/
 Author: Llumo
@@ -24,6 +24,10 @@ Requires-Dist: openai==1.75.0
 Requires-Dist: tqdm==4.67.1
 Requires-Dist: google-generativeai==0.8.5
 Requires-Dist: websocket-client==1.8.0
+Requires-Dist: pandas
+Requires-Dist: python-dateutil
+Requires-Dist: numpy
+Requires-Dist: langchain_core
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{llumo-0.2.34.dist-info → llumo-0.2.36.dist-info}/RECORD

@@ -1,20 +1,20 @@
 llumo/__init__.py,sha256=kkuppu7ZPiVZFdnYzJ9BM3syMbYHOSZLpwKwAvGHsnY,311
-llumo/callback.py,sha256=K8O_bXgeIOXr5gougWF2Y7wnXTf7c3gSEvJZFiVoCmA,23829
+llumo/callback.py,sha256=6y9TeVD8qjEXSsLUsIvkK29hbzgIefpBngyA_u9y9LU,23981
 llumo/callbacks-0.py,sha256=TEIOCWRvk2UYsTmBMBsnlgpqWvr-2y3a6d0w_e96NRM,8958
 llumo/chains.py,sha256=6lCgLseh04RUgc6SahhmvQj82quay2Mi1j8gPUlx8Es,2923
-llumo/client.py,sha256=ZFey1s56-bdYEUQwDfH9crx4ylNt7XHZFUI0WGofjmY,73029
+llumo/client.py,sha256=EmgnIlVVs83Cne-4i5liyokaY3QEmmDOiQnh4oZyHD0,73053
 llumo/exceptions.py,sha256=1OyhN9YL9LcyUPUsqYHq6Rret0udATZAwMVJaio2_Ec,2123
 llumo/execution.py,sha256=nWbJ7AvWuUPcOb6i-JzKRna_PvF-ewZTiK8skS-5n3w,1380
 llumo/functionCalling.py,sha256=D5jYapu1rIvdIJNUYPYMTyhQ1H-6nkwoOLMi6eekfUE,7241
 llumo/google.py,sha256=6y9YnDFDRHv6-sQNT5LIsV9p31BCN0B9eow5KTRBWfM,2185
 llumo/helpingFuntions.py,sha256=jhB14o5e0YuRp-lCnu1c4vXjoG_y8ZinFHxUrZsnZAk,27284
-llumo/llumoLogger.py,sha256=grdjhu6Ngxg7nhnrMOP5Pd5ALR7U2ROws48yhf_N7y0,1912
-llumo/llumoSessionContext.py,sha256=s1yu8dP_lrVPdiPQdg80n7Pm_IXoc6_XR2-6oBq5UtI,12585
+llumo/llumoLogger.py,sha256=7DZR2_QHy0fencng9Nnf9UPmEx8-OZzhvz1QRUp9w6c,2190
+llumo/llumoSessionContext.py,sha256=si7T66D4bsea9vrCaQiko3ZcBcx7zy9W0lf0bpitaLY,12587
 llumo/models.py,sha256=aVEZsOOoQx5LeNtwSyBxqvrINq0izH3QWu_YjsMPE6o,2910
 llumo/openai.py,sha256=VstBzaORe8Tq0feUIIEszzcN1oq6TJfkPviaCr5d3Bw,8950
-llumo/sockets.py,sha256=2zp9xzIV46naR3kuotroIvZfJp13TTxpgg2iSxtJofU,6114
-llumo-0.2.34.dist-info/licenses/LICENSE,sha256=tF9yAcfPV9xGT3ViWmC8hPvOo8BEk4ZICbUfcEo8Dlk,182
-llumo-0.2.34.dist-info/METADATA,sha256=3eiB92kNLh_IvOlesPckMDo92v3kPTa5qhNJndbGrHQ,1558
-llumo-0.2.34.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-llumo-0.2.34.dist-info/top_level.txt,sha256=d5zUTMI99llPtLRB8rtSrqELm_bOqX-bNC5IcwlDk88,6
-llumo-0.2.34.dist-info/RECORD,,
+llumo/sockets.py,sha256=pfWz1zTEiwqJhdbSy3i3_Y4WlIdJ3cuac11wMePeBS0,6130
+llumo-0.2.36.dist-info/licenses/LICENSE,sha256=tF9yAcfPV9xGT3ViWmC8hPvOo8BEk4ZICbUfcEo8Dlk,182
+llumo-0.2.36.dist-info/METADATA,sha256=-lPlVSKmlLUjVHpbufy8N4X5b3_247Semgcv7boLfHI,1662
+llumo-0.2.36.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+llumo-0.2.36.dist-info/top_level.txt,sha256=d5zUTMI99llPtLRB8rtSrqELm_bOqX-bNC5IcwlDk88,6
+llumo-0.2.36.dist-info/RECORD,,