From 23ae99241d3678cf17424c0d68cb37c439555320 Mon Sep 17 00:00:00 2001
From: "codegen-sh[bot]" <131295404+codegen-sh[bot]@users.noreply.github.com>
Date: Tue, 25 Mar 2025 15:31:52 +0000
Subject: [PATCH] Fix: Remove reference to non-existent agent module in CLI

---
 src/codegen/cli/cli.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/codegen/cli/cli.py b/src/codegen/cli/cli.py
index 9ac7f69f4..8910f615a 100644
--- a/src/codegen/cli/cli.py
+++ b/src/codegen/cli/cli.py
@@ -1,7 +1,7 @@
 import rich_click as click
 from rich.traceback import install
 
-from codegen.cli.commands.agent.main import agent_command
+# Removed reference to non-existent agent module
 from codegen.cli.commands.config.main import config_command
 from codegen.cli.commands.create.main import create_command
 from codegen.cli.commands.deploy.main import deploy_command
@@ -32,7 +32,7 @@ def main():
 
 
 # Wrap commands with error handler
-main.add_command(agent_command)
+# Removed reference to non-existent agent_command
 main.add_command(init_command)
 main.add_command(logout_command)
 main.add_command(login_command)