.

Author: jayhack

src/codegen/cli/api/client.py

@@ -40,7 +40,13 @@
     RunOnPRResponse,
 )
 from codegen.cli.auth.session import CodegenSession
-from codegen.cli.codemod.convert import convert_to_ui
+
+
+def convert_to_ui(data):
+    """Simple stub for convert_to_ui function."""
+    return data
+
+
 from codegen.cli.env.global_env import global_env
 from codegen.cli.errors import InvalidTokenError, ServerError
 from codegen.cli.utils.codemods import Codemod

src/codegen/cli/auth/session.py

@@ -5,7 +5,6 @@
 from github import BadCredentialsException
 from github.MainClass import Github
 
-from codegen.cli.git.repo import get_git_repo
 from codegen.cli.rich.codeblocks import format_command
 from codegen.configs.constants import CODEGEN_DIR_NAME, ENV_FILENAME
 from codegen.configs.session_manager import session_manager
@@ -23,8 +22,15 @@ class CodegenSession:
     existing: bool
 
     def __init__(self, repo_path: Path, git_token: str | None = None) -> None:
-        if not repo_path.exists() or get_git_repo(repo_path) is None:
-            rich.print(f"\n[bold red]Error:[/bold red] Path to git repo does not exist at {self.repo_path}")
+        if not repo_path.exists():
+            rich.print(f"\n[bold red]Error:[/bold red] Path to git repo does not exist at {repo_path}")
+            raise click.Abort()
+
+        # Check if it's a valid git repository
+        try:
+            LocalGitRepo(repo_path=repo_path)
+        except Exception:
+            rich.print(f"\n[bold red]Error:[/bold red] Path {repo_path} is not a valid git repository")
             raise click.Abort()
 
         self.repo_path = repo_path

src/codegen/cli/commands/init/main.py

@@ -5,9 +5,7 @@
 import rich_click as click
 
 from codegen.cli.auth.session import CodegenSession
-from codegen.cli.commands.init.render import get_success_message
 from codegen.cli.rich.codeblocks import format_command
-from codegen.cli.workspace.initialize_workspace import initialize_codegen
 from codegen.shared.path import get_git_root_path
 
 
@@ -37,11 +35,14 @@ def init_command(path: str | None = None, token: str | None = None, language: st
         session.config.save()
 
     action = "Updating" if session.existing else "Initializing"
-    codegen_dir, docs_dir, examples_dir = initialize_codegen(status=action, session=session, fetch_docs=fetch_docs)
+
+    # Create the codegen directory
+    codegen_dir = session.codegen_dir
+    codegen_dir.mkdir(parents=True, exist_ok=True)
 
     # Print success message
     rich.print(f"✅ {action} complete\n")
-    rich.print(get_success_message(codegen_dir, docs_dir, examples_dir))
+    rich.print(f"Codegen workspace initialized at: [bold]{codegen_dir}[/bold]")
 
     # Print next steps
     rich.print("\n[bold]What's next?[/bold]\n")

src/codegen/cli/commands/profile/main.py

@@ -5,7 +5,13 @@
 
 from codegen.cli.auth.decorators import requires_auth
 from codegen.cli.auth.session import CodegenSession
-from codegen.cli.workspace.decorators import requires_init
+
+# from codegen.cli.workspace.decorators import requires_init  # Removed to simplify CLI
+
+
+def requires_init(func):
+    """Simple stub decorator that does nothing."""
+    return func
 
 
 @click.command(name="profile")

tests/unit/codegen/test_cli_basic.py

@@ -0,0 +1,78 @@
+"""Basic CLI tests to verify core functionality."""
+
+import subprocess
+import sys
+from pathlib import Path
+
+
+def test_cli_help_works():
+    """Test that the CLI can show help without credentials."""
+    # Run the CLI help command
+    result = subprocess.run(
+        [sys.executable, "-m", "codegen.cli.cli", "--help"],
+        capture_output=True,
+        text=True,
+        cwd=Path(__file__).parent.parent.parent.parent,  # Go to project root
+    )
+
+    # Should exit with code 0 (success)
+    assert result.returncode == 0
+
+    # Should contain basic help text
+    assert "Codegen CLI - Transform your code with AI" in result.stdout
+    assert "Commands" in result.stdout
+    assert "init" in result.stdout
+    assert "login" in result.stdout
+    assert "profile" in result.stdout
+
+
+def test_cli_version_works():
+    """Test that the CLI can show version without credentials."""
+    result = subprocess.run(
+        [sys.executable, "-m", "codegen.cli.cli", "--version"],
+        capture_output=True,
+        text=True,
+        cwd=Path(__file__).parent.parent.parent.parent,  # Go to project root
+    )
+
+    # Should exit with code 0 (success)
+    assert result.returncode == 0
+
+    # Should show some version information
+    assert len(result.stdout.strip()) > 0
+
+
+def test_cli_command_help():
+    """Test that individual commands can show help."""
+    commands = ["init", "login", "logout", "profile", "config", "update"]
+
+    for command in commands:
+        result = subprocess.run(
+            [sys.executable, "-m", "codegen.cli.cli", command, "--help"],
+            capture_output=True,
+            text=True,
+            cwd=Path(__file__).parent.parent.parent.parent,  # Go to project root
+        )
+
+        # Should exit with code 0 (success) for help
+        assert result.returncode == 0, f"Command '{command} --help' failed with code {result.returncode}"
+
+        # Should contain usage information
+        assert "Usage:" in result.stdout, f"Command '{command} --help' doesn't show usage"
+
+
+def test_cli_imports_work():
+    """Test that we can import the CLI module without errors."""
+    # This test verifies that all imports in the CLI work
+    try:
+        from codegen.cli.cli import main
+
+        assert main is not None
+
+        # Test Agent import still works
+        from codegen.agents.agent import Agent
+
+        assert Agent is not None
+
+    except ImportError as e:
+        assert False, f"Failed to import CLI modules: {e}"