mem0ai-azure-mysql 0.1.115.1__tar.gz → 0.1.116.1__tar.gz

{mem0ai_azure_mysql-0.1.115.1 → mem0ai_azure_mysql-0.1.116.1}/.gitignore

@@ -164,6 +164,7 @@ cython_debug/
 .todos.md
 
 # Database
+db
 test-db
 !embedchain/embedchain/core/db/
 

mem0ai_azure_mysql-0.1.116.1/PKG-INFO

@@ -0,0 +1,87 @@
+Metadata-Version: 2.4
+Name: mem0ai-azure-mysql
+Version: 0.1.116.1
+Summary: Long-term memory for AI Agents with Azure DefaultAzureCredential authentication and MySQL history database support
+Requires-Python: <4.0,>=3.9
+Requires-Dist: azure-identity>=1.23.1
+Requires-Dist: azure-search-documents>=11.5.3
+Requires-Dist: langchain-neo4j>=0.4.0
+Requires-Dist: openai<1.100.0,>=1.90.0
+Requires-Dist: posthog>=3.5.0
+Requires-Dist: protobuf<6.0.0,>=5.29.0
+Requires-Dist: pydantic>=2.7.3
+Requires-Dist: pymysql
+Requires-Dist: pytz>=2024.1
+Requires-Dist: qdrant-client>=1.9.1
+Requires-Dist: rank-bm25>=0.2.2
+Requires-Dist: sqlalchemy>=2.0.31
+Provides-Extra: dev
+Requires-Dist: isort>=5.13.2; 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'
+Requires-Dist: elasticsearch>=8.0.0; extra == 'extras'
+Requires-Dist: langchain-community>=0.0.0; extra == 'extras'
+Requires-Dist: langchain-memgraph>=0.1.0; extra == 'extras'
+Requires-Dist: opensearch-py>=2.0.0; extra == 'extras'
+Requires-Dist: sentence-transformers>=5.0.0; extra == 'extras'
+Provides-Extra: graph
+Requires-Dist: kuzu>=0.11.0; extra == 'graph'
+Requires-Dist: langchain-aws>=0.2.23; extra == 'graph'
+Requires-Dist: langchain-neo4j>=0.4.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: google-genai>=1.0.0; extra == 'llms'
+Requires-Dist: google-generativeai>=0.3.0; extra == 'llms'
+Requires-Dist: groq>=0.3.0; extra == 'llms'
+Requires-Dist: litellm>=1.74.0; extra == 'llms'
+Requires-Dist: ollama>=0.1.0; extra == 'llms'
+Requires-Dist: openai<1.100.0,>=1.90.0; extra == 'llms'
+Requires-Dist: together>=0.2.10; extra == 'llms'
+Requires-Dist: vertexai>=0.1.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'
+Requires-Dist: chromadb>=0.4.24; extra == 'vector-stores'
+Requires-Dist: databricks-sdk>=0.63.0; extra == 'vector-stores'
+Requires-Dist: faiss-cpu>=1.7.4; extra == 'vector-stores'
+Requires-Dist: pinecone-text>=0.10.0; extra == 'vector-stores'
+Requires-Dist: pinecone<=7.3.0; extra == 'vector-stores'
+Requires-Dist: psycopg>=3.2.8; extra == 'vector-stores'
+Requires-Dist: pymochow>=2.2.9; extra == 'vector-stores'
+Requires-Dist: pymongo>=4.13.2; extra == 'vector-stores'
+Requires-Dist: upstash-vector>=0.1.0; extra == 'vector-stores'
+Requires-Dist: vecs>=0.4.0; extra == 'vector-stores'
+Requires-Dist: weaviate-client<4.15.0,>=4.4.0; 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. Azure Entra ID Authentication
+- **Azure AI Search**: Support for Azure Entra ID (Azure AD) authentication using [`DefaultAzureCredential`](mem0/vector_stores/azure_ai_search.py:114)
+- **Azure OpenAI**: Seamless Entra ID integration for both LLM and embedding services using [`DefaultAzureCredential`](mem0/llms/azure_openai.py:37)
+- **Simplified Authentication**: No need to manage API keys when using managed identities or service principals
+
+### 2. MySQL Database Support
+- **Production-Ready**: Replace SQLite3 with enterprise-grade [`MySQL`](mem0/dbs/mysql.py:18) for scalable memory history storage
+- **Connection Pooling**: Built-in connection pooling and SSL support for secure connections
+- **Migration Support**: Automatic schema migration from existing SQLite databases
+- **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
+```

mem0ai_azure_mysql-0.1.116.1/README.md

@@ -0,0 +1,24 @@
+# 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. Azure Entra ID Authentication
+- **Azure AI Search**: Support for Azure Entra ID (Azure AD) authentication using [`DefaultAzureCredential`](mem0/vector_stores/azure_ai_search.py:114)
+- **Azure OpenAI**: Seamless Entra ID integration for both LLM and embedding services using [`DefaultAzureCredential`](mem0/llms/azure_openai.py:37)
+- **Simplified Authentication**: No need to manage API keys when using managed identities or service principals
+
+### 2. MySQL Database Support
+- **Production-Ready**: Replace SQLite3 with enterprise-grade [`MySQL`](mem0/dbs/mysql.py:18) for scalable memory history storage
+- **Connection Pooling**: Built-in connection pooling and SSL support for secure connections
+- **Migration Support**: Automatic schema migration from existing SQLite databases
+- **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
+```

