example

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

Author: vazirim

.github/workflows/build.yml

@@ -44,9 +44,9 @@ jobs:
     - name: pre-commit checks
       run: pre-commit run -a
     - name: run tests
-      run: py.test -v --capture=tee-sys --ignore=tests/test_examples_run.py --ignore=tests/test_schema.py tests
+      run: py.test -v --capture=tee-sys --ignore=tests/test_examples_run.py --ignore=tests/test_optimizer.py --ignore=tests/test_schema.py tests
       if: matrix.python-version == '3.11'
     - name: run tests
-      run: py.test -v --capture=tee-sys --ignore=tests/test_examples_run.py tests
+      run: py.test -v --capture=tee-sys --ignore=tests/test_examples_run.py --ignore=tests/test_optimizer.py tests
       if: matrix.python-version != '3.11'
 

examples/tutorial/test.pdl

@@ -0,0 +1,14 @@
+defs:
+  m1: 
+    content: Hello
+  m2: 
+    content: Bye
+  m3:
+    content: Hello again
+  
+  mycontext: ${ (m1 + m2) * m3 }
+
+text:
+- model: ollama_chat/granite3.2:2b
+  input: ${ mycontext }
+- ${ mycontext }

src/pdl/pdl_context.py

@@ -4,6 +4,17 @@
 
 from .pdl_lazy import PdlApply, PdlDict, PdlLazy, PdlList
 
+# def _default(self, obj):
+#    return getattr(obj.__class__, "to_json", _default.default)(obj) # pyright: ignore
+
+# _default.default = JSONEncoder().default  # pyright: ignore
+# JSONEncoder.default = _default   # pyright: ignore
+
+# TODO:
+# We could make only DependentContext implement Sequence, IndependentContext could implement Set instead. Review how getItem should be implemented.
+# Should single message (result of a block) be flattened in the interpreter?
+# Serialization
+
 
 class SerializeMode(StrEnum):
     LITELLM = "litellm"
@@ -27,6 +38,9 @@ def __len__(self):
     def __getitem__(self, index: int | slice):  # pyright: ignore
         return []
 
+    # def to_json(self):
+    #    return json.dumps(self.serialize(SerializeMode.LITELLM))
+
 
 class SingletonContext(PDLContext):
     message: PdlLazy[dict[str, Any]]