nlsql-coder 0.0.2__tar.gz → 0.0.3__tar.gz

{nlsql_coder-0.0.2 → nlsql_coder-0.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nlsql-coder
-Version: 0.0.2
+Version: 0.0.3
 Summary: NLSQL's internal AI coding agent CLI tool for interacting with codebases
 Author: NLSQL
 License: MIT License

{nlsql_coder-0.0.2 → nlsql_coder-0.0.3}/nlsql_coder/app/cli/interface.py

@@ -8,6 +8,7 @@ from rich.live import Live
 from rich.align import Align
 from nlsql_coder.data.create_agent_md import create_md
 import time
+import os
 
 
 console = Console()
@@ -17,13 +18,27 @@ Welcome      _   _  _      ____   ___  _
             | \ | || |    / ___| / _ \| |    
        to   |  \| || |    \___ \| | | | |    
             | |\  || |___  ___) | |_| | |___ 
-            |_| \_||_____||____/ \__\_\_____|   Code\n\n[/blue bold]
+            |_| \_||_____||____/ \__\_\_____|   Coder\n\n[/blue bold]
 """
 
+def get_envs():
+    console.print("[bold]To start, we need to set up your environment variables...[/bold]")
+    os.environ["MODEL_PROVIDER"] = "aws"
+    model_name = console.input("\n[bold blue][Model name]:[/bold blue] ")
+    access_key_id = console.input("\n[bold blue][Access Key ID]:[/bold blue] ")
+    secret_key = console.input("\n[bold blue][Secret Access Key]:[/bold blue] ")
+    region = console.input("\n[bold blue][Region]:[/bold blue] ")
+    os.environ["MODEL_NAME"] = model_name
+    os.environ["AWS_ACCESS_KEY_ID"] = access_key_id
+    os.environ["AWS_SECRET_ACCESS_KEY"] = secret_key
+    os.environ["AWS_DEFAULT_REGION"] = region
 
 
 def main_loop():
     try:
+        console.clear()
+        console.print(Align.center(welcome_message))
+        get_envs()
         usage_data = {}
         console.clear()
         with console.status("[bold]Creating AGENTS.md file...\n[/bold]", spinner="dots"):
@@ -39,7 +54,6 @@ def main_loop():
         time.sleep(2)
         console.clear()
         console.print(Align.center(welcome_message))
-
         while True:
 
             if args.usage and usage_data:

{nlsql_coder-0.0.2 → nlsql_coder-0.0.3}/nlsql_coder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nlsql-coder
-Version: 0.0.2
+Version: 0.0.3
 Summary: NLSQL's internal AI coding agent CLI tool for interacting with codebases
 Author: NLSQL
 License: MIT License

{nlsql_coder-0.0.2 → nlsql_coder-0.0.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "nlsql-coder"
-version = "0.0.2"
+version = "0.0.3"
 description = "NLSQL's internal AI coding agent CLI tool for interacting with codebases"
 readme = "README.md"
 requires-python = ">=3.11"