fix ci

Signed-off-by: shruti2522 <[email protected]>
kcl-lang · Jul 8, 2024 · 51ca5ff · 51ca5ff
1 parent 8244340
commit 51ca5ff
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 5 deletions.
diff --git a/kclvm/sema/src/advanced_resolver/node.rs b/kclvm/sema/src/advanced_resolver/node.rs
@@ -1292,17 +1292,29 @@ impl<'ctx> AdvancedResolver<'ctx> {
         } else {
             for (arg, param) in args.iter().zip(func_ty.params.iter()) {
                 self.expr(arg)?;
-
                 let symbol_data = self.gs.get_symbols_mut();
+
                 if let Some(arg_ref) = symbol_data
                     .symbols_info
                     .node_symbol_map
                     .get(&self.ctx.get_node_key(&arg.id))
                 {
-                    if let Some(value) = symbol_data.values.get_mut(arg_ref.get_id()) {
-                        if with_hint {
-                            value.hint = Some(SymbolHint::VarHint(param.name.clone()));
+                    match arg_ref.get_kind() {
+                        crate::core::symbol::SymbolKind::Value => {
+                            if let Some(value) = symbol_data.values.get_mut(arg_ref.get_id()) {
+                                if with_hint {
+                                    value.hint = Some(SymbolHint::VarHint(param.name.clone()));
+                                }
+                            }
+                        }
+                        crate::core::symbol::SymbolKind::Expression => {
+                            if let Some(expr) = symbol_data.exprs.get_mut(arg_ref.get_id()) {
+                                if with_hint {
+                                    expr.hint = Some(SymbolHint::VarHint(param.name.clone()));
+                                }
+                            }
                         }
+                        _ => {}
                     }
                 }
             }

diff --git a/kclvm/sema/src/core/symbol.rs b/kclvm/sema/src/core/symbol.rs
@@ -1959,6 +1959,7 @@ pub struct ExpressionSymbol {
     pub(crate) name: String,
 
     pub(crate) sema_info: SymbolSemanticInfo,
+    pub(crate) hint: Option<SymbolHint>,
 }
 
 impl Symbol for ExpressionSymbol {
@@ -2072,6 +2073,7 @@ impl ExpressionSymbol {
             end,
             sema_info: SymbolSemanticInfo::default(),
             owner,
+            hint: None,
         }
     }
 }

diff --git a/kclvm/tools/src/LSP/src/test_data/inlay_hints/assign_stmt_type_hint/assign_stmt_type_hint.k b/kclvm/tools/src/LSP/src/test_data/inlay_hints/assign_stmt_type_hint/assign_stmt_type_hint.k
@@ -25,4 +25,4 @@ func = lambda x, y, z{
       x * y + z
 }
 
-z = func(a, 1, z = 2)
+value = func(a, 1, z = 2)