Want to build your own software with code generation abilities? CodeChain is a library for generating code with LLMs. Here's how to get started quickly in Python:
pip install codechain openai python-dotenv
Add your OpenAI API key to an environment variable:
echo "OPENAI_API_KEY=sk-XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX" > .env
Create a Python script to run the generation:
from codechain.generation import CompleteCodeChain from langchain.chat_models import ChatOpenAI from dotenv import load_dotenv # Load API key load_dotenv() # Setup the generator generator = ModifyCodeChain.from_instruction( "Return a complete and correct version of the given code.", ChatOpenAI(model="gpt-3.5-turbo", temperature=0.2) ) # Generate result = generator.run(""" def fibonacci(n): # Generate the n-th fibonacci number. """) print(result)
CodeChain is composable with LangChain, thus has support for all Langchain chat models including OpenAI, Anthropic, Azure, Palm, and Llama.