llumo 0.2.15b1__py3-none-any.whl → 0.2.15b2__py3-none-any.whl

llumo/client.py

@@ -10,7 +10,7 @@ import os
 import itertools
 import pandas as pd
 from typing import List, Dict
-from .models import AVAILABLEMODELS, getProviderFromModel
+from .models import AVAILABLEMODELS, getProviderFromModel, Provider
 from .execution import ModelExecutor
 from .exceptions import LlumoAIError
 from .helpingFuntions import *
@@ -50,7 +50,7 @@ class LlumoClient:
         reqBody = {"analytics": [evalName]}
 
         try:
-            print(reqBody)
+            
             response = requests.post(url=validateUrl, json=reqBody, headers=headers)
 
         except requests.exceptions.RequestException as e:
@@ -1151,7 +1151,7 @@ class LlumoClient:
                 # Validate API and dependencies
                 self.validateApiKey(evalName=evalName)
                 metricDependencies = checkDependency(
-                    evalName, list(working_df.columns), tocheck=True
+                    evalName, list(working_df.columns), tocheck=False
                 )
                 if not metricDependencies["status"]:
                     raise LlumoAIError.dependencyError(metricDependencies["message"])
@@ -1189,12 +1189,20 @@ class LlumoClient:
             inputDict = {key: row[key] for key in inputVariables}
             for i, model in enumerate(modelAliases, 1):
                 try:
+
                     provider = getProviderFromModel(model)
+                    if provider == Provider.OPENAI:
+                        print(validateOpenaiKey(apiKey))
+                    elif provider == Provider.GOOGLE:
+                        validateGoogleKey(apiKey)
+
                     filled_template = getInputPopulatedPrompt(prompt_template, inputDict)
                     response = executor.execute(provider, model.value, filled_template, apiKey)
                     df.at[indx, f"output_{i}"] = response
                 except Exception as e:
-                    df.at[indx, f"output_{i}"] = str(e)
+                    # df.at[indx, f"output_{i}"] = str(e)
+                    raise e
+
         return df
 
     def _evaluateForStream(self, df, evals, modelAliases, prompt_template):

llumo/execution.py

@@ -25,15 +25,14 @@ class ModelExecutor:
         return response.choices[0].message.content
 
     def _executeGoogle(self, modelName: str, prompt: str,api_key) -> str:
-        
+
         # Configure GenAI with API Key
         genai.configure(api_key=api_key)
-            
+
         # Select Generative Model
         model = genai.GenerativeModel("gemini-2.0-flash-lite")
         # Generate Response
         response = model.generate_content(prompt)
         return response.text
-        
-    
- 
+
+

llumo/helpingFuntions.py

@@ -8,6 +8,8 @@ import json
 import base64
 import os
 import re
+import openai
+import google.generativeai as genai
 
 
 from .models import  _MODEL_METADATA,  AVAILABLEMODELS
