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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentinit-cli
-Version: 0.1.0
+Version: 0.3.0
 Summary: A CLI tool to scaffold production-ready LLM agent projects
 License: MIT
 Keywords: llm,agents,scaffolding,langgraph,crewai,autogen,google-adk,cli
@@ -18,7 +18,9 @@ Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Software Development :: Code Generators
 Requires-Dist: click (>=8.1.0,<9.0.0)
 Requires-Dist: jinja2 (>=3.1.0,<4.0.0)
-Requires-Dist: rich (>=13.0.0,<14.0.0)
+Requires-Dist: questionary (>=2.0.0,<3.0.0)
+Requires-Dist: requests (>=2.31.0,<3.0.0)
+Requires-Dist: rich (>=13.0.0)
 Project-URL: Homepage, https://github.com/NorthCommits/agentinit
 Project-URL: Repository, https://github.com/NorthCommits/agentinit
 Description-Content-Type: text/markdown

agentinit_cli-0.3.0/agentinit/cli.py

@@ -0,0 +1,134 @@
+import click
+import questionary
+import requests
+from rich.console import Console
+from rich.table import Table
+from agentinit.generator import generate_project, add_agent
+
+console = Console()
+
+SUPPORTED_FRAMEWORKS = ["langgraph", "crewai", "autogen", "google_adk", "openai_agents", "smolagents"]
+SUPPORTED_LLMS = ["openai", "anthropic", "groq", "azure", "bedrock", "gemini"]
+PACKAGE_MANAGERS = ["pip", "uv"]
+API_URL = "https://agentinit-api.onrender.com/analyze"
+
+
+def get_smart_structure(description: str, framework: str) -> dict | None:
+    try:
+        response = requests.post(
+            API_URL,
+            json={"description": description, "framework": framework},
+            timeout=30,
+        )
+        if response.status_code == 200:
+            return response.json()
+    except Exception:
+        pass
+    return None
+
+
+@click.group()
+def app():
+    """agentinit - scaffold LLM agent projects in seconds"""
+    pass
+
+
+@app.command("create")
+@click.argument("project_name")
+def create(project_name):
+    """Scaffold a new LLM agent project interactively."""
+
+    console.print(f"\n[bold green]Creating project:[/bold green] [cyan]{project_name}[/cyan]\n")
+
+    framework = questionary.select(
+        "Select a framework:",
+        choices=SUPPORTED_FRAMEWORKS,
+    ).ask()
+
+    if not framework:
+        raise click.Abort()
+
+    llm = questionary.select(
+        "Select an LLM provider:",
+        choices=SUPPORTED_LLMS,
+    ).ask()
+
+    if not llm:
+        raise click.Abort()
+
+    package_manager = questionary.select(
+        "Select a package manager:",
+        choices=PACKAGE_MANAGERS,
+    ).ask()
+
+    if not package_manager:
+        raise click.Abort()
+
+    description = questionary.text(
+        "Describe your project in one line (press Enter to skip):",
+    ).ask()
+
+    smart_structure = None
+    if description and description.strip():
+        console.print("\n  [dim]Analyzing project description...[/dim]")
+        smart_structure = get_smart_structure(description.strip(), framework)
+        if smart_structure:
+            console.print("  [green]Smart structure generated![/green]")
+        else:
+            console.print("  [yellow]Could not reach API, using default structure.[/yellow]")
+
+    console.print(f"\n  Framework       : [cyan]{framework}[/cyan]")
+    console.print(f"  LLM Provider    : [cyan]{llm}[/cyan]")
+    console.print(f"  Package Manager : [cyan]{package_manager}[/cyan]\n")
+
+    success = generate_project(project_name, framework, llm, package_manager, smart_structure)
+
+    if not success:
+        raise click.Abort()
+
+    console.print(f"\n[bold green]Done![/bold green] Your project is ready at [cyan]./{project_name}[/cyan]")
+    console.print("\nNext steps:")
+    console.print(f"  [yellow]cd {project_name}[/yellow]")
+    console.print("  [yellow]cp .env.example .env[/yellow]")
+
+    if package_manager == "uv":
+        console.print("  [yellow]uv sync[/yellow]")
+        console.print("  [yellow]uv run python main.py[/yellow]\n")
+    else:
+        console.print("  [yellow]pip install -r requirements.txt[/yellow]")
+        console.print("  [yellow]python main.py[/yellow]\n")
+
+
+@app.command("add-agent")
+@click.argument("agent_name")
+@click.option("--framework", "-f", required=True, help="Framework of the existing project")
+def add_agent_cmd(agent_name, framework):
+    """Add a new agent to an existing scaffolded project."""
+    if framework not in SUPPORTED_FRAMEWORKS:
+        console.print(f"[red]Framework '{framework}' is not supported.[/red]")
+        console.print(f"Supported: {', '.join(SUPPORTED_FRAMEWORKS)}")
+        raise click.Abort()
+
+    console.print(f"\n[bold green]Adding agent:[/bold green] {agent_name}")
+    console.print(f"  Framework : [cyan]{framework}[/cyan]\n")
+
+    success = add_agent(agent_name, framework)
+
+    if not success:
+        raise click.Abort()
+
+    console.print(f"\n[bold green]Done![/bold green] Agent [cyan]agents/{agent_name}.py[/cyan] created.\n")
+
+
+@app.command("list-frameworks")
+def list_frameworks():
+    """List all supported frameworks and LLM providers."""
+    table = Table(title="Supported Frameworks & LLM Providers")
+    table.add_column("Frameworks", style="cyan")
+    table.add_column("LLM Providers", style="green")
+
+    rows = list(zip(SUPPORTED_FRAMEWORKS, SUPPORTED_LLMS))
+    for fw, llm in rows:
+        table.add_row(fw, llm)
+
+    console.print(table)    

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

