llm-dialog-manager 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

llm_dialog_manager/__init__.py

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

llm_dialog_manager/agent.py

@@ -275,7 +275,14 @@ if __name__ == "__main__":
     # write a test for detect finding agent
     text = "I think the answer is 42"
 
-    agent = Agent("claude-3-5-sonnet-20241022", "you are an assistant", memory_enabled=True)
+    # from agent.messageloader import information_detector_messages
+    
+    # # Now you can print or use information_detector_messages as needed
+    # information_detector_agent = Agent("gemini-1.5-pro", information_detector_messages)
+    # information_detector_agent.add_message("user", text)
+    # response = information_detector_agent.generate_response()
+    # print(response)
+    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"
@@ -284,7 +291,7 @@ 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[:])
     last_message = agent.history.pop()
     print(last_message)

{llm_dialog_manager-0.2.1.dist-info → llm_dialog_manager-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm_dialog_manager
-Version: 0.2.1
+Version: 0.2.2
 Summary: A Python package for managing LLM chat conversation history
 Author-email: xihajun <work@2333.fun>
 License: MIT
@@ -25,25 +25,25 @@ Requires-Dist: google-generativeai>=0.1.0
 Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: typing-extensions>=4.0.0
 Requires-Dist: uuid>=1.30
-Provides-Extra: all
-Requires-Dist: pytest>=8.0.0; extra == "all"
-Requires-Dist: pytest-asyncio>=0.21.1; extra == "all"
-Requires-Dist: pytest-cov>=4.1.0; extra == "all"
-Requires-Dist: black>=23.9.1; extra == "all"
-Requires-Dist: isort>=5.12.0; extra == "all"
 Provides-Extra: dev
 Requires-Dist: pytest>=8.0.0; extra == "dev"
 Requires-Dist: pytest-asyncio>=0.21.1; extra == "dev"
 Requires-Dist: pytest-cov>=4.1.0; extra == "dev"
 Requires-Dist: black>=23.9.1; extra == "dev"
 Requires-Dist: isort>=5.12.0; extra == "dev"
-Provides-Extra: lint
-Requires-Dist: black>=22.0; extra == "lint"
-Requires-Dist: isort>=5.0; extra == "lint"
 Provides-Extra: test
 Requires-Dist: pytest>=6.0; extra == "test"
 Requires-Dist: pytest-asyncio>=0.14.0; extra == "test"
 Requires-Dist: pytest-cov>=2.0; extra == "test"
+Provides-Extra: lint
+Requires-Dist: black>=22.0; extra == "lint"
+Requires-Dist: isort>=5.0; extra == "lint"
+Provides-Extra: all
+Requires-Dist: pytest>=8.0.0; extra == "all"
+Requires-Dist: pytest-asyncio>=0.21.1; extra == "all"
+Requires-Dist: pytest-cov>=4.1.0; extra == "all"
+Requires-Dist: black>=23.9.1; extra == "all"
+Requires-Dist: isort>=5.12.0; extra == "all"
 
 # LLM Dialog Manager
 
@@ -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.2.2.dist-info/RECORD

@@ -0,0 +1,9 @@
+llm_dialog_manager/__init__.py,sha256=RU9PCU54Wt4GLyoKj-Xa3xiVAv2yj9Mobs7FlLOms-E,86
+llm_dialog_manager/agent.py,sha256=aST_n9jU1tuHjMUK0ytCmWq3wYkZp9VHLcg4Q4Y7Tcw,11731
+llm_dialog_manager/chat_history.py,sha256=xKA-oQCv8jv_g8EhXrG9h1S8Icbj2FfqPIhbty5vra4,6033
+llm_dialog_manager/key_manager.py,sha256=shvxmn4zUtQx_p-x1EFyOmnk-WlhigbpKtxTKve-zXk,4421
+llm_dialog_manager-0.2.2.dist-info/LICENSE,sha256=vWGbYgGuWpWrXL8-xi6pNcX5UzD6pWoIAZmcetyfbus,1064
+llm_dialog_manager-0.2.2.dist-info/METADATA,sha256=gRCap4OYPxoyi1qTlg-V5CjUywzVqtJ3wIQkNycURqE,4152
+llm_dialog_manager-0.2.2.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+llm_dialog_manager-0.2.2.dist-info/top_level.txt,sha256=u2EQEXW0NGAt0AAHT7jx1odXZ4rZfjcgbmJhvKFuMkI,19
+llm_dialog_manager-0.2.2.dist-info/RECORD,,

{llm_dialog_manager-0.2.1.dist-info → llm_dialog_manager-0.2.2.dist-info}/WHEEL

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

llm_dialog_manager-0.2.1.dist-info/RECORD

@@ -1,9 +0,0 @@
-llm_dialog_manager/__init__.py,sha256=joo2odAcSJIo-p3bwQbGxYok_vWUlI2RjkaAmAdKX3o,86
-llm_dialog_manager/agent.py,sha256=cLwDTgCpO8teYaedIiee8XWU8vE9icBe_NNoAklm79w,11388
-llm_dialog_manager/chat_history.py,sha256=xKA-oQCv8jv_g8EhXrG9h1S8Icbj2FfqPIhbty5vra4,6033
-llm_dialog_manager/key_manager.py,sha256=shvxmn4zUtQx_p-x1EFyOmnk-WlhigbpKtxTKve-zXk,4421
-llm_dialog_manager-0.2.1.dist-info/LICENSE,sha256=vWGbYgGuWpWrXL8-xi6pNcX5UzD6pWoIAZmcetyfbus,1064
-llm_dialog_manager-0.2.1.dist-info/METADATA,sha256=F-7unvFE1V-Lk6G_nBvHHXxza3RHjXtB-LJlk_iP2rM,5193
-llm_dialog_manager-0.2.1.dist-info/WHEEL,sha256=a7TGlA-5DaHMRrarXjVbQagU3Man_dCnGIWMJr5kRWo,91
-llm_dialog_manager-0.2.1.dist-info/top_level.txt,sha256=u2EQEXW0NGAt0AAHT7jx1odXZ4rZfjcgbmJhvKFuMkI,19
-llm_dialog_manager-0.2.1.dist-info/RECORD,,