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

memvcs/commands/blame.py

@@ -12,53 +12,46 @@ from ..core.objects import Commit, Tree, Blob
 
 class BlameCommand:
     """Show author and commit for each line of a file, or trace semantic facts."""
-    
-    name = 'blame'
-    help = 'Show who changed each line of a memory file, or trace semantic facts'
-    
+
+    name = "blame"
+    help = "Show who changed each line of a memory file, or trace semantic facts"
+
     @staticmethod
     def add_arguments(parser: argparse.ArgumentParser):
+        parser.add_argument("file", nargs="?", help="File to blame (path relative to current/)")
         parser.add_argument(
-            'file',
-            nargs='?',
-            help='File to blame (path relative to current/)'
-        )
-        parser.add_argument(
-            'ref',
-            nargs='?',
-            default='HEAD',
-            help='Commit to blame at (default: HEAD)'
+            "ref", nargs="?", default="HEAD", help="Commit to blame at (default: HEAD)"
         )
         parser.add_argument(
-            '--query', '-q',
-            help='Semantic query to trace (e.g., "Why does agent think X?")'
+            "--query", "-q", help='Semantic query to trace (e.g., "Why does agent think X?")'
         )
         parser.add_argument(
-            '--limit', '-n',
+            "--limit",
+            "-n",
             type=int,
             default=5,
-            help='Number of results to show for semantic blame (default: 5)'
+            help="Number of results to show for semantic blame (default: 5)",
         )
-    
+
     @staticmethod
     def execute(args) -> int:
         repo, code = require_repo()
         if code != 0:
             return code
-        
+
         # Semantic blame mode
         if args.query:
             return BlameCommand._semantic_blame(repo, args.query, args.limit)
-        
+
         # File blame mode
         if not args.file:
             print("Error: Either --query or a file path is required.")
             print("Usage: agmem blame <file> [ref]")
-            print("       agmem blame --query \"Why does agent think X?\"")
+            print('       agmem blame --query "Why does agent think X?"')
             return 1
-        
+
         return BlameCommand._file_blame(repo, args.file, args.ref)
-    
+
     @staticmethod
     def _file_blame(repo, filepath: str, ref: str) -> int:
         """Traditional file-based blame."""
@@ -66,45 +59,45 @@ class BlameCommand:
         if not commit_hash:
             print(f"Error: Unknown revision: {ref}")
             return 1
-        
+
         # Get file content at commit
         tree = repo.get_commit_tree(commit_hash)
         if not tree:
             print("Error: Could not load tree.")
             return 1
-        
+
         # Find file in tree (support path like semantic/user-prefs.md)
         blob_hash = None
         for entry in tree.entries:
-            path = entry.path + '/' + entry.name if entry.path else entry.name
+            path = entry.path + "/" + entry.name if entry.path else entry.name
             if path == filepath:
                 blob_hash = entry.hash
                 break
-        
+
         if not blob_hash:
             print(f"Error: File not found in {ref}: {filepath}")
             return 1
-        
+
         blob = Blob.load(repo.object_store, blob_hash)
         if not blob:
             print("Error: Could not load file content.")
             return 1
-        
-        lines = blob.content.decode('utf-8', errors='replace').splitlines()
+
+        lines = blob.content.decode("utf-8", errors="replace").splitlines()
         commit = Commit.load(repo.object_store, commit_hash)
-        author_short = commit.author.split('<')[0].strip()[:20] if commit else 'unknown'
+        author_short = commit.author.split("<")[0].strip()[:20] if commit else "unknown"
         hash_short = commit_hash[:8]
-        
+
         for i, line in enumerate(lines, 1):
             print(f"{hash_short} ({author_short:20} {i:4}) {line}")
-        
+
         return 0
-    
+
     @staticmethod
     def _semantic_blame(repo, query: str, limit: int) -> int:
         """
         Semantic blame - trace which commit introduced a fact.
-        
+
         Searches the vector store and shows provenance for matching chunks.
         """
         try:
@@ -113,34 +106,34 @@ class BlameCommand:
             print("Error: Vector search requires sqlite-vec.")
             print("Install with: pip install agmem[vector]")
             return 1