{mem0ai_azure_mysql-0.1.115.1 → mem0ai_azure_mysql-0.1.116.1}/mem0/client/main.py

@@ -267,10 +267,14 @@ class MemoryClient:
         Update a memory by ID.
         Args:
             memory_id (str): Memory ID.
-            text (str, optional): Data to update in the memory.
+            text (str, optional): New content to update the memory with.
             metadata (dict, optional): Metadata to update in the memory.
+            
         Returns:
             Dict[str, Any]: The response from the server.
+            
+        Example:
+            >>> client.update(memory_id="mem_123", text="Likes to play tennis on weekends")
         """
         if text is None and metadata is None:
             raise ValueError("Either text or metadata must be provided for update.")
@@ -447,16 +451,13 @@ class MemoryClient:
         """Batch update memories.
 
         Args:
-            memories: List of memory dictionaries to update. Each dictionary
-                      must contain:
+            memories: List of memory dictionaries to update. Each dictionary must contain:
                 - memory_id (str): ID of the memory to update
-                - text (str): New text content for the memory
+                - text (str, optional): New text content for the memory
+                - metadata (dict, optional): New metadata for the memory
 
         Returns:
-            str: Message indicating the success of the batch update.
-
-        Raises:
-            APIError: If the API request fails.
+            Dict[str, Any]: The response from the server.
         """
         response = self.client.put("/v1/batch/", json={"memories": memories})
         response.raise_for_status()
@@ -1057,13 +1058,18 @@ class AsyncMemoryClient:
         self, memory_id: str, text: Optional[str] = None, metadata: Optional[Dict[str, Any]] = None
     ) -> Dict[str, Any]:
         """
-        Update a memory by ID.
+        Update a memory by ID asynchronously.
+        
         Args:
             memory_id (str): Memory ID.
-            text (str, optional): Data to update in the memory.
+            text (str, optional): New content to update the memory with.
             metadata (dict, optional): Metadata to update in the memory.
+            
         Returns:
             Dict[str, Any]: The response from the server.
+            
+        Example:
+            >>> await client.update(memory_id="mem_123", text="Likes to play tennis on weekends")
         """
         if text is None and metadata is None:
             raise ValueError("Either text or metadata must be provided for update.")
@@ -1232,16 +1238,13 @@ class AsyncMemoryClient:
         """Batch update memories.
 
         Args:
