lexsi-sdk 0.1.16__py3-none-any.whl

lexsi_sdk/core/xai.py

@@ -0,0 +1,306 @@
+import json
+import os
+from typing import List, Optional
+import httpx
+import pandas as pd
+from pydantic import BaseModel
+import requests
+from lexsi_sdk.client.client import APIClient
+from lexsi_sdk.common.environment import Environment
+from lexsi_sdk.core.organization import Organization
+from lexsi_sdk.common.xai_uris import (
+    AVAILABLE_BATCH_SERVERS_URI,
+    AVAILABLE_CUSTOM_SERVERS_URI,
+    AVAILABLE_SYNTHETIC_CUSTOM_SERVERS_URI,
+    CLEAR_NOTIFICATIONS_URI,
+    CREATE_ORGANIZATION_URI,
+    GET_CASE_PROFILE_URI,
+    GET_NOTIFICATIONS_URI,
+    LOGIN_URI,
+    UPLOAD_DATA_PROJECT_URI,
+    USER_ORGANIZATION_URI,
+)
+import getpass
+
+
+class XAI(BaseModel):
+    """Base class to connect with Lexsi.ai platform"""
+
+    env: Environment = Environment()
+    api_client: APIClient = APIClient()
+
+    def __init__(self, **kwargs):
+        """Initialize the API client using environment-derived settings."""
+        super().__init__(**kwargs)
+
+        debug = self.env.get_debug()
+        base_url = self.env.get_base_url()
+
+        self.api_client = APIClient(debug=debug, base_url=base_url)
+
+    def login(self):
+        """login to Lexsi.ai platform
+
+        :param api_key: API key, defaults to XAI_ACCESS_TOKEN environment variable
+        """
+        access_token = os.environ.get("XAI_ACCESS_TOKEN", None) or getpass.getpass(
+            "Enter your Lexsi Ai Access Token: "
+        )
+
+        if not access_token:
+            raise ValueError("Either set XAI_ACCESS_TOKEN or pass the Access token")
+
+        res = self.api_client.post(LOGIN_URI, payload={"access_token": access_token})
+        self.api_client.update_headers(res["access_token"])
+        self.api_client.set_access_token(access_token)
+
+        print("Authenticated successfully.")
+
+    def organizations(self) -> pd.DataFrame:
+        """Get all organizations associated with user
+
+        :return: Organization details dataframe
+        """
+
+        res = self.api_client.get(USER_ORGANIZATION_URI)
+
+        if not res["success"]:
+            raise Exception(res.get("details", "Failed to get organizations"))
+
+        res["details"].insert(
+            0,
+            {
+                "name": "Personal",
+                "organization_owner": True,
+                "organization_admin": True,
+                "current_users": 1,
+                "created_by": "you",
+            },
+        )
+
+        organization_df = pd.DataFrame(
+            res["details"],
+            columns=[
+                "name",
+                "organization_owner",
+                "organization_admin",
+                "current_users",
+                "created_by",
+                "created_at",
+            ],
+        )
+
+        return organization_df
+
+    def organization(self, organization_name: str) -> Organization:
+        """Select specific organization
+
+        :param organization_name: Name of the organization to be used
+        :return: Organization object
+        """
+        if organization_name == "personal":
+            return Organization(
+                api_client=self.api_client,
+                **{
+                    "name": "Personal",
+                    "organization_owner": True,
+                    "organization_admin": True,
+                    "current_users": 1,
+                    "created_by": "you",
+                }
+            )
+
+        organizations = self.api_client.get(USER_ORGANIZATION_URI)
+
+        if not organizations["success"]:
+            raise Exception(organizations.get("details", "Failed to get organizations"))
+
+        user_organization = [
+            Organization(api_client=self.api_client, **organization)
+            for organization in organizations["details"]
+        ]
+
+        organization = next(
+            filter(
+                lambda organization: organization.name == organization_name,
+                user_organization,
+            ),
+            None,
+        )
+
+        if organization is None:
+            raise Exception("Organization Not Found")
+
+        return organization
+
+    def create_organization(self, organization_name: str) -> Organization:
+        """Create New Organization
+
+        :param organization_name: Name of the new organization
+        :return: Organization object
+        """
+        payload = {"organization_name": organization_name}
+        res = self.api_client.post(CREATE_ORGANIZATION_URI, payload)
+
+        if not res["success"]:
+            raise Exception(res.get("details", "Failed to create organization"))
+
+        return Organization(api_client=self.api_client, **res["organization_details"])
+
+    def get_notifications(self) -> pd.DataFrame:
+        """get user notifications
+
+        :return: notification details dataFrame
+        """
+        res = self.api_client.get(GET_NOTIFICATIONS_URI)
+
+        if not res["success"]:
+            raise Exception("Error while getting user notifications.")
+
+        notifications = res["details"]
+
+        if not notifications:
+            return "No notifications found."
+
+        return pd.DataFrame(notifications).reindex(
+            columns=["project_name", "message", "time"]
+        )
+
+    def clear_notifications(self) -> str:
+        """clear user notifications
+
+        :return: response
+        """
+        res = self.api_client.post(CLEAR_NOTIFICATIONS_URI)
+
+        if not res["success"]:
+            raise Exception("Error while clearing user notifications.")
+
+        return res["details"]
+
+    def available_batch_servers(self) -> dict:
+        """available custom batch servers
+
+        :return: response
+        """
+        res = self.api_client.get(AVAILABLE_BATCH_SERVERS_URI)
+        return res["details"]
+
+    def available_custom_servers(self) -> dict:
+        """available custom servers
+
+        :return: response
+        """
+        res = self.api_client.get(AVAILABLE_CUSTOM_SERVERS_URI)
+        return res
+
+    def available_synthetic_custom_servers(self) -> dict:
+        """available synthetic custom servers
+
+        :return: response
+        """
+        res = self.api_client.get(AVAILABLE_SYNTHETIC_CUSTOM_SERVERS_URI)
+        return res["details"]
+    
+    def register_case(
+        self,
+        token: str,
+        client_id: str,
+        unique_identifier: Optional[str] = None,
+        project_name: str = None,
+        tag: Optional[str] = None,
+        data: Optional[str] = None,
+        processed_data: Optional[bool] = False,
+        merge: Optional[bool] = False,
+        image_class: Optional[str] = None,
+        prompt: Optional[str] = None,
+        serverless_instance_type: Optional[str] = None,
+        explainability_method: Optional[str] = None,
+        explain_model: Optional[bool] = False,
+        session_id: Optional[str] = None,
+        xai: Optional[str] = None,
+        file_path: Optional[str] = None
+    ):
+        """Register a new case entry with raw or processed payloads."""
+        form_data = {
+            "client_id": client_id,
+            "project_name": project_name,
+            "unique_identifier": unique_identifier,
+            "tag": tag,
+            "data": json.dumps(data) if isinstance(data, list) else data,
+            "processed_data": str(processed_data).lower(),
+            "merge": str(merge).lower(),
+            "image_class": image_class,
+            "prompt": prompt,
+            "serverless_instance_type": serverless_instance_type,
+            "explainability_method": explainability_method,
+            "explain_model": str(explain_model).lower(),
+            "session_id": str(session_id).lower(),
+            "xai": xai
+        }
+        headers = {
+            "x-api-token": token
+        }
+        form_data = {k: v for k, v in form_data.items() if v is not None}
+        files = {}
+        if file_path:
+            files["in_file"] = open(file_path, "rb")
+        # response = requests.post(
+        #     self.env.get_base_url() + "/" + UPLOAD_DATA_PROJECT_URI,
+        #     data=form_data,
+        #     files=files if files else None,
+        #     headers=headers
+        # ).json()
+
+        with httpx.Client(http2=True, timeout=None) as client:
+            response = client.post(
+                self.env.get_base_url() + "/" + UPLOAD_DATA_PROJECT_URI,
+                data=form_data,
+                files=files or None,
+                headers=headers,
+            )
+            response.raise_for_status()
+            response = response.json()
+
+        if files:
+            files["in_file"].close()
+        return response
+
+    def case_profile(
+        self,
+        token: str,
+        client_id: str,
+        unique_identifier: Optional[str] = None,
+        project_name: str = None,
+        tag: str = None,
+        xai: Optional[List[str]] = None,
+        refresh: Optional[bool] = None
+    ):
+        """Fetch case profile details for a given identifier and tag."""
+        headers = {
+            "x-api-token": token
+        }
+        payload = {
+            "client_id": client_id,
+            "project_name": project_name,
+            "unique_identifier": unique_identifier,
+            "tag": tag,
+            "xai": xai,
+            "refresh": refresh
+        }
+        # res = requests.post(
+        #     self.env.get_base_url() + "/" + GET_CASE_PROFILE_URI,
+        #     headers=headers,
+        #     json=payload
+        # ).json()
+
+        with httpx.Client(http2=True, timeout=None) as client:
+            res = client.post(
+                self.env.get_base_url() + "/" + GET_CASE_PROFILE_URI,
+                headers=headers,
+                json=payload,
+            )
+            res.raise_for_status()
+            res = res.json()
+
+        return res["details"]

