From db830bac08c08389605de8eb7aa515ce5d8c6730 Mon Sep 17 00:00:00 2001
From: "codegen-sh[bot]" <131295404+codegen-sh[bot]@users.noreply.github.com>
Date: Tue, 8 Apr 2025 19:03:42 +0000
Subject: [PATCH] Update documentation to highlight top-level Agent import

---
 README.md                    | 5 ++++-
 src/codegen/agents/README.md | 6 +++++-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index f9e94756b..d05448026 100644
--- a/README.md
+++ b/README.md
@@ -75,8 +75,11 @@ codegen notebook
 
 See [Getting Started](https://docs.codegen.com/introduction/getting-started) for a full tutorial.
 
-```
+```python
 from codegen import Codebase
+
+# For AI agent functionality
+from codegen import Agent
 ```
 
 ## Troubleshooting
diff --git a/src/codegen/agents/README.md b/src/codegen/agents/README.md
index 254ed4bc9..8734ba077 100644
--- a/src/codegen/agents/README.md
+++ b/src/codegen/agents/README.md
@@ -15,7 +15,11 @@ pip install codegen
 ### Basic Example
 
 ```python
-from codegen.agents.agent import Agent
+# You can import Agent directly from the top-level package
+from codegen import Agent
+
+# Or use the full import path
+# from codegen.agents.agent import Agent
 
 # Initialize the Agent with your organization ID and API token
 agent = Agent(