cli-mem 0.3.0__tar.gz → 0.3.2__tar.gz

{cli_mem-0.3.0 → cli_mem-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cli-mem
-Version: 0.3.0
+Version: 0.3.2
 Summary: Privacy-first CLI that turns shell history into searchable memory
 Project-URL: GitHub, https://github.com/matinsaurralde/mem
 Author: Matias Insaurralde

{cli_mem-0.3.0 → cli_mem-0.3.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "cli-mem"
-version = "0.3.0"
+version = "0.3.2"
 description = "Privacy-first CLI that turns shell history into searchable memory"
 license = "MIT"
 readme = "README.md"

{cli_mem-0.3.0 → cli_mem-0.3.2}/src/mem/__init__.py

@@ -1,3 +1,3 @@
 """mem — your shell history, understood."""
 
-__version__ = "0.3.0"
+__version__ = "0.3.2"

cli_mem-0.3.2/src/mem/_generable.py

@@ -0,0 +1,35 @@
+"""Apple FM SDK generable types for mem.
+
+This module is intentionally separate from variables.py because
+``from __future__ import annotations`` (PEP 563) converts all type
+hints to strings, which breaks apple-fm-sdk's @fm.generable decorator
+when it tries to resolve nested type references like
+``list[CredentialDetection]`` via ``typing.get_type_hints()``.
+
+By keeping these classes in their own module without deferred
+annotations, the SDK can inspect the concrete types at decoration time.
+"""
+
+try:
+    import apple_fm_sdk as fm
+
+    @fm.generable("Detected credential in a shell command")
+    class CredentialDetection:
+        original_value: str = fm.guide(
+            "The literal sensitive value found in the command"
+        )
+        suggested_name: str = fm.guide(
+            "A descriptive variable name like ACME_API_TOKEN"
+        )
+        reason: str = fm.guide(
+            "Why this looks like a credential (e.g., 'JWT token', 'API key')"
+        )
+
+    @fm.generable("List of credentials detected in a shell command")
+    class CredentialList:
+        credentials: list[CredentialDetection] = fm.guide("All detected credentials")
+
+except ImportError:
+    # SDK not available — classes won't be used, but module still importable.
+    CredentialDetection = None
+    CredentialList = None

{cli_mem-0.3.0 → cli_mem-0.3.2}/src/mem/patterns.py

@@ -95,18 +95,24 @@ async def _generalize_commands(tool: str, unique_commands: list[str]) -> dict[st
     """Generalize each unique command via Apple FM guided generation.
 
     Returns a mapping from concrete command -> generalized pattern.
-    Uses one LanguageModelSession per tool to share context.
+    Uses a fresh session per command to avoid exceeding the on-device
+    model's context window when processing many commands.
     """
     import apple_fm_sdk as fm
 
     GeneralizedCommand = _get_generable_types()
-    session = fm.LanguageModelSession()
     cmd_to_pattern: dict[str, str] = {}
 
     for cmd in unique_commands:
-        prompt = GENERALIZE_PROMPT.format(tool=tool, command=cmd)
-        result = await session.respond(prompt, generating=GeneralizedCommand)
-        cmd_to_pattern[cmd] = result.pattern
+        try:
+            session = fm.LanguageModelSession()
+            prompt = GENERALIZE_PROMPT.format(tool=tool, command=cmd)
+            result = await session.respond(prompt, generating=GeneralizedCommand)
+            cmd_to_pattern[cmd] = result.pattern
+        except Exception:
+            # Skip commands that fail (too long, context issues, etc.)
+            # Fall back to the raw command as its own pattern.
+            cmd_to_pattern[cmd] = cmd
 
     return cmd_to_pattern
 

{cli_mem-0.3.0 → cli_mem-0.3.2}/src/mem/variables.py

@@ -278,21 +278,7 @@ async def _detect_credentials_async(cmd: str) -> list[tuple[str, str, str]]:
     """
     import apple_fm_sdk as fm
 
-    @fm.generable("Detected credential in a shell command")
-    class CredentialDetection:
-        original_value: str = fm.guide(
-            "The literal sensitive value found in the command"
-        )
-        suggested_name: str = fm.guide(
-            "A descriptive variable name like ACME_API_TOKEN"
-        )
-        reason: str = fm.guide(
-            "Why this looks like a credential (e.g., 'JWT token', 'API key')"
-        )
-
-    @fm.generable("List of credentials detected in a shell command")
-    class CredentialList:
-        credentials: list[CredentialDetection] = fm.guide("All detected credentials")
+    from mem._generable import CredentialList
 
     session = fm.LanguageModelSession()
     prompt = (