├── .github └── workflows │ └── test.yml ├── .gitignore ├── CHANGELOG.md ├── LICENSE ├── MANIFEST.in ├── Makefile ├── README.md ├── benchmark_results.png ├── embeddb ├── __init__.py ├── embeddb.py └── tests │ ├── __init__.py │ ├── test_embeddb.py │ └── test_embeddings.py ├── embedding_models_cached └── sentence-transformers_all-MiniLM-L6-v2 │ ├── 1_Pooling │ └── config.json │ ├── README.md │ ├── config.json │ ├── config_sentence_transformers.json │ ├── model.safetensors │ ├── modules.json │ ├── sentence_bert_config.json │ ├── special_tokens_map.json │ ├── tokenizer.json │ ├── tokenizer_config.json │ └── vocab.txt ├── example_db.json ├── examples ├── basic_usage.py ├── benchmark.py ├── benchmark_requirements.txt ├── text_embedding.py └── verify_commands.py ├── pytest.ini ├── requirements-dev.txt ├── setup.py └── text_db.json /.github/workflows/test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/.github/workflows/test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/README.md -------------------------------------------------------------------------------- /benchmark_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/benchmark_results.png -------------------------------------------------------------------------------- /embeddb/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embeddb/__init__.py -------------------------------------------------------------------------------- /embeddb/embeddb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embeddb/embeddb.py -------------------------------------------------------------------------------- /embeddb/tests/__init__.py: -------------------------------------------------------------------------------- 1 | """ 2 | Test package for EmbedDB. 3 | """ -------------------------------------------------------------------------------- /embeddb/tests/test_embeddb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embeddb/tests/test_embeddb.py -------------------------------------------------------------------------------- /embeddb/tests/test_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embeddb/tests/test_embeddings.py -------------------------------------------------------------------------------- /embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/1_Pooling/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/1_Pooling/config.json -------------------------------------------------------------------------------- /embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/README.md -------------------------------------------------------------------------------- /embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/config.json -------------------------------------------------------------------------------- /embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/config_sentence_transformers.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/config_sentence_transformers.json -------------------------------------------------------------------------------- /embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/model.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/model.safetensors -------------------------------------------------------------------------------- /embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/modules.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/modules.json -------------------------------------------------------------------------------- /embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/sentence_bert_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/sentence_bert_config.json -------------------------------------------------------------------------------- /embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/special_tokens_map.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/special_tokens_map.json -------------------------------------------------------------------------------- /embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/tokenizer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/tokenizer.json -------------------------------------------------------------------------------- /embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/tokenizer_config.json -------------------------------------------------------------------------------- /embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/embedding_models_cached/sentence-transformers_all-MiniLM-L6-v2/vocab.txt -------------------------------------------------------------------------------- /example_db.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/example_db.json -------------------------------------------------------------------------------- /examples/basic_usage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/examples/basic_usage.py -------------------------------------------------------------------------------- /examples/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/examples/benchmark.py -------------------------------------------------------------------------------- /examples/benchmark_requirements.txt: -------------------------------------------------------------------------------- 1 | psutil>=5.9.0 2 | matplotlib>=3.5.0 3 | numpy>=1.20.0 -------------------------------------------------------------------------------- /examples/text_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/examples/text_embedding.py -------------------------------------------------------------------------------- /examples/verify_commands.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/examples/verify_commands.py -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/pytest.ini -------------------------------------------------------------------------------- /requirements-dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/requirements-dev.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/setup.py -------------------------------------------------------------------------------- /text_db.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayushgupta4897/embedDB/HEAD/text_db.json --------------------------------------------------------------------------------