agentinit-cli 0.3.0__tar.gz → 0.3.2__tar.gz

{agentinit_cli-0.3.0 → agentinit_cli-0.3.2}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: agentinit-cli
-Version: 0.3.0
+Version: 0.3.2
 Summary: A CLI tool to scaffold production-ready LLM agent projects
 License: MIT
 Keywords: llm,agents,scaffolding,langgraph,crewai,autogen,google-adk,cli
 Author: Swapnil Bhattacharya
-Author-email: your@email.com
+Author-email: swapnil.bhat17@outlook.com
 Requires-Python: >=3.10,<4.0
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent

{agentinit_cli-0.3.0 → agentinit_cli-0.3.2}/agentinit/cli.py

@@ -13,12 +13,13 @@ PACKAGE_MANAGERS = ["pip", "uv"]
 API_URL = "https://agentinit-api.onrender.com/analyze"
 
 
-def get_smart_structure(description: str, framework: str) -> dict | None:
+def get_smart_structure(description: str, framework: str, llm: str) -> dict | None:
     try:
+        console.print("  [dim]Waking up the API, this may take up to 60 seconds on first run...[/dim]")
         response = requests.post(
             API_URL,
-            json={"description": description, "framework": framework},
-            timeout=30,
+            json={"description": description, "framework": framework, "llm": llm},
+            timeout=90,
         )
         if response.status_code == 200:
             return response.json()
@@ -71,7 +72,7 @@ def create(project_name):
     smart_structure = None
     if description and description.strip():
         console.print("\n  [dim]Analyzing project description...[/dim]")
-        smart_structure = get_smart_structure(description.strip(), framework)
+        smart_structure = get_smart_structure(description.strip(), framework, llm)
         if smart_structure:
             console.print("  [green]Smart structure generated![/green]")
         else:
@@ -131,4 +132,4 @@ def list_frameworks():
     for fw, llm in rows:
         table.add_row(fw, llm)
 
-    console.print(table)    
+    console.print(table)

{agentinit_cli-0.3.0 → agentinit_cli-0.3.2}/agentinit/generator.py

@@ -76,8 +76,12 @@ def generate_project(
         write_file(project_path / output_file, template.render(**context))
 
     if package_manager == "pip":
-        template = fw_env.get_template("requirements.txt.j2")
-        write_file(project_path / "requirements.txt", template.render(**context))
+        if smart_structure and smart_structure.get("requirements"):
+            reqs_content = "\n".join(smart_structure["requirements"]) + "\n"
+            write_file(project_path / "requirements.txt", reqs_content)
+        else:
+            template = fw_env.get_template("requirements.txt.j2")
+            write_file(project_path / "requirements.txt", template.render(**context))
     else:
         common_env = Environment(loader=FileSystemLoader(str(TEMPLATES_DIR / "common")))
         template = common_env.get_template("pyproject.toml.j2")
@@ -104,6 +108,9 @@ def generate_project(
         for item in smart_structure.get("utils", []):
             write_stub(project_path / "utils" / f"{item}.py", item, "util")
 
+        for item in smart_structure.get("services", []):
+            write_stub(project_path / "services" / f"{item}.py", item, "service")
+
         for test in smart_structure.get("tests", []):
             write_stub(project_path / "tests" / f"{test}.py", test, "test")
 

{agentinit_cli-0.3.0 → agentinit_cli-0.3.2}/pyproject.toml

@@ -1,8 +1,8 @@
 [tool.poetry]
 name = "agentinit-cli"
-version = "0.3.0"
+version = "0.3.2"
 description = "A CLI tool to scaffold production-ready LLM agent projects"
-authors = ["Swapnil Bhattacharya <your@email.com>"]
+authors = ["Swapnil Bhattacharya <swapnil.bhat17@outlook.com>"]
 readme = "README.md"
 license = "MIT"
 homepage = "https://github.com/NorthCommits/agentinit"