llm-dialog-manager 0.4.7__py3-none-any.whl → 0.5.3__py3-none-any.whl

llm_dialog_manager/formatters/openai.py

@@ -0,0 +1,67 @@
+"""
+Message formatter for OpenAI models
+"""
+import io
+import base64
+from typing import List, Dict, Union, Optional
+from PIL import Image
+
+from .base import BaseMessageFormatter
+
+class OpenAIFormatter(BaseMessageFormatter):
+    """Formatter for OpenAI API messages"""
+    
+    def format_messages(self, messages: List[Dict[str, Union[str, List[Union[str, Image.Image, Dict]]]]]) -> tuple:
+        """
+        Format messages for the OpenAI API.
+        
+        Args:
+            messages: List of message dictionaries in standard format
+            
+        Returns:
+            A tuple containing (None, formatted_messages)
+            since OpenAI handles system messages in the message list
+        """
+        formatted = []
+        
+        for msg in messages:
+            content = msg["content"]
+            if isinstance(content, str):
+                formatted.append({"role": msg["role"], "content": content})
+            elif isinstance(content, list):
+                # For OpenAI with multimodal models like GPT-4V
+                formatted_content = []
+                
+                for block in content:
+                    if isinstance(block, str):
+                        formatted_content.append({"type": "text", "text": block})
+                    elif isinstance(block, Image.Image):
+                        # Convert PIL.Image to base64
+                        buffered = io.BytesIO()
+                        block.save(buffered, format="PNG")
+                        image_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
+                        formatted_content.append({
+                            "type": "image_url",
+                            "image_url": {
+                                "url": f"data:image/png;base64,{image_base64}"
+                            }
+                        })
+                    elif isinstance(block, dict):
+                        if block.get("type") == "image_url":
+                            formatted_content.append({
+                                "type": "image_url",
+                                "image_url": {
+                                    "url": block["image_url"]["url"]
+                                }
+                            })
+                        elif block.get("type") == "image_base64":
+                            formatted_content.append({
+                                "type": "image_url",
+                                "image_url": {
+                                    "url": f"data:{block['image_base64']['media_type']};base64,{block['image_base64']['data']}"
+                                }
+                            })
+                
+                formatted.append({"role": msg["role"], "content": formatted_content})
+                
+        return None, formatted

llm_dialog_manager/formatters/x.py

@@ -0,0 +1,77 @@
+"""
+Message formatter for X.AI (Grok) models
+"""
+from typing import List, Dict, Union, Optional
+from PIL import Image
+import io
+import base64
+
+from .base import BaseMessageFormatter
+
+class XFormatter(BaseMessageFormatter):
+    """Formatter for X.AI (Grok) API messages"""
+    
+    def format_messages(self, messages: List[Dict[str, Union[str, List[Union[str, Image.Image, Dict]]]]]) -> tuple:
+        """
+        Format messages for the X.AI (Grok) API.
+        
+        Args:
+            messages: List of message dictionaries in standard format
+            
+        Returns:
+            A tuple containing (system_message, formatted_messages)
+        """
+        system_msg = None
+        formatted = []
+        
+        for msg in messages:
+            # Extract system message if present, similar to many other APIs
+            if msg["role"] == "system":
+                system_msg = msg["content"] if isinstance(msg["content"], str) else str(msg["content"])
+                continue
+                
+            content = msg["content"]
+            if isinstance(content, str):
+                formatted.append({"role": msg["role"], "content": content})
+            elif isinstance(content, list):
+                # Grok API format may need adjustments as it evolves
+                combined_content = []
+                
+                for block in content:
+                    if isinstance(block, str):
+                        combined_content.append({"type": "text", "text": block})
+                    elif isinstance(block, Image.Image):
+                        # Convert PIL.Image to base64
+                        buffered = io.BytesIO()
+                        block.save(buffered, format="PNG")
+                        image_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
+                        combined_content.append({
+                            "type": "image",
+                            "image": {
+                                "type": "base64",
+                                "media_type": "image/png",
+                                "data": image_base64
+                            }
+                        })
+                    elif isinstance(block, dict):
+                        if block.get("type") == "image_url":
+                            combined_content.append({
+                                "type": "image",
+                                "image": {
+                                    "type": "url",
+                                    "url": block["image_url"]["url"]
+                                }
+                            })
+                        elif block.get("type") == "image_base64":
+                            combined_content.append({
+                                "type": "image",
+                                "image": {
+                                    "type": "base64",
+                                    "media_type": block["image_base64"]["media_type"],
+                                    "data": block["image_base64"]["data"]
+                                }
+                            })
+                
+                formatted.append({"role": msg["role"], "content": combined_content})
+                
+        return system_msg, formatted

