agmem 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

memvcs/commands/checkout.py

@@ -10,27 +10,18 @@ from ..core.repository import Repository
 
 class CheckoutCommand:
     """Switch branches or restore working tree files."""
-    
-    name = 'checkout'
-    help = 'Switch branches or restore working tree files'
-    
+
+    name = "checkout"
+    help = "Switch branches or restore working tree files"
+
     @staticmethod
     def add_arguments(parser: argparse.ArgumentParser):
+        parser.add_argument("ref", help="Branch, tag, or commit to checkout")
+        parser.add_argument("-b", action="store_true", help="Create and checkout a new branch")
         parser.add_argument(
-            'ref',
-            help='Branch, tag, or commit to checkout'
+            "--force", "-f", action="store_true", help="Force checkout (discard local changes)"
         )
-        parser.add_argument(
-            '-b',
-            action='store_true',
-            help='Create and checkout a new branch'
-        )
-        parser.add_argument(
-            '--force', '-f',
-            action='store_true',
-            help='Force checkout (discard local changes)'
-        )
-    
+
     @staticmethod
     def execute(args) -> int:
         repo, code = require_repo()
@@ -40,24 +31,24 @@ class CheckoutCommand:
         # Create and checkout new branch
         if args.b:
             branch_name = args.ref
-            
+
             # Check if branch already exists
             if repo.refs.branch_exists(branch_name):
                 print(f"Error: A branch named '{branch_name}' already exists.")
                 return 1
-            
+
             # Get current HEAD commit
             head = repo.refs.get_head()
-            if head['type'] == 'branch':
-                current_commit = repo.refs.get_branch_commit(head['value'])
+            if head["type"] == "branch":
+                current_commit = repo.refs.get_branch_commit(head["value"])
             else:
-                current_commit = head['value']
-            
+                current_commit = head["value"]
+
             # Create branch
             if not repo.refs.create_branch(branch_name, current_commit):
                 print(f"Error: Could not create branch '{branch_name}'")
                 return 1
-            
+
             # Switch to new branch
             try:
                 repo.refs.set_head_branch(branch_name)
@@ -66,16 +57,16 @@ class CheckoutCommand:
             except Exception as e:
                 print(f"Error switching to branch: {e}")
                 return 1
-        
+
         # Regular checkout
         ref = args.ref
-        
+
         # Check if it's a branch
         is_branch = repo.refs.branch_exists(ref)
-        
+
         try:
             commit_hash = repo.checkout(ref, force=args.force)
-            
+
             if is_branch:
                 print(f"Switched to branch '{ref}'")
             else:
@@ -83,16 +74,20 @@ class CheckoutCommand:
                 if repo.refs.tag_exists(ref):
                     print(f"Note: checking out '{ref}'.")
                     print()
-                    print("You are in 'detached HEAD' state. You can look around, make experimental")
-                    print("changes and commit them, and you can discard any commits you make in this")
+                    print(
+                        "You are in 'detached HEAD' state. You can look around, make experimental"
+                    )
+                    print(
+                        "changes and commit them, and you can discard any commits you make in this"
+                    )
                     print("state without impacting any branches by switching back to a branch.")
                 else:
                     print(f"Note: checking out '{commit_hash[:8]}'.")
                     print()
                     print("You are in 'detached HEAD' state.")
-            
+
             return 0
-        
+
         except ValueError as e:
             print(f"Error: {e}")
             return 1

memvcs/commands/clean.py

@@ -12,52 +12,46 @@ from ..core.repository import Repository
 
 class CleanCommand:
     """Remove untracked files from working directory."""
-    
-    name = 'clean'
-    help = 'Remove untracked files from working directory'
-    
+
+    name = "clean"
+    help = "Remove untracked files from working directory"
+
     @staticmethod
     def add_arguments(parser: argparse.ArgumentParser):
         parser.add_argument(
-            '-n', '--dry-run',
-            action='store_true',
-            help='Show what would be removed without removing'
-        )
-        parser.add_argument(
-            '-f', '--force',
-            action='store_true',
-            help='Required to actually remove files'
+            "-n",
+            "--dry-run",
+            action="store_true",
+            help="Show what would be removed without removing",
         )
         parser.add_argument(
-            '-d',
-            action='store_true',
-            help='Remove untracked directories too'
+            "-f", "--force", action="store_true", help="Required to actually remove files"
         )
-    
+        parser.add_argument("-d", action="store_true", help="Remove untracked directories too")
+
     @staticmethod
     def execute(args) -> int:
         repo, code = require_repo()
         if code != 0:
             return code
 
-        
         status = repo.get_status()
-        untracked = status.get('untracked', [])
-        
+        untracked = status.get("untracked", [])
+
         if not untracked:
             print("Nothing to clean.")
             return 0
-        
+
         if args.dry_run:
             print("Would remove:")
             for p in untracked:
                 print(f"  {p}")
             return 0
-        
+
         if not args.force:
             print("Use -f to force removal of untracked files.")
             return 1
-        
+
         removed = 0
         for rel_path in untracked:
             full_path = repo.current_dir / rel_path
@@ -68,9 +62,10 @@ class CleanCommand:
                     print(f"Removed {rel_path}")
                 elif args.d and full_path.is_dir():
                     import shutil
+
                     shutil.rmtree(full_path)
                     removed += 1
                     print(f"Removed {rel_path}/")
-        
+
         print(f"Removed {removed} file(s)")
         return 0

memvcs/commands/clone.py

@@ -80,12 +80,22 @@ class CloneCommand:
 
         # Set remote origin to source
         import json
