milvus-haystack 0.0.2__tar.gz → 0.0.4__tar.gz

milvus_haystack-0.0.4/.github/mergify.yml

@@ -0,0 +1,50 @@
+pull_request_rules:
+  - name: Add needs-dco label when DCO check failed
+    conditions:
+      - or:
+          - base=main
+          - base~=^2(\.\d+){1,2}$
+      - -status-success=DCO
+    actions:
+      label:
+        remove:
+          - dco-passed
+        add:
+          - needs-dco
+
+  - name: Add dco-passed label when DCO check passed
+    conditions:
+      - or:
+          - base=main
+          - base~=^2(\.\d+){1,2}$
+      - status-success=DCO
+    actions:
+      label:
+        remove:
+          - needs-dco
+        add:
+          - dco-passed
+
+  - name: Test passed for code changed
+    conditions:
+      - or:
+          - base=main
+          - base~=^2(\.\d+){1,2}$
+      - 'status-success=Lint and Test / Run Lint and Tests (3.9) (pull_request) '
+    actions:
+      label:
+        add:
+          - ci-passed
+
+  - name: Remove ci-passed label when test failed
+    conditions:
+      - or:
+        - base=main
+        - base~=^2(\.\d+){1,2}$
+      - or:
+        - 'status-success=Lint and Test / Run Lint and Tests (3.9) (pull_request) '
+    actions:
+      label:
+        remove:
+          - ci-passed
+

milvus_haystack-0.0.4/.github/workflows/test.yml

@@ -0,0 +1,38 @@
+name: Lint and Test
+
+on:
+  pull_request:
+    paths:
+      - "src/**"
+      - "tests/**"
+      - pyproject.toml
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
+  cancel-in-progress: ${{ github.event_name == 'pull_request' }}
+
+jobs:
+  build:
+    name: Run Lint and Tests
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ['3.9']
+    
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v3
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install Hatch
+        run: pip install --upgrade hatch
+
+      - name: Lint
+        run: hatch run lint:all
+
+      - name: Run tests 
+        run: hatch run cov

milvus_haystack-0.0.4/OWNERS

@@ -0,0 +1,10 @@
+filters:
+  ".*":
+    reviewers:
+      - jaelgu
+      - zc277584121
+      - codingjaguar
+    approvers:
+      - jaelgu
+      - zc277584121
+      - codingjaguar

{milvus_haystack-0.0.2 → milvus_haystack-0.0.4}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: milvus-haystack
-Version: 0.0.2
-Project-URL: Documentation, https://github.com/unknown/milvus-haystack#readme
-Project-URL: Issues, https://github.com/unknown/milvus-haystack/issues
-Project-URL: Source, https://github.com/unknown/milvus-haystack
+Version: 0.0.4
+Project-URL: Documentation, https://github.com/milvus-io/milvus-haystack#readme
+Project-URL: Issues, https://github.com/milvus-io/milvus-haystack/issues
+Project-URL: Source, https://github.com/milvus-io/milvus-haystack
 Author-email: Zilliz <support@zilliz.com>
 License-Expression: Apache-2.0
 License-File: LICENSE.txt
@@ -17,13 +17,13 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: >=3.7
-Requires-Dist: farm-haystack
+Requires-Dist: haystack-ai
+Requires-Dist: milvus
 Requires-Dist: pymilvus
-Provides-Extra: dev
-Requires-Dist: pytest; extra == 'dev'
+Requires-Dist: typing-extensions
 Description-Content-Type: text/markdown
 
-# milvus-documentstore
+# Milvus Document Store for Haystack
 
 
 ## Installation
@@ -41,11 +41,16 @@ Then, to use the `MilvusDocumentStore` in a Haystack pipeline"
 from haystack import Document
 from milvus_haystack import MilvusDocumentStore
 
-ds = MilvusDocumentStore()
-ds.write_documents([Document("Some Content")])
-ds.get_all_documents()  # prints [<Document: {'content': 'foo', 'content_type': 'text', ...>]
+document_store = MilvusDocumentStore()
+documents = [Document(
+    content="A Foo Document",
+    meta={"page": "100", "chapter": "intro"},
+    embedding=[-10.0] * 128,
+)]
+document_store.write_documents(documents)
+document_store.count_documents()  # 1
 ```
 
 ## License
 
-`milvus-documentstore` is distributed under the terms of the [Apache-2.0](https://spdx.org/licenses/Apache-2.0.html) license.
+`milvus-haystack` is distributed under the terms of the [Apache-2.0](https://spdx.org/licenses/Apache-2.0.html) license.

{milvus_haystack-0.0.2 → milvus_haystack-0.0.4}/README.md

@@ -1,4 +1,4 @@
-# milvus-documentstore
+# Milvus Document Store for Haystack
 
 
 ## Installation
@@ -16,11 +16,16 @@ Then, to use the `MilvusDocumentStore` in a Haystack pipeline"
 from haystack import Document
 from milvus_haystack import MilvusDocumentStore
 
-ds = MilvusDocumentStore()
-ds.write_documents([Document("Some Content")])
-ds.get_all_documents()  # prints [<Document: {'content': 'foo', 'content_type': 'text', ...>]
+document_store = MilvusDocumentStore()
+documents = [Document(
+    content="A Foo Document",
+    meta={"page": "100", "chapter": "intro"},
+    embedding=[-10.0] * 128,
+)]
+document_store.write_documents(documents)
+document_store.count_documents()  # 1
 ```
 
 ## License
 