llm_dialog_manager/utils/__init__.py

@@ -0,0 +1,3 @@
+"""
+Utility modules for the LLM dialog manager.
+"""

llm_dialog_manager/utils/environment.py

@@ -0,0 +1,66 @@
+"""
+Environment utilities for LLM Dialog Manager
+"""
+import os
+import logging
+from pathlib import Path
+from dotenv import load_dotenv
+
+logger = logging.getLogger(__name__)
+
+def load_env_vars(env_path=None):
+    """
+    Load environment variables from .env file.
+    
+    Args:
+        env_path: Optional path to .env file
+    
+    Returns:
+        True if env vars were loaded, False otherwise
+    """
+    try:
+        # Default to .env in the current directory or parent directory
+        if not env_path:
+            if os.path.exists(".env"):
+                env_path = ".env"
+            elif os.path.exists("../.env"):
+                env_path = "../.env"
+            else:
+                # Try to find .env in parent directories
+                current_dir = Path.cwd()
+                for parent in current_dir.parents:
+                    potential_path = parent / ".env"
+                    if potential_path.exists():
+                        env_path = str(potential_path)
+                        break
+        
+        if env_path and os.path.exists(env_path):
+            load_dotenv(env_path)
+            logger.info(f"Loaded environment variables from {env_path}")
+            
+            # Log detected providers without showing sensitive data
+            providers = []
+            if os.getenv("OPENAI_API_KEY"):
+                providers.append("OpenAI")
+            if os.getenv("ANTHROPIC_API_KEY"):
+                providers.append("Anthropic")
+            if os.getenv("AWS_ACCESS_KEY_ID") and os.getenv("AWS_SECRET_ACCESS_KEY"):
+                providers.append("AWS Bedrock (for anthropic.* models)")
+            if os.getenv("VERTEX_PROJECT_ID") and os.getenv("VERTEX_REGION"):
+                providers.append("Anthropic Vertex")
+            if os.getenv("GEMINI_API_KEY"):
+                providers.append("Google Gemini")
+            if os.getenv("XAI_API_KEY"):
+                providers.append("X.AI Grok")
+            
+            if providers:
+                logger.info(f"Detected LLM providers: {', '.join(providers)}")
+            
+            return True
+        else:
+            logger.warning(f"Environment file not found: {env_path}")
+            return False
+    
+    except Exception as e:
+        logger.error(f"Error loading environment variables: {e}")
+        return False

llm_dialog_manager/utils/image_tools.py

@@ -0,0 +1,81 @@
+"""
+Image handling utilities for LLM dialog manager
+"""
+import base64
+import io
+from typing import Optional
+import requests
+from PIL import Image
+
+def load_image_from_path(image_path: str) -> Image.Image:
+    """
+    Load an image from a local file path.
+    
+    Args:
+        image_path: Path to the local image file
+        
+    Returns:
+        PIL.Image object
+    """
+    try:
+        image = Image.open(image_path)
+        # Store the filename for reference
+        image.filename = image_path
+        return image
+    except Exception as e:
+        raise ValueError(f"Failed to load image from {image_path}: {e}")
+
+def load_image_from_url(image_url: str) -> Image.Image:
+    """
+    Load an image from a URL.
+    
+    Args:
+        image_url: URL to the image
+        
+    Returns:
+        PIL.Image object
+    """
+    try:
+        response = requests.get(image_url, stream=True)
+        response.raise_for_status()
+        image = Image.open(io.BytesIO(response.content))
+        # Store the URL for reference
+        image.filename = image_url
+        return image
+    except Exception as e:
+        raise ValueError(f"Failed to load image from {image_url}: {e}")
+
+def encode_image_to_base64(image: Image.Image, format: str = "PNG") -> str:
+    """
+    Convert a PIL.Image to base64 string.
+    
+    Args:
+        image: PIL.Image object
+        format: Image format (default: PNG)
+        
+    Returns:
+        Base64-encoded string
+    """
+    buffered = io.BytesIO()
+    image.save(buffered, format=format)
+    return base64.b64encode(buffered.getvalue()).decode("utf-8")
+
+def create_image_content_block(image: Image.Image, media_type: str = "image/png") -> dict:
+    """
+    Create a standard image content block for messages.
+    
+    Args:
+        image: PIL.Image object
+        media_type: MIME type of the image
+        
+    Returns:
+        Dictionary with image information
+    """
+    image_base64 = encode_image_to_base64(image)
+    return {
+        "type": "image_base64",
+        "image_base64": {
+            "media_type": media_type,
+            "data": image_base64
+        }
+    }

