feat: from codegen import Codebase (#55)

# Motivation
<!-- Why is this change necessary? -->

# Content
<!-- Please include a summary of the change -->
# Testing
<!-- How was the change tested? -->
# Please check the following before marking your PR as ready for review

- [ ] I have added tests for my changes
- [ ] I have updated the documentation or added new documentation as
needed
- [ ] I have read and agree to the [Contributor License
Agreement](../CLA.md)

Author: jayhack

README.md

@@ -1,7 +1,6 @@
 # Codegen
 
 [![Documentation](https://img.shields.io/badge/docs-docs.codegen.com-blue)](https://docs.codegen.com)
-[![Unit Tests](https://github.com/codegen-sh/codegen-sdk/actions/workflows/unit-tests.yml/badge.svg)](https://github.com/codegen-sh/codegen-sdk/actions/workflows/unit-tests.yml)
 
 [Codegen](https://docs.codegen.com) is a python library for manipulating codebases.
 
@@ -42,7 +41,7 @@ We built Codegen backwards from real-world refactors performed on enterprise cod
 
 - **Natural mental model**: Write transforms that read like your thought process - "move this function", "rename this variable", "add this parameter". No more wrestling with ASTs or manual import management.
 
-- **Battle-tested on complex codebases**: Handle Python, TypeScript, and React codebases with millions of lines of code. Built and validated on refactors at companies like [Ramp](https://ramp.com).
+- **Battle-tested on complex codebases**: Handle Python, TypeScript, and React codebases with millions of lines of code.
 
 - **Built for advanced intelligences**: As AI developers become more sophisticated, they need expressive yet precise tools to manipulate code. Codegen provides a programmatic interface that both humans and AI can use to express complex transformations through code itself.
 

src/codegen/__init__.py

@@ -0,0 +1,5 @@
+from codegen.cli.sdk.decorator import function
+from codegen.cli.sdk.functions import Function
+from codegen.sdk.core.codebase import Codebase
+
+__all__ = ["Codebase", "Function", "function"]

tests/unit/test_imports.py

@@ -0,0 +1,16 @@
+import codegen
+from codegen import Codebase
+
+
+def test_codegen_imports():
+    # Test decorated function
+    @codegen.function(name="sample_codemod")
+    def run(codebase):
+        pass
+
+    # Test class
+    cls = codegen.Function
+    assert cls is not None
+
+    codebase = Codebase("./")
+    assert codebase is not None