The Codegen SDK provides a programmatic interface to code agents provided by Codegen.
from codegen.agents.agent import Agent
# Initialize the Agent with your organization ID and API token
agent = Agent(
org_id="YOUR_ORG_ID", # Find this at codegen.com/developer
token="YOUR_API_TOKEN", # Get this from codegen.com/developer
# base_url="https://codegen-sh-rest-api.modal.run", # Optional - defaults to production
)
# Run an agent with a prompt
task = agent.run(prompt="Implement a new feature to sort users by last login.")
# Check the initial status
print(task.status)
# Refresh the task to get updated status (tasks can take time)
task.refresh()
# Check the updated status
print(task.status)
# Once task is complete, you can access the result
if task.status == "completed":
print(task.result) # Result often contains code, summaries, or links
Install the SDK using pip or uv:
pip install codegen
# or
uv pip install codegen
Get started at codegen.com and get your API token at codegen.com/developer.
You can interact with your AI engineer via API, or chat with it in Slack, Linear, Github, or on our website.
Please see our Contributing Guide for instructions on how to set up the development environment and submit contributions.
For more information on enterprise engagements, please contact us or request a demo.