https://github.com/tensorchord/qtext
https://github.com/tensorchord/qtext
Last synced: about 1 year ago
JSON representation
- Host: GitHub
- URL: https://github.com/tensorchord/qtext
- Owner: tensorchord
- License: apache-2.0
- Created: 2024-03-22T03:37:42.000Z (over 2 years ago)
- Default Branch: main
- Last Pushed: 2024-04-11T07:28:16.000Z (about 2 years ago)
- Last Synced: 2024-04-11T08:55:40.353Z (about 2 years ago)
- Language: Python
- Size: 191 KB
- Stars: 7
- Watchers: 4
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# QText
[](https://github.com/tensorchord/qtext/actions/workflows/check.yml)
QText is a microservices framework for building the RAG pipeline, or semantic search engine on top of Postgres. It provides a simple API to add, query, and highlight the text in your existing database.
The main features include:
- Full-text search with Postgres GIN index.
- Vector and sparse search with [pgvecto.rs](https://github.com/tensorchord/pgvecto.rs)
- Reranking with cross-encoder model, cohere reranking API, or other methods.
- Semantic highlight
Besides this, qtext also provides a dashboard to visualize the vector search, sparse vector search, full text search, and reranking results.
[](https://asciinema.org/a/653540)
## Design goals
- **Simple**: easy to deploy and use.
- **Customizable**: can be integrated into your existing databases.
- **Extensible**: can be extended with new features.
## How to use
To start all the services with [docker compose](https://docs.docker.com/compose/):
```bash
docker compose -f docker/compose.yaml up -d server
```
Some of the dependent services can be opt-out:
- `emb`: used to generate embedding for query and documents
- `sparse`: used to generate sparse embedding for query and documents (this requires a HuggingFace token that signed the agreement for [prithivida/Splade_PP_en_v1](https://huggingface.co/prithivida/Splade_PP_en_v1))
- `highlight`: used to provide the semantic highlight feature
- `encoder`: rerank with cross-encoder model, you can choose other methods or other online services
For the client example, check:
- [test.py](./test.py): simple demo.
- [test_cohere_wiki.py](./test_cohere_wiki.py): a Wikipedia dataset with Cohere embedding.
## API
We provide a simple sync/async [client](./qtext/client.py). You can also refer to the OpenAPI and build your own client.
- `/api/namespace` POST: create a new namespace and configure the index
- `/api/doc` POST: add a new doc
- `/api/query` POST: query the docs
- `/api/highlight` POST: semantic highlight
- `/metrics` GET: open metrics
Check the [OpenAPI documentation](http://127.0.0.1:8000/openapi/redoc) for more information (this requires the qtext service).
## Terminal UI
We provide a simple terminal UI powered by [Textual](https://github.com/textualize/textual) for you to interact with the service.
```bash
pip install textual
# need to run the qtext service first
python tui/main.py $QTEXT_PORT
```
## Configurations
Check the [config.py](./qtext/config.py) for more detail. It will read the `$HOME/.config/qtext/config.json` if this file exists.
## Integrate to the RAG pipeline
This project has most of the components you need for the RAG except for the last LLM generation step. You can send the retrieval + reranked docs to any LLM providers to get the final result.
## Customize the table schema
> [!NOTE]
> If you already have the table in Postgres, you will be responsible for the text-indexing and vector-indexing part.
1. Define a `dataclass` that includes the **necessary** columns as class attributes
- annotate the `primary_key`, `text_index`, `vector_index`, `sparse_index` with metadata (not all of them are required, only the necessary ones)
- attributes without default value or default factory is treated as required when you add new docs
2. Implement the `to_record` and `from_record` methods to be used in the reranking stage
3. Change the `config.vector_store.schema` to the class you have defined
Check the [schema.py](/qtext/schema.py) for more details.