mem0ai-azure-mysql 0.1.115__py3-none-any.whl

mem0/client/utils.py

@@ -0,0 +1,29 @@
+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

mem0/configs/base.py

@@ -0,0 +1,90 @@
+import os
+from typing import Any, Dict, Optional
+
+from pydantic import BaseModel, Field
+
+from mem0.embeddings.configs import EmbedderConfig
+from mem0.graphs.configs import GraphStoreConfig
+from mem0.llms.configs import LlmConfig
+from mem0.vector_stores.configs import VectorStoreConfig
+from mem0.dbs.configs import DBConfig
+
+# Set up the directory path
+home_dir = os.path.expanduser("~")
+mem0_dir = os.environ.get("MEM0_DIR") or os.path.join(home_dir, ".mem0")
+
+
+class MemoryItem(BaseModel):
+    id: str = Field(..., description="The unique identifier for the text data")
+    memory: str = Field(
+        ..., description="The memory deduced from the text data"
+    )  # TODO After prompt changes from platform, update this
+    hash: Optional[str] = Field(None, description="The hash of the memory")
+    # The metadata value can be anything and not just string. Fix it
+    metadata: Optional[Dict[str, Any]] = Field(None, description="Additional metadata for the text data")
+    score: Optional[float] = Field(None, description="The score associated with the text data")
+    created_at: Optional[str] = Field(None, description="The timestamp when the memory was created")
+    updated_at: Optional[str] = Field(None, description="The timestamp when the memory was updated")
+
+
+class MemoryConfig(BaseModel):
+    vector_store: VectorStoreConfig = Field(
+        description="Configuration for the vector store",
+        default_factory=VectorStoreConfig,
+    )
+    llm: LlmConfig = Field(
+        description="Configuration for the language model",
+        default_factory=LlmConfig,
+    )
+    embedder: EmbedderConfig = Field(
+        description="Configuration for the embedding model",
+        default_factory=EmbedderConfig,
+    )
+    history_db_path: str = Field(
+        description="Path to the history database",
+        default=os.path.join(mem0_dir, "history.db"),
+    )
+    db: DBConfig = Field(
+        description="Configuration for the database",
+        default_factory=DBConfig,
+    )
+    graph_store: GraphStoreConfig = Field(
+        description="Configuration for the graph",
+        default_factory=GraphStoreConfig,
+    )
+    version: str = Field(
+        description="The version of the API",
+        default="v1.1",
+    )
+    custom_fact_extraction_prompt: Optional[str] = Field(
+        description="Custom prompt for the fact extraction",
+        default=None,
+    )
+    custom_update_memory_prompt: Optional[str] = Field(
+        description="Custom prompt for the update memory",
+        default=None,
+    )
+
+
+class AzureConfig(BaseModel):
+    """
+    Configuration settings for Azure.
+
+    Args:
+        api_key (str): The API key used for authenticating with the Azure service.
+        azure_deployment (str): The name of the Azure deployment.
+        azure_endpoint (str): The endpoint URL for the Azure service.
+        api_version (str): The version of the Azure API being used.
+        default_headers (Dict[str, str]): Headers to include in requests to the Azure API.
+    """
+
+    api_key: str = Field(
+        description="The API key used for authenticating with the Azure service.",
+        default=None,
+    )
+    azure_deployment: str = Field(description="The name of the Azure deployment.", default=None)
+    azure_endpoint: str = Field(description="The endpoint URL for the Azure service.", default=None)
+    api_version: str = Field(description="The version of the Azure API being used.", default=None)
+    default_headers: Optional[Dict[str, str]] = Field(
+        description="Headers to include in requests to the Azure API.", default=None
+    )

mem0/configs/dbs/__init__.py

@@ -0,0 +1,4 @@
+from mem0.configs.dbs.base import BaseDBConfig
+from mem0.configs.dbs.mysql import MySQLConfig
+
+__all__ = ["BaseDBConfig", "MySQLConfig"]

mem0/configs/dbs/base.py

