code-muse 0.1.3__tar.gz → 0.1.7__tar.gz

{code_muse-0.1.3 → code_muse-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-muse
-Version: 0.1.3
+Version: 0.1.7
 Summary: Muse — eternal guide of creators in the arts and sciences. An elegant AI coding assistant.
 Project-URL: Repository, https://github.com/asx8678/muse
 Project-URL: Homepage, https://github.com/asx8678/muse

{code_muse-0.1.3 → code_muse-0.1.7}/code_muse/plugins/history_store/register_callbacks.py

@@ -53,7 +53,7 @@ def _on_message_history_processor_start(
     agent_name: str | None = None,
     session_id: str | None = None,
     message_history: list | None = None,
-    **kwargs,
+    incoming_messages: list | None = None,
 ) -> None:
     """No-op — could be used to restore from store in future."""
     pass

{code_muse-0.1.3 → code_muse-0.1.7}/code_muse/tools/agent_tools.py

@@ -61,6 +61,21 @@ def _generate_session_hash_suffix() -> str:
     return hashlib.sha1(timestamp.encode()).hexdigest()[:6]
 
 
+def _sanitize_for_session_id(value: str) -> str:
+    """Coerce an arbitrary string into kebab-case suitable for a session_id.
+
+    Lowercases everything, replaces any non ``[a-z0-9]`` runs with a single
+    hyphen, and strips leading/trailing hyphens.  This lets us safely embed
+    agent names like ``"LPZ-Main-Coder"`` or ``"My_Agent"`` into auto-
+    generated session IDs without tripping the kebab-case validator.
+    """
+    lowered = value.lower()
+    # Replace any run of disallowed chars with a single hyphen
+    cleaned = re.sub(r"[^a-z0-9]+", "-", lowered)
+    # Strip leading/trailing hyphens
+    return cleaned.strip("-")
+
+
 # Regex pattern for kebab-case session IDs
 SESSION_ID_PATTERN = re.compile(r"^[a-z0-9]+(-[a-z0-9]+)*$")
 SESSION_ID_MAX_LENGTH = 128
@@ -368,13 +383,19 @@ def register_invoke_agent(agent):
         if session_id is None:
             # Auto-generate a session ID with hash suffix for uniqueness
             # Example: "qa-expert-session-a3f2b1"
+            # Sanitize agent_name to kebab-case so capitalised names like
+            # "LPZ-Main-Coder" don't produce invalid session IDs.
             hash_suffix = _generate_session_hash_suffix()
-            session_id = f"{agent_name}-session-{hash_suffix}"
+            safe_agent_name = _sanitize_for_session_id(agent_name) or "agent"
+            session_id = f"{safe_agent_name}-session-{hash_suffix}"
         elif is_new_session:
             # User provided a base name for a NEW session - append hash suffix
             # Example: "review-auth" -> "review-auth-a3f2b1"
+            # Sanitize the user-provided base to be forgiving of casing/
+            # underscores while still producing a valid kebab-case ID.
             hash_suffix = _generate_session_hash_suffix()
-            session_id = f"{session_id}-{hash_suffix}"
+            safe_base = _sanitize_for_session_id(session_id) or "session"
+            session_id = f"{safe_base}-{hash_suffix}"
         # else: continuing existing session, use session_id as-is
 
         # Lazy imports to avoid circular dependency

{code_muse-0.1.3 → code_muse-0.1.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-muse"
-version = "0.1.3"
+version = "0.1.7"
 description = "Muse — eternal guide of creators in the arts and sciences. An elegant AI coding assistant."
 readme = "README.md"
 requires-python = ">=3.14,<3.16"

{code_muse-0.1.3 → code_muse-0.1.7}/code_muse/agents/_runtime.py

@@ -483,8 +483,6 @@ async def run(
             return RunOutcome(False)
         return outcome
 
-    agent_task = asyncio.create_task(run_agent_task())
-
     try:
         await on_agent_run_start(
             agent_name=agent.name,
@@ -495,6 +493,8 @@ async def run(
         # Hook failures never block the agent.
         pass
 
+    agent_task = asyncio.create_task(run_agent_task())
+
     loop = asyncio.get_running_loop()
 
     def schedule_agent_cancel() -> None: