PDL Python decorator (#1166)

* pdl generative

Signed-off-by: Mandana Vaziri <[email protected]>

Author: vazirim

examples/tutorial/sdk/lib.pdl

@@ -0,0 +1,3 @@
+defs:
+  hello: 
+    data: Hello there!

examples/tutorial/sdk/run_pdl.py

@@ -0,0 +1,34 @@
+from pdl.pdl import pdl
+
+
+@pdl
+def my_pdl_program(scope):
+    """
+    model: ${ model }
+    input: ${ input }
+    """
+    return
+
+
+@pdl
+def another(scope):
+    """
+    defs:
+        lib:
+            import: lib
+    text:
+    - ${ lib.hello }
+    """
+
+
+def main():
+    result1 = my_pdl_program(
+        scope={"model": "ollama_chat/granite3.2:2b", "input": "Hello\n"}
+    )
+    result2 = another(scope={})
+    print(result1)
+    print(result2)
+
+
+if __name__ == "__main__":
+    main()

src/pdl/pdl.py

@@ -160,6 +160,14 @@ def exec_file(
     return result
 
 
+def pdl(func):
+    def pdl_wrapper(scope):
+        result = exec_str(prog=func.__doc__, scope=scope)
+        return result
+
+    return pdl_wrapper
+
+
 def main():
     parser = argparse.ArgumentParser("")
     parser.add_argument(

tests/test_examples_run.yaml

@@ -35,6 +35,7 @@ skip:
   - examples/optimizer/eval_levenshtein.pdl
   - examples/requirements/email.pdl
   - examples/skeleton-of-thought/tips.pdl
+  - examples/tutorial/sdk/lib.pdl
 with_inputs:
   examples/tutorial/programs/chatbot.pdl:
     stdin: |