@@ -17,7 +17,6 @@ LLM_CONFIG = {
 
 FRAMEWORK_FILES = [
     ("main.py.j2",              "main.py"),
-    ("requirements.txt.j2",     "requirements.txt"),
     ("agents/base_agent.py.j2", "agents/base_agent.py"),
     ("tools/sample_tool.py.j2", "tools/sample_tool.py"),
 ]
@@ -35,7 +34,27 @@ def write_file(output_path: Path, content: str):
     console.print(f"  [green]created[/green] {output_path}")
 
 
-def generate_project(project_name: str, framework: str, llm: str) -> bool:
+def write_stub(output_path: Path, name: str, kind: str):
+    """Write a clean stub file with just a docstring and pass."""
+    class_name = "".join(word.capitalize() for word in name.split("_"))
+    if kind == "agent":
+        content = f'"""\n{class_name} agent stub.\nReplace this with your agent logic.\n"""\n\n\nclass {class_name}:\n    """Agent: {name}"""\n\n    def run(self, *args, **kwargs):\n        raise NotImplementedError\n'
+    elif kind == "tool":
+        content = f'"""\n{class_name} tool stub.\nReplace this with your tool logic.\n"""\n\n\ndef {name}(*args, **kwargs):\n    """{class_name} tool. Replace with real logic."""\n    raise NotImplementedError\n'
+    elif kind == "test":
+        content = f'"""\nTests for {name}.\n"""\n\n\ndef {name}():\n    """TODO: implement test"""\n    pass\n'
+    else:
+        content = f'"""\n{class_name} stub.\nReplace this with your logic.\n"""\n\n\nclass {class_name}:\n    """Module: {name}"""\n    pass\n'
+    write_file(output_path, content)
+
+
+def generate_project(
+    project_name: str,
+    framework: str,
+    llm: str,
+    package_manager: str = "pip",
+    smart_structure: dict | None = None,
+) -> bool:
     project_path = Path(project_name)
 
     if project_path.exists():
@@ -46,14 +65,24 @@ def generate_project(project_name: str, framework: str, llm: str) -> bool:
         "project_name": project_name,
         "framework": framework,
         "llm_provider": llm,
+        "package_manager": package_manager,
         **LLM_CONFIG[llm],
     }
 
     fw_env = Environment(loader=FileSystemLoader(str(TEMPLATES_DIR / framework)))
+
     for template_file, output_file in FRAMEWORK_FILES:
         template = fw_env.get_template(template_file)
         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))
+    else:
+        common_env = Environment(loader=FileSystemLoader(str(TEMPLATES_DIR / "common")))
+        template = common_env.get_template("pyproject.toml.j2")
+        write_file(project_path / "pyproject.toml", template.render(**context))
+
     common_env = Environment(loader=FileSystemLoader(str(TEMPLATES_DIR / "common")))
     for template_file, output_file in COMMON_FILES:
         template = common_env.get_template(template_file)
@@ -61,6 +90,23 @@ def generate_project(project_name: str, framework: str, llm: str) -> bool:
 
     env_content = build_env_file(llm)
     write_file(project_path / ".env.example", env_content)
+
+    if smart_structure:
+        for agent in smart_structure.get("agents", []):
+            write_stub(project_path / "agents" / f"{agent}.py", agent, "agent")
+
+        for tool in smart_structure.get("tools", []):
+            write_stub(project_path / "tools" / f"{tool}.py", tool, "tool")
+
+        for item in smart_structure.get("core", []):
+            write_stub(project_path / "core" / f"{item}.py", item, "core")
+
+        for item in smart_structure.get("utils", []):
+            write_stub(project_path / "utils" / f"{item}.py", item, "util")
+
+        for test in smart_structure.get("tests", []):
+            write_stub(project_path / "tests" / f"{test}.py", test, "test")
+
     return True
 
 
