Merge pull request #12 from ryandansby/ryand/fix_key_values

Fix key values for updated dbt
rfdearborn · Dec 28, 2022 · 5b896f7 · 5b896f7
2 parents 23f2764 + 1a30b81
commit 5b896f7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dbt_docs_to_notion.py b/dbt_docs_to_notion.py
@@ -334,7 +334,7 @@ def main():
               {
                 "type": "text",
                 "text": {
-                  "content": data['raw_sql'][:2000]
+                  "content": data['raw_code'][:2000] if 'raw_code' in data else data['raw_sql'][:2000]
                 }
               }
             ],
@@ -362,7 +362,7 @@ def main():
               {
                 "type": "text",
                 "text": {
-                  "content": data['compiled_sql'][:2000]
+                  "content": data['compiled_code'][:2000] if 'compiled_code' in data else data['compiled_sql'][:2000]
                 }
               }
             ],