@@ -0,0 +1,41 @@
+from abc import ABC
+from typing import Optional
+
+
+class BaseDBConfig(ABC):
+    """
+    Config for Database.
+    """
+
+    def __init__(
+        self,
+        host: Optional[str] = None,
+        port: Optional[int] = None,
+        user: Optional[str] = None,
+        password: Optional[str] = None,
+        database: Optional[str] = None,
+        ssl_enabled: bool = False,
+    ):
+        """
+        Initializes a configuration class instance for the Database.
+
+        :param host: Database host, defaults to None
+        :type host: Optional[str], optional
+        :param port: Database port, defaults to None
+        :type port: Optional[int], optional
+        :param user: Database user, defaults to None
+        :type user: Optional[str], optional
+        :param password: Database password, defaults to None
+        :type password: Optional[str], optional
+        :param database: Database name, defaults to None
+        :type database: Optional[str], optional
+        :param ssl_enabled: Whether to use SSL for the connection, defaults to False
+        :type ssl_enabled: bool, optional
+        """
+
+        self.host = host
+        self.port = port
+        self.user = user
+        self.password = password
+        self.database = database
+        self.ssl_enabled = ssl_enabled

mem0/configs/dbs/mysql.py

@@ -0,0 +1,25 @@
+from typing import Optional
+
+from mem0.configs.dbs.base import BaseDBConfig
+
+
+class MySQLConfig(BaseDBConfig):
+    """Configuration for MySQL database."""
+    
+    def __init__(
+        self,
+        host: Optional[str] = "localhost",
+        port: Optional[int] = 3306,
+        user: Optional[str] = None,
+        password: Optional[str] = None,
+        database: Optional[str] = None,
+        ssl_enabled: bool = False,
+        **kwargs
+    ):
+        self.host = host
+        self.port = port
+        self.user = user
+        self.password = password
+        self.database = database
+        self.ssl_enabled = ssl_enabled
+        self.connection_params = kwargs

mem0/configs/embeddings/base.py

@@ -0,0 +1,108 @@
+from abc import ABC
+from typing import Dict, Optional, Union
+
+import httpx
+
+from mem0.configs.base import AzureConfig
+
+
+class BaseEmbedderConfig(ABC):
+    """
+    Config for Embeddings.
+    """
+
+    def __init__(
+        self,
+        model: Optional[str] = None,
+        api_key: Optional[str] = None,
+        embedding_dims: Optional[int] = None,
+        # Ollama specific
+        ollama_base_url: Optional[str] = None,
+        # Openai specific
+        openai_base_url: Optional[str] = None,
+        # Huggingface specific
+        model_kwargs: Optional[dict] = None,
+        huggingface_base_url: Optional[str] = None,
+        # AzureOpenAI specific
+        azure_kwargs: Optional[AzureConfig] = {},
+        http_client_proxies: Optional[Union[Dict, str]] = None,
+        # VertexAI specific
+        vertex_credentials_json: Optional[str] = None,
+        memory_add_embedding_type: Optional[str] = None,
+        memory_update_embedding_type: Optional[str] = None,
+        memory_search_embedding_type: Optional[str] = None,
+        # Gemini specific
+        output_dimensionality: Optional[str] = None,
+        # LM Studio specific
+        lmstudio_base_url: Optional[str] = "http://localhost:1234/v1",
+        # AWS Bedrock specific
+        aws_access_key_id: Optional[str] = None,
+        aws_secret_access_key: Optional[str] = None,
+        aws_region: Optional[str] = "us-west-2",
+    ):
+        """
+        Initializes a configuration class instance for the Embeddings.
+
+        :param model: Embedding model to use, defaults to None
+        :type model: Optional[str], optional
+        :param api_key: API key to be use, defaults to None
+        :type api_key: Optional[str], optional
+        :param embedding_dims: The number of dimensions in the embedding, defaults to None
+        :type embedding_dims: Optional[int], optional
+        :param ollama_base_url: Base URL for the Ollama API, defaults to None
+        :type ollama_base_url: Optional[str], optional
+        :param model_kwargs: key-value arguments for the huggingface embedding model, defaults a dict inside init
+        :type model_kwargs: Optional[Dict[str, Any]], defaults a dict inside init
+        :param huggingface_base_url: Huggingface base URL to be use, defaults to None
+        :type huggingface_base_url: Optional[str], optional
+        :param openai_base_url: Openai base URL to be use, defaults to "https://api.openai.com/v1"
+        :type openai_base_url: Optional[str], optional
+        :param azure_kwargs: key-value arguments for the AzureOpenAI embedding model, defaults a dict inside init
+        :type azure_kwargs: Optional[Dict[str, Any]], defaults a dict inside init
+        :param http_client_proxies: The proxy server settings used to create self.http_client, defaults to None
+        :type http_client_proxies: Optional[Dict | str], optional
+        :param vertex_credentials_json: The path to the Vertex AI credentials JSON file, defaults to None
+        :type vertex_credentials_json: Optional[str], optional
+        :param memory_add_embedding_type: The type of embedding to use for the add memory action, defaults to None
+        :type memory_add_embedding_type: Optional[str], optional
+        :param memory_update_embedding_type: The type of embedding to use for the update memory action, defaults to None
+        :type memory_update_embedding_type: Optional[str], optional
+        :param memory_search_embedding_type: The type of embedding to use for the search memory action, defaults to None
+        :type memory_search_embedding_type: Optional[str], optional
+        :param lmstudio_base_url: LM Studio base URL to be use, defaults to "http://localhost:1234/v1"
+        :type lmstudio_base_url: Optional[str], optional
+        """
+
+        self.model = model
+        self.api_key = api_key
+        self.openai_base_url = openai_base_url
+        self.embedding_dims = embedding_dims
+
+        # AzureOpenAI specific
+        self.http_client = httpx.Client(proxies=http_client_proxies) if http_client_proxies else None
+
+        # Ollama specific
+        self.ollama_base_url = ollama_base_url
+
+        # Huggingface specific
+        self.model_kwargs = model_kwargs or {}
+        self.huggingface_base_url = huggingface_base_url
+        # AzureOpenAI specific
+        self.azure_kwargs = AzureConfig(**azure_kwargs) or {}
+
+        # VertexAI specific
+        self.vertex_credentials_json = vertex_credentials_json
+        self.memory_add_embedding_type = memory_add_embedding_type
+        self.memory_update_embedding_type = memory_update_embedding_type
+        self.memory_search_embedding_type = memory_search_embedding_type
+
+        # Gemini specific
+        self.output_dimensionality = output_dimensionality
+
+        # LM Studio specific
+        self.lmstudio_base_url = lmstudio_base_url
+
+        # AWS Bedrock specific
+        self.aws_access_key_id = aws_access_key_id
+        self.aws_secret_access_key = aws_secret_access_key
+        self.aws_region = aws_region