@@ -91,13 +137,13 @@ def add_agent(agent_name: str, framework: str) -> bool:
         console.print(f"[red]Agent '{agent_name}.py' already exists. Aborting.[/red]")
         return False
 
-    template_path = TEMPLATES_DIR / framework / "agents" / "base_agent.py.j2"
-    if not template_path.exists():
-        console.print(f"[red]No template found for framework '{framework}'.[/red]")
-        return False
-
     env = Environment(loader=FileSystemLoader(str(TEMPLATES_DIR / framework / "agents")))
     template = env.get_template("base_agent.py.j2")
-    content = template.render(project_name=agent_name, framework=framework, llm_provider="openai", **LLM_CONFIG["openai"])
+    content = template.render(
+        project_name=agent_name,
+        framework=framework,
+        llm_provider="openai",
+        **LLM_CONFIG["openai"]
+    )
     write_file(output_path, content)
     return True

agentinit_cli-0.3.0/agentinit/templates/common/pyproject.toml.j2

@@ -0,0 +1,54 @@
+[project]
+name = "{{ project_name }}"
+version = "0.1.0"
+description = "A {{ framework }} agent project scaffolded with agentinit"
+requires-python = ">=3.10"
+dependencies = [
+{% if framework == "langgraph" %}
+    "langgraph",
+    {% if llm_provider == "openai" %}"openai", "langchain-openai",{% endif %}
+    {% if llm_provider == "anthropic" %}"anthropic", "langchain-anthropic",{% endif %}
+    {% if llm_provider == "groq" %}"groq", "langchain-groq",{% endif %}
+    {% if llm_provider == "azure" %}"openai", "langchain-openai", "azure-identity",{% endif %}
+    {% if llm_provider == "bedrock" %}"boto3", "anthropic[bedrock]",{% endif %}
+    {% if llm_provider == "gemini" %}"google-generativeai",{% endif %}
+{% elif framework == "crewai" %}
+    "crewai",
+    {% if llm_provider == "openai" %}"openai",{% endif %}
+    {% if llm_provider == "anthropic" %}"anthropic",{% endif %}
+    {% if llm_provider == "groq" %}"groq",{% endif %}
+    {% if llm_provider == "azure" %}"openai", "azure-identity",{% endif %}
+    {% if llm_provider == "bedrock" %}"boto3", "anthropic[bedrock]",{% endif %}
+    {% if llm_provider == "gemini" %}"google-generativeai",{% endif %}
+{% elif framework == "autogen" %}
+    "pyautogen",
+    {% if llm_provider == "openai" %}"openai",{% endif %}
+    {% if llm_provider == "anthropic" %}"anthropic",{% endif %}
+    {% if llm_provider == "groq" %}"groq",{% endif %}
+    {% if llm_provider == "azure" %}"openai", "azure-identity",{% endif %}
+    {% if llm_provider == "bedrock" %}"boto3", "anthropic[bedrock]",{% endif %}
+    {% if llm_provider == "gemini" %}"google-generativeai",{% endif %}
+{% elif framework == "google_adk" %}
+    "google-adk",
+    {% if llm_provider == "openai" %}"openai",{% endif %}
+    {% if llm_provider == "anthropic" %}"anthropic",{% endif %}
+    {% if llm_provider == "groq" %}"groq",{% endif %}
+    {% if llm_provider == "azure" %}"openai", "azure-identity",{% endif %}
+    {% if llm_provider == "bedrock" %}"boto3", "anthropic[bedrock]",{% endif %}
+    {% if llm_provider == "gemini" %}"google-generativeai",{% endif %}
+{% elif framework == "openai_agents" %}
+    "openai-agents",
+    {% if llm_provider == "azure" %}"azure-identity",{% endif %}
+    {% if llm_provider == "bedrock" %}"boto3", "anthropic[bedrock]",{% endif %}
+    {% if llm_provider == "gemini" %}"google-generativeai",{% endif %}
+{% elif framework == "smolagents" %}
+    "smolagents",
+    {% if llm_provider == "openai" %}"openai",{% endif %}
+    {% if llm_provider == "anthropic" %}"anthropic",{% endif %}
+    {% if llm_provider == "groq" %}"groq",{% endif %}
+    {% if llm_provider == "azure" %}"openai", "azure-identity",{% endif %}
+    {% if llm_provider == "bedrock" %}"boto3", "anthropic[bedrock]",{% endif %}
+    {% if llm_provider == "gemini" %}"google-generativeai",{% endif %}
+{% endif %}
+    "python-dotenv",
+]

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agentinit-cli"
-version = "0.1.0"
+version = "0.3.0"
 description = "A CLI tool to scaffold production-ready LLM agent projects"
 authors = ["Swapnil Bhattacharya <your@email.com>"]
 readme = "README.md"
