emdash-cli 0.1.4__py3-none-any.whl → 0.1.25__py3-none-any.whl

emdash_cli/__init__.py

@@ -1,3 +1,8 @@
 """EmDash CLI - Command-line interface for code intelligence."""
 
-__version__ = "0.1.3"
+from importlib.metadata import version, PackageNotFoundError
+
+try:
+    __version__ = version("emdash-cli")
+except PackageNotFoundError:
+    __version__ = "0.0.0-dev"

emdash_cli/client.py

@@ -1,10 +1,16 @@
 """HTTP client for emdash-core API."""
 
+import os
 from typing import Any, Iterator, Optional
 
 import httpx
 
 
+def _get_max_iterations() -> int:
+    """Get max iterations from env var with default."""
+    return int(os.getenv("EMDASH_MAX_ITERATIONS", "100"))
+
+
 class EmdashClient:
     """HTTP client for interacting with emdash-core API.
 
@@ -44,7 +50,7 @@ class EmdashClient:
         message: str,
         model: Optional[str] = None,
         session_id: Optional[str] = None,
-        max_iterations: int = 20,
+        max_iterations: int = _get_max_iterations(),
         options: Optional[dict] = None,
     ) -> Iterator[str]:
         """Stream agent chat response via SSE.
@@ -82,14 +88,18 @@ class EmdashClient:
         if session_id:
             payload["session_id"] = session_id
 
-        with self._client.stream(
-            "POST",
-            f"{self.base_url}/api/agent/chat",
-            json=payload,
-        ) as response:
-            response.raise_for_status()
-            for line in response.iter_lines():
-                yield line
+        try:
+            with self._client.stream(
+                "POST",
+                f"{self.base_url}/api/agent/chat",
+                json=payload,
+            ) as response:
+                response.raise_for_status()
+                for line in response.iter_lines():
+                    yield line
+        except GeneratorExit:
+            # Stream was closed early (interrupted)
+            pass
 
     def agent_continue_stream(
         self,
@@ -107,14 +117,18 @@ class EmdashClient:
         """
         payload = {"message": message}
 
-        with self._client.stream(
-            "POST",
-            f"{self.base_url}/api/agent/chat/{session_id}/continue",
-            json=payload,
-        ) as response:
-            response.raise_for_status()
-            for line in response.iter_lines():
-                yield line
+        try:
+            with self._client.stream(
+                "POST",
+                f"{self.base_url}/api/agent/chat/{session_id}/continue",
+                json=payload,
+            ) as response:
+                response.raise_for_status()
+                for line in response.iter_lines():
+                    yield line
+        except GeneratorExit:
+            # Stream was closed early (interrupted)
+            pass
 
     def list_sessions(self) -> list[dict]:
         """List active agent sessions.
@@ -476,7 +490,7 @@ class EmdashClient:
     def research_stream(
         self,
         goal: str,
-        max_iterations: int = 10,
+        max_iterations: int = _get_max_iterations(),
         budget: int = 50,
         model: Optional[str] = None,
     ) -> Iterator[str]:

emdash_cli/commands/__init__.py

@@ -10,6 +10,7 @@ from .plan import plan
 from .rules import rules
 from .search import search
 from .server import server
+from .skills import skills
 from .team import team
 from .swarm import swarm
 from .projectmd import projectmd
@@ -28,6 +29,7 @@ __all__ = [
     "rules",
     "search",
     "server",
+    "skills",
     "team",
     "swarm",
     "projectmd",