diff --git a/README.md b/README.md index e3cf81d..c34eff6 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,5 @@ +
+ # kotaemon An open-source clean & customizable RAG UI for chatting with your documents. Built with both end users and @@ -5,12 +7,14 @@ developers in mind. ![Preview](https://raw.githubusercontent.com/Cinnamon/kotaemon/main/docs/images/preview-graph.png) -[Live Demo](https://huggingface.co/spaces/cin-model/kotaemon-demo) | -[Source Code](https://github.com/Cinnamon/kotaemon) +Cinnamon%2Fkotaemon | Trendshift +[Live Demo](https://huggingface.co/spaces/cin-model/kotaemon-demo) | +[Online Install](https://cinnamon.github.io/kotaemon/online_install/) | [User Guide](https://cinnamon.github.io/kotaemon/) | [Developer Guide](https://cinnamon.github.io/kotaemon/development/) | -[Feedback](https://github.com/Cinnamon/kotaemon/issues) +[Feedback](https://github.com/Cinnamon/kotaemon/issues) | +[Contact](mailto:kotaemon.support@cinnamon.is) [![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/release/python-31013/) [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black) @@ -20,22 +24,13 @@ developers in mind. Featured|HelloGitHub -Cinnamon%2Fkotaemon | Trendshift +
## Introduction This project serves as a functional RAG UI for both end users who want to do QA on their documents and developers who want to build their own RAG pipeline. - -- For end users: - - A clean & minimalistic UI for RAG-based QA. - - Supports LLM API providers (OpenAI, AzureOpenAI, Cohere, etc) and local LLMs - (via `ollama` and `llama-cpp-python`). - - Easy installation scripts. -- For developers: - - A framework for building your own RAG-based document QA pipeline. - - Customize and see your RAG pipeline in action with the provided UI (built with Gradio ). - - If you use Gradio for development, check out our theme here: [kotaemon-gradio-theme](https://github.com/lone17/kotaemon-gradio-theme). +
```yml +----------------------------------------------------------------------------+ @@ -52,162 +47,163 @@ documents and developers who want to build their own RAG pipeline. +----------------------------------------------------------------------------+ ``` -This repository is under active development. Feedback, issues, and PRs are highly -appreciated. +### For end users + +- **Clean & Minimalistic UI**: A user-friendly interface for RAG-based QA. +- **Support for Various LLMs**: Compatible with LLM API providers (OpenAI, AzureOpenAI, Cohere, etc.) and local LLMs (via `ollama` and `llama-cpp-python`). +- **Easy Installation**: Simple scripts to get you started quickly. + +### For developers + +- **Framework for RAG Pipelines**: Tools to build your own RAG-based document QA pipeline. +- **Customizable UI**: See your RAG pipeline in action with the provided UI, built with Gradio . +- **Gradio Theme**: If you use Gradio for development, check out our theme here: [kotaemon-gradio-theme](https://github.com/lone17/kotaemon-gradio-theme). ## Key Features -- **Host your own document QA (RAG) web-UI**. Support multi-user login, organize your files in private / public collections, collaborate and share your favorite chat with others. +- **Host your own document QA (RAG) web-UI**: Support multi-user login, organize your files in private/public collections, collaborate and share your favorite chat with others. -- **Organize your LLM & Embedding models**. Support both local LLMs & popular API providers (OpenAI, Azure, Ollama, Groq). +- **Organize your LLM & Embedding models**: Support both local LLMs & popular API providers (OpenAI, Azure, Ollama, Groq). -- **Hybrid RAG pipeline**. Sane default RAG pipeline with hybrid (full-text & vector) retriever + re-ranking to ensure best retrieval quality. +- **Hybrid RAG pipeline**: Sane default RAG pipeline with hybrid (full-text & vector) retriever and re-ranking to ensure best retrieval quality. -- **Multi-modal QA support**. Perform Question Answering on multiple documents with figures & tables support. Support multi-modal document parsing (selectable options on UI). +- **Multi-modal QA support**: Perform Question Answering on multiple documents with figures and tables support. Support multi-modal document parsing (selectable options on UI). -- **Advance citations with document preview**. By default the system will provide detailed citations to ensure the correctness of LLM answers. View your citations (incl. relevant score) directly in the _in-browser PDF viewer_ with highlights. Warning when retrieval pipeline return low relevant articles. +- **Advanced citations with document preview**: By default the system will provide detailed citations to ensure the correctness of LLM answers. View your citations (incl. relevant score) directly in the _in-browser PDF viewer_ with highlights. Warning when retrieval pipeline return low relevant articles. -- **Support complex reasoning methods**. Use question decomposition to answer your complex / multi-hop question. Support agent-based reasoning with ReAct, ReWOO and other agents. +- **Support complex reasoning methods**: Use question decomposition to answer your complex/multi-hop question. Support agent-based reasoning with `ReAct`, `ReWOO` and other agents. -- **Configurable settings UI**. You can adjust most important aspects of retrieval & generation process on the UI (incl. prompts). +- **Configurable settings UI**: You can adjust most important aspects of retrieval & generation process on the UI (incl. prompts). -- **Extensible**. Being built on Gradio, you are free to customize / add any UI elements as you like. Also, we aim to support multiple strategies for document indexing & retrieval. `GraphRAG` indexing pipeline is provided as an example. +- **Extensible**: Being built on Gradio, you are free to customize or add any UI elements as you like. Also, we aim to support multiple strategies for document indexing & retrieval. `GraphRAG` indexing pipeline is provided as an example. ![Preview](https://raw.githubusercontent.com/Cinnamon/kotaemon/main/docs/images/preview.png) ## Installation -### For end users +> If you are not a developer and just want to use the app, please check out our easy-to-follow [User Guide](https://cinnamon.github.io/kotaemon/). Download the `.zip` file from the [latest release](https://github.com/Cinnamon/kotaemon/releases/latest) to get all the newest features and bug fixes. -This document is intended for developers. If you just want to install and use the app as -it is, please follow the non-technical [User Guide](https://cinnamon.github.io/kotaemon/). -Use the most recent release `.zip` to include latest features and bug-fixes. +### System requirements -### For developers +1. [Python](https://www.python.org/downloads/) >= 3.10 +2. [Docker](https://www.docker.com/): optional, if you [install with Docker](#with-docker-recommended) +3. [Unstructured](https://docs.unstructured.io/open-source/installation/full-installation#full-installation) if you want to process files other than `.pdf`, `.html`, `.mhtml`, and `.xlsx` documents. Installation steps differ depending on your operating system. Please visit the link and follow the specific instructions provided there. -#### System requirements +### With Docker (recommended) -1. Python >=3.10 -2. (optional) [Docker](https://www.docker.com/) +1. We support both `lite` & `full` version of Docker images. With `full`, the extra packages of `unstructured` will be installed as well, it can support additional file types (`.doc`, `.docx`, ...) but the cost is larger docker image size. For most users, the `lite` image should work well in most cases. -##### If you would like to process files other than .pdf, .html, .mhtml, and .xlsx documents + - To use the `lite` version. -You will need to install the system dependencies of [unstructured](https://docs.unstructured.io/open-source/installation/full-installation#full-installation). The installations vary by operating system, so please go to the link and follow the instructions there. + ```bash + docker run \ + -e GRADIO_SERVER_NAME=0.0.0.0 \ + -e GRADIO_SERVER_PORT=7860 \ + -p 7860:7860 -it --rm \ + ghcr.io/cinnamon/kotaemon:main-lite + ``` -#### With Docker (recommended) + - To use the `full` version. -We support both `lite` & `full` version of Docker images. With `full`, the extra packages of `unstructured` will be installed as -well, it can support additional file types (.doc, .docx, ...) but the cost is larger docker image size. For most users, the `lite` image should work well in most cases. + ```bash + docker run \ + -e GRADIO_SERVER_NAME=0.0.0.0 \ + -e GRADIO_SERVER_PORT=7860 \ + -p 7860:7860 -it --rm \ + ghcr.io/cinnamon/kotaemon:main-full + ``` -- To use the `lite` version. +2. We currently support and test two platforms: `linux/amd64` and `linux/arm64` (for newer Mac). You can specify the platform by passing `--platform` in the `docker run` command. For example: -``` -docker run \ --e GRADIO_SERVER_NAME=0.0.0.0 \ --e GRADIO_SERVER_PORT=7860 \ --p 7860:7860 -it --rm \ -ghcr.io/cinnamon/kotaemon:main-lite -``` + ```bash + # To run docker with platform linux/arm64 + docker run \ + -e GRADIO_SERVER_NAME=0.0.0.0 \ + -e GRADIO_SERVER_PORT=7860 \ + -p 7860:7860 -it --rm \ + --platform linux/arm64 \ + ghcr.io/cinnamon/kotaemon:main-lite + ``` -- To use the `full` version. +3. Once everything is set up correctly, you can go to `http://localhost:7860/` to access the WebUI. -``` -docker run \ --e GRADIO_SERVER_NAME=0.0.0.0 \ --e GRADIO_SERVER_PORT=7860 \ --p 7860:7860 -it --rm \ -ghcr.io/cinnamon/kotaemon:main-full -``` +4. We use [GHCR](https://docs.github.com/en/packages/working-with-a-github-packages-registry/working-with-the-container-registry) to store docker images, all images can be found [here.](https://github.com/Cinnamon/kotaemon/pkgs/container/kotaemon) -Currently, two platforms: `linux/amd64` and `linux/arm64` (for newer Mac) are provided & tested. User can specify the platform by passing `--platform` in the docker run command. For example: +### Without Docker -``` -# To run docker with platform linux/arm64 -docker run \ --e GRADIO_SERVER_NAME=0.0.0.0 \ --e GRADIO_SERVER_PORT=7860 \ --p 7860:7860 -it --rm \ ---platform linux/arm64 \ -ghcr.io/cinnamon/kotaemon:main-lite -``` +1. Clone and install required packages on a fresh python environment. -If everything is set up fine, navigate to `http://localhost:7860/` to access the web UI. + ```shell + # optional (setup env) + conda create -n kotaemon python=3.10 + conda activate kotaemon -We use [GHCR](https://docs.github.com/en/packages/working-with-a-github-packages-registry/working-with-the-container-registry) to store docker images, all images can be found [here.](https://github.com/Cinnamon/kotaemon/pkgs/container/kotaemon) + # clone this repo + git clone https://github.com/Cinnamon/kotaemon + cd kotaemon -#### Without Docker + pip install -e "libs/kotaemon[all]" + pip install -e "libs/ktem" + ``` -- Clone and install required packages on a fresh python environment. +2. Create a `.env` file in the root of this project. Use `.env.example` as a template -```shell -# optional (setup env) -conda create -n kotaemon python=3.10 -conda activate kotaemon + The `.env` file is there to serve use cases where users want to pre-config the models before starting up the app (e.g. deploy the app on HF hub). The file will only be used to populate the db once upon the first run, it will no longer be used in consequent runs. -# clone this repo -git clone https://github.com/Cinnamon/kotaemon -cd kotaemon - -pip install -e "libs/kotaemon[all]" -pip install -e "libs/ktem" - -``` - -- Create a .env file in the root of this project. Use .env.example as a template - -The .env file is there to serve use cases where users want to pre-config the models before starting up the app (e.g. deploy the app on HF hub). The file will only be used to populate the db once upon the first run, it will no longer be used in consequent runs. - -- (Optional) To enable in-browser PDF_JS viewer, download [PDF_JS_DIST](https://github.com/mozilla/pdf.js/releases/download/v4.0.379/pdfjs-4.0.379-dist.zip) and extract it to `libs/ktem/ktem/assets/prebuilt` +3. (Optional) To enable in-browser `PDF_JS` viewer, download [PDF_JS_DIST](https://github.com/mozilla/pdf.js/releases/download/v4.0.379/pdfjs-4.0.379-dist.zip) then extract it to `libs/ktem/ktem/assets/prebuilt` pdf-setup -- Start the web server: +4. Start the web server: -```shell -python app.py -``` + ```shell + python app.py + ``` -The app will be automatically launched in your browser. + - The app will be automatically launched in your browser. + - Default username and password are both `admin`. You can set up additional users directly through the UI. -Default username / password are: `admin` / `admin`. You can setup additional users directly on the UI. + ![Chat tab](https://raw.githubusercontent.com/Cinnamon/kotaemon/main/docs/images/chat-tab.png) -![Chat tab](https://raw.githubusercontent.com/Cinnamon/kotaemon/main/docs/images/chat-tab.png) +5. Check the `Resources` tab and `LLMs and Embeddings` and ensure that your `api_key` value is set correctly from your `.env` file. If it is not set, you can set it there. -- Check the Resources tab and LLMs and Embeddings and ensure that your `api_key` value is set correctly from your `.env`. file. If it is not set, you can set it here. - -## Setup GraphRAG +### Setup GraphRAG > [!NOTE] > Currently GraphRAG feature only works with OpenAI or Ollama API. -- [*If you are not using Docker installation*], install GraphRAG with `pip install graphrag future` -- To use GraphRAG retriever feature, make sure to set `GRAPHRAG_API_KEY` environment variables (or in the `.env` file). -- To use GraphRAG with local models (Ollama), or setup custom default LLM / other configurations, set `USE_CUSTOMIZED_GRAPHRAG_SETTING=true` and tweak your GraphRAG settings in `settings.yaml.example`. +- **Non-Docker Installation**: If you are not using Docker, install GraphRAG with the following command: -## Setup local models (for local / private RAG) + ```shell + pip install graphrag future + ``` + +- **Setting Up API KEY**: To use the GraphRAG retriever feature, ensure you set the `GRAPHRAG_API_KEY` environment variable. You can do this directly in your environment or by adding it to a `.env` file. +- **Using Local Models and Custom Settings**: If you want to use GraphRAG with local models (like `Ollama`) or customize the default LLM and other configurations, set the `USE_CUSTOMIZED_GRAPHRAG_SETTING` environment variable to true. Then, adjust your settings in the `settings.yaml.example` file. + +### Setup Local Models (for local/private RAG) See [Local model setup](docs/local_model.md). -## Customize your application +### Customize your application -By default, all application data are stored in `./ktem_app_data` folder. You can backup or copy this folder to move your installation to a new machine. +- By default, all application data is stored in the `./ktem_app_data` folder. You can back up or copy this folder to transfer your installation to a new machine. -For advance users or specific use-cases, you can customize those files: +- For advanced users or specific use cases, you can customize these files: -- `flowsettings.py` -- `.env` + - `flowsettings.py` + - `.env` -### `flowsettings.py` +#### `flowsettings.py` This file contains the configuration of your application. You can use the example -[here](flowsettings.py) as the -starting point. +[here](flowsettings.py) as the starting point.
Notable settings -``` +```python # setup your preferred document store (with full-text search capabilities) KH_DOCSTORE=(Elasticsearch | LanceDB | SimpleFileDocumentStore) @@ -224,109 +220,98 @@ KH_REASONINGS = [ "ktem.reasoning.react.ReactAgentPipeline", "ktem.reasoning.rewoo.RewooAgentPipeline", ] -) ```
-### `.env` +#### `.env` This file provides another way to configure your models and credentials. -
+
Configure model via the .env file -Alternatively, you can configure the models via the `.env` file with the information needed to connect to the LLMs. This file is located in -the folder of the application. If you don't see it, you can create one. +- Alternatively, you can configure the models via the `.env` file with the information needed to connect to the LLMs. This file is located in the folder of the application. If you don't see it, you can create one. -Currently, the following providers are supported: +- Currently, the following providers are supported: -#### OpenAI + - **OpenAI** -In the `.env` file, set the `OPENAI_API_KEY` variable with your OpenAI API key in order -to enable access to OpenAI's models. There are other variables that can be modified, -please feel free to edit them to fit your case. Otherwise, the default parameter should -work for most people. + In the `.env` file, set the `OPENAI_API_KEY` variable with your OpenAI API key in order + to enable access to OpenAI's models. There are other variables that can be modified, + please feel free to edit them to fit your case. Otherwise, the default parameter should + work for most people. -```shell -OPENAI_API_BASE=https://api.openai.com/v1 -OPENAI_API_KEY= -OPENAI_CHAT_MODEL=gpt-3.5-turbo -OPENAI_EMBEDDINGS_MODEL=text-embedding-ada-002 -``` + ```shell + OPENAI_API_BASE=https://api.openai.com/v1 + OPENAI_API_KEY= + OPENAI_CHAT_MODEL=gpt-3.5-turbo + OPENAI_EMBEDDINGS_MODEL=text-embedding-ada-002 + ``` -#### Azure OpenAI + - **Azure OpenAI** -For OpenAI models via Azure platform, you need to provide your Azure endpoint and API -key. Your might also need to provide your developments' name for the chat model and the -embedding model depending on how you set up Azure development. + For OpenAI models via Azure platform, you need to provide your Azure endpoint and API + key. Your might also need to provide your developments' name for the chat model and the + embedding model depending on how you set up Azure development. -```shell -AZURE_OPENAI_ENDPOINT= -AZURE_OPENAI_API_KEY= -OPENAI_API_VERSION=2024-02-15-preview -AZURE_OPENAI_CHAT_DEPLOYMENT=gpt-35-turbo -AZURE_OPENAI_EMBEDDINGS_DEPLOYMENT=text-embedding-ada-002 -``` + ```shell + AZURE_OPENAI_ENDPOINT= + AZURE_OPENAI_API_KEY= + OPENAI_API_VERSION=2024-02-15-preview + AZURE_OPENAI_CHAT_DEPLOYMENT=gpt-35-turbo + AZURE_OPENAI_EMBEDDINGS_DEPLOYMENT=text-embedding-ada-002 + ``` -#### Local models + - **Local Models** -##### Using ollama OpenAI compatible server + - Using `ollama` OpenAI compatible server: -Install [ollama](https://github.com/ollama/ollama) and start the application. + - Install [ollama](https://github.com/ollama/ollama) and start the application. -Pull your model (e.g): + - Pull your model, for example: -``` -ollama pull llama3.1:8b -ollama pull nomic-embed-text -``` + ```shell + ollama pull llama3.1:8b + ollama pull nomic-embed-text + ``` -Set the model names on web UI and make it as default. + - Set the model names on web UI and make it as default: -![Models](https://raw.githubusercontent.com/Cinnamon/kotaemon/main/docs/images/models.png) + ![Models](https://raw.githubusercontent.com/Cinnamon/kotaemon/main/docs/images/models.png) -##### Using GGUF with llama-cpp-python + - Using `GGUF` with `llama-cpp-python` -You can search and download a LLM to be ran locally from the [Hugging Face -Hub](https://huggingface.co/models). Currently, these model formats are supported: + You can search and download a LLM to be ran locally from the [Hugging Face Hub](https://huggingface.co/models). Currently, these model formats are supported: -- GGUF + - GGUF -You should choose a model whose size is less than your device's memory and should leave -about 2 GB. For example, if you have 16 GB of RAM in total, of which 12 GB is available, -then you should choose a model that takes up at most 10 GB of RAM. Bigger models tend to -give better generation but also take more processing time. + You should choose a model whose size is less than your device's memory and should leave + about 2 GB. For example, if you have 16 GB of RAM in total, of which 12 GB is available, + then you should choose a model that takes up at most 10 GB of RAM. Bigger models tend to + give better generation but also take more processing time. -Here are some recommendations and their size in memory: + Here are some recommendations and their size in memory: -- [Qwen1.5-1.8B-Chat-GGUF](https://huggingface.co/Qwen/Qwen1.5-1.8B-Chat-GGUF/resolve/main/qwen1_5-1_8b-chat-q8_0.gguf?download=true): - around 2 GB + - [Qwen1.5-1.8B-Chat-GGUF](https://huggingface.co/Qwen/Qwen1.5-1.8B-Chat-GGUF/resolve/main/qwen1_5-1_8b-chat-q8_0.gguf?download=true): around 2 GB -Add a new LlamaCpp model with the provided model name on the web uI. + Add a new LlamaCpp model with the provided model name on the web UI. -
+
-## Adding your own RAG pipeline +### Adding your own RAG pipeline -#### Custom reasoning pipeline +#### Custom Reasoning Pipeline -First, check the default pipeline implementation in -[here](libs/ktem/ktem/reasoning/simple.py). You can make quick adjustment to how the default QA pipeline work. +1. Check the default pipeline implementation in [here](libs/ktem/ktem/reasoning/simple.py). You can make quick adjustment to how the default QA pipeline work. +2. Add new `.py` implementation in `libs/ktem/ktem/reasoning/` and later include it in `flowssettings` to enable it on the UI. -Next, if you feel comfortable adding new pipeline, add new `.py` implementation in `libs/ktem/ktem/reasoning/` and later include it in `flowssettings` to enable it on the UI. +#### Custom Indexing Pipeline -#### Custom indexing pipeline +- Check sample implementation in `libs/ktem/ktem/index/file/graph` -Check sample implementation in `libs/ktem/ktem/index/file/graph` - -(more instruction WIP). - -## Developer guide - -Please refer to the [Developer Guide](https://cinnamon.github.io/kotaemon/development/) -for more details. +> (more instruction WIP). ## Star History @@ -337,3 +322,11 @@ for more details. Star History Chart + +## Contribution + +Since our project is actively being developed, we greatly value your feedback and contributions. Please see our [Contributing Guide](https://github.com/Cinnamon/kotaemon/blob/main/CONTRIBUTING.md) to get started. Thank you to all our contributors! + + + + diff --git a/docs/images/chat-tab.png b/docs/images/chat-tab.png index 6f21f7d..3d294b8 100644 Binary files a/docs/images/chat-tab.png and b/docs/images/chat-tab.png differ