emdash-core 0.1.16__tar.gz → 0.1.50__tar.gz

{emdash_core-0.1.16 → emdash_core-0.1.50}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emdash-core
-Version: 0.1.16
+Version: 0.1.50
 Summary: EmDash Core - FastAPI server for code intelligence
 Author: Em Dash Team
 Requires-Python: >=3.10,<4.0
@@ -10,26 +10,27 @@ 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: graph
 Requires-Dist: astroid (>=3.0.1,<4.0.0)
 Requires-Dist: beautifulsoup4 (>=4.12.0)
+Requires-Dist: claude-agent-sdk (>=0.1.19)
+Requires-Dist: cmake (>=3.25.0) ; extra == "graph"
 Requires-Dist: duckduckgo-search (>=6.0.0)
 Requires-Dist: fastapi (>=0.109.0)
 Requires-Dist: gitpython (>=3.1.40,<4.0.0)
 Requires-Dist: httpx (>=0.25.0)
-Requires-Dist: kuzu (>=0.4.0)
+Requires-Dist: kuzu (>=0.4.0) ; extra == "graph"
 Requires-Dist: loguru (>=0.7.2,<0.8.0)
 Requires-Dist: networkx (>=3.2.1,<4.0.0)
 Requires-Dist: numpy (>=1.26.0)
 Requires-Dist: openai (>=1.0.0)
-Requires-Dist: pillow (>=10.0.0,<11.0.0)
 Requires-Dist: pydantic (>=2.5.0,<3.0.0)
 Requires-Dist: pydantic-settings (>=2.0.0,<3.0.0)
 Requires-Dist: pygithub (>=2.1.1,<3.0.0)
+Requires-Dist: pypng (>=0.0.21,<0.0.22)
 Requires-Dist: python-dotenv (>=1.0.0,<2.0.0)
 Requires-Dist: python-louvain (>=0.16,<0.17)
-Requires-Dist: scipy (>=1.11.4,<2.0.0)
-Requires-Dist: sentence-transformers (>=2.2.0)
+Requires-Dist: pyyaml (>=6.0,<7.0)
 Requires-Dist: sse-starlette (>=2.0.0)
-Requires-Dist: supabase (>=2.0.0)
 Requires-Dist: tqdm (>=4.66.1,<5.0.0)
 Requires-Dist: uvicorn[standard] (>=0.27.0)

{emdash_core-0.1.16 → emdash_core-0.1.50}/emdash_core/agent/__init__.py

@@ -19,6 +19,9 @@ def __getattr__(name: str):
     elif name == "AgentRunner":
         from .runner import AgentRunner
         return AgentRunner
+    elif name == "SafeJSONEncoder":
+        from .runner import SafeJSONEncoder
+        return SafeJSONEncoder
     elif name == "ToolResult":
         from .tools.base import ToolResult
         return ToolResult
@@ -32,6 +35,7 @@ __all__ = [
     "AgentToolkit",
     "AgentSession",
     "AgentRunner",
+    "SafeJSONEncoder",
     "ToolResult",
     "ToolCategory",
 ]

{emdash_core-0.1.16 → emdash_core-0.1.50}/emdash_core/agent/agents.py

@@ -2,16 +2,74 @@
 
 Allows users to define custom agent configurations with
 specialized system prompts and tool selections.
+
+Example agent file:
+```markdown
+---
+description: GitHub integration agent
+model: claude-sonnet-4-20250514
+tools: [grep, glob, read_file]
+mcp_servers:
+  github:
+    command: github-mcp-server
+    args: []
+    env:
+      GITHUB_TOKEN: ${GITHUB_TOKEN}
+    enabled: true
+  filesystem:
+    command: npx
+    args: [-y, "@anthropic/mcp-server-filesystem", "/tmp"]
+    enabled: false  # Disabled - won't be started
+---
+
+# System Prompt
+
+You are a GitHub integration specialist...
+```
 """
 
 from dataclasses import dataclass, field
 from pathlib import Path
-from typing import Optional
+from typing import Any, Optional
 import re
 
+import yaml
+
 from ..utils.logger import log
 
 
+@dataclass
+class AgentMCPServerConfig:
+    """MCP server configuration for a custom agent.
+
+    Attributes:
+        name: Server name (key in mcp_servers dict)
+        command: Command to run the server
+        args: Arguments to pass to the command
+        env: Environment variables (supports ${VAR} syntax)
+        enabled: Whether this server is enabled (default: True)
+        timeout: Timeout in seconds for tool calls
+    """
+    name: str
+    command: str
+    args: list[str] = field(default_factory=list)
+    env: dict[str, str] = field(default_factory=dict)
+    enabled: bool = True
+    timeout: int = 30
+
+    @classmethod
+    def from_dict(cls, name: str, data: dict[str, Any]) -> "AgentMCPServerConfig":
+        """Create from dictionary parsed from YAML."""
+        return cls(
+            name=name,
+            command=data.get("command", ""),
+            args=data.get("args", []),
+            env=data.get("env", {}),
+            enabled=data.get("enabled", True),
+            timeout=data.get("timeout", 30),
+        )
+
+
 @dataclass
 class CustomAgent:
     """A custom agent configuration loaded from markdown.