+
         config_file = target / ".mem" / "config.json"
         config = json.loads(config_file.read_text()) if config_file.exists() else {}
         if "remotes" not in config:
             config["remotes"] = {}
-        config["remotes"]["origin"] = {"url": url if url.startswith("file://") else f"file://{remote_path}"}
+        config["remotes"]["origin"] = {
+            "url": url if url.startswith("file://") else f"file://{remote_path}"
+        }
         config_file.write_text(json.dumps(config, indent=2))
 
+        # Copy remote's public key to .mem/keys/remotes/origin.pub for trust store
+        remote_keys = remote_mem / "keys" / "public.pem"
+        if remote_keys.exists():
+            keys_remotes = target / ".mem" / "keys" / "remotes"
+            keys_remotes.mkdir(parents=True, exist_ok=True)
+            shutil.copy2(remote_keys, keys_remotes / "origin.pub")
+
         print(f"Cloned into {target}")
         return 0

memvcs/commands/commit.py

@@ -7,88 +7,82 @@ from datetime import datetime
 
 from ..commands.base import require_repo
 from ..core.schema import SchemaValidator
-from ..core.hooks import run_pre_commit_hooks
+from ..core.hooks import run_pre_commit_hooks, compute_suggested_importance
 
 
 class CommitCommand:
     """Create a commit from staged changes."""
-    
-    name = 'commit'
-    help = 'Save staged changes as a memory snapshot'
-    
+
+    name = "commit"
+    help = "Save staged changes as a memory snapshot"
+
     @staticmethod
     def add_arguments(parser: argparse.ArgumentParser):
         parser.add_argument(
-            '-m', '--message',
-            required=True,
-            help='Commit message describing the changes'
-        )
-        parser.add_argument(
-            '--author',
-            help='Override default author'
+            "-m", "--message", required=True, help="Commit message describing the changes"
         )
+        parser.add_argument("--author", help="Override default author")
         parser.add_argument(
-            '--no-verify',
-            action='store_true',
-            help='Skip pre-commit hooks and schema validation'
+            "--no-verify", action="store_true", help="Skip pre-commit hooks and schema validation"
         )
+        parser.add_argument("--strict", action="store_true", help="Treat schema warnings as errors")
         parser.add_argument(
-            '--strict',
-            action='store_true',
-            help='Treat schema warnings as errors'
+            "--run-tests", action="store_true", help="Run memory tests before committing"
         )
         parser.add_argument(
-            '--run-tests',
-            action='store_true',
-            help='Run memory tests before committing'
+            "--importance",
+            type=float,
+            metavar="SCORE",
+            help="Importance score 0.0-1.0 for recall/decay weighting",
         )
-    
+
     @staticmethod
     def _get_blob_hash(file_info) -> str:
         """Get blob hash from StagedFile or dict (for hooks that pass either)."""
-        if hasattr(file_info, 'blob_hash'):
+        if hasattr(file_info, "blob_hash"):
             return file_info.blob_hash
         if isinstance(file_info, dict):
-            return file_info.get('blob_hash') or file_info.get('hash') or ''
-        return ''
+            return file_info.get("blob_hash") or file_info.get("hash") or ""
+        return ""
 
     @staticmethod
     def _validate_staged_files(repo, staged: dict, strict: bool) -> tuple:
         """
         Validate staged files for schema compliance.
-        
+
         Returns:
             Tuple of (success, validation_results)
         """
         validation_results = {}
         has_errors = False
-        
+
         for filepath, file_info in staged.items():
             blob_hash = CommitCommand._get_blob_hash(file_info)
             if not blob_hash:
                 continue
-            
+
             # Read content from object store
             from ..core.objects import Blob
+
             blob = Blob.load(repo.object_store, blob_hash)
             if not blob:
                 continue
-            
+
             try:
-                content = blob.content.decode('utf-8')
+                content = blob.content.decode("utf-8")
             except UnicodeDecodeError:
                 # Skip binary files
                 continue
-            
+
             # Validate the file
             result = SchemaValidator.validate(content, filepath, strict=strict)
             validation_results[filepath] = result
-            
+
             if not result.valid:
                 has_errors = True
-        
+
         return not has_errors, validation_results
-    
+
     @staticmethod
     def _print_validation_results(results: dict) -> None:
         """Print validation results in a readable format."""
@@ -126,6 +120,7 @@ class CommitCommand:
         """Run memory tests if available. Returns 1 on failure, 0 on success or skip."""
         try:
             from ..core.test_runner import TestRunner
+
             test_runner = TestRunner(repo)
             test_result = test_runner.run_all()
             if not test_result.passed:
@@ -156,13 +151,19 @@ class CommitCommand:
         if args.run_tests:
             if CommitCommand._run_memory_tests(repo) != 0:
                 return 1
-        metadata = {
-            'files_changed': len(staged),
-            'timestamp': datetime.utcnow().isoformat() + 'Z'
-        }
+        metadata = {"files_changed": len(staged), "timestamp": datetime.utcnow().isoformat() + "Z"}
+        # Importance scoring: explicit --importance or auto from heuristics
+        if args.importance is not None:
+            if not (0.0 <= args.importance <= 1.0):
+                print("Error: --importance must be between 0.0 and 1.0")
+                return 1
+            metadata["importance"] = args.importance
+        else:
+            suggested = compute_suggested_importance(repo, staged, args.message, metadata)
+            metadata["importance"] = suggested
         if args.author:
             config = repo.get_config()
-            config['author']['name'] = args.author
+            config["author"]["name"] = args.author
             repo.set_config(config)
         try:
             commit_hash = repo.commit(args.message, metadata)