Merge branch 'main' into baz/doaf-make-extract-output-optional

airbytehq · Jan 22, 2025 · 2364ce7 · 2364ce7
2 parents f8c45cc + d04ea8a
commit 2364ce7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/protocol-models/bin/generate-python-dataclasses.sh b/protocol-models/bin/generate-python-dataclasses.sh
@@ -27,6 +27,7 @@ for f in "$ROOT_DIR/$YAML_DIR"/*.yaml; do
   echo "from .$filename_wo_ext import *" >> "$ROOT_DIR/$OUTPUT_DIR"/__init__.py
 
   datamodel-codegen \
+    --input-file-type jsonschema \
     --input "$ROOT_DIR/$YAML_DIR/$filename_wo_ext.yaml" \
     --output "$ROOT_DIR/$OUTPUT_DIR/$filename_wo_ext.py" \
     --output-model-type dataclasses.dataclass \

diff --git a/protocol-models/bin/generate-python-pydantic-v2.sh b/protocol-models/bin/generate-python-pydantic-v2.sh
@@ -27,6 +27,7 @@ for f in "$ROOT_DIR/$YAML_DIR"/*.yaml; do
   echo "from .$filename_wo_ext import *" >> "$ROOT_DIR/$OUTPUT_DIR"/__init__.py
 
   datamodel-codegen \
+    --input-file-type jsonschema \
     --input "$ROOT_DIR/$YAML_DIR/$filename_wo_ext.yaml" \
     --output "$ROOT_DIR/$OUTPUT_DIR/$filename_wo_ext.py" \
     --output-model-type pydantic_v2.BaseModel \