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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cli-mem
-Version: 0.3.0
+Version: 0.3.1
 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.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "cli-mem"
-version = "0.3.0"
+version = "0.3.1"
 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.1}/src/mem/__init__.py

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

{cli_mem-0.3.0 → cli_mem-0.3.1}/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