mem0/configs/enums.py

@@ -0,0 +1,7 @@
+from enum import Enum
+
+
+class MemoryType(Enum):
+    SEMANTIC = "semantic_memory"
+    EPISODIC = "episodic_memory"
+    PROCEDURAL = "procedural_memory"

mem0/configs/llms/base.py

@@ -0,0 +1,152 @@
+from abc import ABC
+from typing import Dict, Optional, Union
+
+import httpx
+
+from mem0.configs.base import AzureConfig
+
+
+class BaseLlmConfig(ABC):
+    """
+    Config for LLMs.
+    """
+
+    def __init__(
+        self,
+        model: Optional[Union[str, Dict]] = None,
+        temperature: float = 0.1,
+        api_key: Optional[str] = None,
+        max_tokens: int = 2000,
+        top_p: float = 0.1,
+        top_k: int = 1,
+        enable_vision: bool = False,
+        vision_details: Optional[str] = "auto",
+        # Openrouter specific
+        models: Optional[list[str]] = None,
+        route: Optional[str] = "fallback",
+        openrouter_base_url: Optional[str] = None,
+        # Openai specific
+        openai_base_url: Optional[str] = None,
+        site_url: Optional[str] = None,
+        app_name: Optional[str] = None,
+        # Ollama specific
+        ollama_base_url: Optional[str] = None,
+        # AzureOpenAI specific
+        azure_kwargs: Optional[AzureConfig] = {},
+        # AzureOpenAI specific
+        http_client_proxies: Optional[Union[Dict, str]] = None,
+        # DeepSeek specific
+        deepseek_base_url: Optional[str] = None,
+        # XAI specific
+        xai_base_url: Optional[str] = None,
+        # Sarvam specific
+        sarvam_base_url: Optional[str] = "https://api.sarvam.ai/v1",
+        # LM Studio specific
+        lmstudio_base_url: Optional[str] = "http://localhost:1234/v1",
+        lmstudio_response_format: dict = None,
+        # vLLM specific
+        vllm_base_url: Optional[str] = "http://localhost:8000/v1",
+        # AWS Bedrock specific
+        aws_access_key_id: Optional[str] = None,
+        aws_secret_access_key: Optional[str] = None,
+        aws_region: Optional[str] = "us-west-2",
+    ):
+        """
+        Initializes a configuration class instance for the LLM.
+
+        :param model: Controls the OpenAI model used, defaults to None
+        :type model: Optional[str], optional
+        :param temperature:  Controls the randomness of the model's output.
+        Higher values (closer to 1) make output more random, lower values make it more deterministic, defaults to 0
+        :type temperature: float, optional
+        :param api_key: OpenAI API key to be use, defaults to None
+        :type api_key: Optional[str], optional
+        :param max_tokens: Controls how many tokens are generated, defaults to 2000
+        :type max_tokens: int, optional
+        :param top_p: Controls the diversity of words. Higher values (closer to 1) make word selection more diverse,
+        defaults to 1
+        :type top_p: float, optional
+        :param top_k: Controls the diversity of words. Higher values make word selection more diverse, defaults to 0
+        :type top_k: int, optional
+        :param enable_vision: Enable vision for the LLM, defaults to False
+        :type enable_vision: bool, optional
+        :param vision_details: Details of the vision to be used [low, high, auto], defaults to "auto"
+        :type vision_details: Optional[str], optional
+        :param models: Openrouter models to use, defaults to None
+        :type models: Optional[list[str]], optional
+        :param route: Openrouter route to be used, defaults to "fallback"
+        :type route: Optional[str], optional
+        :param openrouter_base_url: Openrouter base URL to be use, defaults to "https://openrouter.ai/api/v1"
+        :type openrouter_base_url: Optional[str], optional
+        :param site_url: Openrouter site URL to use, defaults to None
+        :type site_url: Optional[str], optional
+        :param app_name: Openrouter app name to use, defaults to None
+        :type app_name: Optional[str], optional
+        :param ollama_base_url: The base URL of the LLM, defaults to None
+        :type ollama_base_url: Optional[str], optional
+        :param openai_base_url: Openai base URL to be use, defaults to "https://api.openai.com/v1"
+        :type openai_base_url: Optional[str], optional
+        :param azure_kwargs: key-value arguments for the AzureOpenAI LLM model, defaults a dict inside init
+        :type azure_kwargs: Optional[Dict[str, Any]], defaults a dict inside init
+        :param http_client_proxies: The proxy server(s) settings used to create self.http_client, defaults to None
+        :type http_client_proxies: Optional[Dict | str], optional
+        :param deepseek_base_url: DeepSeek base URL to be use, defaults to None
+        :type deepseek_base_url: Optional[str], optional
+        :param xai_base_url: XAI base URL to be use, defaults to None
+        :type xai_base_url: Optional[str], optional
+        :param sarvam_base_url: Sarvam base URL to be use, defaults to "https://api.sarvam.ai/v1"
+        :type sarvam_base_url: Optional[str], optional
+        :param lmstudio_base_url: LM Studio base URL to be use, defaults to "http://localhost:1234/v1"
+        :type lmstudio_base_url: Optional[str], optional
+        :param lmstudio_response_format: LM Studio response format to be use, defaults to None
+        :type lmstudio_response_format: Optional[Dict], optional
+        :param vllm_base_url: vLLM base URL to be use, defaults to "http://localhost:8000/v1"
+        :type vllm_base_url: Optional[str], optional
+        """
+
+        self.model = model
+        self.temperature = temperature
+        self.api_key = api_key
+        self.max_tokens = max_tokens
+        self.top_p = top_p
+        self.top_k = top_k
+        self.enable_vision = enable_vision
+        self.vision_details = vision_details
+
+        # AzureOpenAI specific
+        self.http_client = httpx.Client(proxies=http_client_proxies) if http_client_proxies else None
+
+        # Openrouter specific
+        self.models = models
+        self.route = route
+        self.openrouter_base_url = openrouter_base_url
+        self.openai_base_url = openai_base_url
+        self.site_url = site_url
+        self.app_name = app_name
+
+        # Ollama specific
+        self.ollama_base_url = ollama_base_url
+
+        # DeepSeek specific
+        self.deepseek_base_url = deepseek_base_url
+
+        # AzureOpenAI specific
+        self.azure_kwargs = AzureConfig(**azure_kwargs) or {}
+
+        # XAI specific
+        self.xai_base_url = xai_base_url
+
+        # Sarvam specific
+        self.sarvam_base_url = sarvam_base_url
+
+        # LM Studio specific
+        self.lmstudio_base_url = lmstudio_base_url
+        self.lmstudio_response_format = lmstudio_response_format
+
+        # vLLM specific
+        self.vllm_base_url = vllm_base_url
+
+        # AWS Bedrock specific
+        self.aws_access_key_id = aws_access_key_id
+        self.aws_secret_access_key = aws_secret_access_key
+        self.aws_region = aws_region