mdbt 0.4.33__tar.gz → 0.4.34__tar.gz

{mdbt-0.4.33 → mdbt-0.4.34}/PKG-INFO

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

{mdbt-0.4.33 → mdbt-0.4.34}/mdbt/ai_core.py

@@ -16,7 +16,7 @@ from mdbt.prompts import Prompts
 
 class AiCore(Core):
 
-    def __init__(self, model: str = "gpt-4.1", test_mode: bool = False):
+    def __init__(self, model: str = "gpt-5", test_mode: bool = False):
         super().__init__(test_mode=test_mode)
         self.model = model
         # Make sure you have OPENAI_API_KEY set in your environment variables.

mdbt-0.4.34/mdbt/build_dbt_docs_ai.py

@@ -0,0 +1,145 @@
+import subprocess
+
+import pyperclip
+from dotenv import find_dotenv
+from dotenv import load_dotenv
+
+from mdbt.ai_core import AiCore
+from mdbt.prompts import Prompts
+
+load_dotenv(find_dotenv("../.env"))
+load_dotenv(find_dotenv(".env"))
+
+
+class BuildDBTDocs(AiCore):
+    """
+    # Make sure you have OPENAI_API_KEY set in your environment variables.
+    """
+
+    def __init__(self):
+        super().__init__()
+
+    def main(self, select, sys_context, is_new=False):
+        if not is_new:
+            print(
+                """
+            1) Build new DBT documentation.
+            2) Check existing DBT documentation against model for missing definitions.
+            """
+            )
+            mode = int(input())
+        else:
+            mode = 1
+        print("Getting file.")
+        models = self.get_models(select)
+        if not models:
+            raise ValueError(f"No models found for select '{select}'")
+
+        for model in models:
+            original_file_path = model["original_file_path"]
+            model_name = model["name"]
+            if "l4" in original_file_path.lower() or "l3" in original_file_path.lower():
+                system_instructions = Prompts().dbt_docs_gte_l3_prompt
+            else:
+                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"
+
+            sample_data = self._get_sample_data_from_snowflake([model_name])
+
+            system_instructions = system_instructions + sample_data[model_name]
+
+            # Might bring this back in the future.
+            extra_info = ""
+
+            if mode == 1:
+                # Build new documentation
+                user_input = self.build_user_msg_mode_1(original_file_path, extra_info)
+                yml_file_path = original_file_path.replace(".sql", ".yml")
+            elif mode == 2:
+                # Check existing documentation
+                yml_file_path = original_file_path[:-4] + ".yml"
+                user_input = self.build_user_msg_mode_2(
+                    original_file_path, yml_file_path, extra_info
+                )
+            else:
+                print(mode)
+                raise ValueError("Invalid mode")
+
+            messages = [
+                {"role": "user", "content": system_instructions + "\n" + user_input}
+            ]
+
+            assistant_responses = []
+            result = self.send_message(messages)
+            assistant_responses.append(result)
+
+            messages.append({"role": "assistant", "content": assistant_responses[0]})
+            print(assistant_responses[0])
+            output = assistant_responses[0]
+            # Check for ``` at end of output (str) and remove
+            # Remove trailing markdown code fences if present
+            lines = output.split('\n')
+            new_lines = []
+            for line in lines:
+                if '```' not in line:
+                    new_lines.append(line)
+            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:"
+                )
+            else:
+                clip_or_file = "2"
+
+            if clip_or_file == "1":
+                print("Output copied to clipboard")
+                pyperclip.copy(output)
+            elif clip_or_file == "2":
+                if mode == 2:
+                    # Make a backup of the current YML file.
+                    self.backup_existing_yml_file(yml_file_path)
+                with open(yml_file_path, "w") as file:
+                    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: ")
+                    else:
+                        commit_file = "1"
+
+                    if commit_file == "1":
+                        subprocess.run(["git", "add", yml_file_path])
+
+    @staticmethod
+    def backup_existing_yml_file(yml_file_path):
+        with open(yml_file_path, "r") as file:
+            yml_content = file.read()
+        with open(yml_file_path + ".bak", "w") as file:
+            file.write(yml_content)
+
+    def build_user_msg_mode_1(self, _sql_file_path: str, extra_info: str) -> str:
+        self.read_file(_sql_file_path)
+        model_name = _sql_file_path.split("/")[-1].split(".")[0]
+        prompt_str = f"Build new DBT documentation for the following SQL query with model name {model_name}"
+        if len(extra_info):
+            prompt_str += f"\n{extra_info}"
+
+        return prompt_str
+
+    def build_user_msg_mode_2(
+        self, _sql_file_path: str, _yml_file_path: str, extra_info: str
+    ) -> str:
+        self.read_file(_sql_file_path)
+        yml = self.read_file(_yml_file_path)
+        model_name = _sql_file_path.split("/")[-1].split(".")[0]
+        prompt_str = f"Check for missing columns in the following DBT documentation for the following SQL query with model name {model_name}. Identify any columns in the DBT documentation that do not exist in the SQL and comment them out."
+        if len(extra_info):
+            prompt_str += f"\n {extra_info}"
+        prompt_str += f"\nYML File Contents:\n{yml}"
+
+        return prompt_str
+
+
+if __name__ == "__main__":
+    BuildDBTDocs().main("revenue_by_dvm")