-            memories: List of memory dictionaries to update. Each dictionary
-                      must contain:
+            memories: List of memory dictionaries to update. Each dictionary must contain:
                 - memory_id (str): ID of the memory to update
-                - text (str): New text content for the memory
+                - text (str, optional): New text content for the memory
+                - metadata (dict, optional): New metadata for the memory
 
         Returns:
-            str: Message indicating the success of the batch update.
-
-        Raises:
-            APIError: If the API request fails.
+            Dict[str, Any]: The response from the server.
         """
         response = await self.async_client.put("/v1/batch/", json={"memories": memories})
         response.raise_for_status()

{mem0ai_azure_mysql-0.1.115.1 → mem0ai_azure_mysql-0.1.116.1}/mem0/configs/base.py

@@ -78,13 +78,27 @@ class AzureConfig(BaseModel):
         default_headers (Dict[str, str]): Headers to include in requests to the Azure API.
     """
 
-    api_key: str = Field(
+    api_key: str | None = 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)
+    azure_ad_token: str | None = Field(
+        description="The Azure AD token used for authentication.",
+        default=None,
+    )
+    azure_deployment: str | None = Field(
+        description="The name of the Azure deployment.",
+        default=None,
+    )
+    azure_endpoint: str | None = Field(
+        description="The endpoint URL for the Azure service.",
+        default=None,
+    )
+    api_version: str | None = 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
+        description="Headers to include in requests to the Azure API.",
+        default=None
     )

mem0ai_azure_mysql-0.1.116.1/mem0/configs/llms/anthropic.py

@@ -0,0 +1,56 @@
+from typing import Optional
+
+from mem0.configs.llms.base import BaseLlmConfig
+
+
+class AnthropicConfig(BaseLlmConfig):
+    """
+    Configuration class for Anthropic-specific parameters.
+    Inherits from BaseLlmConfig and adds Anthropic-specific settings.
+    """
+
+    def __init__(
+        self,
+        # Base parameters
+        model: Optional[str] = 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",
+        http_client_proxies: Optional[dict] = None,
+        # Anthropic-specific parameters
+        anthropic_base_url: Optional[str] = None,
+    ):
+        """
+        Initialize Anthropic configuration.
+
+        Args:
+            model: Anthropic model to use, defaults to None
+            temperature: Controls randomness, defaults to 0.1
+            api_key: Anthropic API key, defaults to None
+            max_tokens: Maximum tokens to generate, defaults to 2000
+            top_p: Nucleus sampling parameter, defaults to 0.1
+            top_k: Top-k sampling parameter, defaults to 1
+            enable_vision: Enable vision capabilities, defaults to False
+            vision_details: Vision detail level, defaults to "auto"
+            http_client_proxies: HTTP client proxy settings, defaults to None
+            anthropic_base_url: Anthropic API base URL, defaults to None
+        """
+        # Initialize base parameters
+        super().__init__(
+            model=model,
+            temperature=temperature,
+            api_key=api_key,
+            max_tokens=max_tokens,
+            top_p=top_p,
+            top_k=top_k,
+            enable_vision=enable_vision,
+            vision_details=vision_details,
+            http_client_proxies=http_client_proxies,
+        )
+
+        # Anthropic-specific parameters
+        self.anthropic_base_url = anthropic_base_url

mem0ai_azure_mysql-0.1.116.1/mem0/configs/llms/aws_bedrock.py