-        
+
         try:
-            vs = VectorStore(repo.root / '.mem')
+            vs = VectorStore(repo.root / ".mem")
             results = vs.search_with_provenance(query, limit=limit)
         except Exception as e:
             print(f"Error: Vector search failed: {e}")
             print("Try running 'agmem search --rebuild' to rebuild the index.")
             return 1
-        
+
         if not results:
             print("No matching facts found in memory.")
             print("Try rebuilding the index with 'agmem search --rebuild'")
             return 0
-        
-        print(f"Semantic blame for: \"{query}\"")
+
+        print(f'Semantic blame for: "{query}"')
         print("=" * 60)
-        
+
         for i, result in enumerate(results, 1):
-            path = result['path']
-            content = result['content']
-            similarity = result['similarity']
-            commit_hash = result['commit_hash']
-            author = result['author']
-            indexed_at = result['indexed_at']
-            
+            path = result["path"]
+            content = result["content"]
+            similarity = result["similarity"]
+            commit_hash = result["commit_hash"]
+            author = result["author"]
+            indexed_at = result["indexed_at"]
+
             print(f"\n[{i}] {path}")
             print(f"    Similarity: {similarity:.2%}")
-            
+
             if commit_hash:
                 # Try to get commit details
                 commit = Commit.load(repo.object_store, commit_hash)
@@ -157,13 +150,13 @@ class BlameCommand:
                 print("    Commit: (not tracked)")
                 if indexed_at:
                     print(f"    Indexed: {indexed_at}")
-            
+
             # Show content preview
             print(f"\n    Content preview:")
-            for line in content.split('\n')[:5]:
+            for line in content.split("\n")[:5]:
                 print(f"      {line[:70]}")
-            if len(content.split('\n')) > 5:
+            if len(content.split("\n")) > 5:
                 print("      ...")
-        
+
         print()
         return 0

memvcs/commands/branch.py

@@ -10,43 +10,23 @@ from ..core.repository import Repository
 
 class BranchCommand:
     """Manage branches."""
-    
-    name = 'branch'
-    help = 'List, create, or delete branches'
-    
+
+    name = "branch"
+    help = "List, create, or delete branches"
+
     @staticmethod
     def add_arguments(parser: argparse.ArgumentParser):
+        parser.add_argument("name", nargs="?", help="Branch name to create or delete")
+        parser.add_argument("--delete", "-d", action="store_true", help="Delete a branch")
         parser.add_argument(
-            'name',
-            nargs='?',
-            help='Branch name to create or delete'
-        )
-        parser.add_argument(
-            '--delete', '-d',
-            action='store_true',
-            help='Delete a branch'
+            "--force", "-D", action="store_true", help="Force delete a branch (even if not merged)"
         )
+        parser.add_argument("--list", "-l", action="store_true", help="List all branches")
         parser.add_argument(
-            '--force', '-D',
-            action='store_true',
-            help='Force delete a branch (even if not merged)'
+            "--all", "-a", action="store_true", help="List all branches including remote"
         )
-        parser.add_argument(
-            '--list', '-l',
-            action='store_true',
-            help='List all branches'
-        )
-        parser.add_argument(
-            '--all', '-a',
-            action='store_true',
-            help='List all branches including remote'
-        )
-        parser.add_argument(
-            'start_point',
-            nargs='?',
-            help='Commit to start the new branch from'
-        )
-    
+        parser.add_argument("start_point", nargs="?", help="Commit to start the new branch from")
+
     @staticmethod
     def execute(args) -> int:
         repo, code = require_repo()
@@ -60,13 +40,13 @@ class BranchCommand:
             if not args.name:
                 print("Error: Branch name required for deletion")
                 return 1
-            
+
             current = repo.refs.get_current_branch()
             if args.name == current:
                 print(f"Error: Cannot delete current branch '{args.name}'")
                 print("Switch to another branch first.")
                 return 1
-            
+
             if repo.refs.delete_branch(args.name):
                 print(f"Deleted branch {args.name}")
                 return 0

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,11 +80,14 @@ 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))
 
         print(f"Cloned into {target}")

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)