mdbt 0.5.1__py3-none-any.whl → 0.5.2__py3-none-any.whl

mdbt/build_dbt_docs_ai.py

@@ -46,19 +46,23 @@ class BuildDBTDocs(AiCore):
                 system_instructions = Prompts().dbt_docs_lte_l2_prompt
 
             if sys_context:
-                system_instructions += f"\nContext about system docs are generated for: \n{sys_context}\n"
+                system_instructions += (
+                    f"\nContext about system docs are generated for: \n{sys_context}\n"
+                )
 
             sample_data = self._get_sample_data_from_snowflake([model_name])
 
             system_instructions = system_instructions + sample_data[model_name]
 
             # Get existing column descriptions from neighboring YML files
-            existing_descriptions = self.get_existing_column_descriptions(original_file_path)
+            existing_descriptions = self.get_existing_column_descriptions(
+                original_file_path
+            )
             if existing_descriptions:
-                system_instructions += f'Here are some existing column descriptions from neighboring models. These may or may not be relevant to the current model. Assess, and use as a source if it seems to match the sample data: \n{existing_descriptions}'
+                system_instructions += f"Here are some existing column descriptions from neighboring models. These may or may not be relevant to the current model. Assess, and use as a source if it seems to match the sample data: \n{existing_descriptions}"
 
             # Might bring this back in the future.
-            extra_info = ''
+            extra_info = ""
 
             if mode == 1:
                 # Build new documentation
@@ -87,12 +91,12 @@ class BuildDBTDocs(AiCore):
             output = assistant_responses[0]
             # Check for ``` at end of output (str) and remove
             # Remove trailing markdown code fences if present
-            lines = output.split('\n')
+            lines = output.split("\n")
             new_lines = []
             for line in lines:
-                if '```' not in line:
+                if "```" not in line:
                     new_lines.append(line)
