raxe 0.4.1__tar.gz → 0.4.3__tar.gz

{raxe-0.4.1/src/raxe.egg-info → raxe-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: raxe
-Version: 0.4.1
+Version: 0.4.3
 Summary: RAXE Community Edition - AI Security for Everyone. 460+ threat detection rules, L2 CPU-based ML, always free.
 Author-email: "RAXE Technologies, Inc." <hello@raxe.ai>
 License: Proprietary - See LICENSE file
@@ -57,15 +57,37 @@ Requires-Dist: tomli-w<2.0,>=1.0; extra == "config-tools"
 Provides-Extra: wrappers
 Requires-Dist: openai>=1.0; extra == "wrappers"
 Requires-Dist: anthropic>=0.5; extra == "wrappers"
-Requires-Dist: langchain>=0.1; extra == "wrappers"
+Provides-Extra: langchain
+Requires-Dist: langchain>=0.1; extra == "langchain"
+Requires-Dist: langchain-core>=0.1; extra == "langchain"
+Provides-Extra: crewai
+Requires-Dist: crewai>=0.30; extra == "crewai"
+Provides-Extra: autogen
+Requires-Dist: pyautogen>=0.2; extra == "autogen"
+Provides-Extra: llamaindex
+Requires-Dist: llama-index-core>=0.10; extra == "llamaindex"
 Provides-Extra: litellm
 Requires-Dist: litellm>=1.0; extra == "litellm"
 Provides-Extra: dspy
 Requires-Dist: dspy>=2.0; extra == "dspy"
+Provides-Extra: portkey
+Requires-Dist: openai>=1.0; extra == "portkey"
+Provides-Extra: agents
+Requires-Dist: langchain>=0.1; extra == "agents"
+Requires-Dist: langchain-core>=0.1; extra == "agents"
+Requires-Dist: crewai>=0.30; extra == "agents"
+Requires-Dist: pyautogen>=0.2; extra == "agents"
+Requires-Dist: llama-index-core>=0.10; extra == "agents"
+Requires-Dist: litellm>=1.0; extra == "agents"
+Requires-Dist: dspy>=2.0; extra == "agents"
 Provides-Extra: all
 Requires-Dist: openai>=1.0; extra == "all"
 Requires-Dist: anthropic>=0.5; extra == "all"
 Requires-Dist: langchain>=0.1; extra == "all"
+Requires-Dist: langchain-core>=0.1; extra == "all"
+Requires-Dist: crewai>=0.30; extra == "all"
+Requires-Dist: pyautogen>=0.2; extra == "all"
+Requires-Dist: llama-index-core>=0.10; extra == "all"
 Requires-Dist: litellm>=1.0; extra == "all"
 Requires-Dist: dspy>=2.0; extra == "all"
 Requires-Dist: prompt-toolkit<4.0,>=3.0; extra == "all"

{raxe-0.4.1 → raxe-0.4.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "raxe"
-version = "0.4.1"
+version = "0.4.3"
 description = "RAXE Community Edition - AI Security for Everyone. 460+ threat detection rules, L2 CPU-based ML, always free."
 authors = [{name = "RAXE Technologies, Inc.", email = "hello@raxe.ai"}]
 license = {text = "Proprietary - See LICENSE file"}
@@ -81,24 +81,59 @@ config-tools = [
 wrappers = [
     "openai>=1.0",
     "anthropic>=0.5",
+]
+
+# Framework integrations
+langchain = [
     "langchain>=0.1",
+    "langchain-core>=0.1",
+]
+
+crewai = [
+    "crewai>=0.30",
+]
+
+autogen = [
+    "pyautogen>=0.2",
+]
+
+llamaindex = [
+    "llama-index-core>=0.10",
 ]
 
-# LiteLLM integration
 litellm = [
     "litellm>=1.0",
 ]
 
-# DSPy integration
 dspy = [
     "dspy>=2.0",
 ]
 
-# All optional features (ML is now in core dependencies)
+# Portkey uses OpenAI client
+portkey = [
+    "openai>=1.0",
+]
+
+# All framework integrations
+agents = [
+    "langchain>=0.1",
+    "langchain-core>=0.1",
+    "crewai>=0.30",
+    "pyautogen>=0.2",
+    "llama-index-core>=0.10",
+    "litellm>=1.0",
+    "dspy>=2.0",
+]
+
+# All optional features
 all = [
     "openai>=1.0",
     "anthropic>=0.5",
     "langchain>=0.1",
+    "langchain-core>=0.1",
+    "crewai>=0.30",
+    "pyautogen>=0.2",
+    "llama-index-core>=0.10",
     "litellm>=1.0",
     "dspy>=2.0",
     "prompt-toolkit>=3.0,<4.0",

{raxe-0.4.1 → raxe-0.4.3}/src/raxe/sdk/integrations/litellm.py

@@ -49,6 +49,19 @@ from dataclasses import dataclass
 from datetime import datetime
 from typing import TYPE_CHECKING, Any
 
+try:
+    from litellm.integrations.custom_logger import CustomLogger
+
+    LITELLM_AVAILABLE = True
+except ImportError:
+    # Define a stub if litellm not installed
+    class CustomLogger:  # type: ignore[no-redef]
+        """Stub CustomLogger for when litellm is not installed."""
+
+        pass
+
+    LITELLM_AVAILABLE = False
+
 from raxe.sdk.agent_scanner import (
     AgentScannerConfig,
     MessageType,
@@ -103,11 +116,11 @@ class LiteLLMConfig:
 # =============================================================================
 
 
-class RaxeLiteLLMCallback:
+class RaxeLiteLLMCallback(CustomLogger):
     """LiteLLM callback handler for RAXE security scanning.
 
-    This class implements the LiteLLM CustomLogger interface to provide
-    automatic security scanning of all LLM API calls.
+    This class extends the LiteLLM CustomLogger to provide automatic
+    security scanning of all LLM API calls.
 
     Attributes:
         raxe: Raxe client for scanning
@@ -140,6 +153,8 @@ class RaxeLiteLLMCallback:
             raxe: Raxe client for scanning. Created if not provided.
             config: Integration configuration. Uses defaults if not provided.
         """
+        super().__init__()
+
         if raxe is None:
             from raxe.sdk.client import Raxe
 

{raxe-0.4.1 → raxe-0.4.3/src/raxe.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: raxe
-Version: 0.4.1
+Version: 0.4.3
 Summary: RAXE Community Edition - AI Security for Everyone. 460+ threat detection rules, L2 CPU-based ML, always free.
 Author-email: "RAXE Technologies, Inc." <hello@raxe.ai>
 License: Proprietary - See LICENSE file
@@ -57,15 +57,37 @@ Requires-Dist: tomli-w<2.0,>=1.0; extra == "config-tools"
 Provides-Extra: wrappers
 Requires-Dist: openai>=1.0; extra == "wrappers"
 Requires-Dist: anthropic>=0.5; extra == "wrappers"
-Requires-Dist: langchain>=0.1; extra == "wrappers"
+Provides-Extra: langchain
+Requires-Dist: langchain>=0.1; extra == "langchain"
+Requires-Dist: langchain-core>=0.1; extra == "langchain"
+Provides-Extra: crewai
+Requires-Dist: crewai>=0.30; extra == "crewai"
+Provides-Extra: autogen
+Requires-Dist: pyautogen>=0.2; extra == "autogen"
+Provides-Extra: llamaindex
+Requires-Dist: llama-index-core>=0.10; extra == "llamaindex"
 Provides-Extra: litellm
 Requires-Dist: litellm>=1.0; extra == "litellm"
 Provides-Extra: dspy
 Requires-Dist: dspy>=2.0; extra == "dspy"
+Provides-Extra: portkey
+Requires-Dist: openai>=1.0; extra == "portkey"
+Provides-Extra: agents
+Requires-Dist: langchain>=0.1; extra == "agents"
+Requires-Dist: langchain-core>=0.1; extra == "agents"
+Requires-Dist: crewai>=0.30; extra == "agents"
+Requires-Dist: pyautogen>=0.2; extra == "agents"
+Requires-Dist: llama-index-core>=0.10; extra == "agents"
+Requires-Dist: litellm>=1.0; extra == "agents"
+Requires-Dist: dspy>=2.0; extra == "agents"
 Provides-Extra: all
 Requires-Dist: openai>=1.0; extra == "all"
 Requires-Dist: anthropic>=0.5; extra == "all"
 Requires-Dist: langchain>=0.1; extra == "all"
+Requires-Dist: langchain-core>=0.1; extra == "all"
+Requires-Dist: crewai>=0.30; extra == "all"
+Requires-Dist: pyautogen>=0.2; extra == "all"
+Requires-Dist: llama-index-core>=0.10; extra == "all"
 Requires-Dist: litellm>=1.0; extra == "all"
 Requires-Dist: dspy>=2.0; extra == "all"
 Requires-Dist: prompt-toolkit<4.0,>=3.0; extra == "all"

{raxe-0.4.1 → raxe-0.4.3}/src/raxe.egg-info/requires.txt

@@ -14,18 +14,37 @@ numpy<2.0,>=1.24.0
 [:python_version < "3.11"]
 tomli<3.0,>=2.0
 
+[agents]
+langchain>=0.1
+langchain-core>=0.1
+crewai>=0.30
+pyautogen>=0.2
+llama-index-core>=0.10
+litellm>=1.0
+dspy>=2.0
+
 [all]
 openai>=1.0
 anthropic>=0.5
 langchain>=0.1
+langchain-core>=0.1
+crewai>=0.30
+pyautogen>=0.2
+llama-index-core>=0.10
 litellm>=1.0
 dspy>=2.0
 prompt-toolkit<4.0,>=3.0
 tomli-w<2.0,>=1.0
 
+[autogen]
+pyautogen>=0.2
+
 [config-tools]
 tomli-w<2.0,>=1.0
 
+[crewai]
+crewai>=0.30
+
 [dev]
 pytest>=7.4
 pytest-cov>=4.1
@@ -43,13 +62,22 @@ build>=1.0
 [dspy]
 dspy>=2.0
 
+[langchain]
+langchain>=0.1
+langchain-core>=0.1
+
 [litellm]
 litellm>=1.0
 
+[llamaindex]
+llama-index-core>=0.10
+
+[portkey]
+openai>=1.0
+
 [repl]
 prompt-toolkit<4.0,>=3.0
 
 [wrappers]
 openai>=1.0
 anthropic>=0.5
-langchain>=0.1