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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nlsql-coder
-Version: 0.0.2
+Version: 0.0.4
 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.4}/nlsql_coder/app/agent/helpers/get_agent.py

@@ -8,6 +8,7 @@ from importlib import resources
 
 provider_config = ProviderConfig()
 agent = None
+llm = None
 prompt_path = resources.files("nlsql_coder.app.agent.helpers") / "system_prompt.md"
 system_prompt = prompt_path.read_text()
 agents_md = Path(args.workspace, "AGENTS.md").resolve()

nlsql_coder-0.0.4/nlsql_coder/app/cli/arg_parsing.py

@@ -0,0 +1,28 @@
+import argparse
+
+arg_parser = argparse.ArgumentParser()
+
+arg_parser.add_argument(
+    "command",
+    nargs="?",
+    default=None
+)
+
+arg_parser.add_argument(
+    "-w",
+    "--workspace",
+    default=".",
+    help="Path to the target project's workspace."
+)
+
+arg_parser.add_argument(
+    "-u",
+    "--usage",
+    action="store_true",
+    help="Displayed usage data after each conversation turn."
+)
+
+args = arg_parser.parse_args()
+
+if args.command == "init":
+    args.workspace = "."

{nlsql_coder-0.0.2 → nlsql_coder-0.0.4}/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.4}/nlsql_coder/data/create_agent_md.py

@@ -7,6 +7,9 @@ from pathlib import Path
 # Get relevant project files for context
 def create_md():
     try:
+        if not llm:
+            raise EnvironmentError("Environment variables not present or incorrect.")
+            return
         root = Path(args.workspace)
         if Path(root, "AGENTS.md").exists():
             return "exists"

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nlsql-coder
-Version: 0.0.2
+Version: 0.0.4
 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.4}/pyproject.toml

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

nlsql_coder-0.0.2/nlsql_coder/app/cli/arg_parsing.py

@@ -1,7 +0,0 @@
-import argparse
-
-arg_parser = argparse.ArgumentParser()
-arg_parser.add_argument("-w", "--workspace", default=".", help="Path to the targe project's workspace.")
-arg_parser.add_argument("-u", "--usage", action="store_true", help="Displayed usage data after each conversation turn.")
-
-args = arg_parser.parse_args()