dmddl 0.2.13__tar.gz → 0.2.15__tar.gz

{dmddl-0.2.13 → dmddl-0.2.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dmddl
-Version: 0.2.13
+Version: 0.2.15
 Summary: cli tool for creating insert script from ddl script
 License: MIT
 Author: HoJLter

{dmddl-0.2.13 → dmddl-0.2.15}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "dmddl"
-version = "0.2.13"
+version = "0.2.15"
 description = "cli tool for creating insert script from ddl script"
 authors = [
     {name = "HoJLter",email = "hojlter.work@gmail.com"}

{dmddl-0.2.13 → dmddl-0.2.15}/src/dmddl/cli.py

@@ -30,17 +30,16 @@ def ask_api_key():
 
 def make_query(provider, api_key, prompt):
     console = Console()
-    print(provider)
     with console.status("[bold blue]Making query. Wait for result..."):
-        if provider == "OpenAI":
-            response = openai_request(base_prompt+prompt, api_key)
-            return response
+        if provider:
+            if provider == "OpenAI":
+                response = openai_request(base_prompt+prompt, api_key)
+                return response
 
-        elif provider is None:
+            raise Exception("LLM Provider not found")
+        else:
             raise Exception("Use -c (--config) to configurate app and set LLM provider.")
 
-        raise Exception("LLM Provider not found")
-
 
 def write_output_file(data):
     with open("output.txt", 'w') as file:
@@ -95,12 +94,14 @@ def main():
                                            api_key=api_key,
                                            prompt=user_prompt)
             write_output_file(response)
-            syntax = Syntax(response, 'sql', line_numbers=True)
             print("\n\n[yellow bold]OUTPUT.TXT\n",)
-            console.print(syntax)
             if success:
+                syntax = Syntax(response, 'sql', line_numbers=True)
+                console.print(syntax)
                 print("[green bold] Your DML script is ready! Check output.txt")
             if not success:
+                syntax = Syntax(response, 'python', line_numbers=True)
+                console.print(syntax)
                 print("[red bold] Error has occurred... Check output.txt")