Kani: A Lightweight and Highly Hackable Framework for Building Language Model Applications

Published: 09 Oct 2023, Last Modified: 20 Oct 2023NLP-OSS 2023EveryoneRevisionsBibTeX
Keywords: tools, library, framework, chatgpt, functions, llama, open-source
TL;DR: We introduce Kani, a lightweight and highly hackable open-source software package for building language model applications.
Abstract: Language model applications are becoming increasingly popular and complex, often including features like tool usage and retrieval augmentation. However, existing frameworks for such applications are often opinionated, deciding for developers how their prompts ought to be formatted and imposing limitations on customizability and reproducibility. To solve this we present Kani: a lightweight, flexible, and model-agnostic open-source framework for building language model applications. Kani helps developers implement a variety of complex features by supporting the core building blocks of chat interaction: model interfacing, chat management, and robust function calling. All Kani core functions are easily overridable and well documented to empower developers to customize functionality for their own needs. Kani thus serves as a useful tool for researchers, hobbyists, and industry professionals alike to accelerate their development while retaining interoperability and fine-grained control.
Submission Number: 12