llumo 0.2.34__tar.gz → 0.2.36__tar.gz

{llumo-0.2.34/llumo.egg-info → llumo-0.2.36}/PKG-INFO

@@ -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

{llumo-0.2.34 → llumo-0.2.36}/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-0.2.34 → llumo-0.2.36}/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-0.2.34 → llumo-0.2.36}/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-0.2.34 → llumo-0.2.36}/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-0.2.34 → llumo-0.2.36}/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 → llumo-0.2.36/llumo.egg-info}/PKG-INFO

@@ -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

{llumo-0.2.34 → llumo-0.2.36}/setup.py

@@ -39,6 +39,10 @@ def read_requirements():
             "tqdm==4.67.1",
             "google-generativeai==0.8.5",
             "websocket-client==1.8.0",
+            "pandas",
+            "python-dateutil",
+            "numpy",
+            "langchain_core",
             
         ]