Skip to content

Commit

Permalink
chore(core): publish to pypi
Browse files Browse the repository at this point in the history
  • Loading branch information
jotyy committed Oct 23, 2024
1 parent b0280c4 commit b0fed10
Show file tree
Hide file tree
Showing 7 changed files with 2,016 additions and 1,684 deletions.
2 changes: 1 addition & 1 deletion .github/ISSUE_TEMPLATE/question.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ assignees:

<!-- Mark with an `x` all the checkboxes that apply (like `[x]`) -->

- [ ] I've searched the project's [`issues`](https://github.com/denser_org/denser-retriever/issues?q=is%3Aissue).
- [ ] I've searched the project's [`issues`](https://github.com/denser-org/denser-retriever/issues?q=is%3Aissue).

## ❓ Question

Expand Down
4 changes: 2 additions & 2 deletions .github/PULL_REQUEST_TEMPLATE.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,8 @@

<!-- Mark with an `x` all the checkboxes that apply (like `[x]`) -->

- [ ] I've read the [`CODE_OF_CONDUCT.md`](https://github.com/denser_org/denser-retriever/blob/main/CODE_OF_CONDUCT.md) document.
- [ ] I've read the [`CONTRIBUTING.md`](https://github.com/denser_org/denser-retriever/blob/main/CONTRIBUTING.md) guide.
- [ ] I've read the [`CODE_OF_CONDUCT.md`](https://github.com/denser-org/denser-retriever/blob/main/CODE_OF_CONDUCT.md) document.
- [ ] I've read the [`CONTRIBUTING.md`](https://github.com/denser-org/denser-retriever/blob/main/CONTRIBUTING.md) guide.
- [ ] I've updated the code style using `make codestyle`.
- [ ] I've written tests for all new methods and classes that I created.
- [ ] I've written the docstring in Google format for all the methods and classes that I used.
2 changes: 1 addition & 1 deletion cookiecutter-config-file.yml
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ default_context:
organization: "denser-org"
license: "MIT"
minimal_python_version: 3.8
github_name: "denser_org"
github_name: "denser-org"
email: "[email protected]"
version: "0.1.0"
line_length: "120"
Expand Down
24 changes: 24 additions & 0 deletions denser_retriever/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,16 @@
"""Enterprise-grade AI retriever solution that seamlessly integrates to enhance your AI applications."""

import sys
from .embeddings import (
DenserEmbeddings,
SentenceTransformerEmbeddings,
VoyageAPIEmbeddings,
)
from .keyword import DenserKeywordSearch, ElasticKeywordSearch
from .reranker import DenserReranker, HFReranker, CohereReranker
from .retriever import DenserRetriever
from .vectordb.base import DenserVectorDB
from .vectordb.milvus import MilvusDenserVectorDB

if sys.version_info >= (3, 8):
from importlib import metadata as importlib_metadata
Expand All @@ -17,3 +27,17 @@ def get_version() -> str:


version: str = get_version()

__all__ = [
"DenserEmbeddings",
"SentenceTransformerEmbeddings",
"VoyageAPIEmbeddings",
"DenserKeywordSearch",
"ElasticKeywordSearch",
"DenserReranker",
"HFReranker",
"CohereReranker",
"DenserRetriever",
"DenserVectorDB",
"MilvusDenserVectorDB",
]
4 changes: 4 additions & 0 deletions denser_retriever/vectordb/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
from .base import DenserVectorDB
from .milvus import MilvusDenserVectorDB

__all__ = ["DenserVectorDB", "MilvusDenserVectorDB"]
3,653 changes: 1,978 additions & 1,675 deletions poetry.lock

Large diffs are not rendered by default.

11 changes: 6 additions & 5 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -5,13 +5,13 @@ build-backend = "poetry.core.masonry.api"

[tool.poetry]
name = "denser-retriever"
version = "0.1.0"
version = "0.1.2"
description = "Enterprise-grade AI retriever solution that seamlessly integrates to enhance your AI applications."
readme = "README.md"
authors = ["denser-org <[email protected]>"]
license = "MIT"
repository = "https://github.com/denser_org/denser-retriever"
homepage = "https://github.com/denser_org/denser-retriever"
repository = "https://github.com/denser-org/denser-retriever"
homepage = "https://github.com/denser-org/denser-retriever"
packages = [{include = "denser_retriever"}]

# Keywords description https://python-poetry.org/docs/pyproject/#keywords
Expand Down Expand Up @@ -42,8 +42,9 @@ rich = "^10.14.0"
pytrec-eval = "^0.5"

sentence-transformers = "^2.7.0" # Specify the version of sentence-transformers
torch = [{markers = "sys_platform == 'darwin'", url = "https://download.pytorch.org/whl/cpu/torch-1.13.1-cp310-none-macosx_11_0_arm64.whl"},
{markers = "sys_platform == 'linux'", url="https://download.pytorch.org/whl/cu117/torch-1.13.1%2Bcu117-cp310-cp310-linux_x86_64.whl"}]
# torch = [{markers = "sys_platform == 'darwin'", url = "https://download.pytorch.org/whl/cpu/torch-1.13.1-cp310-none-macosx_11_0_arm64.whl"},
# {markers = "sys_platform == 'linux'", url="https://download.pytorch.org/whl/cu117/torch-1.13.1%2Bcu117-cp310-cp310-linux_x86_64.whl"}]
torch = "^1.13.1"
elasticsearch = "^8.13.0"
pymilvus = "^2.4.4"
datasets = "^2.18.0"
Expand Down

0 comments on commit b0fed10

Please sign in to comment.