mem0ai-azure-mysql 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

mem0/client/utils.py

@@ -1,29 +0,0 @@
-import logging
-
-import httpx
-
-logger = logging.getLogger(__name__)
-
-
-class APIError(Exception):
-    """Exception raised for errors in the API."""
-
-    pass
-
-
-def api_error_handler(func):
-    """Decorator to handle API errors consistently."""
-    from functools import wraps
-
-    @wraps(func)
-    def wrapper(*args, **kwargs):
-        try:
-            return func(*args, **kwargs)
-        except httpx.HTTPStatusError as e:
-            logger.error(f"HTTP error occurred: {e}")
-            raise APIError(f"API request failed: {e.response.text}")
-        except httpx.RequestError as e:
-            logger.error(f"Request error occurred: {e}")
-            raise APIError(f"Request failed: {str(e)}")
-
-    return wrapper

mem0ai_azure_mysql-0.2.0.dist-info/METADATA

@@ -1,221 +0,0 @@
-Metadata-Version: 2.4
-Name: mem0ai-azure-mysql
-Version: 0.2.0
-Summary: Long-term memory for AI Agents with Azure DefaultAzureCredential authentication and MySQL history database support. Async-only API.
-Requires-Python: <4.0,>=3.10
-Requires-Dist: azure-identity>=1.23.1
-Requires-Dist: azure-search-documents>=11.5.3
-Requires-Dist: cohere>=5.17.0
-Requires-Dist: langchain-neo4j==0.6.0
-Requires-Dist: openai<1.100.0,>=1.90.0
-Requires-Dist: posthog>=3.5.0
-Requires-Dist: protobuf>=5.29.0
-Requires-Dist: pydantic>=2.7.3
-Requires-Dist: pymysql
-Requires-Dist: pytz>=2024.1
-Requires-Dist: rank-bm25>=0.2.2
-Requires-Dist: sqlalchemy>=2.0.31
-Provides-Extra: dev
-Requires-Dist: azure-keyvault-secrets; extra == 'dev'
-Requires-Dist: isort>=5.13.2; extra == 'dev'
-Requires-Dist: pytest-asyncio>=0.23.7; extra == 'dev'
-Requires-Dist: pytest>=8.2.2; extra == 'dev'
-Requires-Dist: ruff>=0.6.5; extra == 'dev'
-Provides-Extra: extras
-Requires-Dist: boto3>=1.34.0; extra == 'extras'
-Provides-Extra: graph
-Requires-Dist: langchain-neo4j==0.6.0; extra == 'graph'
-Requires-Dist: neo4j>=5.23.1; extra == 'graph'
-Requires-Dist: rank-bm25>=0.2.2; extra == 'graph'
-Provides-Extra: llms
-Requires-Dist: openai<1.100.0,>=1.90.0; extra == 'llms'
-Provides-Extra: test
-Requires-Dist: pytest-asyncio>=0.23.7; extra == 'test'
-Requires-Dist: pytest-mock>=3.14.0; extra == 'test'
-Requires-Dist: pytest>=8.2.2; extra == 'test'
-Provides-Extra: vector-stores
-Requires-Dist: azure-identity>=1.24.0; extra == 'vector-stores'
-Requires-Dist: azure-search-documents>=11.4.0b8; extra == 'vector-stores'
-Description-Content-Type: text/markdown
-
-# Mem0 - Azure Enhanced Fork
-
-This repository is an enhanced fork of [mem0ai/mem0](https://github.com/mem0ai/mem0.git) that provides enterprise-grade improvements for Azure environments and production deployments.
-
-## Key Enhancements
-
-### 1. Async-Only API
-- **AsyncMemory**: This fork provides only the async `AsyncMemory` class for better performance and scalability
-- **Simplified Codebase**: Removed synchronous `Memory` class to reduce code complexity and maintenance burden
-- **Modern Python**: Built for async/await patterns with full asyncio support
-
-### 2. Azure Entra ID Authentication
-- **Azure AI Search**: Support for Azure Entra ID (Azure AD) authentication using `DefaultAzureCredential`
-- **Azure OpenAI**: Seamless Entra ID integration for both LLM and embedding services
-- **Simplified Authentication**: No need to manage API keys when using managed identities or service principals
-
-### 3. MySQL Database Support
-- **Production-Ready**: Replace SQLite3 with enterprise-grade MySQL for scalable memory history storage
-- **Connection Pooling**: Built-in connection pooling and SSL support for secure connections
-- **Thread-Safe**: Thread-safe operations with proper connection management
-
-## Installation
-
-Install the enhanced package with Azure and MySQL dependencies:
-
-```bash
-pip install mem0ai-azure-mysql
-```
-
-Or with uv:
-
-```bash
-uv add mem0ai-azure-mysql
-```
-
-## Quick Start
-
-```python
-import asyncio
-from mem0 import AsyncMemory
-
-config = {
-    "vector_store": {
-        "provider": "azure_ai_search",
-        "config": {
-            "collection_name": "mem0",
-            "service_name": "your-search-service",
-            "embedding_model_dims": 1536,
-            "azure_ad_token": "<your-token>"  # Or use DefaultAzureCredential
-        },
-    },
-    "llm": {
-        "provider": "azure_openai",
-        "config": {
-            "model": "gpt-4",
-            "azure_kwargs": {
-                "api_version": "2024-12-01-preview",
-                "azure_deployment": "gpt-4",
-                "azure_endpoint": "https://your-endpoint.openai.azure.com/",
-                "azure_ad_token": "<your-token>",
-            },
-        },
-    },
-    "embedder": {
-        "provider": "azure_openai",
-        "config": {
-            "model": "text-embedding-3-small",
-            "embedding_dims": 1536,
-            "azure_kwargs": {
-                "api_version": "2024-12-01-preview",
-                "azure_deployment": "text-embedding-3-small",
-                "azure_endpoint": "https://your-endpoint.openai.azure.com/",
-                "azure_ad_token": "<your-token>",
-            },
-        },
-    },
-    "db": {
-        "provider": "mysql",
-        "config": {
-            "host": "your-mysql-server.mysql.database.azure.com",
-            "port": 3306,
-            "user": "mem0",
-            "password": "<your-password>",
-            "database": "mem0",
-            "ssl_enabled": True,
-        },
-    },
-}
-
-async def main():
-    memory = await AsyncMemory.from_config(config)
-
-    # Add memories
-    result = await memory.add(
-        "I love playing tennis on weekends",
-        user_id="user123"
-    )
-    print(result)
-
-    # Search memories
-    results = await memory.search("What sports do I like?", user_id="user123")
-    print(results)
-
-asyncio.run(main())
-```
-
-## Development
-
-### Setup with uv (Recommended)
-
-```bash
-# Install dependencies with dev extras
-uv sync --extra dev
-
-# Run example
-uv run python example.py
-
-# Run tests
-uv run pytest tests/
-```
-
-### Setup with hatch
-
-```bash
-# Create environment
-make install
-
-# Run tests
-make test
-```
-
-### Available Commands
-
-```bash
-make format    # Format code with ruff
-make lint      # Lint code with ruff
-make test      # Run tests
-make build     # Build package
-make clean     # Clean build artifacts
-```
-
-## API Reference
-
-### AsyncMemory
-
-The main class for interacting with the memory system.
-
-```python
-from mem0 import AsyncMemory
-
-# Create from config
-memory = await AsyncMemory.from_config(config)
-
-# Add memories
-await memory.add(messages, user_id="user123")
-
-# Search memories
-await memory.search(query, user_id="user123")
-
-# Get all memories
-await memory.get_all(user_id="user123")
-
-# Delete memories
-await memory.delete_all(user_id="user123")
-
-# Get memory by ID
-await memory.get(memory_id)
-
-# Update memory
-await memory.update(memory_id, new_data)
-
-# Get history
-await memory.history(memory_id)
-
-# Reset all
-await memory.reset()
-```
-
-## License
-
-This project is licensed under the Apache License 2.0 - see the original [mem0](https://github.com/mem0ai/mem0) repository for details.