morphsdk 0.2.5__py3-none-any.whl

morphsdk/resources/compact.py

@@ -0,0 +1,133 @@
+"""Compact resource -- context compression with line-range awareness."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from morphsdk._constants import API_BASE_URL, COMPACT_MODEL, COMPACT_TIMEOUT
+from morphsdk.types.compact import CompactResult
+
+if TYPE_CHECKING:
+    from morphsdk._base import AsyncBaseClient, BaseClient
+
+
+def _build_compact_body(
+    *,
+    input: str | list[dict[str, Any]] | None,
+    messages: list[dict[str, Any]] | None,
+    query: str | None,
+    compression_ratio: float,
+    preserve_recent: int,
+    include_line_ranges: bool,
+    include_markers: bool,
+    model: str | None,
+) -> dict[str, Any]:
+    """Build the ``/v1/compact`` request body.
+
+    Normalizes the *messages*/*input* variants: *messages* takes priority,
+    then a string *input*, then a list *input*. Raises
+    :class:`~morphsdk._errors.ValidationError` if neither is provided.
+    """
+    body: dict[str, Any] = {
+        "compression_ratio": compression_ratio,
+        "preserve_recent": preserve_recent,
+        "model": model or COMPACT_MODEL,
+        "include_line_ranges": include_line_ranges,
+        "include_markers": include_markers,
+    }
+
+    if query is not None:
+        body["query"] = query
+
+    if messages is not None:
+        body["messages"] = messages
+    elif isinstance(input, str):
+        body["input"] = input
+    elif isinstance(input, list):
+        body["messages"] = input
+    else:
+        from morphsdk._errors import ValidationError
+
+        raise ValidationError("Either 'input' or 'messages' must be provided")
+
+    return body
+
+
+def compact(
+    client: BaseClient,
+    *,
+    input: str | list[dict[str, Any]] | None = None,
+    messages: list[dict[str, Any]] | None = None,
+    query: str | None = None,
+    compression_ratio: float = 0.5,
+    preserve_recent: int = 2,
+    include_line_ranges: bool = True,
+    include_markers: bool = True,
+    model: str | None = None,
+    timeout: float | None = None,
+) -> CompactResult:
+    """Compact messages or text via ``/v1/compact``.
+
+    Accepts either a plain *input* string, a list of message dicts via
+    *input*, or a separate *messages* keyword.  Returns per-message
+    ``compacted_line_ranges`` showing which lines were removed.
+
+    *include_line_ranges* controls whether the response includes
+    ``compacted_line_ranges`` metadata (default ``True``).  *include_markers*
+    controls whether ``(filtered N lines)`` text markers are inserted; when
+    ``False``, removed gaps collapse to empty lines (default ``True``).
+    """
+    body = _build_compact_body(
+        input=input,
+        messages=messages,
+        query=query,
+        compression_ratio=compression_ratio,
+        preserve_recent=preserve_recent,
+        include_line_ranges=include_line_ranges,
+        include_markers=include_markers,
+        model=model,
+    )
+
+    response = client._request(
+        "POST",
+        f"{API_BASE_URL}/v1/compact",
+        json=body,
+        timeout=timeout or COMPACT_TIMEOUT,
+    )
+
+    return CompactResult.model_validate(response.json())
+
+
+async def acompact(
+    client: AsyncBaseClient,
+    *,
+    input: str | list[dict[str, Any]] | None = None,
+    messages: list[dict[str, Any]] | None = None,
+    query: str | None = None,
+    compression_ratio: float = 0.5,
+    preserve_recent: int = 2,
+    include_line_ranges: bool = True,
+    include_markers: bool = True,
+    model: str | None = None,
+    timeout: float | None = None,
+) -> CompactResult:
+    """Async counterpart of :func:`compact`. Same wire contract."""
+    body = _build_compact_body(
+        input=input,
+        messages=messages,
+        query=query,
+        compression_ratio=compression_ratio,
+        preserve_recent=preserve_recent,
+        include_line_ranges=include_line_ranges,
+        include_markers=include_markers,
+        model=model,
+    )
+
+    response = await client._request(
+        "POST",
+        f"{API_BASE_URL}/v1/compact",
+        json=body,
+        timeout=timeout or COMPACT_TIMEOUT,
+    )
+
+    return CompactResult.model_validate(response.json())

morphsdk/resources/edit.py

@@ -0,0 +1,506 @@
+"""Fast Apply resource -- AI-powered code editing with intelligent merge."""
+
+from __future__ import annotations
+
+import difflib
+import logging
+import os
+from pathlib import Path
+from typing import TYPE_CHECKING, Any
+
+from morphsdk._constants import API_BASE_URL, FAST_APPLY_MODEL_FAST, FAST_APPLY_MODEL_LARGE
+from morphsdk.types.edit import ApplyEditResult, EditChanges, EditFileResult
+
+if TYPE_CHECKING:
+    from morphsdk._base import AsyncBaseClient, BaseClient
+
+logger = logging.getLogger("morphsdk")
+
+
+def _generate_udiff(original: str, modified: str, filepath: str) -> str:
+    """Generate a unified diff between *original* and *modified*."""
+    return "".join(
+        difflib.unified_diff(
+            original.splitlines(keepends=True),
+            modified.splitlines(keepends=True),
+            fromfile=filepath,
+            tofile=filepath,
+            lineterm="",
+        )
+    )
+
+
+def _count_changes(original: str, modified: str) -> EditChanges:
+    """Count added / removed / modified lines from a unified diff."""
+    diff = list(
+        difflib.unified_diff(
+            original.splitlines(keepends=True),
+            modified.splitlines(keepends=True),
+            lineterm="",
+        )
+    )
+
+    added = 0
+    removed = 0
+    for line in diff:
+        if line.startswith("+") and not line.startswith("+++"):
+            added += 1
+        elif line.startswith("-") and not line.startswith("---"):
+            removed += 1
+
+    modified_count = min(added, removed)
+    return EditChanges(
+        lines_added=added - modified_count,
+        lines_removed=removed - modified_count,
+        lines_modified=modified_count,
+    )
+
+
+def _build_message(instruction: str, original_code: str, code_edit: str) -> str:
+    return (
+        f"<instruction>{instruction}</instruction>\n"
+        f"<code>{original_code}</code>\n"
+        f"<update>{code_edit}</update>"
+    )
+
+
+def _format_edit_result(result: EditFileResult) -> str:
+    """Human-readable summary of an applied edit (shared by tool adapters)."""
+    c = result.changes
+    return (
+        f"Successfully applied changes to {result.path}. "
+        f"Added {c.lines_added}, removed {c.lines_removed}, modified {c.lines_modified}."
+    )
+
+
+def _build_apply_body(
+    original_code: str, code_edit: str, instruction: str
+) -> dict[str, Any]:
+    """Build the ``/v1/chat/completions`` body for a Fast Apply merge."""
+    message = _build_message(instruction, original_code, code_edit)
+    model = (
+        FAST_APPLY_MODEL_LARGE
+        if os.environ.get("MORPH_LARGE_APPLY", "true") != "false"
+        else FAST_APPLY_MODEL_FAST
+    )
+    return {"model": model, "messages": [{"role": "user", "content": message}]}
+
+
+def _parse_apply_response(data: dict[str, Any]) -> tuple[str, str | None]:
+    """Parse a Fast Apply chat-completions response into ``(merged_code, id)``."""
+    content = data["choices"][0]["message"]["content"]
+    if not content:
+        from morphsdk._errors import MorphError
+
+        raise MorphError("Morph API returned empty response")
+
+    return content, data.get("id")
+
+
+def _resolve_edit_path(path: str, base_dir: str | None) -> str | None:
+    """Resolve *path* under *base_dir* (or cwd), guarding against escape.
+
+    Returns the normalized absolute path, or ``None`` if it escapes *base_dir*.
+    """
+    base = base_dir or os.getcwd()
+    full_path = os.path.normpath(os.path.join(base, path))
+    if not full_path.startswith(os.path.normpath(base)):
+        return None
+    return full_path
+
+
+def _read_original(full_path: str) -> str:
+    """Read existing file content, or an empty string for new files."""
+    try:
+        return Path(full_path).read_text(encoding="utf-8")
+    except FileNotFoundError:
+        return ""
+
+
+def _write_merged(full_path: str, merged_code: str) -> None:
+    """Write *merged_code* to *full_path*, creating parent directories."""
+    Path(full_path).parent.mkdir(parents=True, exist_ok=True)
+    Path(full_path).write_text(merged_code, encoding="utf-8")
+
+
+def _assemble_file_result(
+    *,
+    path: str,
+    original_code: str,
+    merged_code: str,
+    completion_id: str | None,
+    generate_udiff: bool,
+) -> EditFileResult:
+    """Build the public :class:`EditFileResult` from a merge."""
+    udiff = _generate_udiff(original_code, merged_code, path) if generate_udiff else None
+    return EditFileResult(
+        path=path,
+        udiff=udiff,
+        changes=_count_changes(original_code, merged_code),
+        completion_id=completion_id,
+    )
+
+
+def _assemble_code_result(
+    *,
+    original_code: str,
+    merged_code: str,
+    completion_id: str | None,
+    generate_udiff: bool,
+) -> ApplyEditResult:
+    """Build the public :class:`ApplyEditResult` from a merge."""
+    udiff = _generate_udiff(original_code, merged_code, "file") if generate_udiff else None
+    return ApplyEditResult(
+        merged_code=merged_code,
+        udiff=udiff,
+        changes=_count_changes(original_code, merged_code),
+        completion_id=completion_id,
+    )
+
+
+class EditResource:
+    """AI-powered code editing via Morph Fast Apply."""
+
+    def __init__(self, client: BaseClient) -> None:
+        self._client = client
+
+    # ------------------------------------------------------------------
+    # file()  -- read from disk, merge, write back
+    # ------------------------------------------------------------------
+
+    def file(
+        self,
+        *,
+        path: str,
+        instruction: str,
+        code_edit: str,
+        base_dir: str | None = None,
+        generate_udiff: bool = True,
+        auto_write: bool = True,
+        timeout: float | None = None,
+    ) -> EditFileResult:
+        """Edit a file on disk using Morph Fast Apply.
+
+        1. Resolves *path* relative to *base_dir* (or cwd).
+        2. Reads current content (empty string for new files).
+        3. Calls the Morph merge API.
+        4. Optionally writes the merged result and generates a udiff.
+        """
+        full_path = _resolve_edit_path(path, base_dir)
+        if full_path is None:
+            return EditFileResult(
+                path=path,
+                changes=EditChanges(lines_added=0, lines_removed=0, lines_modified=0),
+            )
+
+        original_code = _read_original(full_path)
+
+        merged_code, completion_id = self._call_api(
+            original_code, code_edit, instruction, timeout=timeout
+        )
+
+        if auto_write:
+            _write_merged(full_path, merged_code)
+
+        return _assemble_file_result(
+            path=path,
+            original_code=original_code,
+            merged_code=merged_code,
+            completion_id=completion_id,
+            generate_udiff=generate_udiff,
+        )
+
+    # ------------------------------------------------------------------
+    # code()  -- pure code-in / code-out, no file I/O
+    # ------------------------------------------------------------------
+
+    def code(
+        self,
+        *,
+        original_code: str,
+        code_edit: str,
+        instruction: str,
+        generate_udiff: bool = True,
+        timeout: float | None = None,
+    ) -> ApplyEditResult:
+        """Apply an edit to code directly without file I/O."""
+        merged_code, completion_id = self._call_api(
+            original_code, code_edit, instruction, timeout=timeout
+        )
+
+        return _assemble_code_result(
+            original_code=original_code,
+            merged_code=merged_code,
+            completion_id=completion_id,
+            generate_udiff=generate_udiff,
+        )
+
+    # ------------------------------------------------------------------
+    # Internal
+    # ------------------------------------------------------------------
+
+    # ------------------------------------------------------------------
+    # Framework adapters
+    # ------------------------------------------------------------------
+
+    def as_openai_tool(
+        self,
+        *,
+        base_dir: str | None = None,
+        description: str | None = None,
+    ) -> Any:
+        """Create an OpenAI ChatCompletionTool for edit_file."""
+        from morphsdk.adapters.openai import (
+            EDIT_FILE_SYSTEM_PROMPT,
+            OpenAITool,
+            edit_file_tool_def,
+        )
+
+        effective_base_dir = base_dir
+
+        def executor(
+            target_filepath: str,
+            instruction: str,
+            code_edit: str,
+        ) -> EditFileResult:
+            return self.file(
+                path=target_filepath,
+                instruction=instruction,
+                code_edit=code_edit,
+                base_dir=effective_base_dir,
+            )
+
+        return OpenAITool(
+            definition=edit_file_tool_def(description=description),
+            executor=executor,
+            formatter=_format_edit_result,
+            system_prompt=EDIT_FILE_SYSTEM_PROMPT,
+        )
+
+    def as_anthropic_tool(
+        self,
+        *,
+        base_dir: str | None = None,
+        description: str | None = None,
+    ) -> Any:
+        """Create an Anthropic Tool for edit_file."""
+        from morphsdk.adapters.anthropic import AnthropicTool, edit_file_tool_def
+        from morphsdk.adapters.openai import EDIT_FILE_SYSTEM_PROMPT
+
+        effective_base_dir = base_dir
+
+        def executor(
+            target_filepath: str,
+            instruction: str,
+            code_edit: str,
+        ) -> EditFileResult:
+            return self.file(
+                path=target_filepath,
+                instruction=instruction,
+                code_edit=code_edit,
+                base_dir=effective_base_dir,
+            )
+
+        return AnthropicTool(
+            definition=edit_file_tool_def(description=description),
+            executor=executor,
+            formatter=_format_edit_result,
+            system_prompt=EDIT_FILE_SYSTEM_PROMPT,
+        )
+
+    # ------------------------------------------------------------------
+    # Internal
+    # ------------------------------------------------------------------
+
+    def _call_api(
+        self,
+        original_code: str,
+        code_edit: str,
+        instruction: str,
+        *,
+        timeout: float | None = None,
+    ) -> tuple[str, str | None]:
+        """Call the Morph Fast Apply chat completions endpoint.
+
+        Returns ``(merged_code, completion_id)``.
+        """
+        response = self._client._request(
+            "POST",
+            f"{API_BASE_URL}/v1/chat/completions",
+            json=_build_apply_body(original_code, code_edit, instruction),
+            timeout=timeout,
+        )
+
+        return _parse_apply_response(response.json())
+
+
+class AsyncEditResource:
+    """Async AI-powered code editing via Morph Fast Apply."""
+
+    def __init__(self, client: AsyncBaseClient) -> None:
+        self._client = client
+
+    # ------------------------------------------------------------------
+    # file()  -- read from disk, merge, write back
+    # ------------------------------------------------------------------
+
+    async def file(
+        self,
+        *,
+        path: str,
+        instruction: str,
+        code_edit: str,
+        base_dir: str | None = None,
+        generate_udiff: bool = True,
+        auto_write: bool = True,
+        timeout: float | None = None,
+    ) -> EditFileResult:
+        """Edit a file on disk using Morph Fast Apply.
+
+        1. Resolves *path* relative to *base_dir* (or cwd).
+        2. Reads current content (empty string for new files).
+        3. Calls the Morph merge API.
+        4. Optionally writes the merged result and generates a udiff.
+        """
+        full_path = _resolve_edit_path(path, base_dir)
+        if full_path is None:
+            return EditFileResult(
+                path=path,
+                changes=EditChanges(lines_added=0, lines_removed=0, lines_modified=0),
+            )
+
+        original_code = _read_original(full_path)
+
+        merged_code, completion_id = await self._call_api(
+            original_code, code_edit, instruction, timeout=timeout
+        )
+
+        if auto_write:
+            _write_merged(full_path, merged_code)
+
+        return _assemble_file_result(
+            path=path,
+            original_code=original_code,
+            merged_code=merged_code,
+            completion_id=completion_id,
+            generate_udiff=generate_udiff,
+        )
+
+    # ------------------------------------------------------------------
+    # code()  -- pure code-in / code-out, no file I/O
+    # ------------------------------------------------------------------
+
+    async def code(
+        self,
+        *,
+        original_code: str,
+        code_edit: str,
+        instruction: str,
+        generate_udiff: bool = True,
+        timeout: float | None = None,
+    ) -> ApplyEditResult:
+        """Apply an edit to code directly without file I/O."""
+        merged_code, completion_id = await self._call_api(
+            original_code, code_edit, instruction, timeout=timeout
+        )
+
+        return _assemble_code_result(
+            original_code=original_code,
+            merged_code=merged_code,
+            completion_id=completion_id,
+            generate_udiff=generate_udiff,
+        )
+
+    # ------------------------------------------------------------------
+    # Framework adapters
+    # ------------------------------------------------------------------
+
+    def as_openai_tool(
+        self,
+        *,
+        base_dir: str | None = None,
+        description: str | None = None,
+    ) -> Any:
+        """Create an OpenAI ChatCompletionTool for edit_file with an async executor."""
+        from morphsdk.adapters.openai import (
+            EDIT_FILE_SYSTEM_PROMPT,
+            OpenAITool,
+            edit_file_tool_def,
+        )
+
+        effective_base_dir = base_dir
+
+        async def executor(
+            target_filepath: str,
+            instruction: str,
+            code_edit: str,
+        ) -> EditFileResult:
+            return await self.file(
+                path=target_filepath,
+                instruction=instruction,
+                code_edit=code_edit,
+                base_dir=effective_base_dir,
+            )
+
+        return OpenAITool(
+            definition=edit_file_tool_def(description=description),
+            executor=executor,
+            formatter=_format_edit_result,
+            system_prompt=EDIT_FILE_SYSTEM_PROMPT,
+        )
+
+    def as_anthropic_tool(
+        self,
+        *,
+        base_dir: str | None = None,
+        description: str | None = None,
+    ) -> Any:
+        """Create an Anthropic Tool for edit_file with an async executor."""
+        from morphsdk.adapters.anthropic import AnthropicTool, edit_file_tool_def
+        from morphsdk.adapters.openai import EDIT_FILE_SYSTEM_PROMPT
+
+        effective_base_dir = base_dir
+
+        async def executor(
+            target_filepath: str,
+            instruction: str,
+            code_edit: str,
+        ) -> EditFileResult:
+            return await self.file(
+                path=target_filepath,
+                instruction=instruction,
+                code_edit=code_edit,
+                base_dir=effective_base_dir,
+            )
+
+        return AnthropicTool(
+            definition=edit_file_tool_def(description=description),
+            executor=executor,
+            formatter=_format_edit_result,
+            system_prompt=EDIT_FILE_SYSTEM_PROMPT,
+        )
+
+    # ------------------------------------------------------------------
+    # Internal
+    # ------------------------------------------------------------------
+
+    async def _call_api(
+        self,
+        original_code: str,
+        code_edit: str,
+        instruction: str,
+        *,
+        timeout: float | None = None,
+    ) -> tuple[str, str | None]:
+        """Call the Morph Fast Apply chat completions endpoint.
+
+        Returns ``(merged_code, completion_id)``.
+        """
+        response = await self._client._request(
+            "POST",
+            f"{API_BASE_URL}/v1/chat/completions",
+            json=_build_apply_body(original_code, code_edit, instruction),
+            timeout=timeout,
+        )
+
+        return _parse_apply_response(response.json())