@@ -19,16 +77,26 @@ class CustomAgent:
     Attributes:
         name: Agent name (from filename)
         description: Brief description
+        model: Model to use for this agent (optional, uses default if not set)
         system_prompt: Custom system prompt
         tools: List of tools to enable
+        mcp_servers: MCP server configurations for this agent
+        rules: List of rule names to apply (references .emdash/rules/)
+        skills: List of skill names to enable (references .emdash/skills/)
+        verifiers: List of verifier names to use (references .emdash/verifiers.json)
         examples: Example interactions
         file_path: Source file path
     """
 
     name: str
     description: str = ""
+    model: Optional[str] = None
     system_prompt: str = ""
     tools: list[str] = field(default_factory=list)
+    mcp_servers: list[AgentMCPServerConfig] = field(default_factory=list)
+    rules: list[str] = field(default_factory=list)
+    skills: list[str] = field(default_factory=list)
+    verifiers: list[str] = field(default_factory=list)
     examples: list[dict] = field(default_factory=list)
     file_path: Optional[Path] = None
 
@@ -121,46 +189,48 @@ def _parse_agent_file(file_path: Path) -> Optional[CustomAgent]:
     if system_prompt.startswith("# System Prompt"):
         system_prompt = system_prompt[len("# System Prompt") :].strip()
 
+    # Parse MCP servers from frontmatter
+    mcp_servers = []
+    mcp_servers_data = frontmatter.get("mcp_servers", {})
+    if isinstance(mcp_servers_data, dict):
+        for server_name, server_config in mcp_servers_data.items():
+            if isinstance(server_config, dict):
+                mcp_servers.append(
+                    AgentMCPServerConfig.from_dict(server_name, server_config)
+                )
+
     return CustomAgent(
         name=file_path.stem,
         description=frontmatter.get("description", ""),
+        model=frontmatter.get("model"),
         system_prompt=system_prompt,
         tools=frontmatter.get("tools", []),
+        mcp_servers=mcp_servers,
+        rules=frontmatter.get("rules", []),
+        skills=frontmatter.get("skills", []),
+        verifiers=frontmatter.get("verifiers", []),
         examples=examples,
         file_path=file_path,
     )
 
 
 def _parse_frontmatter(frontmatter_str: str) -> dict:
-    """Parse YAML-like frontmatter.
+    """Parse YAML frontmatter.
 
-    Simple parser for key: value pairs.
+    Uses PyYAML for proper nested structure parsing.
 
     Args:
-        frontmatter_str: Frontmatter string
+        frontmatter_str: Frontmatter string (YAML format)
 
     Returns:
         Dict of parsed values
     """
-    result = {}
-
-    for line in frontmatter_str.strip().split("\n"):
-        if ":" not in line:
-            continue
-
-        key, value = line.split(":", 1)
-        key = key.strip()
-        value = value.strip()
-
-        # Parse list values
-        if value.startswith("[") and value.endswith("]"):
-            # Simple list parsing
-            items = value[1:-1].split(",")
-            result[key] = [item.strip().strip("'\"") for item in items if item.strip()]
-        else:
-            result[key] = value.strip("'\"")
-
-    return result
+    try:
+        result = yaml.safe_load(frontmatter_str)
+        return result if isinstance(result, dict) else {}
+    except yaml.YAMLError as e:
+        log.warning(f"Failed to parse frontmatter as YAML: {e}")
+        return {}
 
 
 def _parse_examples(examples_str: str) -> list[dict]: