mirror of
https://github.com/khoj-ai/khoj.git
synced 2024-11-24 07:55:07 +01:00
102 lines
4.4 KiB
Org Mode
102 lines
4.4 KiB
Org Mode
[[https://github.com/debanjum/semantic-search/actions/workflows/test.yml/badge.svg]] [[https://github.com/debanjum/semantic-search/actions/workflows/build.yml/badge.svg]]
|
|
|
|
* Semantic Search
|
|
/Allow natural language search on user content like notes, images, transactions using transformer based models/
|
|
|
|
All search is done locally. User can interface with semantic-search app via [[./src/interface/emacs/semantic-search.el][Emacs]], API or Commandline
|
|
|
|
** Setup
|
|
|
|
*** Setup using Docker
|
|
|
|
**** 1. Clone Repository
|
|
#+begin_src shell
|
|
git clone https://github.com/debanjum/semantic-search && cd semantic-search
|
|
#+end_src
|
|
|
|
**** 2. Configure
|
|
- Add Content Directories for Semantic Search to Docker-Compose
|
|
- Update [[./docker-compose.yml][docker-compose.yml]] to mount your images, org-mode notes, ledger/beancount directories
|
|
- If required, edit config settings in [[./docker_sample_config.yml][docker_sample_config.yml]].
|
|
|
|
**** 3. Run
|
|
#+begin_src shell
|
|
docker-compose up -d
|
|
#+end_src
|
|
|
|
***** Troubleshooting
|
|
- The first run will take time. Let it run, it's mostly not hung
|
|
- Symptom: Errors out with "Killed" in error message
|
|
- Fix: Increase RAM available to Docker Containers in Docker Settings
|
|
- Refer: [[https://stackoverflow.com/a/50770267][StackOverflow Solution]], [[https://docs.docker.com/desktop/mac/#resources][Configure Resources on Docker for Mac]]
|
|
- Symptom: Errors out complaining about Tensors mismatch, null etc
|
|
- Mitigation: Delete content-type > image section from docker_sample_config.yml
|
|
|
|
*** Setup on Local Machine
|
|
|
|
**** 1. Install Dependencies
|
|
1. Install Python3 [Required[
|
|
2. [[https://docs.conda.io/projects/conda/en/latest/user-guide/install/index.html][Install Conda]] [Required]
|
|
3. Install Exiftool [Optional]
|
|
#+begin_src shell
|
|
sudo apt-get -y install libimage-exiftool-perl
|
|
#+end_src
|
|
|
|
**** 2. Install Semantic Search
|
|
#+begin_src shell
|
|
git clone https://github.com/debanjum/semantic-search && cd semantic-search
|
|
conda env create -f environment.yml
|
|
conda activate semantic-search
|
|
#+end_src
|
|
|
|
**** 3. Configure
|
|
- Configure files/directories to search in ~content-type~ section of ~sample_config.yml~
|
|
- To run application on test data, update file paths containing ~/data/~ to ~tests/data/~ in ~sample_config.yml~
|
|
- Example replace ~/data/notes/*.org~ with ~tests/data/notes/*.org~
|
|
|
|
**** 4. Run
|
|
Load ML model, generate embeddings and expose API to query notes, images, transactions etc specified in config YAML
|
|
|
|
#+begin_src shell
|
|
python3 -m src.main -c=sample_config.yml -vv
|
|
#+end_src
|
|
|
|
** Use
|
|
- *Semantic Search via Emacs*
|
|
- [[https://github.com/debanjum/semantic-search/tree/master/src/interface/emacs#installation][Install]] [[./src/interface/emacs/semantic-search.el][semantic-search.el]]
|
|
- Run ~M-x semantic-search <user-query>~
|
|
|
|
- *Semantic Search via API*
|
|
- Query: ~GET~ [[http://localhost:8000/search?q=%22what%20is%20the%20meaning%20of%20life%22][http://localhost:8000/search?q="What is the meaning of life"&t=notes]]
|
|
- Regenerate Embeddings: ~GET~ [[http://localhost:8000/regenerate][http://localhost:8000/regenerate]]
|
|
- [[http://localhost:8000/docs][Semantic Search API Docs]]
|
|
|
|
- *UI to Edit Config*
|
|
- [[https://localhost:8000/ui][Config UI]]
|
|
|
|
** Upgrade
|
|
|
|
*** On Docker
|
|
#+begin_src shell
|
|
docker-compose build
|
|
#+end_src
|
|
|
|
*** On Local Machine
|
|
#+begin_src shell
|
|
cd semantic-search
|
|
git pull origin master
|
|
conda deactivate semantic-search
|
|
conda env update -f environment.yml
|
|
conda activate semantic-search
|
|
#+end_src
|
|
|
|
** Miscellaneous
|
|
- The experimental /chat API endpoint uses the [[https://openai.com/api/][OpenAI API]]
|
|
- It is disabled by default
|
|
- To use it add your ~openai-api-key~ to config.yml
|
|
|
|
** Acknowledgments
|
|
- [[https://huggingface.co/sentence-transformers/msmarco-MiniLM-L-6-v3][MiniLM Model]] for Asymmetric Text Search. See [[https://www.sbert.net/examples/applications/retrieve_rerank/README.html][SBert Documentation]]
|
|
- [[https://github.com/openai/CLIP][OpenAI CLIP Model]] for Image Search. See [[https://www.sbert.net/examples/applications/image-search/README.html][SBert Documentation]]
|
|
- Charles Cave for [[http://members.optusnet.com.au/~charles57/GTD/orgnode.html][OrgNode Parser]]
|
|
- Sven Marnach for [[https://github.com/smarnach/pyexiftool/blob/master/exiftool.py][PyExifTool]]
|