@@ -0,0 +1,191 @@
+from typing import Optional, Dict, Any, List
+from mem0.configs.llms.base import BaseLlmConfig
+import os
+
+
+class AWSBedrockConfig(BaseLlmConfig):
+    """
+    Configuration class for AWS Bedrock LLM integration.
+
+    Supports all available Bedrock models with automatic provider detection.
+    """
+
+    def __init__(
+        self,
+        model: Optional[str] = None,
+        temperature: float = 0.1,
+        max_tokens: int = 2000,
+        top_p: float = 0.9,
+        top_k: int = 1,
+        aws_access_key_id: Optional[str] = None,
+        aws_secret_access_key: Optional[str] = None,
+        aws_region: str = "us-west-2",
+        aws_session_token: Optional[str] = None,
+        aws_profile: Optional[str] = None,
+        model_kwargs: Optional[Dict[str, Any]] = None,
+        **kwargs,
+    ):
+        """
+        Initialize AWS Bedrock configuration.
+
+        Args:
+            model: Bedrock model identifier (e.g., "amazon.nova-3-mini-20241119-v1:0")
+            temperature: Controls randomness (0.0 to 2.0)
+            max_tokens: Maximum tokens to generate
+            top_p: Nucleus sampling parameter (0.0 to 1.0)
+            top_k: Top-k sampling parameter (1 to 40)
+            aws_access_key_id: AWS access key (optional, uses env vars if not provided)
+            aws_secret_access_key: AWS secret key (optional, uses env vars if not provided)
+            aws_region: AWS region for Bedrock service
+            aws_session_token: AWS session token for temporary credentials
+            aws_profile: AWS profile name for credentials
+            model_kwargs: Additional model-specific parameters
+            **kwargs: Additional arguments passed to base class
+        """
+        super().__init__(
+            model=model or "anthropic.claude-3-5-sonnet-20240620-v1:0",
+            temperature=temperature,
+            max_tokens=max_tokens,
+            top_p=top_p,
+            top_k=top_k,
+            **kwargs,
+        )
+
+        self.aws_access_key_id = aws_access_key_id
+        self.aws_secret_access_key = aws_secret_access_key
+        self.aws_region = aws_region
+        self.aws_session_token = aws_session_token
+        self.aws_profile = aws_profile
+        self.model_kwargs = model_kwargs or {}
+
+    @property
+    def provider(self) -> str:
+        """Get the provider from the model identifier."""
+        if not self.model or "." not in self.model:
+            return "unknown"
+        return self.model.split(".")[0]
+
+    @property
+    def model_name(self) -> str:
+        """Get the model name without provider prefix."""
+        if not self.model or "." not in self.model:
+            return self.model
+        return ".".join(self.model.split(".")[1:])
+
+    def get_model_config(self) -> Dict[str, Any]:
+        """Get model-specific configuration parameters."""
+        base_config = {
+            "temperature": self.temperature,
+            "max_tokens": self.max_tokens,
+            "top_p": self.top_p,
+            "top_k": self.top_k,
+        }
+
+        # Add custom model kwargs
+        base_config.update(self.model_kwargs)
+
+        return base_config
+
+    def get_aws_config(self) -> Dict[str, Any]:
+        """Get AWS configuration parameters."""
+        config = {
+            "region_name": self.aws_region,
+        }
+
+        if self.aws_access_key_id:
+            config["aws_access_key_id"] = self.aws_access_key_id or os.getenv("AWS_ACCESS_KEY_ID")
+            
+        if self.aws_secret_access_key:
+            config["aws_secret_access_key"] = self.aws_secret_access_key or os.getenv("AWS_SECRET_ACCESS_KEY")
+            
+        if self.aws_session_token:
+            config["aws_session_token"] = self.aws_session_token or os.getenv("AWS_SESSION_TOKEN")
+            
+        if self.aws_profile:
+            config["profile_name"] = self.aws_profile or os.getenv("AWS_PROFILE")
+
+        return config
+
+    def validate_model_format(self) -> bool:
+        """
+        Validate that the model identifier follows Bedrock naming convention.
+        
+        Returns:
+            True if valid, False otherwise
+        """
+        if not self.model:
+            return False
+            
+        # Check if model follows provider.model-name format
+        if "." not in self.model:
+            return False
+            
+        provider, model_name = self.model.split(".", 1)
+        
+        # Validate provider
+        valid_providers = [
+            "ai21", "amazon", "anthropic", "cohere", "meta", "mistral", 
+            "stability", "writer", "deepseek", "gpt-oss", "perplexity", 
+            "snowflake", "titan", "command", "j2", "llama"
+        ]
+        
+        if provider not in valid_providers:
+            return False
+            
+        # Validate model name is not empty
+        if not model_name:
+            return False
+            
+        return True
+
+    def get_supported_regions(self) -> List[str]:
+        """Get list of AWS regions that support Bedrock."""
+        return [
+            "us-east-1",
+            "us-west-2",
+            "us-east-2",
+            "eu-west-1",
+            "ap-southeast-1",
+            "ap-northeast-1",
+        ]
+
+    def get_model_capabilities(self) -> Dict[str, Any]:
+        """Get model capabilities based on provider."""
+        capabilities = {
+            "supports_tools": False,
+            "supports_vision": False,
+            "supports_streaming": False,
+            "supports_multimodal": False,
+        }
+        
+        if self.provider == "anthropic":
+            capabilities.update({
+                "supports_tools": True,
+                "supports_vision": True,
+                "supports_streaming": True,
+                "supports_multimodal": True,
+            })
+        elif self.provider == "amazon":
+            capabilities.update({
+                "supports_tools": True,
+                "supports_vision": True,
+                "supports_streaming": True,
+                "supports_multimodal": True,
+            })
+        elif self.provider == "cohere":
+            capabilities.update({
+                "supports_tools": True,
+                "supports_streaming": True,
+            })
+        elif self.provider == "meta":
+            capabilities.update({
+                "supports_vision": True,
+                "supports_streaming": True,
+            })
+        elif self.provider == "mistral":
+            capabilities.update({
+                "supports_vision": True,
+                "supports_streaming": True,
+            })
+            
+        return capabilities

