(no commit message)

README.md

@@ -1,2 +0,0 @@
-# doc-extract
-

auto_classes.json

@@ -0,0 +1,4 @@
+{
+  "AutoConfig": "modaic.PrecompiledConfig",
+  "AutoProgram": "program.DocExtract"
+}

config.json

@@ -0,0 +1,3 @@
+{
+  "model": null
+}

program.json

@@ -0,0 +1,32 @@
+{
+  "cot.predict": {
+    "traces": [],
+    "train": [],
+    "demos": [],
+    "signature": {
+      "instructions": "Classify a document snippet into one of the supported label buckets.",
+      "fields": [
+        {
+          "prefix": "Context:",
+          "description": "Document text or snippet to classify"
+        },
+        {
+          "prefix": "Reasoning: Let's think step by step in order to",
+          "description": "${reasoning}"
+        },
+        {
+          "prefix": "Label:",
+          "description": "Return exactly one of: legal, healthcare, personal, tax, or other"
+        }
+      ]
+    },
+    "lm": null
+  },
+  "metadata": {
+    "dependency_versions": {
+      "python": "3.11",
+      "dspy": "3.1.3",
+      "cloudpickle": "3.1"
+    }
+  }
+}

program.py

@@ -0,0 +1,28 @@
+# Here you can define your dspy.Module as a modaic.PrecompiledProgram
+from typing import Literal
+
+import dspy
+from modaic import PrecompiledProgram
+
+DocumentLabel = Literal["legal", "healthcare", "personal", "tax", "other"]
+
+
+class ExtractSignature(dspy.Signature):
+    """Classify a document snippet into one of the supported label buckets."""
+
+    context: str = dspy.InputField(desc="Document text or snippet to classify")
+    label: DocumentLabel = dspy.OutputField(
+        desc="Return exactly one of: legal, healthcare, personal, tax, or other"
+    )
+
+
+class DocExtract(PrecompiledProgram):
+    def __init__(self):
+        super().__init__()
+        self.cot = dspy.ChainOfThought(ExtractSignature)
+
+    def forward(self, context: str) -> dspy.Prediction:
+        return self.cot(context=context)
+
+
+DocExtract().push_to_hub("tyrin/doc-extract")

pyproject.toml

@@ -0,0 +1,7 @@
+[project]
+name = "doc-extract"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.11"
+dependencies = ["dspy>=3.1.3", "modaic>=0.29.1"]