snowflake-code-unit-registry 0.8.41__tar.gz → 0.8.49__tar.gz

{snowflake_code_unit_registry-0.8.41 → snowflake_code_unit_registry-0.8.49}/Cargo.lock

@@ -758,12 +758,6 @@ version = "1.0.7"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "3f9eec918d3f24069decb9af1554cad7c880e2da24a9afd88aca000531ab82c1"
 
-[[package]]
-name = "foldhash"
-version = "0.1.5"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "d9c4f5dac5e15c24eb999c26181a6ca40b39fe946cbe4c263c7209467bc83af2"
-
 [[package]]
 name = "foldhash"
 version = "0.2.0"
@@ -930,15 +924,13 @@ dependencies = [
 
 [[package]]
 name = "getrandom"
-version = "0.4.2"
+version = "0.4.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "0de51e6874e94e7bf76d726fc5d13ba782deca734ff60d5bb2fb2607c7406555"
+checksum = "300e883d756b2e4ec94e02791f39b04b522276138852cfc41d9fb7e904106099"
 dependencies = [
  "cfg-if",
  "libc",
  "r-efi 6.0.0",
- "wasip2",
- "wasip3",
 ]
 
 [[package]]
@@ -1007,15 +999,6 @@ dependencies = [
  "zerocopy",
 ]
 
-[[package]]
-name = "hashbrown"
-version = "0.15.5"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "9229cfe53dfd69f0609a49f65461bd93001ea1ef889cd5529dd176593f5338a1"
-dependencies = [
- "foldhash 0.1.5",
-]
-
 [[package]]
 name = "hashbrown"
 version = "0.16.1"
@@ -1024,7 +1007,7 @@ checksum = "841d1cc9bed7f9236f321df977030373f4a4163ae1a7dbfe1a51a2c1a51d9100"
 dependencies = [
  "allocator-api2",
  "equivalent",
- "foldhash 0.2.0",
+ "foldhash",
 ]
 
 [[package]]
@@ -1258,12 +1241,6 @@ dependencies = [
  "zerovec",
 ]
 
-[[package]]
-name = "id-arena"
-version = "2.3.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "3d3067d79b975e8844ca9eb072e16b31c3c1c36928edf9c6789548c524d0d954"
-
 [[package]]
 name = "ident_case"
 version = "1.0.1"
@@ -1335,8 +1312,6 @@ checksum = "d466e9454f08e4a911e14806c24e16fba1b4c121d1ea474396f396069cf949d9"
 dependencies = [
  "equivalent",
  "hashbrown 0.17.1",
- "serde",
- "serde_core",
 ]
 
 [[package]]
@@ -1550,12 +1525,6 @@ version = "1.5.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "bbd2bcb4c963f2ddae06a2efc7e9f3591312473c50c6685e1f298068316e66fe"
 
-[[package]]
-name = "leb128fmt"
-version = "0.1.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "09edd9e8b54e49e587e4f6295a7d29c3ea94d469cb40ab8ca70b288248a81db2"
-
 [[package]]
 name = "libc"
 version = "0.2.186"
@@ -2622,7 +2591,7 @@ dependencies = [
 
 [[package]]
 name = "scai-error-derive"
-version = "0.8.41"
+version = "0.8.49"
 dependencies = [
  "err_code",
  "heck",
@@ -2636,7 +2605,7 @@ dependencies = [
 
 [[package]]
 name = "scai-log"
-version = "0.8.41"
+version = "0.8.49"
 dependencies = [
  "chrono",
  "serde",
@@ -2645,7 +2614,7 @@ dependencies = [
 
 [[package]]
 name = "scai-state-core"
-version = "0.8.41"
+version = "0.8.49"
 dependencies = [
  "chrono",
  "err_code",
@@ -2671,7 +2640,7 @@ dependencies = [
 
 [[package]]
 name = "scai-state-csharp"
-version = "0.8.41"
+version = "0.8.49"
 dependencies = [
  "interoptopus",
  "interoptopus_backend_csharp",
@@ -2683,7 +2652,7 @@ dependencies = [
 
 [[package]]
 name = "scai-state-node"
-version = "0.8.41"
+version = "0.8.49"
 dependencies = [
  "napi",
  "napi-build",
@@ -2696,7 +2665,7 @@ dependencies = [
 
 [[package]]
 name = "scai-state-perf"
-version = "0.8.41"
+version = "0.8.49"
 dependencies = [
  "chrono",
  "criterion",
@@ -2712,7 +2681,7 @@ dependencies = [
 
 [[package]]
 name = "scai-state-python"
-version = "0.8.41"
+version = "0.8.49"
 dependencies = [
  "pyo3",
  "pythonize",
@@ -2963,9 +2932,9 @@ dependencies = [
 
 [[package]]
 name = "sqlparser"
-version = "0.61.0"
+version = "0.62.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "dbf5ea8d4d7c808e1af1cbabebca9a2abe603bcefc22294c5b95018d53200cb7"
+checksum = "13c6d1b651dc4edf07eead2a0c6c78016ce971bc2c10da5266861b13f25e7cec"
 dependencies = [
  "log",
  "recursive",
@@ -3033,9 +3002,9 @@ dependencies = [
 
 [[package]]
 name = "syn"
-version = "2.0.117"
+version = "2.0.118"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "e665b8803e7b1d2a727f4023456bbbbe74da67099c585258af0ad9c5013b9b99"
+checksum = "1b9ae57f904213ebb649ce6895b8a66c66f0203b9319718f69a5612a065b1422"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -3075,7 +3044,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "32497e9a4c7b38532efcdebeef879707aa9f794296a4f0244f6f69e9bc8574bd"
 dependencies = [
  "fastrand",
- "getrandom 0.4.2",
+ "getrandom 0.4.3",
  "once_cell",
  "rustix",
  "windows-sys 0.61.2",
@@ -3360,12 +3329,6 @@ version = "1.13.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "c6f5d3c3b1bf09027a88a6bc961fc00497d651009560b5463668dc81b0fa87a8"
 
-[[package]]
-name = "unicode-xid"
-version = "0.2.6"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "ebc1c04c71510c7f702b52b7c350734c9ff1295c464a03335b00bb84fc54f853"
-
 [[package]]
 name = "untrusted"
 version = "0.9.0"
@@ -3396,7 +3359,7 @@ version = "1.23.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "144d6b123cef80b301b8f72a9e2ca4370ddec21950d0a103dd22c437006d2db7"
 dependencies = [
- "getrandom 0.4.2",
+ "getrandom 0.4.3",
  "js-sys",
  "wasm-bindgen",
 ]
@@ -3454,16 +3417,7 @@ version = "1.0.4+wasi-0.2.12"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "b67efb37e106e55ce722a510d6b5f9c17f083e5fc79afc2badeb12cc313d9487"
 dependencies = [
- "wit-bindgen 0.57.1",
-]
-
-[[package]]
-name = "wasip3"
-version = "0.4.0+wasi-0.3.0-rc-2026-01-06"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "5428f8bf88ea5ddc08faddef2ac4a67e390b88186c703ce6dbd955e1c145aca5"
-dependencies = [
- "wit-bindgen 0.51.0",
+ "wit-bindgen",
 ]
 
 [[package]]
@@ -3521,40 +3475,6 @@ dependencies = [
  "unicode-ident",
 ]
 
-[[package]]
-name = "wasm-encoder"
-version = "0.244.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "990065f2fe63003fe337b932cfb5e3b80e0b4d0f5ff650e6985b1048f62c8319"
-dependencies = [
- "leb128fmt",
- "wasmparser",
-]
-
-[[package]]
-name = "wasm-metadata"
-version = "0.244.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "bb0e353e6a2fbdc176932bbaab493762eb1255a7900fe0fea1a2f96c296cc909"
-dependencies = [
- "anyhow",
- "indexmap",
- "wasm-encoder",
- "wasmparser",
-]
-
-[[package]]
-name = "wasmparser"
-version = "0.244.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "47b807c72e1bac69382b3a6fb3dbe8ea4c0ed87ff5629b8685ae6b9a611028fe"
-dependencies = [
- "bitflags 2.13.0",
- "hashbrown 0.15.5",
- "indexmap",
- "semver",
-]
-
 [[package]]
 name = "web-sys"
 version = "0.3.102"
@@ -3567,9 +3487,9 @@ dependencies = [
 
 [[package]]
 name = "webpki-root-certs"
-version = "1.0.7"
+version = "1.0.8"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "f31141ce3fc3e300ae89b78c0dd67f9708061d1d2eda54b8209346fd6be9a92c"
+checksum = "0d46a5a140e6f7afeccd8eae97eff335163939eac8b929834875168b29b3d267"
 dependencies = [
  "rustls-pki-types",
 ]
@@ -3746,100 +3666,12 @@ version = "0.52.6"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "589f6da84c646204747d1270a2a5661ea66ed1cced2631d546fdfb155959f9ec"
 
-[[package]]
-name = "wit-bindgen"
-version = "0.51.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "d7249219f66ced02969388cf2bb044a09756a083d0fab1e566056b04d9fbcaa5"
-dependencies = [
- "wit-bindgen-rust-macro",
-]
-
 [[package]]
 name = "wit-bindgen"
 version = "0.57.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "1ebf944e87a7c253233ad6766e082e3cd714b5d03812acc24c318f549614536e"
 
-[[package]]
-name = "wit-bindgen-core"
-version = "0.51.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "ea61de684c3ea68cb082b7a88508a8b27fcc8b797d738bfc99a82facf1d752dc"
-dependencies = [
- "anyhow",
- "heck",
- "wit-parser",
-]
-
-[[package]]
-name = "wit-bindgen-rust"
-version = "0.51.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "b7c566e0f4b284dd6561c786d9cb0142da491f46a9fbed79ea69cdad5db17f21"
-dependencies = [
- "anyhow",
- "heck",
- "indexmap",
- "prettyplease",
- "syn",
- "wasm-metadata",
- "wit-bindgen-core",
- "wit-component",
-]
-
-[[package]]
-name = "wit-bindgen-rust-macro"
-version = "0.51.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "0c0f9bfd77e6a48eccf51359e3ae77140a7f50b1e2ebfe62422d8afdaffab17a"
-dependencies = [
- "anyhow",
- "prettyplease",
- "proc-macro2",
- "quote",
- "syn",
- "wit-bindgen-core",
- "wit-bindgen-rust",
-]
-
-[[package]]
-name = "wit-component"
-version = "0.244.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "9d66ea20e9553b30172b5e831994e35fbde2d165325bec84fc43dbf6f4eb9cb2"
-dependencies = [
- "anyhow",
- "bitflags 2.13.0",
- "indexmap",
- "log",
- "serde",
- "serde_derive",
- "serde_json",
- "wasm-encoder",
- "wasm-metadata",
- "wasmparser",
- "wit-parser",
-]
-
-[[package]]
-name = "wit-parser"
-version = "0.244.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "ecc8ac4bc1dc3381b7f59c34f00b67e18f910c2c0f50015669dde7def656a736"
-dependencies = [
- "anyhow",
- "id-arena",
- "indexmap",
- "log",
- "semver",
- "serde",
- "serde_derive",
- "serde_json",
- "unicode-xid",
- "wasmparser",
-]
-
 [[package]]
 name = "writeable"
 version = "0.6.3"

{snowflake_code_unit_registry-0.8.41 → snowflake_code_unit_registry-0.8.49}/Cargo.toml

@@ -3,7 +3,7 @@ resolver = "2"
 members = ["crates/scai-error-derive", "crates/scai-log", "crates/scai-state-core", "crates/scai-state-python"]
 
 [workspace.package]
-version = "0.8.41"
+version = "0.8.49"
 edition = "2021"
 license = "SEE LICENSE IN LICENSE"
 repository = "https://github.com/snowflake-eng/scai-state"
@@ -15,7 +15,7 @@ snafu = "0.9"
 err_code = "0.1"
 uuid = { version = "1.0", features = ["v4"] }
 fs2 = "0.4"
-sqlparser = "0.61"
+sqlparser = "0.62"
 criterion = { version = "0.5", features = ["html_reports"] }
 
 [profile.profiling]

{snowflake_code_unit_registry-0.8.41 → snowflake_code_unit_registry-0.8.49}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-code-unit-registry
-Version: 0.8.41
+Version: 0.8.49
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: Apache Software License

{snowflake_code_unit_registry-0.8.41 → snowflake_code_unit_registry-0.8.49}/crates/scai-state-core/src/dependency_view.rs

@@ -51,13 +51,8 @@ pub(crate) fn depends_on_with_origin(
     root.chain(parts)
 }
 
-/// Convenience: every `dependsOn` edge, origin discarded.
-pub(crate) fn depends_on(unit: &CodeUnit) -> impl Iterator<Item = &Dependency> {
-    depends_on_with_origin(unit).map(|(_, dep)| dep)
-}
-
 /// Mutable iterator over every `dependsOn` edge. Mirrors the order of
-/// [`depends_on`] exactly.
+/// [`depends_on_with_origin`] exactly (with origin discarded).
 pub(crate) fn depends_on_mut(unit: &mut CodeUnit) -> impl Iterator<Item = &mut Dependency> {
     dep_blocks_mut(unit).flat_map(|d| d.depends_on.iter_mut())
 }

{snowflake_code_unit_registry-0.8.41 → snowflake_code_unit_registry-0.8.49}/crates/scai-state-core/src/generated/schema_constraints.rs

@@ -103,13 +103,14 @@ pub struct RequiredWhen {
     pub when: &'static [(&'static str, &'static [&'static str])],
 }
 
-pub const REQUIRED_WHEN_CONSTRAINTS: &[RequiredWhen] = &[
-    RequiredWhen {
-        parent_pointer: "/properties/target",
-        leaf_name: "format",
-        when: &[("codeStatus.conversion.status", &["completed"]), ("kind", &["script"])],
-    },
-];
+pub const REQUIRED_WHEN_CONSTRAINTS: &[RequiredWhen] = &[RequiredWhen {
+    parent_pointer: "/properties/target",
+    leaf_name: "format",
+    when: &[
+        ("codeStatus.conversion.status", &["completed"]),
+        ("kind", &["script"]),
+    ],
+}];
 
 /// One `x-allowed-fields` constraint at a `$ref`'d use site.
 ///
@@ -122,9 +123,7 @@ pub struct AllowedFieldConstraint {
     pub allowed_fields: &'static [&'static str],
 }
 
-pub const ALLOWED_FIELD_CONSTRAINTS: &[AllowedFieldConstraint] = &[
-    AllowedFieldConstraint {
-        pointer: "/properties/parts/items/properties/cloudStatus",
-        allowed_fields: &["testing", "deployment"],
-    },
-];
+pub const ALLOWED_FIELD_CONSTRAINTS: &[AllowedFieldConstraint] = &[AllowedFieldConstraint {
+    pointer: "/properties/parts/items/properties/cloudStatus",
+    allowed_fields: &["testing", "deployment"],
+}];

{snowflake_code_unit_registry-0.8.41 → snowflake_code_unit_registry-0.8.49}/crates/scai-state-core/src/registry/graph.rs

@@ -11,6 +11,8 @@
 //!
 //! `predecessors(A)` yields B — follows `dependsOn` (what must come before A)
 //! `successors(B)`   yields A — follows `requiredBy` (what comes after B)
+//! Root-level `dependsOn` self-references are ignored for deployment ordering
+//! (topological rank and cycle detection) but remain in stored `dependsOn`.
 
 use std::collections::{HashMap, HashSet, VecDeque};
 
@@ -98,15 +100,29 @@ impl<'a> RegistryGraph<'a> {
 
 // ── Traversal utilities ─────────────────────────────────────────────────
 
+/// Indices of this unit's predecessors (units it depends on).
+/// A depends on B means B is A's predecessor.
+fn is_root_self_dep(
+    origin: dependency_view::Origin<'_>,
+    unit_id: Option<&str>,
+    dep_id: Option<&str>,
+) -> bool {
+    matches!(origin, dependency_view::Origin::Root) && unit_id.is_some() && dep_id == unit_id
+}
+
 /// Indices of this unit's predecessors (units it depends on).
 /// A depends on B means B is A's predecessor.
 fn predecessors<'a>(
     unit: &'a CodeUnit,
     id_to_index: &'a HashMap<String, usize>,
 ) -> impl Iterator<Item = usize> + 'a {
-    dependency_view::depends_on(unit)
-        .filter_map(|dep| dep.id.as_deref())
-        .filter_map(|id| id_to_index.get(id).copied())
+    dependency_view::depends_on_with_origin(unit).filter_map(move |(origin, dep)| {
+        let id = dep.id.as_deref()?;
+        if is_root_self_dep(origin, unit.id.as_deref(), Some(id)) {
+            return None;
+        }
+        id_to_index.get(id).copied()
+    })
 }
 
 /// Indices of this unit's successors (units that depend on it),
@@ -263,14 +279,16 @@ impl RefreshState {
         let mut resolved_predecessors = Vec::new();
         let mut saw_any = false;
 
-        for dep in dependency_view::depends_on(unit) {
+        for (origin, dep) in dependency_view::depends_on_with_origin(unit) {
             saw_any = true;
             let missing = match dep.id.as_deref() {
                 None => true,
                 Some(id) => match id_to_index.get(id) {
                     None => true,
                     Some(&dep_idx) => {
-                        resolved_predecessors.push(dep_idx);
+                        if !is_root_self_dep(origin, unit.id.as_deref(), dep.id.as_deref()) {
+                            resolved_predecessors.push(dep_idx);
+                        }
                         units[dep_idx].is_missing
                     }
                 },
@@ -595,6 +613,22 @@ mod tests {
         let mut units = vec![unit_with_dependencies("a", &[Some("a")])];
         let idx = id_index(&units);
 
+        RegistryGraph::new(&mut units, &idx).refresh(true).unwrap();
+
+        assert_eq!(rank_of(&units[0]), Some(0));
+        assert_eq!(required_by_of(&units[0]), Vec::<String>::new());
+        assert_eq!(predecessor_is_missing_of(&units[0]), vec![Some(false)]);
+        assert!(!transitive_missing_of(&units[0]));
+    }
+
+    #[test]
+    fn refresh_self_loop_does_not_mask_real_cycle() {
+        let mut units = vec![
+            unit_with_dependencies("a", &[Some("a"), Some("b")]),
+            unit_with_dependencies("b", &[Some("a")]),
+        ];
+        let idx = id_index(&units);
+
         let err = RegistryGraph::new(&mut units, &idx)
             .refresh(true)
             .unwrap_err();
@@ -603,7 +637,9 @@ mod tests {
         let mut g = RegistryGraph::new(&mut units, &idx);
         g.refresh(false).unwrap();
         assert_eq!(rank_of(&units[0]), Some(-1));
+        assert_eq!(rank_of(&units[1]), Some(-1));
         assert!(transitive_missing_of(&units[0]));
+        assert!(transitive_missing_of(&units[1]));
     }
 
     #[test]

{snowflake_code_unit_registry-0.8.41 → snowflake_code_unit_registry-0.8.49}/crates/scai-state-core/src/registry/tests.rs

@@ -1445,6 +1445,35 @@ fn create_auto_refresh_best_effort_on_cycle() {
     );
 }
 
+#[test]
+fn create_auto_refresh_self_reference() {
+    let dir = temp_dir();
+    let registry = CodeUnitRegistry::init(dir.path()).unwrap();
+
+    let id = registry
+        .create(&mut make_unit_with_deps("a", &["a"]), None)
+        .unwrap();
+    assert_eq!(id, "a");
+
+    let a = read_unit(&registry, "a");
+    assert_eq!(a.planning.unwrap().topological_rank, Some(0));
+    assert_eq!(
+        a.dependencies.as_ref().unwrap().depends_on[0].is_missing,
+        Some(false)
+    );
+    assert!(
+        !a.dependencies
+            .as_ref()
+            .unwrap()
+            .has_transitive_missing_dependencies
+    );
+
+    registry.refresh_dependencies().unwrap();
+
+    let a_after = read_unit(&registry, "a");
+    assert_eq!(a_after.planning.unwrap().topological_rank, Some(0));
+}
+
 #[test]
 fn create_with_corrupt_sibling_succeeds_without_hooks() {
     let dir = temp_dir();

{snowflake_code_unit_registry-0.8.41 → snowflake_code_unit_registry-0.8.49}/crates/scai-state-python/tests/test_refresh.py

@@ -81,6 +81,23 @@ def test_create_best_effort_on_cycle(registry_dir: str):
     assert y.dependencies.hasTransitiveMissingDependencies is True
 
 
+def test_create_self_reference_is_not_cycle(registry_dir: str):
+    registry = CodeUnitRegistry.init(registry_dir)
+
+    unit_id = registry.create(_unit("a", ["a"]))
+    assert unit_id == "a"
+
+    a = registry.get_by_id("a")
+    assert a.planning.topologicalRank == 0
+    assert a.dependencies.dependsOn[0].isMissing is False
+    assert a.dependencies.hasTransitiveMissingDependencies is False
+
+    registry.refresh_dependencies()
+
+    a_after = registry.get_by_id("a")
+    assert a_after.planning.topologicalRank == 0
+
+
 def test_delete_auto_refreshes_missing_flags(registry_dir: str):
     registry = CodeUnitRegistry.init(registry_dir)
 

{snowflake_code_unit_registry-0.8.41 → snowflake_code_unit_registry-0.8.49}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "snowflake-code-unit-registry"
-version = "0.8.41"
+version = "0.8.49"
 description = "SCAI state management library for database migrations"
 readme = "README.md"
 requires-python = ">=3.9"

{snowflake_code_unit_registry-0.8.41 → snowflake_code_unit_registry-0.8.49}/python/snowflake_code_unit_registry/__init__.py

@@ -46,7 +46,7 @@ from snowflake_code_unit_registry.types import CodeUnit
 if TYPE_CHECKING:
     from collections.abc import Sequence
 
-__version__ = "0.8.41"
+__version__ = "0.8.49"
 __all__ = [
     "CodeUnitRegistry",
     "ScaiLogger",

{snowflake_code_unit_registry-0.8.41 → snowflake_code_unit_registry-0.8.49}/python/snowflake_code_unit_registry/types.py

@@ -4,7 +4,7 @@
 from __future__ import annotations
 
 from enum import Enum
-from typing import Any, Literal
+from typing import Any, Dict, Literal
 
 from pydantic import AwareDatetime, BaseModel, ConfigDict, Field, RootModel, constr
 
@@ -406,6 +406,10 @@ class CloudStatus(BaseModel):
     )
 
 
+CloudStatus.__annotations__['__pydantic_extra__'] = Dict[str, Any]
+CloudStatus.model_rebuild(force=True)
+
+
 class PartTarget(BaseModel):
     """
     Target object metadata for a single ETL part
@@ -744,6 +748,10 @@ class CodeStatus(BaseModel):
     )
 
 
+CodeStatus.__annotations__['__pydantic_extra__'] = Dict[str, Any]
+CodeStatus.model_rebuild(force=True)
+
+
 class Issues(RootModel[list[Issue]]):
     root: list[Issue] = Field(
         ..., description='Derived issues from converted code scanning', title='Issues'