emdash-cli 0.1.8__tar.gz → 0.1.30__tar.gz

{emdash_cli-0.1.8 → emdash_cli-0.1.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emdash-cli
-Version: 0.1.8
+Version: 0.1.30
 Summary: EmDash CLI - Command-line interface for code intelligence
 Author: Em Dash Team
 Requires-Python: >=3.10,<4.0
@@ -10,8 +10,10 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
+Provides-Extra: images
 Requires-Dist: click (>=8.1.7,<9.0.0)
-Requires-Dist: emdash-core (>=0.1.8)
+Requires-Dist: emdash-core (>=0.1.30)
 Requires-Dist: httpx (>=0.25.0)
+Requires-Dist: pillow (>=10.0.0) ; extra == "images"
 Requires-Dist: prompt_toolkit (>=3.0.43,<4.0.0)
 Requires-Dist: rich (>=13.7.0)

emdash_cli-0.1.30/emdash_cli/__init__.py

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

{emdash_cli-0.1.8 → emdash_cli-0.1.30}/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,8 +50,9 @@ 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,
+        images: Optional[list[dict]] = None,
     ) -> Iterator[str]:
         """Stream agent chat response via SSE.
 
@@ -55,6 +62,7 @@ class EmdashClient:
             session_id: Session ID for continuity (optional)
             max_iterations: Max agent iterations
             options: Additional options (mode, save, no_graph_tools, etc.)
+            images: List of images [{"data": base64_str, "format": "png"}]
 
         Yields:
             SSE lines from the response
@@ -81,40 +89,54 @@ class EmdashClient:
             payload["model"] = model
         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
+        if images:
+            payload["images"] = images
+
+        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,
         session_id: str,
         message: str,
+        images: Optional[list[dict]] = None,
     ) -> Iterator[str]:
         """Continue an existing agent session.
 
         Args:
             session_id: Existing session ID
             message: Continuation message
+            images: List of images [{"data": base64_str, "format": "png"}]
 
         Yields:
             SSE lines from the response
         """
         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
+        if images:
+            payload["images"] = images
+
+        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 +498,7 @@ class EmdashClient:
     def research_stream(
         self,
         goal: str,
-        max_iterations: int = 50,
+        max_iterations: int = _get_max_iterations(),
         budget: int = 50,
         model: Optional[str] = None,
     ) -> Iterator[str]:

emdash_cli-0.1.30/emdash_cli/clipboard.py

@@ -0,0 +1,123 @@
+"""Clipboard utilities for image handling."""
+
+import base64
+import io
+from typing import Optional, Tuple
+
+
+def get_clipboard_image() -> Optional[Tuple[str, str]]:
+    """Get image from clipboard if available.
+
+    Returns:
+        Tuple of (base64_data, format) if image found, None otherwise.
+    """
+    try:
+        from PIL import ImageGrab, Image
+
+        # Try to grab image from clipboard
+        image = ImageGrab.grabclipboard()
+
+        if image is None:
+            return None
+
+        # Handle list of file paths (Windows)
+        if isinstance(image, list):
+            # It's a list of file paths
+            if image and isinstance(image[0], str):
+                try:
+                    image = Image.open(image[0])
+                except Exception:
+                    return None
+            else:
+                return None
+
+        # Convert to PNG bytes
+        if isinstance(image, Image.Image):
+            buffer = io.BytesIO()
+            # Convert to RGB if necessary (for RGBA images)
+            if image.mode in ('RGBA', 'LA') or (image.mode == 'P' and 'transparency' in image.info):
+                # Keep as PNG to preserve transparency
+                image.save(buffer, format='PNG')
+                img_format = 'png'
+            else:
+                # Convert to JPEG for smaller size
+                if image.mode != 'RGB':
+                    image = image.convert('RGB')
+                image.save(buffer, format='JPEG', quality=85)
+                img_format = 'jpeg'
+
+            buffer.seek(0)
+            base64_data = base64.b64encode(buffer.read()).decode('utf-8')
+            return base64_data, img_format
+
+    except ImportError:
+        # PIL not available
+        return None
+    except Exception:
+        # Any other error (no clipboard access, etc.)
+        return None
+
+    return None
+
+
+def get_image_from_path(path: str) -> Optional[Tuple[str, str]]:
+    """Load image from file path.
+
+    Args:
+        path: Path to image file
+
+    Returns:
+        Tuple of (base64_data, format) if successful, None otherwise.
+    """
+    try:
+        from PIL import Image
+
+        image = Image.open(path)
+        buffer = io.BytesIO()
+
+        # Determine format from file extension
+        ext = path.lower().split('.')[-1]
+        if ext in ('jpg', 'jpeg'):
+            if image.mode != 'RGB':
+                image = image.convert('RGB')
+            image.save(buffer, format='JPEG', quality=85)
+            img_format = 'jpeg'
+        elif ext == 'png':
+            image.save(buffer, format='PNG')
+            img_format = 'png'
+        elif ext == 'gif':
+            image.save(buffer, format='GIF')
+            img_format = 'gif'
+        elif ext == 'webp':
+            image.save(buffer, format='WEBP')
+            img_format = 'webp'
+        else:
+            # Default to PNG
+            image.save(buffer, format='PNG')
+            img_format = 'png'
+
+        buffer.seek(0)
+        base64_data = base64.b64encode(buffer.read()).decode('utf-8')
+        return base64_data, img_format
+
+    except Exception:
+        return None
+
+
+def get_image_dimensions(base64_data: str) -> Optional[Tuple[int, int]]:
+    """Get dimensions of base64-encoded image.
+
+    Args:
+        base64_data: Base64-encoded image data
+
+    Returns:
+        Tuple of (width, height) if successful, None otherwise.
+    """
+    try:
+        from PIL import Image
+
+        image_bytes = base64.b64decode(base64_data)
+        image = Image.open(io.BytesIO(image_bytes))
+        return image.size
+    except Exception:
+        return None

{emdash_cli-0.1.8 → emdash_cli-0.1.30}/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",