@@ -378,12 +380,12 @@ def createColumn(workspaceID, dataframe, playgroundID, promptText=None,queryColN
         row_dict = {}
 
         # For each column, we need to map the column ID to the corresponding value in the row
-        print(dataframe.columns)
+        
         for col in dataframe.columns:
             columnID = columnIDMapping[col]
 
             if any(col.startswith(eval + "_") or col == eval for eval in allEvals) and not " Reason"  in col and promptText!=None:
-                print(col)
+                
                 row_dict[columnID] = {
 
                     "value": row[col],
@@ -632,3 +634,20 @@ def validateModels(model_aliases):
 
 
 
+def validateOpenaiKey(api_key):
+    try:
+        client = openai.OpenAI(api_key=api_key)
+        _ = client.models.list()  # Light call to list models
+    except openai.AuthenticationError:
+        raise ValueError("❌ Invalid OpenAI API key.")
+    except Exception as e:
+        raise RuntimeError(f"⚠️ Error validating OpenAI key: {e}")
+
+def validateGoogleKey(api_key):
+    try:
+        genai.configure(api_key=api_key)
+        _ = genai.GenerativeModel("gemini-2.0").generate_content("test")
+    except Exception as e:
+        if "PERMISSION_DENIED" in str(e) or "API key not valid" in str(e):
+            raise ValueError("❌ Invalid Google API key.")
+        raise RuntimeError(f"⚠️ Error validating Gemini key: {e}")

{llumo-0.2.15b1.dist-info → llumo-0.2.15b2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: llumo
-Version: 0.2.15b1
+Version: 0.2.15b2
 Summary: Python SDK for interacting with the Llumo ai API.
 Home-page: https://www.llumo.ai/
 Author: Llumo

llumo-0.2.15b2.dist-info/RECORD

@@ -0,0 +1,13 @@
+llumo/__init__.py,sha256=O04b4yW1BnOvcHzxWFddAKhtdBEhBNhLdb6xgnpHH_Q,205
+llumo/client.py,sha256=60RSxhk-9wzK9KgBz8dfbUd3-AaKiljxqbHI5UL8GIw,54021
+llumo/exceptions.py,sha256=Vp_MnanHbnd1Yjuoi6WLrKiwwZbJL3znCox2URMmGU4,2032
+llumo/execution.py,sha256=nWbJ7AvWuUPcOb6i-JzKRna_PvF-ewZTiK8skS-5n3w,1380
+llumo/functionCalling.py,sha256=D5jYapu1rIvdIJNUYPYMTyhQ1H-6nkwoOLMi6eekfUE,7241
+llumo/helpingFuntions.py,sha256=BZfUIgTO0PJchppHn0wDRF1wcYSuMST5ry95HBPN5SQ,23534
+llumo/models.py,sha256=aVEZsOOoQx5LeNtwSyBxqvrINq0izH3QWu_YjsMPE6o,2910
+llumo/sockets.py,sha256=I2JO_eNEctRo_ikgvFVp5zDd-m0VDu04IEUhhsa1Tic,5950
+llumo-0.2.15b2.dist-info/licenses/LICENSE,sha256=tF9yAcfPV9xGT3ViWmC8hPvOo8BEk4ZICbUfcEo8Dlk,182
+llumo-0.2.15b2.dist-info/METADATA,sha256=vbXwSwhuxnO0CSMz4uJ45AepuwVMl7irZlHmYkqRYbY,1521
+llumo-0.2.15b2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+llumo-0.2.15b2.dist-info/top_level.txt,sha256=d5zUTMI99llPtLRB8rtSrqELm_bOqX-bNC5IcwlDk88,6
+llumo-0.2.15b2.dist-info/RECORD,,

llumo-0.2.15b1.dist-info/RECORD

@@ -1,13 +0,0 @@
-llumo/__init__.py,sha256=O04b4yW1BnOvcHzxWFddAKhtdBEhBNhLdb6xgnpHH_Q,205
-llumo/client.py,sha256=XljwD5mZxjyrXHhu8YhN0cGsd-O_LyKbPzrhS8zbqZo,53778
-llumo/exceptions.py,sha256=Vp_MnanHbnd1Yjuoi6WLrKiwwZbJL3znCox2URMmGU4,2032
-llumo/execution.py,sha256=x88wQV8eL99wNN5YtjFaAMCIfN1PdfQVlAZQb4vzgQ0,1413
-llumo/functionCalling.py,sha256=D5jYapu1rIvdIJNUYPYMTyhQ1H-6nkwoOLMi6eekfUE,7241
-llumo/helpingFuntions.py,sha256=0-ZwG0fnbfP4DP1JTMewM8LdXzz_-p1gRqhPsX0Zmpk,22785
-llumo/models.py,sha256=aVEZsOOoQx5LeNtwSyBxqvrINq0izH3QWu_YjsMPE6o,2910
-llumo/sockets.py,sha256=I2JO_eNEctRo_ikgvFVp5zDd-m0VDu04IEUhhsa1Tic,5950
-llumo-0.2.15b1.dist-info/licenses/LICENSE,sha256=tF9yAcfPV9xGT3ViWmC8hPvOo8BEk4ZICbUfcEo8Dlk,182
-llumo-0.2.15b1.dist-info/METADATA,sha256=yDLkiD46Qq44PA3ylKK2dzsXZmnuE23yxH0RmoqizOk,1521
-llumo-0.2.15b1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-llumo-0.2.15b1.dist-info/top_level.txt,sha256=d5zUTMI99llPtLRB8rtSrqELm_bOqX-bNC5IcwlDk88,6
-llumo-0.2.15b1.dist-info/RECORD,,