@@ -20,7 +20,9 @@ packages = [{include = "agentinit"}]
 python = "^3.10"
 click = "^8.1.0"
 jinja2 = "^3.1.0"
-rich = "^13.0.0"
+rich = ">=13.0.0"
+questionary = "^2.0.0"
+requests = "^2.31.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.0.0"
@@ -30,4 +32,4 @@ agentinit = "agentinit.cli:app"
 
 [build-system]
 requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"   
+build-backend = "poetry.core.masonry.api"

agentinit_cli-0.1.0/agentinit/cli.py

@@ -1,83 +0,0 @@
-import click
-from rich.console import Console
-from rich.table import Table
-from agentinit.generator import generate_project, add_agent
-
-console = Console()
-
-SUPPORTED_FRAMEWORKS = ["langgraph", "crewai", "autogen", "google_adk", "openai_agents", "smolagents"]
-SUPPORTED_LLMS = ["openai", "anthropic", "groq", "azure", "bedrock", "gemini"]
-
-
-@click.group()
-def app():
-    """agentinit - scaffold LLM agent projects in seconds"""
-    pass
-
-
-@app.command("init")
-@click.argument("project_name")
-@click.option("--framework", "-f", required=True, help="Agent framework to use")
-@click.option("--llm", "-l", required=True, help="LLM provider to use")
-def init(project_name, framework, llm):
-    """Scaffold a new LLM agent project."""
-    if framework not in SUPPORTED_FRAMEWORKS:
-        console.print(f"[red]Framework '{framework}' is not supported.[/red]")
-        console.print(f"Supported: {', '.join(SUPPORTED_FRAMEWORKS)}")
-        raise click.Abort()
-
-    if llm not in SUPPORTED_LLMS:
-        console.print(f"[red]LLM provider '{llm}' is not supported.[/red]")
-        console.print(f"Supported: {', '.join(SUPPORTED_LLMS)}")
-        raise click.Abort()
-
-    console.print(f"\n[bold green]Scaffolding project:[/bold green] {project_name}")
-    console.print(f"  Framework : [cyan]{framework}[/cyan]")
-    console.print(f"  LLM       : [cyan]{llm}[/cyan]\n")
-
-    success = generate_project(project_name, framework, llm)
-
-    if not success:
-        raise click.Abort()
-
-    console.print(f"\n[bold green]Done![/bold green] Your project is ready at [cyan]./{project_name}[/cyan]")
-    console.print("\nNext steps:")
-    console.print(f"  [yellow]cd {project_name}[/yellow]")
-    console.print("  [yellow]cp .env.example .env[/yellow]")
-    console.print("  [yellow]pip install -r requirements.txt[/yellow]")
-    console.print("  [yellow]python main.py[/yellow]\n")
-
-
-@app.command("add-agent")
-@click.argument("agent_name")
-@click.option("--framework", "-f", required=True, help="Framework of the existing project")
-def add_agent_cmd(agent_name, framework):
-    """Add a new agent to an existing scaffolded project."""
-    if framework not in SUPPORTED_FRAMEWORKS:
-        console.print(f"[red]Framework '{framework}' is not supported.[/red]")
-        console.print(f"Supported: {', '.join(SUPPORTED_FRAMEWORKS)}")
-        raise click.Abort()
-
-    console.print(f"\n[bold green]Adding agent:[/bold green] {agent_name}")
-    console.print(f"  Framework : [cyan]{framework}[/cyan]\n")
-
-    success = add_agent(agent_name, framework)
-
-    if not success:
-        raise click.Abort()
-
-    console.print(f"\n[bold green]Done![/bold green] Agent [cyan]agents/{agent_name}.py[/cyan] created.\n")
-
-
-@app.command("list-frameworks")
-def list_frameworks():
-    """List all supported frameworks and LLM providers."""
-    table = Table(title="Supported Frameworks & LLM Providers")
-    table.add_column("Frameworks", style="cyan")
-    table.add_column("LLM Providers", style="green")
-
-    rows = list(zip(SUPPORTED_FRAMEWORKS, SUPPORTED_LLMS))
-    for fw, llm in rows:
-        table.add_row(fw, llm)
-
-    console.print(table)