axonflow 0.4.0__py3-none-any.whl

axonflow/utils/retry.py

@@ -0,0 +1,111 @@
+"""Retry utilities for AxonFlow SDK."""
+
+from __future__ import annotations
+
+from typing import Any, Callable, TypeVar
+
+from tenacity import (
+    RetryCallState,
+    retry,
+    retry_if_exception_type,
+    stop_after_attempt,
+    wait_exponential,
+)
+
+from axonflow.types import RetryConfig
+
+T = TypeVar("T")
+
+
+class RetryHandler:
+    """Handles retry logic with exponential backoff."""
+
+    def __init__(self, config: RetryConfig) -> None:
+        """Initialize retry handler.
+
+        Args:
+            config: Retry configuration
+        """
+        self.config = config
+
+    def create_decorator(
+        self,
+        retry_on: tuple[type[Exception], ...],
+    ) -> Callable[[Callable[..., T]], Callable[..., T]]:
+        """Create retry decorator based on config.
+
+        Args:
+            retry_on: Exception types to retry on
+
+        Returns:
+            Decorator function
+        """
+        if not self.config.enabled:
+            return lambda f: f
+
+        return retry(
+            stop=stop_after_attempt(self.config.max_attempts),
+            wait=wait_exponential(
+                multiplier=self.config.initial_delay,
+                max=self.config.max_delay,
+                exp_base=self.config.exponential_base,
+            ),
+            retry=retry_if_exception_type(retry_on),
+            reraise=True,
+        )
+
+    @staticmethod
+    def log_retry(retry_state: RetryCallState) -> None:
+        """Log retry attempt.
+
+        Args:
+            retry_state: Current retry state
+        """
+        if retry_state.outcome and retry_state.outcome.failed:
+            exception = retry_state.outcome.exception()
+            attempt = retry_state.attempt_number
+            print(f"Retry attempt {attempt} failed: {exception}")
+
+
+def with_retry(
+    max_attempts: int = 3,
+    initial_delay: float = 1.0,
+    max_delay: float = 30.0,
+    exponential_base: float = 2.0,
+    retry_on: tuple[type[Exception], ...] = (Exception,),
+) -> Callable[[Callable[..., T]], Callable[..., T]]:
+    """Decorator for adding retry logic to a function.
+
+    Args:
+        max_attempts: Maximum retry attempts
+        initial_delay: Initial delay between retries
+        max_delay: Maximum delay between retries
+        exponential_base: Exponential backoff base
+        retry_on: Exception types to retry on
+
+    Returns:
+        Decorator function
+    """
+    config = RetryConfig(
+        enabled=True,
+        max_attempts=max_attempts,
+        initial_delay=initial_delay,
+        max_delay=max_delay,
+        exponential_base=exponential_base,
+    )
+    handler = RetryHandler(config)
+    return handler.create_decorator(retry_on)
+
+
+def create_retry_decorator(config: RetryConfig, retry_on: tuple[type[Exception], ...]) -> Any:
+    """Create a retry decorator from config.
+
+    Args:
+        config: Retry configuration
+        retry_on: Exception types to retry on
+
+    Returns:
+        Retry decorator
+    """
+    handler = RetryHandler(config)
+    return handler.create_decorator(retry_on)

axonflow-0.4.0.dist-info/METADATA