mem0ai_azure_mysql-0.1.116.1/mem0/configs/llms/azure.py

@@ -0,0 +1,57 @@
+from typing import Any, Dict, Optional
+
+from mem0.configs.base import AzureConfig
+from mem0.configs.llms.base import BaseLlmConfig
+
+
+class AzureOpenAIConfig(BaseLlmConfig):
+    """
+    Configuration class for Azure OpenAI-specific parameters.
+    Inherits from BaseLlmConfig and adds Azure OpenAI-specific settings.
+    """
+
+    def __init__(
+        self,
+        # Base parameters
+        model: Optional[str] = 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",
+        http_client_proxies: Optional[dict] = None,
+        # Azure OpenAI-specific parameters
+        azure_kwargs: Optional[Dict[str, Any]] = None,
+    ):
+        """
+        Initialize Azure OpenAI configuration.
+
+        Args:
+            model: Azure OpenAI model to use, defaults to None
+            temperature: Controls randomness, defaults to 0.1
+            api_key: Azure OpenAI API key, defaults to None
+            max_tokens: Maximum tokens to generate, defaults to 2000
+            top_p: Nucleus sampling parameter, defaults to 0.1
+            top_k: Top-k sampling parameter, defaults to 1
+            enable_vision: Enable vision capabilities, defaults to False
+            vision_details: Vision detail level, defaults to "auto"
+            http_client_proxies: HTTP client proxy settings, defaults to None
+            azure_kwargs: Azure-specific configuration, defaults to None
+        """
+        # Initialize base parameters
+        super().__init__(
+            model=model,
+            temperature=temperature,
+            api_key=api_key,
+            max_tokens=max_tokens,
+            top_p=top_p,
+            top_k=top_k,
+            enable_vision=enable_vision,
+            vision_details=vision_details,
+            http_client_proxies=http_client_proxies,
+        )
+
+        # Azure OpenAI-specific parameters
+        self.azure_kwargs = AzureConfig(**(azure_kwargs or {}))

mem0ai_azure_mysql-0.1.116.1/mem0/configs/llms/base.py

