mirror of
https://github.com/khoaliber/khoj.git
synced 2026-03-02 13:18:18 +00:00
Create script to install semantic-search as a program
This commit is contained in:
25
install.py
Normal file
25
install.py
Normal file
@@ -0,0 +1,25 @@
|
||||
#!/usr/bin/env python3
|
||||
import pathlib
|
||||
import argparse
|
||||
|
||||
if __name__ == '__main__':
|
||||
# Setup Argument Parser
|
||||
parser = argparse.ArgumentParser(description="Setup the semantic search program")
|
||||
parser.add_argument('--script-dir', '-s', default="./", type=pathlib.Path, help="The project directory. Default: Current Directory")
|
||||
parser.add_argument('--install-path', '-i', default="/usr/bin/semantic-search", type=pathlib.Path, help="The directory to install the script. Default: ./embeddings.pt")
|
||||
parser.add_argument('--model-dir', '-m', default="./", type=pathlib.Path, help="The directory to store the model in. Default: Current Directory")
|
||||
args = parser.parse_args()
|
||||
|
||||
run_script_content = f'''#!/bin/bash
|
||||
|
||||
# Arrange
|
||||
eval "$(conda shell.bash hook)"
|
||||
conda activate samvayati
|
||||
cd {args.script_dir.expanduser().absolute()}
|
||||
|
||||
# Act
|
||||
python3 asymmetric.py -j {args.model_dir.expanduser().absolute()}/notes.jsonl.gz -e {args.model_dir.expanduser().absolute()}/notes_embeddings.pt -n 5 --interactive
|
||||
'''
|
||||
|
||||
with args.install_path.open(mode='w') as run_script:
|
||||
run_script.write(run_script_content)
|
||||
Reference in New Issue
Block a user