llm-dialog-manager 0.3.1__tar.gz → 0.3.4__tar.gz

{llm_dialog_manager-0.3.1 → llm_dialog_manager-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm_dialog_manager
-Version: 0.3.1
+Version: 0.3.4
 Summary: A Python package for managing LLM chat conversation history
 Author-email: xihajun <work@2333.fun>
 License: MIT

{llm_dialog_manager-0.3.1 → llm_dialog_manager-0.3.4}/llm_dialog_manager/__init__.py

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

{llm_dialog_manager-0.3.1 → llm_dialog_manager-0.3.4}/llm_dialog_manager/agent.py

@@ -6,6 +6,9 @@ from typing import List, Dict, Optional
 import logging
 from pathlib import Path
 import random
+import requests
+import zipfile
+import io
 
 # Third-party imports
 import anthropic
@@ -234,8 +237,15 @@ class Agent:
         self.history = ChatHistory(messages)
         self.memory_enabled = memory_enabled
         self.api_key = api_key
+        self.repo_content = []
 
     def add_message(self, role, content):
+        repo_content = ""
+        while self.repo_content:
+            repo = self.repo_content.pop()
+            repo_content += f"<repo>\n{repo}\n</repo>\n"
+        
+        content = repo_content + content
         self.history.add_message(content, role)
 
     def generate_response(self, max_tokens=3585, temperature=0.7):
@@ -271,6 +281,29 @@ class Agent:
             messages = json.load(file)
             self.history = ChatHistory(messages)
 
+    def add_repo(self, repo_url: Optional[str] = None, username: Optional[str] = None, repo_name: Optional[str] = None, commit_hash: Optional[str] = None):
+        if username and repo_name:
+            if commit_hash:
+                repo_url = f"https://github.com/{username}/{repo_name}/archive/{commit_hash}.zip"
+            else:
+                repo_url = f"https://github.com/{username}/{repo_name}/archive/refs/heads/main.zip"
+        
+        if not repo_url:
+            raise ValueError("Either repo_url or both username and repo_name must be provided")
+        
+        response = requests.get(repo_url)
+        if response.status_code == 200:
+            repo_content = ""
+            with zipfile.ZipFile(io.BytesIO(response.content)) as z:
+                for file_info in z.infolist():
+                    if not file_info.is_dir() and file_info.filename.endswith(('.py', '.txt')):
+                        with z.open(file_info) as f:
+                            content = f.read().decode('utf-8')
+                            repo_content += f"{file_info.filename}\n```\n{content}\n```\n"
+            self.repo_content.append(repo_content)
+        else:
+            raise ValueError(f"Failed to download repository from {repo_url}")
+
 if __name__ == "__main__":
 
     # write a test for detect finding agent

{llm_dialog_manager-0.3.1 → llm_dialog_manager-0.3.4}/llm_dialog_manager.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm_dialog_manager
-Version: 0.3.1
+Version: 0.3.4
 Summary: A Python package for managing LLM chat conversation history
 Author-email: xihajun <work@2333.fun>
 License: MIT

{llm_dialog_manager-0.3.1 → llm_dialog_manager-0.3.4}/pyproject.toml

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

llm_dialog_manager-0.3.4/tests/test_agent.py

@@ -0,0 +1,60 @@
+import pytest
+from llm_dialog_manager.agent import Agent
+
+def test_agent_initialization(env_setup):
+    # Test valid model initialization
+    agent = Agent("claude-2.1")
+    assert agent.model_name == "claude-2.1"
+    
+    # Test invalid model
+    with pytest.raises(ValueError):
+        Agent("invalid-model")
+
+    # Test add_repo method
+    agent.add_repo("https://github.com/some/repo/archive/refs/heads/main.zip")
+    assert "some content from the repo" in agent.history.messages[-1]["content"]
+
+    # Test add_repo method with username and repo name
+    agent.add_repo(repo_url="", username="someuser", repo_name="somerepo")
+    assert "some content from the repo" in agent.history.messages[-1]["content"]
+
+def test_agent_message_handling(env_setup):
+    agent = Agent("claude-2.1", memory_enabled=True)
+    
+    # Test adding messages
+    agent.add_message("system", "You are a helpful assistant")
+    agent.add_message("user", "Hello!")
+    
+    assert len(agent.history) == 2
+    assert agent.history.messages[-1]["role"] == "user"
+
+    # Test add_repo method
+    agent.add_repo("https://github.com/some/repo/archive/refs/heads/main.zip")
+    assert "some content from the repo" in agent.history.messages[-1]["content"]
+
+    # Test add_repo method with username and repo name
+    agent.add_repo(repo_url="", username="someuser", repo_name="somerepo")
+    assert "some content from the repo" in agent.history.messages[-1]["content"]
+
+def test_conversation_save_load(env_setup, tmp_path):
+    agent = Agent("claude-2.1")
+    agent.add_message("system", "Test message")
+    
+    # Save conversation
+    agent.save_conversation()
+    
+    # Create new agent and load conversation
+    new_agent = Agent("claude-2.1")
+    new_agent.load_conversation()
+    
+    assert len(new_agent.history) == 1
+    assert new_agent.history.messages[0]["content"] == "Test message"
+
+def test_add_repo_with_github_repo(env_setup):
+    agent = Agent("claude-2.1", memory_enabled=True)
+    
+    # Test add_repo method with a real GitHub repository
+    agent.add_repo(repo_url="https://github.com/xihajun/llm_dialog_manager/archive/refs/heads/main.zip")
+    
+    # Check if .env.example file content is in the repo_content
+    assert any(".env.example" in content for content in agent.repo_content)

llm_dialog_manager-0.3.1/tests/test_agent.py

@@ -1,35 +0,0 @@
-import pytest
-from llm_dialog_manager.agent import Agent
-
-def test_agent_initialization(env_setup):
-    # Test valid model initialization
-    agent = Agent("claude-2.1")
-    assert agent.model_name == "claude-2.1"
-    
-    # Test invalid model
-    with pytest.raises(ValueError):
-        Agent("invalid-model")
-
-def test_agent_message_handling(env_setup):
-    agent = Agent("claude-2.1", memory_enabled=True)
-    
-    # Test adding messages
-    agent.add_message("system", "You are a helpful assistant")
-    agent.add_message("user", "Hello!")
-    
-    assert len(agent.history) == 2
-    assert agent.history.messages[-1]["role"] == "user"
-
-def test_conversation_save_load(env_setup, tmp_path):
-    agent = Agent("claude-2.1")
-    agent.add_message("system", "Test message")
-    
-    # Save conversation
-    agent.save_conversation()
-    
-    # Create new agent and load conversation
-    new_agent = Agent("claude-2.1")
-    new_agent.load_conversation()
-    
-    assert len(new_agent.history) == 1
-    assert new_agent.history.messages[0]["content"] == "Test message"