mrmd-ai 0.1.0__py3-none-any.whl

mrmd_ai/signatures/text.py

@@ -0,0 +1,134 @@
+"""Signature definitions for text transformation programs."""
+
+import dspy
+from typing import Optional, List
+
+
+class GetSynonymsSignature(dspy.Signature):
+    """
+    Find synonyms for a SINGLE WORD that fit the context.
+
+    Guidelines:
+    - Provide 3-6 alternative words that could replace the input word
+    - Consider the surrounding context to suggest appropriate synonyms
+    - Preserve the part of speech (noun, verb, adjective, etc.)
+    - Order synonyms by relevance to the context
+    - Synonyms should be single words or very short (1-2 words max)
+    """
+
+    text: str = dspy.InputField(
+        desc="The single word to find synonyms for"
+    )
+    local_context: str = dspy.InputField(
+        desc="Surrounding text to understand usage context"
+    )
+    document_context: Optional[str] = dspy.InputField(
+        desc="Full document for topic understanding",
+        default=None,
+    )
+    original: str = dspy.OutputField(
+        desc="The original word (echoed back)"
+    )
+    synonyms: List[str] = dspy.OutputField(
+        desc="List of 3-6 synonym alternatives (prefer single words)"
+    )
+
+
+class GetPhraseSynonymsSignature(dspy.Signature):
+    """
+    Find alternative phrases for a multi-word expression that fit the context.
+
+    Guidelines:
+    - Provide 3-6 alternative phrases that could replace the input phrase
+    - Preserve the meaning and grammatical structure
+    - Consider the surrounding context for appropriate alternatives
+    - Alternatives should be complete phrases that can directly replace the original
+    - Order by relevance to the context
+    """
+
+    phrase: str = dspy.InputField(
+        desc="The multi-word phrase to find alternatives for (e.g., 'executable notebooks')"
+    )
+    local_context: str = dspy.InputField(
+        desc="Surrounding text to understand usage context"
+    )
+    document_context: Optional[str] = dspy.InputField(
+        desc="Full document for topic understanding",
+        default=None,
+    )
+    original: str = dspy.OutputField(
+        desc="The original phrase (echoed back)"
+    )
+    alternatives: List[str] = dspy.OutputField(
+        desc="List of 3-6 alternative phrases that preserve meaning"
+    )
+
+
+class ReformatMarkdownSignature(dspy.Signature):
+    """
+    Clean up and reformat markdown text while preserving meaning.
+
+    CRITICAL: You must ONLY reformat the `text` field. Do NOT include text from `local_context`.
+
+    Guidelines:
+    - Fix inconsistent heading levels
+    - Normalize list formatting (bullets, numbering)
+    - Clean up whitespace and line breaks
+    - Fix broken links and formatting
+    - Improve paragraph structure
+    - Preserve all content and meaning exactly
+    - Output must be EXACTLY the `text` input reformatted
+    """
+
+    text: str = dspy.InputField(
+        desc="The EXACT markdown text to reformat. Output must contain ONLY this text reformatted."
+    )
+    local_context: str = dspy.InputField(
+        desc="Surrounding text for style context ONLY - do NOT include this in output"
+    )
+    document_context: Optional[str] = dspy.InputField(
+        desc="Full document for understanding structure - do NOT include in output",
+        default=None,
+    )
+    reformatted_text: str = dspy.OutputField(
+        desc="ONLY the `text` input reformatted. Must have same content, cleaner formatting."
+    )
+
+
+class IdentifyReplacementSignature(dspy.Signature):
+    """
+    Identify the exact phrase to replace when applying a synonym.
+
+    Given an original word and a chosen synonym, analyze the context to determine
+    what COMPLETE phrase should be replaced.
+
+    CRITICAL RULES:
+    1. If the synonym is "adjective noun" (e.g., "interactive notebooks"), find if the original
+       word in context also has an adjective (e.g., "executable notebooks") and replace the WHOLE phrase
+    2. The text_to_replace MUST be copied EXACTLY from the context - character for character
+    3. The replacement should result in grammatically correct text
+
+    Examples:
+    - original_word="notebooks", synonym="interactive notebooks", context has "executable notebooks"
+      → text_to_replace="executable notebooks", replacement="interactive notebooks"
+    - original_word="notebooks", synonym="notebooks", context has "executable notebooks"
+      → text_to_replace="notebooks", replacement="notebooks"
+    - original_word="big", synonym="large", context has "big house"
+      → text_to_replace="big", replacement="large"
+    """
+
+    original_word: str = dspy.InputField(
+        desc="The word that was identified for synonym replacement"
+    )
+    chosen_synonym: str = dspy.InputField(
+        desc="The synonym the user chose (may be single or multi-word like 'interactive notebooks')"
+    )
+    context: str = dspy.InputField(
+        desc="The text surrounding the original word - search here for the phrase to replace"
+    )
+    text_to_replace: str = dspy.OutputField(
+        desc="The EXACT phrase copied from context that should be replaced. If synonym is 'adj noun', find 'adj noun' pattern in context containing original_word."
+    )
+    replacement: str = dspy.OutputField(
+        desc="The text to insert - usually the chosen_synonym, possibly adjusted for grammar"
+    )

mrmd_ai/utils/__init__.py

@@ -0,0 +1 @@
+"""Utility functions for MRMD AI programs."""

mrmd_ai-0.1.0.dist-info/METADATA