-            output = '\n'.join(new_lines)
+            output = "\n".join(new_lines)
             if not is_new:
                 clip_or_file = input(
                     f"1. to copy to clipboard\n2, to write to file ({yml_file_path}\n:"
@@ -111,7 +115,9 @@ class BuildDBTDocs(AiCore):
                     file.write(output)
                 if not self.is_file_committed(yml_file_path):
                     if not is_new:
-                        commit_file = input("Press 1 to add to git, any other key to byapss: ")
+                        commit_file = input(
+                            "Press 1 to add to git, any other key to byapss: "
+                        )
                     else:
                         commit_file = "1"
 
@@ -120,9 +126,9 @@ class BuildDBTDocs(AiCore):
 
     @staticmethod
     def backup_existing_yml_file(yml_file_path):
-        with open(yml_file_path, 'r') as file:
+        with open(yml_file_path, "r") as file:
             yml_content = file.read()
-        with open(yml_file_path + '.bak', 'w') as file:
+        with open(yml_file_path + ".bak", "w") as file:
             file.write(yml_content)
 
     def get_existing_column_descriptions(self, model_file_path: str) -> str:
@@ -138,14 +144,14 @@ class BuildDBTDocs(AiCore):
             in neighboring YML files.
         """
         folder_path = Path(model_file_path).parent
-        current_yml_path = model_file_path.replace('.sql', '.yml')
+        current_yml_path = model_file_path.replace(".sql", ".yml")
         column_descriptions = {}
 
         yaml = YAML()
         yaml.preserve_quotes = True
 
         # Find all .yml files in the same folder
-        yml_files = list(folder_path.glob('*.yml'))
+        yml_files = list(folder_path.glob("*.yml"))
 
         for yml_file in yml_files:
             # Skip the current model's yml file if it exists
@@ -153,60 +159,64 @@ class BuildDBTDocs(AiCore):
                 continue
 
             try:
-                with open(yml_file, 'r') as f:
+                with open(yml_file, "r") as f:
                     yml_content = yaml.load(f)
 
                 if not yml_content:
                     continue
 
                 # Extract column descriptions from models section
-                models = yml_content.get('models', [])
+                models = yml_content.get("models", [])
                 for model in models:
-                    columns = model.get('columns', [])
+                    columns = model.get("columns", [])
                     for column in columns:
-                        col_name = column.get('name')
-                        col_desc = column.get('description')
+                        col_name = column.get("name")
+                        col_desc = column.get("description")
                         if col_name and col_desc:
                             # Only store if not already seen, or prefer longer descriptions
-                            if col_name not in column_descriptions or \
-                               len(str(col_desc)) > len(str(column_descriptions[col_name])):
+                            if col_name not in column_descriptions or len(
+                                str(col_desc)
+                            ) > len(str(column_descriptions[col_name])):
                                 column_descriptions[col_name] = str(col_desc).strip()
 
             except Exception as e:
-                print(f'Warning: Could not parse {yml_file}: {e}')
+                print(f"Warning: Could not parse {yml_file}: {e}")
                 continue
 
         if not column_descriptions:
-            return ''
+            return ""
 
         # Filter out ID columns to reduce noise (columns ending in _id or named id)
         non_id_descriptions = {
-            col_name: desc for col_name, desc in column_descriptions.items()
-            if not col_name.lower().endswith('_id') and col_name.lower() != 'id'
+            col_name: desc
+            for col_name, desc in column_descriptions.items()
+            if not col_name.lower().endswith("_id") and col_name.lower() != "id"
         }
 
         # Limit to first 300 unique columns to avoid oversized prompts
         max_columns = 300
         limited_descriptions = dict(
-            list(sorted(non_id_descriptions.items()))[:max_columns]
+            sorted(non_id_descriptions.items())[:max_columns]
         )
 
         if not limited_descriptions:
-            return ''
+            return ""
 
         # Format the output
         lines = [
-            '\n--- Existing Column Descriptions from Neighboring Models ---',
-            'Use these existing descriptions for consistency when the same '
-            'or similar columns appear in the new model:\n'
+            "\n--- Existing Column Descriptions from Neighboring Models ---",
+            "Use these existing descriptions for consistency when the same "
+            "or similar columns appear in the new model:\n",
         ]
         for col_name, description in limited_descriptions.items():
             # Truncate very long descriptions for context
-            desc_preview = description[:200] + '...' if len(description) > 200 else description
-            lines.append(f'  - {col_name}: {desc_preview}')
+            desc_preview = (
+                description[:200] + "..." if len(description) > 200 else description
+            )
+            lines.append(f"  - {col_name}: {desc_preview}")
 
-        lines.append('--- End of Existing Column Descriptions ---\n')
-        return '\n'.join(lines)
+        lines.append("--- End of Existing Column Descriptions ---\n")
+        return "\n".join(lines)
 
     def build_user_msg_mode_1(self, _sql_file_path: str, extra_info: str) -> str:
         self.read_file(_sql_file_path)

mdbt/cmdline.py

@@ -5,7 +5,6 @@ import click
 from mdbt.build_dbt_docs_ai import BuildDBTDocs
 from mdbt.build_unit_test_data_ai import BuildUnitTestDataAI
 from mdbt.expectations_output_builder import ExpectationsOutputBuilder
-from mdbt.lightdash import Lightdash
 from mdbt.main import MDBT
 from mdbt.precommit_format import PrecommitFormat
 from mdbt.recce import Recce
@@ -243,7 +242,7 @@ def gbuild(ctx, main, full_refresh, threads):
     "--is_new",
     "-n",
     is_flag=True,
-    help="Passing this flag will bypass the questions such as 'is this a new model,' and 'add to git"
+    help="Passing this flag will bypass the questions such as 'is this a new model,' and 'add to git",
 )
 @click.pass_context
 def build_docs(ctx, select, sys_context, is_new):
@@ -251,6 +250,7 @@ def build_docs(ctx, select, sys_context, is_new):
     dbt_docs = BuildDBTDocs()
     dbt_docs.main(select, sys_context, is_new)
 
+
 @mdbt.command()
 @click.option(
     "--select",
@@ -266,53 +266,6 @@ def build_unit(ctx, select):
     build_unit_test_data.main(select)
 
 
-@mdbt.command()
-@click.option(
-    "--select",
-    "-s",
-    type=str,
-    help="Name of the model to start a lightdash preview for. If not provided, all models will be previewed.",
-)
-@click.option(
-    "--name",
-    "-n",
-    type=str,
-    help="Name of the lightdash preview. If no name given, the preview will take the name of the current branch.",
-)
-@click.option(
-    "--l43",
-    is_flag=True,
-    help="Include L3 and L4 models in the preview. Default is False.",
-)
-@click.pass_context
-def ld_preview(ctx, select, name, l43):
-    """Start a lightdash preview for a model."""
-    preview_name = name
-    Lightdash().lightdash_start_preview(ctx, select, preview_name, l43)
-
-
-@mdbt.command()
-@click.option("--select", "-s", type=str, help="Names of the model(s) to clean.")
-@click.option(
-    "--split-names", is_flag=True, help="Split names like isupdated into is_updated."
-)
-@click.option(
-    "--remove-airbyte",
-    is_flag=True,
-    help="Whether to remove Airbyte specific lines. Default is True.",
-)
-@click.option(
-    "--overwrite",
-    is_flag=True,
-    help="Will overwrite the files. If not set, files will be saved to a folder.",
-)
-@click.pass_context
-def clean_stg(select, split_names, remove_airbyte, overwrite):
-    """Designed to clean files in the L1_stg folders only"""
-    sql_model_cleaner = SQLModelCleaner()
-    sql_model_cleaner.main(select, split_names, remove_airbyte, overwrite)
-
-
 @mdbt.command()
 @click.option("--select", "-s", type=str, help="Name of model to sort YML columns for.")
 @click.option("--all-files", is_flag=True, help="Sort all YML files in the project.")
@@ -321,11 +274,13 @@ def sort_yaml(select, all_files, overwrite):
     sy = SortYAML()
     sy.main(select, all_files, overwrite)
 
+
 @mdbt.command()
 def sort_sql():
     c = ColumnSorter()
     c.main()
 
+
 @mdbt.command()
 @click.pass_context
 def pre_commit(ctx):

mdbt/core.py

@@ -5,11 +5,12 @@ import subprocess
 import sys
 import typing as t
 
+import snowflake.connector as snow
 from cryptography.hazmat.backends import default_backend
 from cryptography.hazmat.primitives import serialization
-import snowflake.connector as snow
 from dotenv import find_dotenv
 from dotenv import load_dotenv
+
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 load_dotenv(find_dotenv("../.env"))
 load_dotenv(find_dotenv(".env"))
@@ -29,12 +30,12 @@ class Core:
         self.dbt_execute_command_output = ""
 
     def _create_snowflake_connection(self):
-        rsa_key_str = os.environ.get('MAIN__RSA_KEY')
+        rsa_key_str = os.environ.get("MAIN__RSA_KEY")
         if not rsa_key_str:
-            raise ValueError('MAIN__RSA_KEY environment variable not set')
+            raise ValueError("MAIN__RSA_KEY environment variable not set")
         # The RSA key from the environment variable may be in single line or multi-line string
-        if '\\n' in rsa_key_str:
-            rsa_key_str = rsa_key_str.replace('\\n', '\n')
+        if "\\n" in rsa_key_str:
+            rsa_key_str = rsa_key_str.replace("\\n", "\n")
         rsa_key_bytes = rsa_key_str.encode()
         # Load private key object
         private_key = serialization.load_pem_private_key(
@@ -43,13 +44,13 @@ class Core:
             backend=default_backend(),
         )
         self._conn = snow.connect(
-            user=os.getenv('MAIN__USER'),
-            account=os.getenv('MAIN__ACCOUNT'),
-            database=os.getenv('MAIN__DATABASE'),
+            user=os.getenv("MAIN__USER"),
+            account=os.getenv("MAIN__ACCOUNT"),
+            database=os.getenv("MAIN__DATABASE"),
             private_key=private_key,
-            schema=os.getenv('MAIN__SCHEMA'),
-            warehouse=os.getenv('MAIN__WAREHOUSE'),
-            role='MDBT',
+            schema=os.getenv("MAIN__SCHEMA"),
+            warehouse=os.getenv("MAIN__WAREHOUSE"),
+            role="MDBT",
         )
 
         self._cur = self._conn.cursor()
@@ -123,7 +124,9 @@ class Core:
         raise Exception(f"Failure while running command: {' '.join(e.cmd)}")
         # sys.exit(e.returncode)
 
-    def get_models(self, select: str, all_files: bool = False) -> t.List[t.Dict[str, t.Any]]:
+    def get_models(
+        self, select: str, all_files: bool = False
+    ) -> t.List[t.Dict[str, t.Any]]:
         if not all_files:
             args = [
                 "--select",
@@ -138,4 +141,4 @@ class Core:
             ]
         ls_json = self.dbt_ls_to_json(args)
 
-        return ls_json
+        return ls_json

mdbt/prompts.py

@@ -5,7 +5,7 @@ class Prompts:
         return """
 You will help build DBT documentation YML files for a given SQL query. Sometimes you will be asked to generate a description from scratch, other times you will be asked to fill in missing columns that exist in the model, but not in the documentation.
 
-These models are built for Marki Microwave, a RF semicoductor company that designed and built RF and microwave components. The models are built to help the company understand their business better. The models are built in DBT and are used to generate reports and dashboards. For acronyms, or other terminology, consider in context of the RF semiconductor industry. 
+These models are built for Marki Microwave, a RF semicoductor company that designed and built RF and microwave components. The models are built to help the company understand their business better. The models are built in DBT and are used to generate reports and dashboards. For acronyms, or other terminology, consider in context of the RF semiconductor industry.
 
 Primary DBT Guidelines:
 
@@ -27,7 +27,7 @@ Primary DBT Guidelines:
     9. Reorder or order the column descriptions in the YML file in the same order they appear in the SQL query. If you are modifying an existing YML file, still re-order the elements, don't comment out the old element location and put a new element in.
     10. If modifying an existing YML, leave the value of materialized as is. Do not change it to `table` if it is `view` or vice versa.
     11. Use lowercase for all column names, metric, and dimension names. The sample data will come back with uppercase column names, but the YML file should have lowercase names.
-   
+
     12. If there is an primary key ID column as the first field, then add a
        ```
        data_tests:
@@ -73,7 +73,7 @@ models:
 
       - name: network_name
         description: "The name of the network associated with the order."
-        
+
       - name: medical_revenue
         description: "Total revenue from medical services."
 
@@ -90,7 +90,7 @@ This is a CSV data sample from the model:
         return """
 You will help build DBT documentation YML files for a given SQL query. Sometimes you will be asked to generate a description from scratch, other times you will be asked to fill in missing columns that exist in the model, but not in the documentation.
 
-These models are built for Marki Microwave, a RF semicoductor company that designed and built RF and microwave components. The models are built to help the company understand their business better. The models are built in DBT and are used to generate reports and dashboards. For acronyms, or other terminology, consider in context of the RF semiconductor industry. 
+These models are built for Marki Microwave, a RF semicoductor company that designed and built RF and microwave components. The models are built to help the company understand their business better. The models are built in DBT and are used to generate reports and dashboards. For acronyms, or other terminology, consider in context of the RF semiconductor industry.
 
 Primary DBT Guidelines:
 

mdbt/sort_yaml_fields.py

@@ -66,8 +66,6 @@ class SortYAML(AiCore):
             else:
                 self.save_yml_to_clipboard(updated_schema)
 
-
-
     @staticmethod
     def _get_schema_path_and_table(
         original_file_path: str, model_name: str

mdbt/sql_sorter.py

@@ -1,13 +1,14 @@
-
-
 #!/usr/bin/env python3
 import re
 import sys
-from typing import List, Tuple
+from typing import List
+from typing import Tuple
 
 import pyperclip
+
 from mdbt.core import Core
 
+
 class ColumnSorter(Core):
     """Sorts SQL select lines from the clipboard based on alias and data‐type rules."""
 
@@ -25,7 +26,9 @@ class ColumnSorter(Core):
         try:
             clipboard_content = pyperclip.paste()
         except pyperclip.PyperclipException:
-            sys.stderr.write('Error: could not access clipboard. Is pyperclip installed and supported?\n')
+            sys.stderr.write(
+                "Error: could not access clipboard. Is pyperclip installed and supported?\n"
+            )
             sys.exit(1)
 
         # 2) Split into individual lines
@@ -36,13 +39,15 @@ class ColumnSorter(Core):
         sorted_lines = sorter.sort_lines(lines)
 
         # 4) Join back together
-        result = '\n'.join(sorted_lines)
+        result = "\n".join(sorted_lines)
 
         # 5) Copy sorted result back to clipboard
         try:
             pyperclip.copy(result)
         except pyperclip.PyperclipException:
-            sys.stderr.write('Warning: could not write back to clipboard. Outputting to stdout instead.\n')
+            sys.stderr.write(
+                "Warning: could not write back to clipboard. Outputting to stdout instead.\n"
+            )
             print(result)
             sys.exit(0)
 
@@ -66,28 +71,28 @@ class ColumnSorter(Core):
             - alias: the alias (e.g. 'foo_id') or base column name if no AS.
         """
         # 1) Extract prefix (leading comma + whitespace), if present
-        m = re.match(r'^(\s*,\s*)(.*)$', line)
+        m = re.match(r"^(\s*,\s*)(.*)$", line)
         if m:
             prefix = m.group(1)
             rest = m.group(2).strip()
         else:
             # No leading comma, treat everything as rest
-            prefix = ''
+            prefix = ""
             rest = line.strip()
 
         # 2) Look for 'AS' (case-insensitive), split into left/right
         #    Use regex to split on whitespace+as+whitespace, max once
         lower_rest = rest.lower()
-        if re.search(r'\s+as\s+', lower_rest):
-            parts = re.split(r'\s+as\s+', rest, maxsplit=1, flags=re.IGNORECASE)
+        if re.search(r"\s+as\s+", lower_rest):
+            parts = re.split(r"\s+as\s+", rest, maxsplit=1, flags=re.IGNORECASE)
             expression_part = parts[0].strip()
             alias = parts[1].strip()
         else:
             # No AS: take expression exactly as rest; derive alias from expression
             expression_part = rest
             # If there's a '::', drop the cast and use the part before it
-            if '::' in expression_part:
-                alias = expression_part.split('::', 1)[0].strip()
+            if "::" in expression_part:
+                alias = expression_part.split("::", 1)[0].strip()
             else:
                 # If nothing to split, alias is simply the whole rest
                 alias = expression_part
@@ -114,14 +119,14 @@ class ColumnSorter(Core):
         a = alias.lower()
         expr_lower = expression.lower()
 
-        if a.endswith('_id'):
+        if a.endswith("_id"):
             return 0
-        if a.endswith('_at'):
+        if a.endswith("_at"):
             return 1
-        if a.startswith('is_'):
+        if a.startswith("is_"):
             return 2
         # Check for VARIANT cast
-        if re.search(r'::\s*variant\b', expr_lower):
+        if re.search(r"::\s*variant\b", expr_lower):
             return 4
         # Otherwise, everything else is group 3
         return 3
@@ -153,13 +158,13 @@ class ColumnSorter(Core):
             # Determine if original expr already contained ' as alias_lower' (case-insensitive)
             # We can check if expr.lower().endswith(alias_lower) but that fails if casting was present.
             # Instead, if alias_lower != expr.split('::')[0].strip().lower(), we assume original used AS.
-            base_no_cast = expr.split('::', 1)[0].strip().lower()
+            base_no_cast = expr.split("::", 1)[0].strip().lower()
             if base_no_cast != alias_lower:
                 # original must have had an explicit alias, so we add ' as alias'
-                line_text = f'{prefix}{expr} as {alias_lower}'
+                line_text = f"{prefix}{expr} as {alias_lower}"
             else:
                 # no AS needed
-                line_text = f'{prefix}{expr}'
+                line_text = f"{prefix}{expr}"
             sorted_lines.append(line_text)
 
         return sorted_lines

{mdbt-0.5.1.dist-info → mdbt-0.5.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mdbt
-Version: 0.5.1
+Version: 0.5.2
 Summary: A CLI tool to manage dbt builds with state handling and manifest management
 Author: Craig Lathrop
 Author-email: info@markimicrowave.com

mdbt-0.5.2.dist-info/RECORD

@@ -0,0 +1,19 @@
+mdbt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mdbt/ai_core.py,sha256=EyuEciA6EIRfZ2vj4YxCLkdfmkke7tbZWPjBb_JOuNw,3961
+mdbt/build_dbt_docs_ai.py,sha256=Ci7ilLR-YQDMLWo-K4AOJ_SKDe6BxINXJ5UA9NTGOR8,9440
+mdbt/build_unit_test_data_ai.py,sha256=SJlxEjwrH3zyXbZFLQFl3_BZYIM3WLdubA8eUEYRbQU,4290
+mdbt/cmdline.py,sha256=WTx1z-vgL40zUOsgF5Q-NZXPSprNovOnTZ-sYqhzwtc,9378
+mdbt/core.py,sha256=kWcMZhYCeSAey1YK7UsimzD4UZIY_dnX6V1W5XmZtMI,5105
+mdbt/expectations_output_builder.py,sha256=AXKEM-WO7FecYzfMLwzsOnQnVf7AiHBi_khyidE2lJs,3195
+mdbt/main.py,sha256=UO3r7zOXmVpjnAIz2eeZVeQMFmgXa698Gm42Wo2qhRU,16939
+mdbt/precommit_format.py,sha256=9HC10mh4QQIgaQSxdAwaCXbsiHT9cCrLdbc3PAQkotc,2845
+mdbt/prompts.py,sha256=sqDqbrxfF7eFm5CcoiEAkwsBEVmX_1qVgLMeVoqzqNY,13979
+mdbt/recce.py,sha256=P14CvWd7lRgTPUW7BVMLmcclSqby-_uSgpoI3r5VjTA,2327
+mdbt/sort_yaml_fields.py,sha256=rrPW-DqX52Dd1iFyKLluapbMl0jXiKxfuoq3L3M66vo,5055
+mdbt/sql_sorter.py,sha256=UVn7vFzomC4wGuzBBXLAP9SJtg40-O5fND_Fbcnzw44,6545
+mdbt-0.5.2.dist-info/licenses/LICENSE,sha256=DrJpgQEYhttwpwcE56BzrGZ1aEfR_tqfaxsI5NlsYOE,1072
+mdbt-0.5.2.dist-info/METADATA,sha256=TyHT9ATJRCqViiIuy4ZZ9f19KztKQk0czfHCljJ2-ME,933
+mdbt-0.5.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+mdbt-0.5.2.dist-info/entry_points.txt,sha256=fVft1CYVP2MtZMMtsXN67S1T5RszfgKtAuaeoKLdCow,43
+mdbt-0.5.2.dist-info/top_level.txt,sha256=-PP7vAl9EgVjRTzBovElczsPNjOfja6kjZssNmv5vo0,5
+mdbt-0.5.2.dist-info/RECORD,,

mdbt/lightdash.py

@@ -1,84 +0,0 @@
-import json
-import os
-import subprocess
-import sys
-
-from click.core import Context
-
-from mdbt.core import Core
-
-
-class Lightdash(Core):
-
-    def __init__(self, test_mode=False):
-        super().__init__(test_mode=test_mode)
-
-    def lightdash_start_preview(
-        self, ctx: Context, select: str, preview_name: str, l43: bool
-    ):
-        # Check to make sure the LIGHTDASH_PROJECT env variable is set
-        if not os.getenv("LIGHTDASH_PROJECT"):
-            print(
-                "LIGHTDASH_PROJECT environment variable not set. Set this key to the ID of the project you will "
-                "promote charts to."
-            )
-            sys.exit(1)
-        else:
-            print(f"Building for LIGHTDASH_PROJECT: {os.getenv('LIGHTDASH_PROJECT')}")
-
-        self._check_lightdash_for_updates()
-        if not preview_name:
-            # If no preview name, use the current name of the git branch
-            result = subprocess.run(
-                ["git", "branch", "--show-current"], stdout=subprocess.PIPE, text=True
-            )
-            preview_name = result.stdout.strip()
-
-        args = ["lightdash", "start-preview", "--name", preview_name]
-
-        if l43:
-            args = args + ["-s", "tag:l3 tag:l4"]
-
-        if select:
-            args = args + ["--select", select]
-
-        try:
-            print(f'Running command: {" ".join(args)}')
-            subprocess.run(args, check=True)
-        except subprocess.CalledProcessError as e:
-            self.handle_cmd_line_error(e)
-
-    @staticmethod
-    def _check_lightdash_for_updates():
-        api_str = 'curl -s "https://app.lightdash.cloud/api/v1/health"'
-
-        try:
-            result = subprocess.run(
-                api_str, shell=True, check=True, text=True, capture_output=True
-            )
-            # Convert to JSON
-            result_json = json.loads(result.stdout)
-        except subprocess.CalledProcessError as e:
-            print(f"Failure while running command: {api_str}")
-            print(e.stderr)
-            print(e.stdout)
-            sys.exit(e.returncode)
-
-        api_version = result_json["results"]["version"]
-
-        result = subprocess.run(
-            ["lightdash", "--version"], check=True, text=True, capture_output=True
-        )
-
-        current_version = result.stdout.strip()
-
-        if api_version != current_version:
-            print(
-                f"API version {api_version} does not match current version {current_version}. Upgrading."
-            )
-            args = ["npm", "install", "-g", f"@lightdash/cli@{api_version}"]
-            subprocess.run(args, check=True)
-        else:
-            print(
-                f"API version {api_version} matches current version {current_version}."
-            )

mdbt-0.5.1.dist-info/RECORD

@@ -1,20 +0,0 @@
-mdbt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mdbt/ai_core.py,sha256=EyuEciA6EIRfZ2vj4YxCLkdfmkke7tbZWPjBb_JOuNw,3961
-mdbt/build_dbt_docs_ai.py,sha256=Yx0AYdsl67WKUJmA_hPeEb8CKRoWrWAxbHqNcJjPiyM,9248
-mdbt/build_unit_test_data_ai.py,sha256=SJlxEjwrH3zyXbZFLQFl3_BZYIM3WLdubA8eUEYRbQU,4290
-mdbt/cmdline.py,sha256=meNATu3BzP_4Htt5VcoT923mlh9NsfK8og0JQgn9PCE,10822
-mdbt/core.py,sha256=XO6_KwMnVumtilj0OWZBcfPM7SwaUOba2lkw1apxx6w,5089
-mdbt/expectations_output_builder.py,sha256=AXKEM-WO7FecYzfMLwzsOnQnVf7AiHBi_khyidE2lJs,3195
-mdbt/lightdash.py,sha256=qJBJ-pc5mN8GBA1MZElRhtA8aldrX-AgvHtha4iOA-Y,2745
-mdbt/main.py,sha256=UO3r7zOXmVpjnAIz2eeZVeQMFmgXa698Gm42Wo2qhRU,16939
-mdbt/precommit_format.py,sha256=9HC10mh4QQIgaQSxdAwaCXbsiHT9cCrLdbc3PAQkotc,2845
-mdbt/prompts.py,sha256=2vCvh9hamgop92kGGaMKtap11F2MZiM7hHKjcwX4lhQ,13992
-mdbt/recce.py,sha256=P14CvWd7lRgTPUW7BVMLmcclSqby-_uSgpoI3r5VjTA,2327
-mdbt/sort_yaml_fields.py,sha256=QF-zwXgeZ4iuaV3CfLxz4w30EqC7y34uBuowO4DbYUA,5057
-mdbt/sql_sorter.py,sha256=8bd6svrtcXp7ePT2g4FTGLTW55qbsVjXgUmba7L-G-4,6467
-mdbt-0.5.1.dist-info/licenses/LICENSE,sha256=DrJpgQEYhttwpwcE56BzrGZ1aEfR_tqfaxsI5NlsYOE,1072
-mdbt-0.5.1.dist-info/METADATA,sha256=i_8GKD1UTQwN3nX_xnpnm0EhdoXMDosc5BImrPOoTtM,933
-mdbt-0.5.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-mdbt-0.5.1.dist-info/entry_points.txt,sha256=fVft1CYVP2MtZMMtsXN67S1T5RszfgKtAuaeoKLdCow,43
-mdbt-0.5.1.dist-info/top_level.txt,sha256=-PP7vAl9EgVjRTzBovElczsPNjOfja6kjZssNmv5vo0,5
-mdbt-0.5.1.dist-info/RECORD,,