Add Agent class to top-level imports

This change allows importing the Agent class directly from the codegen package:
from codegen import Agent

Also adds a unit test to verify this functionality.

Co-authored-by: Jay Hack <[email protected]>

Author: codegen-sh[bot]

src/codegen/__init__.py

@@ -1,7 +1,7 @@
 # file generated by setuptools-scm
 # don't change, don't track in version control
 
-__all__ = ["__version__", "__version_tuple__", "version", "version_tuple"]
+__all__ = ["__version__", "__version_tuple__", "version", "version_tuple", "Agent"]
 
 TYPE_CHECKING = False
 if TYPE_CHECKING:
@@ -18,3 +18,6 @@
 
 __version__ = version = "0.55.8.dev50+gd8c9a1fa"
 __version_tuple__ = version_tuple = (0, 55, 8, "dev50", "gd8c9a1fa")
+
+# Import Agent class for top-level access
+from codegen.agents.agent import Agent

tests/unit/codegen/test_imports.py

@@ -0,0 +1,16 @@
+"""Tests for top-level imports in the codegen package."""
+
+
+class TestTopLevelImports:
+    """Test that we can properly import classes from the top level."""
+
+    def test_can_import_agent_from_top_level(self):
+        """Test that we can import Agent directly from codegen."""
+        from codegen import Agent
+
+        assert Agent is not None
+        
+        # Verify it's the same class as the one in codegen.agents.agent
+        from codegen.agents.agent import Agent as AgentFromModule
+        assert Agent is AgentFromModule
+