Updating tutorial to match new codegen (#85)

* Updating tutorial to match new codegen * Adding support for better doc strings * Updating examples to work with http and orm changes * fixing type errors
rmyers · Nov 17, 2024 · 3650ac6 · 3650ac6
1 parent e41ee69
commit 3650ac6
Show file tree

Hide file tree

Showing 28 changed files with 1,046 additions and 746 deletions.
diff --git a/cannula/contrib/orm.py b/cannula/contrib/orm.py
diff --git a/cannula/datasource/base.py b/cannula/datasource/base.py
@@ -47,6 +47,9 @@ def expected_fields(obj: typing.Any) -> set[str]:
     if dataclasses.is_dataclass(obj):
         return {field.name for field in dataclasses.fields(obj)}
 
+    if hasattr(obj, "model_fields"):
+        return {field for field in obj.model_fields}
+
     raise ValueError(
         "Invalid model for 'GraphModel' must be a dataclass or pydantic model"
     )