-`milvus-documentstore` is distributed under the terms of the [Apache-2.0](https://spdx.org/licenses/Apache-2.0.html) license.
+`milvus-haystack` is distributed under the terms of the [Apache-2.0](https://spdx.org/licenses/Apache-2.0.html) license.

milvus_haystack-0.0.4/pyproject.toml

@@ -0,0 +1,183 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "milvus-haystack"
+dynamic = ["version"]
+description = ''
+readme = "README.md"
+requires-python = ">=3.7"
+license = "Apache-2.0"
+keywords = []
+authors = [
+  { name = "Zilliz", email = "support@zilliz.com" },
+]
+classifiers = [
+  "Development Status :: 4 - Beta",
+  "Programming Language :: Python",
+  "Programming Language :: Python :: 3.7",
+  "Programming Language :: Python :: 3.8",
+  "Programming Language :: Python :: 3.9",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: Implementation :: CPython",
+  "Programming Language :: Python :: Implementation :: PyPy",
+]
+dependencies = [
+  "haystack-ai",
+  "typing_extensions",
+  "pymilvus",
+  "milvus",
+]
+
+[project.urls]
+Documentation = "https://github.com/milvus-io/milvus-haystack#readme"
+Issues = "https://github.com/milvus-io/milvus-haystack/issues"
+Source = "https://github.com/milvus-io/milvus-haystack"
+
+[tool.hatch.version]
+path = "src/milvus_haystack/__about__.py"
+
+[tool.hatch.envs.default]
+dependencies = [
+  "coverage[toml]>=6.5",
+  "pytest",
+]
+[tool.hatch.envs.default.scripts]
+test = "pytest {args:tests}"
+test-cov = "coverage run -m pytest {args:tests}"
+cov-report = [
+  "- coverage combine",
+  "coverage report",
+]
+cov = [
+  "test-cov",
+  "cov-report",
+]
+
+[[tool.hatch.envs.all.matrix]]
+python = ["3.7", "3.8", "3.9", "3.10", "3.11"]
+
+[tool.hatch.envs.lint]
+detached = true
+dependencies = [
+  "black>=23.1.0",
+  "mypy>=1.0.0",
+  "ruff>=0.0.243",
+]
+[tool.hatch.envs.lint.scripts]
+typing = "mypy --install-types --non-interactive --explicit-package-bases {args:src/ tests}"
+style = [
+  "ruff {args:.}",
+  "black --check --diff {args:.}",
+]
+fmt = [
+  "black {args:.}",
+  "ruff --fix {args:.}",
+  "style",
+]
+all = [
+  "style",
+  "typing",
+]
+
+[tool.hatch.metadata]
+allow-direct-references = true
+
+[tool.black]
+target-version = ["py37"]
+line-length = 120
+skip-string-normalization = true
+
+[tool.ruff]
+target-version = "py37"
+line-length = 120
+select = [
+  "A",
+  "ARG",
+  "B",
+  "C",
+  "DTZ",
+  "E",
+  "EM",
+  "F",
+  "FBT",
+  "I",
+  "ICN",
+  "ISC",
+  "N",
+  "PLC",
+  "PLE",
+  "PLR",
+  "PLW",
+  "Q",
+  "RUF",
+  "S",
+  "T",
+  "TID",
+  "UP",
+  "W",
+  "YTT",
+]
+ignore = [
+  # Allow non-abstract empty methods in abstract base classes
+  "B027",
+  # Allow boolean positional values in function calls, like `dict.get(... True)`
+  "FBT003",
+  # Ignore checks for possible passwords
+  "S105", "S106", "S107",
+  # Ignore complexity
+  "C901", "PLR0911", "PLR0912", "PLR0913", "PLR0915",
+]
+unfixable = [
+  # Don't touch unused imports
+  "F401",
+]
+
+[tool.ruff.isort]
+known-first-party = ["milvus_haystack"]
+
+[tool.ruff.flake8-tidy-imports]
+ban-relative-imports = "all"
+
+[tool.ruff.per-file-ignores]
+# Tests can use magic values, assertions, and relative imports
+"tests/**/*" = ["PLR2004", "S101", "TID252"]
+
+[tool.coverage.run]
+source_pkgs = ["milvus_haystack", "tests"]
+branch = true
+parallel = true
+omit = [
+  "src/milvus_haystack/__about__.py",
+]
+
+[tool.coverage.paths]
+milvus_haystack = ["src/milvus_haystack", "*/milvus-haystack/src/milvus_haystack"]
+tests = ["tests", "*/milvus-haystack/tests"]
+
+[tool.coverage.report]
+exclude_lines = [
+  "no cov",
+  "if __name__ == .__main__.:",
+  "if TYPE_CHECKING:",
+]
+
+[tool.pytest.ini_options]
+minversion = "6.0"
+markers = [
+  "unit: unit tests",
+  "integration: integration tests"
+]
+
+[[tool.mypy.overrides]]
+module = [
+  "haystack.*",
+  "milvus_haystack.*",
+  "pymilvus.*",
+  "numpy",
+  "milvus",
+  "pytest.*"
+]
+ignore_missing_imports = true

{milvus_haystack-0.0.2 → milvus_haystack-0.0.4}/src/milvus_haystack/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2023-present Tuana Celik <tuana.celik@deepset.ai>
 #
-# SPDX-License-Identifier: MIT
-__version__ = "0.0.2"
+# SPDX-License-Identifier: Apache-2.0
+__version__ = "0.0.4"

milvus_haystack-0.0.4/src/milvus_haystack/__init__.py

@@ -0,0 +1,7 @@
+# SPDX-FileCopyrightText: 2023-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+from .document_store import MilvusDocumentStore  # noqa: TID252
+from .milvus_embedding_retriever import MilvusEmbeddingRetriever  # noqa: TID252
+
+__all__ = ["MilvusDocumentStore", "MilvusEmbeddingRetriever"]