llumo 0.2.16b1__tar.gz → 0.2.16b2__tar.gz

{llumo-0.2.16b1/llumo.egg-info → llumo-0.2.16b2}/PKG-INFO

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

llumo-0.2.16b2/llumo/google.py

@@ -0,0 +1,66 @@
+from google import generativeai as _genai
+from .client import LlumoClient
+
+
+def evaluate_multiple(data, api_key=None, evals=["Response Correctness"]):
+    client = LlumoClient(api_key=api_key)
+    results = client.evaluateMultiple(
+        data,
+        evals=evals,
+        createExperiment=False,
+        prompt_template="Give answer to the query: {{query}}, using context: {{context}}",
+        getDataFrame=False
+    )
+    return results
+
+
+class ChatCompletionWithEval:
+    def __init__(self, response, evaluation):
+        self._response = response
+        self.evaluation = evaluation
+
+    def __getattr__(self, name):
+        return getattr(self._response, name)
+
+    def __getitem__(self, key):
+        return self._response[key]
+
+    def __repr__(self):
+        return repr(self._response)
+
+
+class genai:
+    class GenerativeModel:
+        def __init__(self, api_key: str, model: str = "gemini-2.5-flash"):
+            _genai.configure(api_key=api_key)
+            self._api_key = api_key
+            self._model_name = model
+            self._model_instance = _genai.GenerativeModel(model_name=model)
+
+        def generate_content(self, contents: str | list[str], **kwargs):
+            context = kwargs.pop("context", None)
+            evals = kwargs.pop("evals", [])
+            llumo_key = kwargs.pop("llumo_key", None)
+
+            # Run Gemini generation
+            response = self._model_instance.generate_content(contents=contents, **kwargs)
+            output = response.text
+
+            eval_input = [{
+                "query": contents,
+                "context": context or contents,
+                "output": output,
+            }]
+
+            evaluation = None
+            try:
+                evaluation = evaluate_multiple(data=eval_input, evals=evals, api_key=llumo_key)
+            except Exception as e:
+                evaluation = None
+            
+            if evaluation is None:
+                    print("Cannot process your request, please check your api and try again later.")
+                    return response
+            
+
+            return ChatCompletionWithEval(response, evaluation)

{llumo-0.2.16b1 → llumo-0.2.16b2}/llumo/openai.py

@@ -5,7 +5,6 @@ from .client import LlumoClient
 def evaluate_multiple(data, api_key=None,evals=["Response Correctness"]):
     client = LlumoClient(api_key=api_key)
     results= client.evaluateMultiple(data, evals=evals,createExperiment=False,prompt_template="Give answer to the query: {{query}}, using context: {{context}}",getDataFrame=False)
-    print(results)
     return results
 
 # Wrapper around ChatCompletion to allow custom fields like `.evaluation`
@@ -70,7 +69,7 @@ class openai(OpenAIClient):
 
                 # If evaluation is None, just return normal response
                 if evaluation is None:
-                    print("All server are busy for evaluation ")
+                    print("Cannot process your request for evaluation, please check your api and try again later.")
                     return response
 
                 # Otherwise wrap with evaluation attached

{llumo-0.2.16b1 → llumo-0.2.16b2/llumo.egg-info}/PKG-INFO

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

llumo-0.2.16b1/llumo/google.py

@@ -1,34 +0,0 @@
-from google import generativeai as _genai
-
-class genai:
-    """
-    Top-level wrapper module to mimic:
-    >>> from google import genai
-    >>> client = genai.Client(api_key=...)
-    """
-
-    class Client:
-        def __init__(self, api_key: str, default_model: str = "gemini-2.5-flash"):
-            _genai.configure(api_key=api_key)
-            self._defaultModel = default_model
-            self._defaultModelInstance = _genai.GenerativeModel(model_name=default_model)
-
-            class Models:
-                def __init__(self, outer):
-                    self._outer = outer
-
-                def generate_content(self, contents: str | list[str], model: str = None, **kwargs):
-                    model_name = model or self._outer._defaultModel
-                    model_instance = _genai.GenerativeModel(model_name=model_name)
-                    return model_instance.generate_content(contents=contents, **kwargs)
-
-            self.models = Models(self)
-
-        def generate(self, prompt: str | list[str], **kwargs):
-            """Convenience shortcut for single-line generation."""
-            return self._defaultModelInstance.generate_content(prompt, **kwargs)
-
-        def setDefaultModel(self, model_name: str):
-            """Change the default model at runtime."""
-            self._defaultModel = model_name
-            self._defaultModelInstance = _genai.GenerativeModel(model_name=model_name)