@@ -0,0 +1,316 @@
+Metadata-Version: 2.4
+Name: axonflow
+Version: 0.4.0
+Summary: AxonFlow Python SDK - Enterprise AI Governance in 3 Lines of Code
+Author-email: AxonFlow <dev@getaxonflow.com>
+Maintainer-email: AxonFlow <dev@getaxonflow.com>
+License: MIT
+Project-URL: Homepage, https://getaxonflow.com
+Project-URL: Documentation, https://docs.getaxonflow.com/sdk/python
+Project-URL: Repository, https://github.com/getaxonflow/axonflow-sdk-python
+Project-URL: Changelog, https://github.com/getaxonflow/axonflow-sdk-python/blob/main/CHANGELOG.md
+Project-URL: Issues, https://github.com/getaxonflow/axonflow-sdk-python/issues
+Keywords: ai,governance,llm,openai,anthropic,bedrock,policy,compliance,enterprise,mcp,multi-agent
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Typing :: Typed
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: httpx>=0.25.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: tenacity>=8.0.0
+Requires-Dist: structlog>=23.0.0
+Requires-Dist: cachetools>=5.0.0
+Requires-Dist: eval_type_backport>=0.2.0; python_version < "3.10"
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Requires-Dist: pytest-httpx>=0.22.0; extra == "dev"
+Requires-Dist: mypy>=1.5.0; extra == "dev"
+Requires-Dist: types-cachetools>=5.0.0; extra == "dev"
+Requires-Dist: ruff>=0.1.0; extra == "dev"
+Requires-Dist: black>=23.0.0; extra == "dev"
+Requires-Dist: isort>=5.12.0; extra == "dev"
+Requires-Dist: pre-commit>=3.0.0; extra == "dev"
+Provides-Extra: docs
+Requires-Dist: sphinx>=7.0.0; extra == "docs"
+Requires-Dist: sphinx-rtd-theme>=1.3.0; extra == "docs"
+Requires-Dist: sphinx-autodoc-typehints>=1.24.0; extra == "docs"
+Requires-Dist: myst-parser>=2.0.0; extra == "docs"
+Provides-Extra: openai
+Requires-Dist: openai>=1.0.0; extra == "openai"
+Provides-Extra: anthropic
+Requires-Dist: anthropic>=0.18.0; extra == "anthropic"
+Provides-Extra: all
+Requires-Dist: openai>=1.0.0; extra == "all"
+Requires-Dist: anthropic>=0.18.0; extra == "all"
+Dynamic: license-file
+
+# AxonFlow Python SDK
+
+Enterprise AI Governance in 3 Lines of Code.
+
+[![PyPI version](https://badge.fury.io/py/axonflow.svg)](https://badge.fury.io/py/axonflow)
+[![Python 3.9+](https://img.shields.io/badge/python-3.9+-blue.svg)](https://www.python.org/downloads/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![Type hints](https://img.shields.io/badge/type%20hints-mypy-brightgreen.svg)](http://mypy-lang.org/)
+
+## Installation
+
+```bash
+pip install axonflow
+```
+
+With LLM provider support:
+```bash
+pip install axonflow[openai]      # OpenAI integration
+pip install axonflow[anthropic]   # Anthropic integration
+pip install axonflow[all]         # All integrations
+```
+
+## Quick Start
+
+### Async Usage (Recommended)
+
+```python
+import asyncio
+from axonflow import AxonFlow
+
+async def main():
+    async with AxonFlow(
+        agent_url="https://your-agent.axonflow.com",
+        client_id="your-client-id",
+        client_secret="your-client-secret"
+    ) as client:
+        # Execute a governed query
+        response = await client.execute_query(
+            user_token="user-jwt-token",
+            query="What is AI governance?",
+            request_type="chat"
+        )
+        print(response.data)
+
+asyncio.run(main())
+```
+
+### Sync Usage
+
+```python
+from axonflow import AxonFlow
+
+with AxonFlow.sync(
+    agent_url="https://your-agent.axonflow.com",
+    client_id="your-client-id",
+    client_secret="your-client-secret"
+) as client:
+    response = client.execute_query(
+        user_token="user-jwt-token",
+        query="What is AI governance?",
+        request_type="chat"
+    )
+    print(response.data)
+```
+
+## Features
+
+### Gateway Mode
+
+For lowest-latency LLM calls with full governance and audit compliance:
+
+```python
+from axonflow import AxonFlow, TokenUsage
+
+async with AxonFlow(...) as client:
+    # 1. Pre-check: Get policy approval
+    ctx = await client.get_policy_approved_context(
+        user_token="user-jwt",
+        query="Find patient records",
+        data_sources=["postgres"]
+    )
+
+    if not ctx.approved:
+        raise Exception(f"Blocked: {ctx.block_reason}")
+
+    # 2. Make LLM call directly (your code)
+    llm_response = await openai.chat.completions.create(
+        model="gpt-4",
+        messages=[{"role": "user", "content": str(ctx.approved_data)}]
+    )
+
+    # 3. Audit the call
+    await client.audit_llm_call(
+        context_id=ctx.context_id,
+        response_summary=llm_response.choices[0].message.content[:100],
+        provider="openai",
+        model="gpt-4",
+        token_usage=TokenUsage(
+            prompt_tokens=llm_response.usage.prompt_tokens,
+            completion_tokens=llm_response.usage.completion_tokens,
+            total_tokens=llm_response.usage.total_tokens
+        ),
+        latency_ms=250
+    )
+```
+
+### OpenAI Integration
+
+Transparent governance for existing OpenAI code:
+
+```python
+from openai import OpenAI
+from axonflow import AxonFlow
+from axonflow.interceptors.openai import wrap_openai_client
+
+openai = OpenAI()
+axonflow = AxonFlow(...)
+
+# Wrap client - governance is now automatic
+wrapped = wrap_openai_client(openai, axonflow, user_token="user-123")
+
+# Use as normal
+response = wrapped.chat.completions.create(
+    model="gpt-4",
+    messages=[{"role": "user", "content": "Hello!"}]
+)
+```
+
+### MCP Connectors
+
+Query data through MCP connectors:
+
+```python
+# List available connectors
+connectors = await client.list_connectors()
+
+# Query a connector
+result = await client.query_connector(
+    user_token="user-jwt",
+    connector_name="postgres",
+    operation="query",
+    params={"sql": "SELECT * FROM users LIMIT 10"}
+)
+```
+
+### Multi-Agent Planning
+
+Generate and execute multi-agent plans:
+
+```python
+# Generate a plan
+plan = await client.generate_plan(
+    query="Book a flight and hotel for my trip to Paris",
+    domain="travel"
+)
+
+print(f"Plan has {len(plan.steps)} steps")
+
+# Execute the plan
+result = await client.execute_plan(plan.plan_id)
+print(f"Result: {result.result}")
+```
+
+## Configuration
+
+```python
+from axonflow import AxonFlow, Mode, RetryConfig
+
+client = AxonFlow(
+    agent_url="https://your-agent.axonflow.com",
+    client_id="your-client-id",
+    client_secret="your-client-secret",
+    license_key="optional-license-key",      # For enterprise features
+    mode=Mode.PRODUCTION,                     # or Mode.SANDBOX
+    debug=True,                               # Enable debug logging
+    timeout=60.0,                             # Request timeout in seconds
+    retry_config=RetryConfig(                 # Retry configuration
+        enabled=True,
+        max_attempts=3,
+        initial_delay=1.0,
+        max_delay=30.0,
+    ),
+    cache_enabled=True,                       # Enable response caching
+    cache_ttl=60.0,                           # Cache TTL in seconds
+)
+```
+
+## Error Handling
+
+```python
+from axonflow.exceptions import (
+    AxonFlowError,
+    PolicyViolationError,
+    AuthenticationError,
+    RateLimitError,
+    TimeoutError,
+)
+
+try:
+    response = await client.execute_query(...)
+except PolicyViolationError as e:
+    print(f"Blocked by policy: {e.block_reason}")
+except RateLimitError as e:
+    print(f"Rate limited: {e.limit}/{e.remaining}, resets at {e.reset_at}")
+except AuthenticationError:
+    print("Invalid credentials")
+except TimeoutError:
+    print("Request timed out")
+except AxonFlowError as e:
+    print(f"AxonFlow error: {e.message}")
+```
+
+## Response Types
+
+All responses are Pydantic models with full type hints:
+
+```python
+from axonflow import (
+    ClientResponse,
+    PolicyApprovalResult,
+    PlanResponse,
+    ConnectorResponse,
+)
+
+# Full autocomplete and type checking support
+response: ClientResponse = await client.execute_query(...)
+print(response.success)
+print(response.data)
+print(response.policy_info.policies_evaluated)
+```
+
+## Development
+
+```bash
+# Install dev dependencies
+pip install -e ".[dev]"
+
+# Run tests
+pytest
+
+# Run linting
+ruff check .
+ruff format .
+
+# Run type checking
+mypy axonflow
+```
+
+## Documentation
+
+- [API Reference](https://docs.getaxonflow.com/sdk/python/api)
+- [Gateway Mode Guide](https://docs.getaxonflow.com/sdk/python/gateway-mode)
+- [Examples](https://github.com/getaxonflow/axonflow/tree/main/sdk/python/examples)
+
+## License
+
+Apache 2.0 - See [LICENSE](LICENSE) for details.

axonflow-0.4.0.dist-info/RECORD

@@ -0,0 +1,22 @@
+axonflow/__init__.py,sha256=nBRiCru0KqddQ24e9kwXGDiSRvPfkZZc8zoCsQvzOsw,3523
+axonflow/client.py,sha256=qL91qxJOQcux6cESqmjTf7oWw-cWPQQoKJMxSF8QLag,54078
+axonflow/exceptions.py,sha256=TYxBlqORwQTVJuaS5yQlghUO13w0KoYo6W5HvpypRUI,2415
+axonflow/policies.py,sha256=f_Hc-qV5cKtxwDVH_rO7GRskebkLCBLexgIj4yUW3Ok,8417
+axonflow/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+axonflow/types.py,sha256=nDhE9CwClSgE3oC5lNfDS8707G7PZuwHDhECtGFtcRM,6946
+axonflow/interceptors/__init__.py,sha256=Yynmu2HMItrZzbN8AnRNgCEQDr9IfUYCdRc-BZUkCLk,660
+axonflow/interceptors/anthropic.py,sha256=ZJW-Tbl2mIzWKckvDoHrDbV0Q0_x2H1sAhgiHjKRzAk,6086
+axonflow/interceptors/base.py,sha256=aNfN6WESl2c19tkIJ6qTe-uhNoFRdMAxg9YgJb7To1c,1469
+axonflow/interceptors/bedrock.py,sha256=FIhlQjoo9nxjCu3Ke9V8I0ZzHcOXpz1aVOWpL73JX1U,7691
+axonflow/interceptors/gemini.py,sha256=eFQv3dmg0oM1eNAIg1ywubpbdYx_Cfk9Q6X0XWfrZKI,9131
+axonflow/interceptors/ollama.py,sha256=hOBaR5LTSBrKOXFjoA3wPQwL15B8WD01a_pcPIF7Txg,7824
+axonflow/interceptors/openai.py,sha256=_brrpE2PBoZzQxq_zkz2gbcfbx4saeIbxB8gcaswa2A,5057
+axonflow/utils/__init__.py,sha256=_a6AC8YB9zkCsxVnS3jekrBnK9Vr81mqpkeW07j06s8,286
+axonflow/utils/cache.py,sha256=OI7jXj3wL6vhA9WtZiJyTP4wumMUke1p6nFR9PCayBk,2381
+axonflow/utils/logging.py,sha256=dfBz1rrm3LAAGN6f6VUSYp--bEPk7Wtum2gZhfdW0Jc,2509
+axonflow/utils/retry.py,sha256=JVIFKWzao2HOrjfJ7ZPZl7ztUMzhl1LJyX8hX7Rm4zM,3003
+axonflow-0.4.0.dist-info/licenses/LICENSE,sha256=kpfaU6d8Kw06TmnV5V2B8kTl5X73i9r0bgR86SWkFzU,1068
+axonflow-0.4.0.dist-info/METADATA,sha256=YUIKPfX_-1o13LLRy7sH67-Jp96MyrYcjlK36u4HeWI,8977
+axonflow-0.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+axonflow-0.4.0.dist-info/top_level.txt,sha256=PTnp0kT26lEgaU4UsWCJG_sPgxHh4mrw8ZTb6jH4OV4,9
+axonflow-0.4.0.dist-info/RECORD,,

axonflow-0.4.0.dist-info/WHEEL

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

axonflow-0.4.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 getaxonflow
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

axonflow-0.4.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+axonflow