{mdbt-0.4.33 → mdbt-0.4.34}/mdbt/build_unit_test_data_ai.py

@@ -23,7 +23,7 @@ logging.getLogger("snowflake.connector").setLevel(logging.WARNING)
 class BuildUnitTestDataAI(AiCore):
 
     def __init__(self):
-        super().__init__(model="o3-mini")
+        super().__init__(model="gpt-5")
 
     def main(self, model_name: str):
 

{mdbt-0.4.33 → mdbt-0.4.34}/mdbt/core.py

@@ -122,3 +122,20 @@ class Core:
         print(e.stdout)
         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]]:
+        if not all_files:
+            args = [
+                "--select",
+                select,
+                "--exclude",
+                "resource_type:test resource_type:seed resource_type:snapshot resource_type:source",
+            ]
+        else:
+            args = [
+                "--exclude",
+                "resource_type:test resource_type:seed resource_type:snapshot resource_type:source",
+            ]
+        ls_json = self.dbt_ls_to_json(args)
+
+        return ls_json

{mdbt-0.4.33 → mdbt-0.4.34}/mdbt/sort_yaml_fields.py

@@ -35,7 +35,7 @@ class SortYAML(AiCore):
         all_files: Optional[bool] = False,
         overwrite: Optional[bool] = False,
     ):