lexsi_sdk/version.py

@@ -0,0 +1,34 @@
+# file generated by setuptools-scm
+# don't change, don't track in version control
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
+
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from typing import Tuple
+    from typing import Union
+
+    VERSION_TUPLE = Tuple[Union[int, str], ...]
+    COMMIT_ID = Union[str, None]
+else:
+    VERSION_TUPLE = object
+    COMMIT_ID = object
+
+version: str
+__version__: str
+__version_tuple__: VERSION_TUPLE
+version_tuple: VERSION_TUPLE
+commit_id: COMMIT_ID
+__commit_id__: COMMIT_ID
+
+__version__ = version = '0.1.16'
+__version_tuple__ = version_tuple = (0, 1, 16)
+
+__commit_id__ = commit_id = None

lexsi_sdk-0.1.16.dist-info/METADATA

@@ -0,0 +1,100 @@
+Metadata-Version: 2.4
+Name: lexsi-sdk
+Version: 0.1.16
+Summary: Full stack ML Observability with Lexsi.ai
+Author-email: Lexsi Labs <support@lexsi.ai>
+License: Lexsi Labs Source Available License (LSAL) v1.0
+Keywords: lexsi,lexsi-sdk,lexsi_sdk,ML observability,observability,machine learning
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Development Status :: 3 - Alpha
+Classifier: Topic :: Scientific/Engineering :: Information Analysis
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: pydantic>=1.10.13
+Requires-Dist: httpx
+Requires-Dist: pyjwt>=2.3.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: requests>=2.31.0
+Requires-Dist: pandas>=1.5.3
+Requires-Dist: plotly>=5.15.0
+Requires-Dist: ipython>=7.34.0
+Requires-Dist: Pillow>=10.4.0
+Requires-Dist: google-genai>=1.24.0
+Requires-Dist: anthropic>=0.57.1
+Requires-Dist: openai>=1.93.1
+Requires-Dist: groq>=0.29.0
+Requires-Dist: xai-sdk>=1.0.0
+Requires-Dist: together>=1.3.14
+Requires-Dist: replicate>=1.0.7
+Requires-Dist: huggingface-hub>=0.33.2
+Requires-Dist: mistralai>=1.9.1
+Requires-Dist: boto3>=1.39.4
+Requires-Dist: opentelemetry-sdk>=1.36.0
+Requires-Dist: opentelemetry-exporter-otlp>=1.36.0
+Requires-Dist: openinference-instrumentation-langchain>=0.1.50
+Requires-Dist: openinference-instrumentation-autogen-agentchat>=0.1.4
+Requires-Dist: openinference-instrumentation-crewai>=0.1.11
+Requires-Dist: openinference-instrumentation-dspy>=0.1.28
+Requires-Dist: openinference-instrumentation-llama-index>=4.3.4
+Requires-Dist: openinference-instrumentation-openai-agents>=1.2.0
+Requires-Dist: openinference-instrumentation-smolagents>=0.1.16
+Provides-Extra: docs
+Requires-Dist: mkdocs>=1.6; extra == "docs"
+Requires-Dist: mkdocs-material>=9.5; extra == "docs"
+Requires-Dist: mkdocstrings[python]>=0.24; extra == "docs"
+Requires-Dist: pymdown-extensions>=10.0; extra == "docs"
+
+## [Lexsi ai](https://lexsi.ai): ML Observability for mission-critical ‘AI’
+
+Lexsi Ai is a full-stack ML Observability platform that integrates with your MLOPs platform to Explain, Monitor, Audit and Improve your ML models.
+
+Lexsi Ai has multiple components to address the complex observability required for mission-critical ‘AI’. 
+
+1. **ML Explainability:** Lexsi Ai offers diverse explainability options like- Bactrace(Specialized for deep learning models), SHAPE, Decision View, Observations (New way to correlate expert functioning vs model functioning) and Similar Cases (reference as explanations).
+2. **ML Monitoring:** Monitor your models for drifts, performance & bias. The tool offers diverse options for drift (data/model) like - PSI, KL Divergence, Chi-square test, 
+3. **Synthetic ‘AI’:** Deploy advanced synthetic ‘AI’ techniques like GPT-2 & GANs on your tabular data to generate high-quality synthetic datasets. Test the quality and privacy of these data sets using our Anonymity tests, column tests etc. 
+4. **ML Risk policies:** Define advanced risk policies on your models. 
+5. **AutoML:** Lexsi Ai also provides fully low-code and no-code options to build ML models on your data. For advanced users, it also provides more options to fine-tune it. 
+
+Lexsi Ai also acts as a common workflow and provides insights acceptable by all stakeholders - Data Science, IT, Risk, Operations and compliance teams, making the rollout and maintenance of AI/ML models seamless and clutter-free.
+
+### Quickstart:
+Get started with Lexsi Ai with a few easy steps:
+
+1. Sign up and log in to your new Lexsi Ai account.
+2. After logging in, generate an Access Token for your user account.
+3. Set the environment variable **XAI_ACCESS_TOKEN** with the generated value. 
+
+Once you've completed these steps, you're all set! Now, you can easily log in and start using the Lexsi Ai SDK:
+
+1. Log in by importing the "xai" object instance from the "lexsi_sdk" package.
+2. Call the "login" method. This method automatically takes the access token value from the "XAI_ACCESS_TOKEN" environment variable and stores the JWT in the object instance. This means that all your future SDK operations will be authorized automatically, making it simple and hassle-free!
+
+
+```
+from lexsi_sdk import xai as lexsi
+
+## login() function authenticates user using token that can be generated in app.lexsi_sdk.com/sdk
+
+
+lexsi.login()
+
+
+Enter your Lexsi ai Access Token: ··········
+Authenticated successfully.
+```
+
+### Cookbook: 
+In this section, you can review the examples of implementation of Lexsi Ai-SDK. 
+
+1. [Full features overview of Lexsi Ai](https://colab.research.google.com/drive/1Dy5eL-FJVnFV0K5yOfGGVoAmiS_Icaz3?usp=sharing)
+2. Using Lexsi Ai in Loan Underwriting (Coming Soon)
+
+
+### Contribution guidelines:
+At Lexsi Ai, we're passionate about open source and value community contributions! Explore our contribution guide for insights into the development workflow and Lexsi Ai library internals. For bug reports or feature requests, head to GitHub Issues or reach out to us at [support@Lexsi Ai.com](support@Lexsi Ai.com).

lexsi_sdk-0.1.16.dist-info/RECORD

@@ -0,0 +1,40 @@
+lexsi_sdk/__init__.py,sha256=yLsuSehO_HEGuddDqDmbyGXRKIdYzGs9En0i-1ELb3s,147
+lexsi_sdk/version.py,sha256=Fs2Vm6Mr4ZbSKXw60jeeSx9iEU9elx4fQcMl_5xXMJI,706
+lexsi_sdk/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lexsi_sdk/client/client.py,sha256=jFnAkiUkSpsECetXroXld1Wf3jj1FyLpfOgaSzTh9Zs,5722
+lexsi_sdk/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lexsi_sdk/common/constants.py,sha256=vqsAhKlcUSeTwLQuXyLX8s4hMHtcDvFyRZ1EqTlDiiY,2705
+lexsi_sdk/common/enums.py,sha256=oVtQBPvly9GFnbnLF9zvhc9UH1j8_mqUnRnkDtF5UIQ,145
+lexsi_sdk/common/environment.py,sha256=zWIreENVA6-EwK5EWv6LYw01wmd8s5X0cFwRRBuT2JE,1181
+lexsi_sdk/common/monitoring.py,sha256=P4wa3Ypxp3DqUiLE5z-zkJlwRGrD3NTlBCluEIoihfI,1797
+lexsi_sdk/common/trigger.py,sha256=-b1SXVG7QsDVtmvdm45fSV2-WeIoKoEu29PDezH-M5Q,1611
+lexsi_sdk/common/types.py,sha256=Nkbo2sMLy-szcvCee2j8xpM9vfpUTC7_Ns95zGmGaLg,3214
+lexsi_sdk/common/utils.py,sha256=qH5PtzOcKPn6u-9hNFSlGNtDeBAGvqJTEWXCBDrw8lk,2933
+lexsi_sdk/common/validation.py,sha256=IjoeOZF7txzMsr0zOZ3QSJwDIT2Q6MHEYhoE27y9RUY,4323
+lexsi_sdk/common/xai_uris.py,sha256=mu-hw_fpx8745qBF3-Feqy4pw4V3EXiV3A5CcIny5Is,10086
+lexsi_sdk/common/config/.env.prod,sha256=PiTaP8HHtGFF7sAgPY0S3wJtjMgGu1EJETZ2h2DkCu8,65
+lexsi_sdk/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lexsi_sdk/core/agent.py,sha256=sFCA3kKBdIyBDxNZT6Fslf_tvTufrFGCAG9N15Hykpw,1699
+lexsi_sdk/core/alert.py,sha256=LV5KBLtpGmU5g2Dc0OkFTTznJp4FHC1Ev46XA-hRuMM,1700
+lexsi_sdk/core/case.py,sha256=xuffHfVRC2AVLBnjPUJK8WmgH_UsTT7HuI1DLvWj5n4,19846
+lexsi_sdk/core/dashboard.py,sha256=ZxbetZ_jmaHGmBY2nNqBzCqJozfd-9Wco8Y3tb9KgFc,4413
+lexsi_sdk/core/model_summary.py,sha256=aILsrhdQCVeVtU71yW3WFXMTTf8sAYjnc9oBkUbdpyw,3487
+lexsi_sdk/core/organization.py,sha256=kUP6-RAf63chITx2rjV-A8zEpWGOj7QzCxgovZatoM4,18972
+lexsi_sdk/core/project.py,sha256=jlQdLxyVwFXGgJro-xdu7DRGhjVgGWIWUj-kISQDqek,186034
+lexsi_sdk/core/synthetic.py,sha256=ItqzctgMZd-G98fmVnNLHi4sO3yGL0y70OVKpfNAP6U,11431
+lexsi_sdk/core/text.py,sha256=bbfu-SIthVnuvwEGTFeruQJAU1YHGSwgBnnMFKGgwTY,20172
+lexsi_sdk/core/tracer.py,sha256=LxcgD6mmlyFdzsqA3aa99iXb0ljMF8nDeNgfPcedfnc,9258
+lexsi_sdk/core/utils.py,sha256=qaxINJKVvhtIV90FDA2NyXyCEmnP82MSipIx5lwgC7Y,1452
+lexsi_sdk/core/workspace.py,sha256=7wbkF9tFwqy0vn0fQekR5eCJbMi7vgF-pAyK2OEESh0,10581
+lexsi_sdk/core/wrapper.py,sha256=Tqwyp_n5wsk50nNSTpE9wL6NzolbWz7JG_OIc7uKZII,33109
+lexsi_sdk/core/xai.py,sha256=SCbHVmWn28IQKU9IVnE4nUYpR2dQEMlc-iAbMBmqMGA,9925
+lexsi_sdk/core/guardrails/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lexsi_sdk/core/guardrails/guard_template.py,sha256=6o48oHEBnYZrWAZ6eMvHZRGX2YOUUykWWgCa1cqFJUw,8572
+lexsi_sdk/core/guardrails/guardrail_autogen.py,sha256=-PKnauWtcA0_z_SVTjzkvMOU5FwxCeF30xX-L4GKWic,25183
+lexsi_sdk/core/guardrails/guardrails_langgraph.py,sha256=lJDXXbjSTb4NvhmdnRs-XLECmZu1E66k6EZc4rP_Dqg,24972
+lexsi_sdk/core/guardrails/guardrails_openai.py,sha256=9eVzNAPry1EzF1H2HVuqBIRzpdmfi65VoEc6qWtXsQE,25062
+lexsi_sdk/core/guardrails/openai_runner.py,sha256=6DV-u9E5MEPq6NSrJYywTipLpc9l5bbgy0RhVBSvY0A,54433
+lexsi_sdk-0.1.16.dist-info/METADATA,sha256=YxZiBERQ6305sBQidlzYHE4EPHJGgHvaO6nrAGhuq6U,5260
+lexsi_sdk-0.1.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lexsi_sdk-0.1.16.dist-info/top_level.txt,sha256=2zF9T78ZDgJfZvOWKkuSFDqR5LZGAmIx9sFPhSsT5Yk,10
+lexsi_sdk-0.1.16.dist-info/RECORD,,

lexsi_sdk-0.1.16.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

lexsi_sdk-0.1.16.dist-info/top_level.txt

@@ -0,0 +1 @@
+lexsi_sdk