llm_dialog_manager/utils/logging.py

@@ -0,0 +1,35 @@
+"""
+Logging configuration utilities
+"""
+import logging
+import sys
+
+def setup_logging(level=logging.INFO, log_file=None):
+    """
+    Configure logging for the application.
+    
+    Args:
+        level: Logging level (default: INFO)
+        log_file: Optional file path to write logs to
+    """
+    # Create logger
+    logger = logging.getLogger('llm_dialog_manager')
+    logger.setLevel(level)
+    
+    # Create formatter
+    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+    
+    # Create console handler
+    console_handler = logging.StreamHandler(sys.stdout)
+    console_handler.setLevel(level)
+    console_handler.setFormatter(formatter)
+    logger.addHandler(console_handler)
+    
+    # Create file handler if log_file specified
+    if log_file:
+        file_handler = logging.FileHandler(log_file)
+        file_handler.setLevel(level)
+        file_handler.setFormatter(formatter)
+        logger.addHandler(file_handler)
+    
+    return logger

{llm_dialog_manager-0.4.7.dist-info → llm_dialog_manager-0.5.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: llm_dialog_manager
-Version: 0.4.7
+Version: 0.5.3
 Summary: A Python package for managing LLM chat conversation history
 Author-email: xihajun <work@2333.fun>
 License: MIT
@@ -103,7 +103,7 @@ XAI_API_KEY=your-x-key
 from llm_dialog_manager import Agent
 
 # Initialize an agent with a specific model
-agent = Agent("claude-2.1", memory_enabled=True)
+agent = Agent("ep-20250319212209-j6tfj-openai", memory_enabled=True)
 
 # Add messages and generate responses
 agent.add_message("system", "You are a helpful assistant")

llm_dialog_manager-0.5.3.dist-info/RECORD

@@ -0,0 +1,25 @@
+llm_dialog_manager/__init__.py,sha256=T4hzIO_4IANFIcjmF0iEJ-_h7LR1_AT13kj5dc3hH-I,463
+llm_dialog_manager/agent.py,sha256=Am2p9fClcHC75_Yjz6b1bof_KeUYmJocATQIQ-KKcr0,6472
+llm_dialog_manager/chat_history.py,sha256=DKKRnj_M6h-4JncnH6KekMTghX7vMgdN3J9uOwXKzMU,10347
+llm_dialog_manager/key_manager.py,sha256=shvxmn4zUtQx_p-x1EFyOmnk-WlhigbpKtxTKve-zXk,4421
+llm_dialog_manager/clients/__init__.py,sha256=f8NFm-9bXUJUrFy37FxpJlyyCqVwHPu3VUtfNdKUVmE,1114
+llm_dialog_manager/clients/anthropic_client.py,sha256=5Zqyosoy60JkWI0cjQAxJ9Uk4bcqHrVEXkauEfC8Egs,5404
+llm_dialog_manager/clients/base.py,sha256=hSkuLLKrdyoKMCRpPav5qbsKLaJLQ9gtMDSxxdAHdII,2107
+llm_dialog_manager/clients/gemini_client.py,sha256=AwIVJubdawrNa4DobCKE0KdtwuLU5XzmOtN_I-SzStY,2540
+llm_dialog_manager/clients/openai_client.py,sha256=8glZbLaUQw2gftOgMeV0TN6aTJ--2LuRxn7wiCK8K4Q,3266
+llm_dialog_manager/clients/x_client.py,sha256=rTaEU9NWTws2XIqfNQt1vysT_fdhWCMAGpfF1onMU6c,2688
+llm_dialog_manager/formatters/__init__.py,sha256=LdvQOhkGbF5_iT7Lu6kTssXKRXtuQTq9LluK4CSplws,787
+llm_dialog_manager/formatters/anthropic.py,sha256=0xFDooeEYq4DO1RWiGxbhuiL_DwtDdUNLEuiJyXbQnQ,3247
+llm_dialog_manager/formatters/base.py,sha256=YBpnfR4Pc7-P4TbCVNe4Q6pOUq9mXwQQR1U9X1NSmwE,711
+llm_dialog_manager/formatters/gemini.py,sha256=im-ZPGOkwDiWcgDFC2d0owKmpdKJ9wiLh5treoV2E-4,2425
+llm_dialog_manager/formatters/openai.py,sha256=i6AAORyEkjRQwXAf9QkKBIxZGtD7a2k0CvcPbilSjBQ,2807
+llm_dialog_manager/formatters/x.py,sha256=qxZdidngZUDp0DazYbolye9BnjsL8MMniLVdlaJ_uik,3248
+llm_dialog_manager/utils/__init__.py,sha256=ta2BdLF2CSzlbk5ZwArzar4E7eip4reQjMAncvY7qhQ,52
+llm_dialog_manager/utils/environment.py,sha256=8jRTe1_JL24Wz4SdX968XKx8vJMA4-NrZtHhXv-9PJM,2322
+llm_dialog_manager/utils/image_tools.py,sha256=Q23gpY-JY0MIGOVWnLl-QS2IrpCMBiyzn2Zu0dWov1c,2110
+llm_dialog_manager/utils/logging.py,sha256=38E_k3OSz7CNFg3ZsildR7IMtn3CzWro7IRY-Sj7dcc,989
+llm_dialog_manager-0.5.3.dist-info/licenses/LICENSE,sha256=vWGbYgGuWpWrXL8-xi6pNcX5UzD6pWoIAZmcetyfbus,1064
+llm_dialog_manager-0.5.3.dist-info/METADATA,sha256=sUFwt6A0CwO5hGGf6iAgctBJwm2T6-CVG2hhWHBU2hk,4236
+llm_dialog_manager-0.5.3.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+llm_dialog_manager-0.5.3.dist-info/top_level.txt,sha256=u2EQEXW0NGAt0AAHT7jx1odXZ4rZfjcgbmJhvKFuMkI,19
+llm_dialog_manager-0.5.3.dist-info/RECORD,,

{llm_dialog_manager-0.4.7.dist-info → llm_dialog_manager-0.5.3.dist-info}/WHEEL

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

llm_dialog_manager-0.4.7.dist-info/RECORD

@@ -1,9 +0,0 @@
-llm_dialog_manager/__init__.py,sha256=uWTAXzyON6m5vqj_MS8_Pe7hga5TZ2YDhkMh0Z-FDew,86
-llm_dialog_manager/agent.py,sha256=LGoX6erGcn9eaM4wRGfs0N5zS8Lop_57nPKzzBWrtk8,27871
-llm_dialog_manager/chat_history.py,sha256=DKKRnj_M6h-4JncnH6KekMTghX7vMgdN3J9uOwXKzMU,10347
-llm_dialog_manager/key_manager.py,sha256=shvxmn4zUtQx_p-x1EFyOmnk-WlhigbpKtxTKve-zXk,4421
-llm_dialog_manager-0.4.7.dist-info/licenses/LICENSE,sha256=vWGbYgGuWpWrXL8-xi6pNcX5UzD6pWoIAZmcetyfbus,1064
-llm_dialog_manager-0.4.7.dist-info/METADATA,sha256=mRNR-f-rRqDR_IFoFJe2oNcTVE27Ah6AN8ENGQwfzVg,4216
-llm_dialog_manager-0.4.7.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-llm_dialog_manager-0.4.7.dist-info/top_level.txt,sha256=u2EQEXW0NGAt0AAHT7jx1odXZ4rZfjcgbmJhvKFuMkI,19
-llm_dialog_manager-0.4.7.dist-info/RECORD,,