-        models = self._get_models(select, all_files)
+        models = self.get_models(select, all_files)
         if len(models) > 1 and not overwrite:
             raise ValueError(
                 "Multiple models found. Default copy to clipboard only works with one model. Use the --overwrite flag "
@@ -66,23 +66,7 @@ class SortYAML(AiCore):
             else:
                 self.save_yml_to_clipboard(updated_schema)
 
-    def _get_models(self, select: str, all_files: bool) -> List[Dict[str, Any]]:
-        mmw = MDBT()
-        if not all_files:
-            args = [
-                "--select",
-                select,
-                "--exclude",
-                "resource_type:test resource_type:seed resource_type:snapshot resource_type:source",
-            ]
-        else:
-            args = [
-                "--exclude",
-                "resource_type:test resource_type:seed resource_type:snapshot resource_type:source",
-            ]
-        ls_json = mmw.dbt_ls_to_json(args)
-
-        return ls_json
+
 
     @staticmethod
     def _get_schema_path_and_table(

{mdbt-0.4.33 → mdbt-0.4.34}/mdbt.egg-info/PKG-INFO

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

{mdbt-0.4.33 → mdbt-0.4.34}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup
 
 setup(
     name="mdbt",
-    version="0.4.33",
+    version="0.4.34",
     description="A CLI tool to manage dbt builds with state handling and manifest management",
     author="Craig Lathrop",
     author_email="info@markimicrowave.com",

mdbt-0.4.33/mdbt/build_dbt_docs_ai.py

@@ -1,142 +0,0 @@
-import subprocess
-
-import pyperclip
-from dotenv import find_dotenv
-from dotenv import load_dotenv
-
-from mdbt.ai_core import AiCore
-from mdbt.prompts import Prompts
-
-load_dotenv(find_dotenv("../.env"))
-load_dotenv(find_dotenv(".env"))
-
-
-class BuildDBTDocs(AiCore):
-    """
-    # Make sure you have OPENAI_API_KEY set in your environment variables.
-    """
-
-    def __init__(self):
-        super().__init__()
-
-    def main(self, model_name, sys_context, is_new=False):
-        if model_name.endswith(".sql"):
-            model_name = model_name[:-4]
-        if not is_new:
-            print(
-                """
-            1) Build new DBT documentation.
-            2) Check existing DBT documentation against model for missing definitions.
-            """
-            )
-            mode = int(input())
-        else:
-            mode = 1
-        print("Getting file.")
-        sql_file_path = self.get_file_path(model_name)
-
-        if "l4" in sql_file_path.lower() or "l3" in sql_file_path.lower():
-            system_instructions = Prompts().dbt_docs_gte_l3_prompt
-        else:
-            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"
-
-        sample_data = self._get_sample_data_from_snowflake([model_name])
-
-        system_instructions = system_instructions + sample_data[model_name]
-
-        # Might bring this back in the future.
-        extra_info = ""
-
-        if mode == 1:
-            # Build new documentation
-            user_input = self.build_user_msg_mode_1(sql_file_path, extra_info)
-            yml_file_path = sql_file_path.replace(".sql", ".yml")
-        elif mode == 2:
-            # Check existing documentation
-            yml_file_path = sql_file_path[:-4] + ".yml"
-            user_input = self.build_user_msg_mode_2(
-                sql_file_path, yml_file_path, extra_info
-            )
-        else:
-            print(mode)
-            raise ValueError("Invalid mode")
-
-        messages = [
-            {"role": "user", "content": system_instructions + "\n" + user_input}
-        ]
-
-        assistant_responses = []
-        result = self.send_message(messages)
-        assistant_responses.append(result)
-
-        messages.append({"role": "assistant", "content": assistant_responses[0]})
-        print(assistant_responses[0])
-        output = assistant_responses[0]
-        # Check for ``` at end of output (str) and remove
-        # Remove trailing markdown code fences if present
-        lines = output.split('\n')
-        new_lines = []
-        for line in lines:
-            if '```' not in line:
-                new_lines.append(line)
-        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:"
-            )
-        else:
-            clip_or_file = "2"
-
-        if clip_or_file == "1":
-            print("Output copied to clipboard")
-            pyperclip.copy(output)
-        elif clip_or_file == "2":
-            if mode == 2:
-                # Make a backup of the current YML file.
-                self.backup_existing_yml_file(yml_file_path)
-            with open(yml_file_path, "w") as file:
-                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: ")
-                else:
-                    commit_file = "1"
-
-                if commit_file == "1":
-                    subprocess.run(["git", "add", yml_file_path])
-
-    @staticmethod
-    def backup_existing_yml_file(yml_file_path):
-        with open(yml_file_path, "r") as file:
-            yml_content = file.read()
-        with open(yml_file_path + ".bak", "w") as file:
-            file.write(yml_content)
-
-    def build_user_msg_mode_1(self, _sql_file_path: str, extra_info: str) -> str:
-        self.read_file(_sql_file_path)
-        model_name = _sql_file_path.split("/")[-1].split(".")[0]
-        prompt_str = f"Build new DBT documentation for the following SQL query with model name {model_name}"
-        if len(extra_info):
-            prompt_str += f"\n{extra_info}"
-
-        return prompt_str
-
-    def build_user_msg_mode_2(
-        self, _sql_file_path: str, _yml_file_path: str, extra_info: str
-    ) -> str:
-        self.read_file(_sql_file_path)
-        yml = self.read_file(_yml_file_path)
-        model_name = _sql_file_path.split("/")[-1].split(".")[0]
-        prompt_str = f"Check for missing columns in the following DBT documentation for the following SQL query with model name {model_name}. Identify any columns in the DBT documentation that do not exist in the SQL and comment them out."
-        if len(extra_info):
-            prompt_str += f"\n {extra_info}"
-        prompt_str += f"\nYML File Contents:\n{yml}"
-
-        return prompt_str
-
-
-if __name__ == "__main__":
-    BuildDBTDocs().main("revenue_by_dvm")