llm-dialog-manager 0.1.2480__tar.gz → 0.2.2__tar.gz

{llm_dialog_manager-0.1.2480 → llm_dialog_manager-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm_dialog_manager
-Version: 0.1.2480
+Version: 0.2.2
 Summary: A Python package for managing LLM chat conversation history
 Author-email: xihajun <work@2333.fun>
 License: MIT
@@ -73,23 +73,29 @@ pip install llm-dialog-manager
 
 ## Quick Start
 
-### Basic Usage
 
-```python
-from llm_dialog_manager import ChatHistory
+### Environment Variables
 
-# Initialize with a system message
-history = ChatHistory("You are a helpful assistant")
+Create a `.env` file in your project root:
 
-# Add messages
-history.add_user_message("Hello!")
-history.add_assistant_message("Hi there! How can I help you today?")
+```bash
+# OpenAI
+OPENAI_API_KEY_1=your-key-1
+OPENAI_API_BASE_1=https://api.openai.com/v1
 
-# Print conversation
-print(history)
+# Anthropic
+ANTHROPIC_API_KEY_1=your-anthropic-key
+ANTHROPIC_API_BASE_1=https://api.anthropic.com
+
+# Google
+GEMINI_API_KEY=your-gemini-key
+
+# X.AI
+XAI_API_KEY=your-x-key
 ```
 
-### Using the AI Agent
+### Basic Usage
+
 
 ```python
 from llm_dialog_manager import Agent
@@ -106,65 +112,15 @@ response = agent.generate_response()
 agent.save_conversation()
 ```
 
-## Advanced Features
-
-### Managing Multiple API Keys
-
-```python
-from llm_dialog_manager import Agent
-
-# Use specific API key
-agent = Agent("gpt-4", api_key="your-api-key")
-
-# Or use environment variables
-# OPENAI_API_KEY_1=key1
-# OPENAI_API_KEY_2=key2
-# The system will automatically handle load balancing
-```
-
-### Conversation Management
-
-```python
-from llm_dialog_manager import ChatHistory
-
-history = ChatHistory()
-
-# Add messages with role validation
-history.add_message("Hello system", "system")
-history.add_message("Hello user", "user")
-history.add_message("Hello assistant", "assistant")
-
-# Search conversations
-results = history.search_for_keyword("hello")
-
-# Get conversation status
-status = history.conversation_status()
-history.display_conversation_status()
-
-# Get conversation snippets
-snippet = history.get_conversation_snippet(1)
-history.display_snippet(1)
-```
-
-## Environment Variables
-
-Create a `.env` file in your project root:
+### Setup Debugging Console
 
 ```bash
-# OpenAI
-OPENAI_API_KEY_1=your-key-1
-OPENAI_API_BASE_1=https://api.openai.com/v1
-
-# Anthropic
-ANTHROPIC_API_KEY_1=your-anthropic-key
-ANTHROPIC_API_BASE_1=https://api.anthropic.com
+python app.py
+# open localhost:8000
+```
+https://github.com/user-attachments/assets/5f640029-24e6-44ea-a3a3-02eb3de0d4df
 
-# Google
-GEMINI_API_KEY=your-gemini-key
 
-# X.AI
-XAI_API_KEY=your-x-key
-```
 
 ## Development
 

{llm_dialog_manager-0.1.2480 → llm_dialog_manager-0.2.2}/README.md

@@ -26,23 +26,29 @@ pip install llm-dialog-manager
 
 ## Quick Start
 
-### Basic Usage
 
-```python
-from llm_dialog_manager import ChatHistory
+### Environment Variables
 
-# Initialize with a system message
-history = ChatHistory("You are a helpful assistant")
+Create a `.env` file in your project root:
 
-# Add messages
-history.add_user_message("Hello!")
-history.add_assistant_message("Hi there! How can I help you today?")
+```bash
+# OpenAI
+OPENAI_API_KEY_1=your-key-1
+OPENAI_API_BASE_1=https://api.openai.com/v1
 
-# Print conversation
-print(history)
+# Anthropic
+ANTHROPIC_API_KEY_1=your-anthropic-key
+ANTHROPIC_API_BASE_1=https://api.anthropic.com
+
+# Google
+GEMINI_API_KEY=your-gemini-key
+
+# X.AI
+XAI_API_KEY=your-x-key
 ```
 
-### Using the AI Agent
+### Basic Usage
+
 
 ```python
 from llm_dialog_manager import Agent
@@ -59,65 +65,15 @@ response = agent.generate_response()
 agent.save_conversation()
 ```
 
-## Advanced Features
-
-### Managing Multiple API Keys
-
-```python
-from llm_dialog_manager import Agent
-
-# Use specific API key
-agent = Agent("gpt-4", api_key="your-api-key")
-
-# Or use environment variables
-# OPENAI_API_KEY_1=key1
-# OPENAI_API_KEY_2=key2
-# The system will automatically handle load balancing
-```
-
-### Conversation Management
-
-```python
-from llm_dialog_manager import ChatHistory
-
-history = ChatHistory()
-
-# Add messages with role validation
-history.add_message("Hello system", "system")
-history.add_message("Hello user", "user")
-history.add_message("Hello assistant", "assistant")
-
-# Search conversations
-results = history.search_for_keyword("hello")
-
-# Get conversation status
-status = history.conversation_status()
-history.display_conversation_status()
-
-# Get conversation snippets
-snippet = history.get_conversation_snippet(1)
-history.display_snippet(1)
-```
-
-## Environment Variables
-
-Create a `.env` file in your project root:
+### Setup Debugging Console
 
 ```bash
-# OpenAI
-OPENAI_API_KEY_1=your-key-1
-OPENAI_API_BASE_1=https://api.openai.com/v1
-
-# Anthropic
-ANTHROPIC_API_KEY_1=your-anthropic-key
-ANTHROPIC_API_BASE_1=https://api.anthropic.com
+python app.py
+# open localhost:8000
+```
+https://github.com/user-attachments/assets/5f640029-24e6-44ea-a3a3-02eb3de0d4df
 
-# Google
-GEMINI_API_KEY=your-gemini-key
 
-# X.AI
-XAI_API_KEY=your-x-key
-```
 
 ## Development
 

{llm_dialog_manager-0.1.2480 → llm_dialog_manager-0.2.2}/llm_dialog_manager/__init__.py

@@ -1,4 +1,4 @@
 from .chat_history import ChatHistory
 from .agent import Agent
 
-__version__ = "0.1.2480" 
+__version__ = "0.2.2" 

{llm_dialog_manager-0.1.2480 → llm_dialog_manager-0.2.2}/llm_dialog_manager/agent.py

@@ -113,27 +113,53 @@ def completion(model: str, messages: List[Dict[str, str]], max_tokens: int = 100
                 return response.content[0].text
 
             elif "gemini" in model:
-                client = openai.OpenAI(
-                    api_key=api_key,
-                    base_url="https://generativelanguage.googleapis.com/v1beta/"
-                )
-                print(api_key)
-                # Remove any system message from the beginning if present
-                if messages and messages[0]["role"] == "system":
-                    system_msg = messages.pop(0)
-                    # Prepend system message to first user message if exists
-                    if messages:
-                        messages[0]["content"] = f"{system_msg['content']}\n\n{messages[0]['content']}"
-                
-                response = client.chat.completions.create(
-                    model=model,
-                    messages=messages,
-                    # max_tokens=max_tokens,
-                    temperature=temperature
-                )
-                print(len(response.choices))
-                
-                return response.choices[0].message.content
+                try:
+                    # First try OpenAI-style API
+                    client = openai.OpenAI(
+                        api_key=api_key,
+                        base_url="https://generativelanguage.googleapis.com/v1beta/"
+                    )
+                    # Remove any system message from the beginning if present
+                    if messages and messages[0]["role"] == "system":
+                        system_msg = messages.pop(0)
+                        # Prepend system message to first user message if exists
+                        if messages:
+                            messages[0]["content"] = f"{system_msg['content']}\n\n{messages[0]['content']}"
+                    
+                    response = client.chat.completions.create(
+                        model=model,
+                        messages=messages,
+                        temperature=temperature
+                    )
+                    
+                    return response.choices[0].message.content
+                    
+                except Exception as e:
+                    # If OpenAI-style API fails, fall back to Google's genai library
+                    logger.info("Falling back to Google's genai library")
+                    genai.configure(api_key=api_key)
+                    
+                    # Convert messages to Gemini format
+                    gemini_messages = []
+                    for msg in messages:
+                        if msg["role"] == "system":
+                            # Prepend system message to first user message if exists
+                            if gemini_messages:
+                                gemini_messages[0].parts[0].text = f"{msg['content']}\n\n{gemini_messages[0].parts[0].text}"
+                        else:
+                            gemini_messages.append({"role": msg["role"], "parts": [{"text": msg["content"]}]})
+                    
+                    # Create Gemini model and generate response
+                    model = genai.GenerativeModel(model_name=model)
+                    response = model.generate_content(
+                        gemini_messages,
+                        generation_config=genai.types.GenerationConfig(
+                            temperature=temperature,
+                            max_output_tokens=max_tokens
+                        )
+                    )
+                    
+                    return response.text
 
             elif "grok" in model:
                 # Randomly choose between OpenAI and Anthropic SDK
@@ -256,7 +282,7 @@ if __name__ == "__main__":
     # information_detector_agent.add_message("user", text)
     # response = information_detector_agent.generate_response()
     # print(response)
-    agent = Agent("claude-3-5-sonnet-20241022", "you are an assistant", memory_enabled=True)
+    agent = Agent("gemini-1.5-pro-002", "you are an assistant", memory_enabled=True)
     
     # Format the prompt to check if the section is the last one in the outline
     prompt = f"Say: {text}\n"
@@ -265,8 +291,8 @@ if __name__ == "__main__":
     agent.add_message("user", prompt)
     agent.add_message("assistant", "the answer")
 
-    print(agent.generate_response(max_tokens=20, temperature=0.0))
+    print(agent.generate_response())
     print(agent.history[:])
-    a = agent.history.pop()
-    print(a)
+    last_message = agent.history.pop()
+    print(last_message)
     print(agent.history[:])

{llm_dialog_manager-0.1.2480 → llm_dialog_manager-0.2.2}/llm_dialog_manager.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm_dialog_manager
-Version: 0.1.2480
+Version: 0.2.2
 Summary: A Python package for managing LLM chat conversation history
 Author-email: xihajun <work@2333.fun>
 License: MIT
@@ -73,23 +73,29 @@ pip install llm-dialog-manager
 
 ## Quick Start
 
-### Basic Usage
 
-```python
-from llm_dialog_manager import ChatHistory
+### Environment Variables
 
-# Initialize with a system message
-history = ChatHistory("You are a helpful assistant")
+Create a `.env` file in your project root:
 
-# Add messages
-history.add_user_message("Hello!")
-history.add_assistant_message("Hi there! How can I help you today?")
+```bash
+# OpenAI
+OPENAI_API_KEY_1=your-key-1
+OPENAI_API_BASE_1=https://api.openai.com/v1
 
-# Print conversation
-print(history)
+# Anthropic
+ANTHROPIC_API_KEY_1=your-anthropic-key
+ANTHROPIC_API_BASE_1=https://api.anthropic.com
+
+# Google
+GEMINI_API_KEY=your-gemini-key
+
+# X.AI
+XAI_API_KEY=your-x-key
 ```
 
-### Using the AI Agent
+### Basic Usage
+
 
 ```python
 from llm_dialog_manager import Agent
@@ -106,65 +112,15 @@ response = agent.generate_response()
 agent.save_conversation()
 ```
 
-## Advanced Features
-
-### Managing Multiple API Keys
-
-```python
-from llm_dialog_manager import Agent
-
-# Use specific API key
-agent = Agent("gpt-4", api_key="your-api-key")
-
-# Or use environment variables
-# OPENAI_API_KEY_1=key1
-# OPENAI_API_KEY_2=key2
-# The system will automatically handle load balancing
-```
-
-### Conversation Management
-
-```python
-from llm_dialog_manager import ChatHistory
-
-history = ChatHistory()
-
-# Add messages with role validation
-history.add_message("Hello system", "system")
-history.add_message("Hello user", "user")
-history.add_message("Hello assistant", "assistant")
-
-# Search conversations
-results = history.search_for_keyword("hello")
-
-# Get conversation status
-status = history.conversation_status()
-history.display_conversation_status()
-
-# Get conversation snippets
-snippet = history.get_conversation_snippet(1)
-history.display_snippet(1)
-```
-
-## Environment Variables
-
-Create a `.env` file in your project root:
+### Setup Debugging Console
 
 ```bash
-# OpenAI
-OPENAI_API_KEY_1=your-key-1
-OPENAI_API_BASE_1=https://api.openai.com/v1
-
-# Anthropic
-ANTHROPIC_API_KEY_1=your-anthropic-key
-ANTHROPIC_API_BASE_1=https://api.anthropic.com
+python app.py
+# open localhost:8000
+```
+https://github.com/user-attachments/assets/5f640029-24e6-44ea-a3a3-02eb3de0d4df
 
-# Google
-GEMINI_API_KEY=your-gemini-key
 
-# X.AI
-XAI_API_KEY=your-x-key
-```
 
 ## Development
 

{llm_dialog_manager-0.1.2480 → llm_dialog_manager-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "llm_dialog_manager"
-version = "0.1.2480"
+version = "0.2.2"
 description = "A Python package for managing LLM chat conversation history"
 readme = "README.md"
 authors = [{ name = "xihajun", email = "work@2333.fun" }]

{llm_dialog_manager-0.1.2480 → llm_dialog_manager-0.2.2}/tests/test_chat_history.py

@@ -1,5 +1,5 @@
 import pytest
-from ai_chat_history import ChatHistory
+from llm_dialog_manager import ChatHistory
 
 def test_chat_history_initialization():
     # Test empty initialization