fix: type upgrade for the assign stmt type annotation for the call ex…

…pression Signed-off-by: peefy <[email protected]>
kcl-lang · Oct 10, 2024 · 29694c6 · 29694c6
1 parent 3e84ea4
commit 29694c6
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 1 deletion.
diff --git a/kclvm/sema/src/resolver/schema.rs b/kclvm/sema/src/resolver/schema.rs
@@ -285,7 +285,12 @@ impl<'ctx> Resolver<'ctx> {
                 self.clear_config_expr_context(init_stack_depth as usize, false);
                 value_ty
             }
-            _ => self.expr(expr),
+            _ => {
+                self.ctx.config_expr_context.push(None);
+                let value_ty = self.expr(expr);
+                self.ctx.config_expr_context.pop();
+                value_ty
+            }
         };
         self.must_assignable_to(
             value_ty.clone(),

diff --git a/test/grammar/types/args/lambda_types_03/main.k b/test/grammar/types/args/lambda_types_03/main.k
@@ -0,0 +1,11 @@
+schema Cluster:
+    name: str
+    extra: int
+
+a: Cluster = { name = "abc", extra = 6 }
+
+enrich = lambda value {
+    [a]
+}
+
+result_typesafe: [Cluster] = enrich({name = "abc", extra = 6 })
diff --git a/test/grammar/types/args/lambda_types_03/stdout.golden b/test/grammar/types/args/lambda_types_03/stdout.golden
@@ -0,0 +1,6 @@
+a:
+  name: abc
+  extra: 6
+result_typesafe:
+- name: abc
+  extra: 6