mrmd-ai 0.1.0__py3-none-any.whl

mrmd_ai/modules/document.py

@@ -0,0 +1,41 @@
+"""Document-level AI modules."""
+
+import dspy
+from ..signatures.document import (
+    DocumentResponseSignature,
+    DocumentSummarySignature,
+    DocumentAnalysisSignature,
+)
+
+
+class DocumentResponsePredict(dspy.Module):
+    """Generate a response to the document (document as prompt)."""
+
+    def __init__(self):
+        super().__init__()
+        self.predict = dspy.Predict(DocumentResponseSignature)
+
+    def forward(self, document: str):
+        return self.predict(document=document)
+
+
+class DocumentSummaryPredict(dspy.Module):
+    """Summarize a document."""
+
+    def __init__(self):
+        super().__init__()
+        self.predict = dspy.Predict(DocumentSummarySignature)
+
+    def forward(self, document: str):
+        return self.predict(document=document)
+
+
+class DocumentAnalysisPredict(dspy.Module):
+    """Analyze a document."""
+
+    def __init__(self):
+        super().__init__()
+        self.predict = dspy.Predict(DocumentAnalysisSignature)
+
+    def forward(self, document: str, analysis_type: str = "general"):
+        return self.predict(document=document, analysis_type=analysis_type)

mrmd_ai/modules/finish.py

@@ -0,0 +1,95 @@
+"""Finish/completion modules - complete sentences, paragraphs, and code."""
+
+import dspy
+from typing import Optional
+
+from mrmd_ai.signatures.finish import (
+    FinishSentenceSignature,
+    FinishParagraphSignature,
+    FinishCodeLineSignature,
+    FinishCodeSectionSignature,
+)
+
+
+class FinishSentencePredict(dspy.Module):
+    """Complete the current sentence naturally."""
+
+    def __init__(self):
+        super().__init__()
+        self.predictor = dspy.Predict(FinishSentenceSignature)
+
+    def forward(
+        self,
+        text_before_cursor: str,
+        local_context: str,
+        document_context: Optional[str] = None,
+    ) -> dspy.Prediction:
+        return self.predictor(
+            document_context=document_context,
+            local_context=local_context,
+            text_before_cursor=text_before_cursor,
+        )
+
+
+class FinishParagraphPredict(dspy.Module):
+    """Complete the current paragraph naturally."""
+
+    def __init__(self):
+        super().__init__()
+        self.predictor = dspy.Predict(FinishParagraphSignature)
+
+    def forward(
+        self,
+        text_before_cursor: str,
+        local_context: str,
+        document_context: Optional[str] = None,
+    ) -> dspy.Prediction:
+        return self.predictor(
+            document_context=document_context,
+            local_context=local_context,
+            text_before_cursor=text_before_cursor,
+        )
+
+
+class FinishCodeLinePredict(dspy.Module):
+    """Complete the current line of code."""
+
+    def __init__(self):
+        super().__init__()
+        self.predictor = dspy.Predict(FinishCodeLineSignature)
+
+    def forward(
+        self,
+        code_before_cursor: str,
+        language: str,
+        local_context: str,
+        document_context: Optional[str] = None,
+    ) -> dspy.Prediction:
+        return self.predictor(
+            document_context=document_context,
+            local_context=local_context,
+            code_before_cursor=code_before_cursor,
+            language=language,
+        )
+
+
+class FinishCodeSectionPredict(dspy.Module):
+    """Complete the current code section (function, class, block)."""
+
+    def __init__(self):
+        super().__init__()
+        self.predictor = dspy.Predict(FinishCodeSectionSignature)
+
+    def forward(
+        self,
+        code_before_cursor: str,
+        language: str,
+        local_context: str,
+        document_context: Optional[str] = None,
+    ) -> dspy.Prediction:
+        return self.predictor(
+            document_context=document_context,
+            local_context=local_context,
+            code_before_cursor=code_before_cursor,
+            language=language,
+        )

mrmd_ai/modules/fix.py

