dmddl 0.2.13__py3-none-any.whl → 0.2.15__py3-none-any.whl

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")
 
 

{dmddl-0.2.13.dist-info → dmddl-0.2.15.dist-info}/METADATA

@@ -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.dist-info → dmddl-0.2.15.dist-info}/RECORD

@@ -1,12 +1,12 @@
 dmddl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dmddl/cli.py,sha256=h8MmS8o8g0gN0QpseW3yfeTtSbSXDHS6ECc9KhhF1ew,3314
+dmddl/cli.py,sha256=h8FMphzSo1koBqSqq1rPV_qNeDmPLuEnvPEshEJt20Q,3431
 dmddl/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dmddl/config/settings.py,sha256=tzxldPnnVv1Shu32XljVO2VuHza43zr61zFcMM_bxvg,1341
 dmddl/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dmddl/models/llm.py,sha256=-0e9tyMXki4t9R-zKEqcNveFbwKhJmMxOVR7VkhMv7g,851
 dmddl/models/prompt.py,sha256=7cjWC-mQJR32jc0a0GsyoYDZWROB57Yk4Wy4nRsiFoM,1923
-dmddl-0.2.13.dist-info/entry_points.txt,sha256=87tyUAFcB9RYRLhYyxroZxcTho-o3IGXnMM4uMGKdV4,40
-dmddl-0.2.13.dist-info/LICENSE,sha256=C--Tlk2EXyLEz1DSTQFKOo-WblCPTsxPcd8t63NCauw,1090
-dmddl-0.2.13.dist-info/METADATA,sha256=wZAgdC5LnC2UycBUTz8ECOgdew90fz-shaV1jhqOmPg,935
-dmddl-0.2.13.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-dmddl-0.2.13.dist-info/RECORD,,
+dmddl-0.2.15.dist-info/entry_points.txt,sha256=87tyUAFcB9RYRLhYyxroZxcTho-o3IGXnMM4uMGKdV4,40
+dmddl-0.2.15.dist-info/LICENSE,sha256=C--Tlk2EXyLEz1DSTQFKOo-WblCPTsxPcd8t63NCauw,1090
+dmddl-0.2.15.dist-info/METADATA,sha256=0ppg2XhSlcwiNarI9pi_MQUm2AggVzT-VgQDjokCVm0,935
+dmddl-0.2.15.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dmddl-0.2.15.dist-info/RECORD,,