Bug Fix

- Updated `requirements-test.txt` to include `psutil` for improved resource management during tests.
- Modified CI workflow to install the package in editable mode, allowing for easier development and testing.
- Added new public method `load_prompts` in `PromptManager` to facilitate reloading prompts from storage.
- Introduced new `__init__.py` files in `agents` and `adapters` directories to encapsulate agent-related functionality.

These changes improve the development experience and enhance the overall functionality of the Promptix library.

Author: Nisarg38

.github/workflows/ci.yml

@@ -28,12 +28,16 @@ jobs:
           python -m pip install --upgrade pip
           pip install -r requirements.txt
           pip install -r requirements-test.txt
-          pip install -e .
+      
+      - name: Install package in editable mode
+        run: |
+          pip install -e . || echo "Editable install failed, will use PYTHONPATH"
       
       - name: Run tests
         env:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
           ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
+          PYTHONPATH: ${{ github.workspace }}/src
         run: |
           pytest --cov=promptix --cov-report=xml
       

requirements-test.txt

@@ -4,4 +4,5 @@ pytest-mock>=3.10.0
 build>=1.0.0
 twine>=4.0.0
 openai>=1.0.0
-anthropic>=0.8.0 
+anthropic>=0.8.0
+psutil>=5.8.0 

src/promptix/core/agents/__init__.py

@@ -0,0 +1,5 @@
+"""
+Agent-related functionality for Promptix.
+
+This module contains agent-related classes and utilities.
+"""

src/promptix/core/agents/adapters/__init__.py

@@ -0,0 +1,5 @@
+"""
+Agent adapter functionality for Promptix.
+
+This module contains agent adapter classes and utilities.
+"""

src/promptix/core/storage/manager.py

@@ -66,6 +66,10 @@ def get_prompt(self, prompt_id: str) -> Dict[str, Any]:
     def list_prompts(self) -> Dict[str, Any]:
         """Return all available prompts."""
         return self.prompts
+    
+    def load_prompts(self) -> None:
+        """Public method to reload prompts from storage."""
+        self._load_prompts()
 
     def _format_prompt_for_storage(self, prompt_data: Dict[str, Any]) -> Dict[str, Any]:
         """Convert multiline prompts to single line with escaped newlines."""

tests/test_components.py

@@ -6,7 +6,7 @@
 """
 
 import pytest
-from unittest.mock import Mock, patch, MagicMock, mock_open
+from unittest.mock import Mock, patch, MagicMock, mock_open, call
 import tempfile
 import os
 import yaml
@@ -40,7 +40,9 @@ def test_yaml_loader_functionality(self, mock_yaml_load, mock_file):
         result = loader.load("/fake/path.yaml")
 
         assert "TestPrompt" in result
-        mock_file.assert_called_once_with("/fake/path.yaml", 'r', encoding='utf-8')
+        # Check that open was called with our specific file (may be called multiple times due to .env loading)
+        expected_call = call("/fake/path.yaml", 'r', encoding='utf-8')
+        assert expected_call in mock_file.call_args_list, f"Expected {expected_call} in {mock_file.call_args_list}"
         mock_yaml_load.assert_called_once()
 
     @patch('builtins.open', side_effect=FileNotFoundError())
@@ -494,8 +496,6 @@ def test_utility_functions(self):
             func = getattr(utils, 'create_default_prompts_file')
             assert callable(func)
 
-            with tempfile.NamedTemporaryFile(suffix='.yaml', delete=False) as tmp:
-            import tempfile
             with tempfile.NamedTemporaryFile(suffix='.yaml', delete=False) as tmp:
                 tmp_path = Path(tmp.name)