@@ -0,0 +1,52 @@
+"""Fix modules - grammar and transcription correction."""
+
+import dspy
+from typing import Optional
+
+from mrmd_ai.signatures.fix import (
+    FixGrammarSignature,
+    FixTranscriptionSignature,
+)
+
+
+class FixGrammarPredict(dspy.Module):
+    """Fix grammar, spelling, and punctuation errors."""
+
+    def __init__(self):
+        super().__init__()
+        self.predictor = dspy.Predict(FixGrammarSignature)
+
+    def forward(
+        self,
+        text_to_fix: str,
+        local_context: str,
+        document_context: Optional[str] = None,
+    ) -> dspy.Prediction:
+        print(f"[FixGrammar] Input: {repr(text_to_fix[:100])}", flush=True)
+        result = self.predictor(
+            document_context=document_context,
+            local_context=local_context,
+            text_to_fix=text_to_fix,
+        )
+        print(f"[FixGrammar] Output: {repr(result.fixed_text[:100] if hasattr(result, 'fixed_text') else 'NO fixed_text')}", flush=True)
+        return result
+
+
+class FixTranscriptionPredict(dspy.Module):
+    """Fix speech-to-text transcription errors."""
+
+    def __init__(self):
+        super().__init__()
+        self.predictor = dspy.Predict(FixTranscriptionSignature)
+
+    def forward(
+        self,
+        text_to_fix: str,
+        local_context: str,
+        document_context: Optional[str] = None,
+    ) -> dspy.Prediction:
+        return self.predictor(
+            document_context=document_context,
+            local_context=local_context,
+            text_to_fix=text_to_fix,
+        )

mrmd_ai/modules/notebook.py

@@ -0,0 +1,15 @@
+"""Notebook-related AI modules."""
+
+import dspy
+from ..signatures.notebook import NotebookNameSignature
+
+
+class NotebookNamePredict(dspy.Module):
+    """Generate a descriptive name for a notebook based on its content."""
+
+    def __init__(self):
+        super().__init__()
+        self.predict = dspy.Predict(NotebookNameSignature)
+
+    def forward(self, document: str, current_name: str = "Untitled"):
+        return self.predict(document=document, current_name=current_name)

mrmd_ai/modules/text.py

@@ -0,0 +1,69 @@
+"""Text transformation modules."""
+
+import dspy
+from ..signatures.text import (
+    GetSynonymsSignature,
+    GetPhraseSynonymsSignature,
+    ReformatMarkdownSignature,
+    IdentifyReplacementSignature,
+)
+
+
+class GetSynonymsPredict(dspy.Module):
+    """Find synonyms for a single word."""
+
+    def __init__(self):
+        super().__init__()
+        self.predict = dspy.Predict(GetSynonymsSignature)
+
+    def forward(self, text: str, local_context: str, document_context: str = None):
+        return self.predict(
+            text=text,
+            local_context=local_context,
+            document_context=document_context,
+        )
+
+
+class GetPhraseSynonymsPredict(dspy.Module):
+    """Find alternative phrases for multi-word expressions."""
+
+    def __init__(self):
+        super().__init__()
+        self.predict = dspy.Predict(GetPhraseSynonymsSignature)
+
+    def forward(self, phrase: str, local_context: str, document_context: str = None):
+        return self.predict(
+            phrase=phrase,
+            local_context=local_context,
+            document_context=document_context,
+        )
+
+
+class ReformatMarkdownPredict(dspy.Module):
+    """Clean up and reformat markdown text."""
+
+    def __init__(self):
+        super().__init__()
+        self.predict = dspy.Predict(ReformatMarkdownSignature)
+
+    def forward(self, text: str, local_context: str, document_context: str = None):
+        return self.predict(
+            text=text,
+            local_context=local_context,
+            document_context=document_context,
+        )
+
+
+class IdentifyReplacementPredict(dspy.Module):
+    """Identify exact phrase to replace for synonym application."""
+
+    def __init__(self):
+        super().__init__()
+        self.predict = dspy.Predict(IdentifyReplacementSignature)
+
+    def forward(self, original_word: str, chosen_synonym: str, context: str):
+        return self.predict(
+            original_word=original_word,
+            chosen_synonym=chosen_synonym,
+            context=context,
+        )

mrmd_ai/optimizers/__init__.py

@@ -0,0 +1 @@
+"""Optimizers for improving MRMD AI program performance."""