@@ -0,0 +1,62 @@
+from abc import ABC
+from typing import Dict, Optional, Union
+
+import httpx
+
+
+class BaseLlmConfig(ABC):
+    """
+    Base configuration for LLMs with only common parameters.
+    Provider-specific configurations should be handled by separate config classes.
+
+    This class contains only the parameters that are common across all LLM providers.
+    For provider-specific parameters, use the appropriate provider config class.
+    """
+
+    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",
+        http_client_proxies: Optional[Union[Dict, str]] = None,
+    ):
+        """
+        Initialize a base configuration class instance for the LLM.
+
+        Args:
+            model: The model identifier to use (e.g., "gpt-4o-mini", "claude-3-5-sonnet-20240620")
+                Defaults to None (will be set by provider-specific configs)
+            temperature: Controls the randomness of the model's output.
+                Higher values (closer to 1) make output more random, lower values make it more deterministic.
+                Range: 0.0 to 2.0. Defaults to 0.1
+            api_key: API key for the LLM provider. If None, will try to get from environment variables.
+                Defaults to None
+            max_tokens: Maximum number of tokens to generate in the response.
+                Range: 1 to 4096 (varies by model). Defaults to 2000
+            top_p: Nucleus sampling parameter. Controls diversity via nucleus sampling.
+                Higher values (closer to 1) make word selection more diverse.
+                Range: 0.0 to 1.0. Defaults to 0.1
+            top_k: Top-k sampling parameter. Limits the number of tokens considered for each step.
+                Higher values make word selection more diverse.
+                Range: 1 to 40. Defaults to 1
+            enable_vision: Whether to enable vision capabilities for the model.
+                Only applicable to vision-enabled models. Defaults to False
+            vision_details: Level of detail for vision processing.
+                Options: "low", "high", "auto". Defaults to "auto"
+            http_client_proxies: Proxy settings for HTTP client.
+                Can be a dict or string. Defaults to None
+        """
+        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
+        self.http_client = httpx.Client(proxies=http_client_proxies) if http_client_proxies else None

mem0ai_azure_mysql-0.1.116.1/mem0/configs/llms/deepseek.py

@@ -0,0 +1,56 @@
+from typing import Optional
+
+from mem0.configs.llms.base import BaseLlmConfig
+
+
+class DeepSeekConfig(BaseLlmConfig):
+    """
+    Configuration class for DeepSeek-specific parameters.
+    Inherits from BaseLlmConfig and adds DeepSeek-specific settings.
+    """
+
+    def __init__(
+        self,
+        # Base parameters
+        model: Optional[str] = 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",
+        http_client_proxies: Optional[dict] = None,
+        # DeepSeek-specific parameters
+        deepseek_base_url: Optional[str] = None,
+    ):
+        """
+        Initialize DeepSeek configuration.
+
+        Args:
+            model: DeepSeek model to use, defaults to None
+            temperature: Controls randomness, defaults to 0.1
+            api_key: DeepSeek API key, defaults to None
+            max_tokens: Maximum tokens to generate, defaults to 2000
+            top_p: Nucleus sampling parameter, defaults to 0.1
+            top_k: Top-k sampling parameter, defaults to 1
+            enable_vision: Enable vision capabilities, defaults to False
+            vision_details: Vision detail level, defaults to "auto"
+            http_client_proxies: HTTP client proxy settings, defaults to None
+            deepseek_base_url: DeepSeek API base URL, defaults to None
+        """
+        # Initialize base parameters
+        super().__init__(
+            model=model,
+            temperature=temperature,
+            api_key=api_key,
+            max_tokens=max_tokens,
+            top_p=top_p,
+            top_k=top_k,
+            enable_vision=enable_vision,
+            vision_details=vision_details,
+            http_client_proxies=http_client_proxies,
+        )
+
+        # DeepSeek-specific parameters
+        self.deepseek_base_url = deepseek_base_url