pattern class init

Author: guicho271828

mellea/stdlib/pattern.py

@@ -0,0 +1,59 @@
+
+from typing import Any
+from dataclasses import dataclass
+from string import Template
+from pydantic import BaseModel, create_model
+
+class Pattern:
+
+    def __init__(self, pattern:str, **types:type[Any]):
+        # e.g.,
+        self.pattern = Template(pattern)
+        self.types   = types
+
+
+    def defaulted_types(self) -> dict[str, type[Any]]:
+        # defaults the variable type to str.
+        # we do not do this defaulting in the constructor because
+        # composition may add in additional type information.
+        # in other words, type defaults are performed in the very last step.
+
+        return {
+            v : (
+                self.types[v]
+                if v in self.types
+                else str
+            )
+            for variable in self.pattern.get_identifiers()
+        }
+
+
+    def model(self) -> type[BaseModel]:
+
+        return create_model("TemplateModel", **self.defaulted_types())  # type: ignore
+
+
+    def format(self) -> str:
+
+        return (
+            "Fill in the blank in the following template, " +
+            "where the placeholders are denoted as $var for a variable named 'var'. " +
+            "Answer in a json schema shown after the template. " +
+            "\nTemplate:\n" +
+            self.pattern.template
+            "\nSchema:\n" +
+            self.model().dump_json_schema()
+        )
+
+
+if __name__ == "__main__":
+
+    from mellea import start_session, SimpleContext
+    m = start_session()
+    p = Pattern("the height of $mountain is $height meters.",
+                mountain=str,
+                height=int)
+
+    print(m.instruct(p.format(),
+                     format=p.model()))
+