@@ -0,0 +1,45 @@
+Metadata-Version: 2.4
+Name: mrmd-ai
+Version: 0.1.0
+Summary: AI programs for MRMD editor - completions, fixes, and corrections
+Requires-Python: >=3.11
+Requires-Dist: dspy>=2.6
+Requires-Dist: fastapi>=0.115
+Requires-Dist: litellm>=1.50
+Requires-Dist: uvicorn>=0.32
+Description-Content-Type: text/markdown
+
+# MRMD AI Server
+
+AI programs for the MRMD editor using DSPy.
+
+## Programs
+
+### Finish (Completion)
+- `FinishSentencePredict` - Complete the current sentence
+- `FinishParagraphPredict` - Complete the current paragraph
+- `FinishCodeLinePredict` - Complete the current code line
+- `FinishCodeSectionPredict` - Complete the current code section
+
+### Fix (Correction)
+- `FixGrammarPredict` - Fix grammar/spelling errors
+- `FixTranscriptionPredict` - Fix speech-to-text errors
+
+### Correct & Finish
+- `CorrectAndFinishLinePredict` - Correct and complete current line
+- `CorrectAndFinishSectionPredict` - Correct and complete current section
+
+## Running
+
+```bash
+cd ai-server
+uv run dspy-cli serve --port 8766
+```
+
+## Configuration
+
+API keys are read from environment variables:
+- `ANTHROPIC_API_KEY`
+- `OPENAI_API_KEY`
+
+See `dspy.config.yaml` for model configuration.

mrmd_ai-0.1.0.dist-info/RECORD

@@ -0,0 +1,26 @@
+mrmd_ai/__init__.py,sha256=RocuoOEEJCUNinxqsXWjm_bIimCu8aDCDha-G5pU4fU,105
+mrmd_ai/juice.py,sha256=0_h7_M89IDhSt4iohdz0J5StcFXOaLF33d3fTxdBO-o,14442
+mrmd_ai/server.py,sha256=Jgl-bsQYbBMtv_tGu_2FwSN9BxgFpt-fmb9_pjIcqNw,14416
+mrmd_ai/metrics/__init__.py,sha256=6BngKqh0a09phOzUdYeWjhsUXznCIx5jEjrEt7DIDu4,62
+mrmd_ai/modules/__init__.py,sha256=4V4IzurDZs_nhBTGlHyXco24iwEkum58kRNrKfm4IjQ,1793
+mrmd_ai/modules/code.py,sha256=8cK6LF0ZTrSp2srt1ltvUnmYjPlt5NZTj1yWctpJ7j0,4099
+mrmd_ai/modules/correct.py,sha256=TWnE1HD_Ip7xZ5yQwJi1n01tNXgBtYNvkTK--kAknak,1478
+mrmd_ai/modules/document.py,sha256=o6iLR2amscn-DHZ95JFQuEwhaj8cLs4bBISf7G9cT9Y,1106
+mrmd_ai/modules/finish.py,sha256=VtyE-45-8iM6iWjNg57wRWL0Ln3mSF0RcZq8CO5CoSk,2638
+mrmd_ai/modules/fix.py,sha256=fb4flKWyyyelheigeb1sI0dixd2scL9HZX-0_M_Uh-o,1506
+mrmd_ai/modules/notebook.py,sha256=w8Dg-NKVL6_kPOKkvb84kGrwgv5zHxvFNWBtXHLHww8,477
+mrmd_ai/modules/text.py,sha256=9MCO__EDalwi-iFf__sd8t7orsUy7WiBBf6Lp4bxxGE,2010
+mrmd_ai/optimizers/__init__.py,sha256=Ay6ZrQu8fLQaG7-dl6hTMruQY5AdGOT_YnlRhhZGgag,60
+mrmd_ai/signatures/__init__.py,sha256=wWT2D8beisIpYMPxN4j4JncQmUqcamD6v5BTKp2zFWc,655
+mrmd_ai/signatures/code.py,sha256=zBM_Nl2NImfOw49fVWCGlXcE_sm8xgWCN1ksDbEa6e8,11245
+mrmd_ai/signatures/correct.py,sha256=tIhYCONgGhuTV0eJCiLSXcGZSAEi06XY35ommtTTsRE,2920
+mrmd_ai/signatures/document.py,sha256=4Y-9SeXJGCq098Vy-PIbb_rexS2dYDlkU-kxnKAPVSU,1828
+mrmd_ai/signatures/finish.py,sha256=x-ZB0U8GQJdNoGGO80FBOxHXjYsCmTFq9fnkXlHDeUY,5294
+mrmd_ai/signatures/fix.py,sha256=LJNvu9_XjPl90Wtt3xn6s-jGXA9GB5rdIL0MeFyRGtE,3042
+mrmd_ai/signatures/notebook.py,sha256=ZBioHA9ZTkLUD_UovdfiRYiDaUKuKOCDhiZP1NDFY8o,1226
+mrmd_ai/signatures/text.py,sha256=GhmFtEZqwivbevPI_NSBzh6AlH6JKLt2rA_LaYGK2lQ,5223
+mrmd_ai/utils/__init__.py,sha256=T4e9jmFWDSj1HOyz5_Qv-JQSC08GwT_9CACcAn37vWg,46
+mrmd_ai-0.1.0.dist-info/METADATA,sha256=4w17Do4YbBpwoT3PFN2i01zswTPJCVzR6rgib0zZXME,1167
+mrmd_ai-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mrmd_ai-0.1.0.dist-info/entry_points.txt,sha256=Bq6nXiXxhNSPEgYWBgrrgJH1DeGjKM6hfCijcWClApw,55
+mrmd_ai-0.1.0.dist-info/RECORD,,

mrmd_ai-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

mrmd_ai-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+mrmd-ai-server = mrmd_ai.server:main