Fix vscode syntax

BoundaryML · Dec 9, 2024 · 35cbcc2 · 35cbcc2
1 parent 6e64ed0
commit 35cbcc2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/integ-tests/baml_src/test-files/functions/output/type-aliases.baml b/integ-tests/baml_src/test-files/functions/output/type-aliases.baml
@@ -37,7 +37,7 @@ function NestedAlias(c: Combination) -> Combination {
 
 // Test attribute merging.
 type Currency = int @check(gt_ten, {{ this > 10 }})
-type Amount = Currency @assert ({{ this > 0 }})
+type Amount = Currency @assert({{ this > 0 }})
 
 class MergeAttrs {
     amount Amount @description("In USD")

diff --git a/typescript/vscode-ext/packages/syntaxes/baml.tmLanguage.json b/typescript/vscode-ext/packages/syntaxes/baml.tmLanguage.json
@@ -731,9 +731,10 @@
         { "include": "#comment" },
         { 
           "begin": "(?<=\\=)\\s*",
-          "end": "(?=$|\\n)",
+          "end": "(?=//|$|\\n)",
           "patterns": [
-            { "include": "#type_definition" }
+            { "include": "#type_definition" },
+            { "include": "#block_attribute" }
           ]
         }
       ]