langchain-nomic 0.1.4__tar.gz → 0.1.5__tar.gz

{langchain_nomic-0.1.4 → langchain_nomic-0.1.5}/PKG-INFO

@@ -1,22 +1,15 @@
 Metadata-Version: 2.1
 Name: langchain-nomic
-Version: 0.1.4
+Version: 0.1.5
 Summary: An integration package connecting Nomic and LangChain
-Home-page: https://github.com/langchain-ai/langchain
 License: MIT
-Requires-Python: >=3.9,<4.0
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: langchain-core (>=0.2.43,<0.4.0,!=0.3.0,!=0.3.1,!=0.3.2,!=0.3.3,!=0.3.4,!=0.3.5,!=0.3.6,!=0.3.7,!=0.3.8,!=0.3.9,!=0.3.10,!=0.3.11,!=0.3.12,!=0.3.13,!=0.3.14)
-Requires-Dist: nomic (>=3.1.2,<4.0.0)
-Requires-Dist: pillow (>=10.3.0,<11.0.0)
-Project-URL: Repository, https://github.com/langchain-ai/langchain
-Project-URL: Release Notes, https://github.com/langchain-ai/langchain/releases?q=tag%3A%22langchain-nomic%3D%3D0%22&expanded=true
 Project-URL: Source Code, https://github.com/langchain-ai/langchain/tree/master/libs/partners/nomic
+Project-URL: Release Notes, https://github.com/langchain-ai/langchain/releases?q=tag%3A%22langchain-nomic%3D%3D0%22&expanded=true
+Project-URL: repository, https://github.com/langchain-ai/langchain
+Requires-Python: >=3.9
+Requires-Dist: langchain-core>=0.3.76
+Requires-Dist: nomic<4.0.0,>=3.5.3
+Requires-Dist: pillow<11.0.0,>=10.3.0
 Description-Content-Type: text/markdown
 
 # langchain-nomic

{langchain_nomic-0.1.4 → langchain_nomic-0.1.5}/README.md

@@ -20,4 +20,4 @@ And you should configure credentials by setting the following environment variab
 from langchain_nomic import NomicEmbeddings
 
 embeddings = NomicEmbeddings()
-embeddings.embed_query("What is the meaning of life?")
+embeddings.embed_query("What is the meaning of life?")

{langchain_nomic-0.1.4 → langchain_nomic-0.1.5}/langchain_nomic/embeddings.py

@@ -1,5 +1,7 @@
+from __future__ import annotations
+
 import os
-from typing import List, Literal, Optional, overload
+from typing import Literal, Optional, overload
 
 import nomic  # type: ignore[import]
 from langchain_core.embeddings import Embeddings
@@ -10,11 +12,13 @@ class NomicEmbeddings(Embeddings):
     """NomicEmbeddings embedding model.
 
     Example:
+
         .. code-block:: python
 
             from langchain_nomic import NomicEmbeddings
 
             model = NomicEmbeddings()
+
     """
 
     @overload
@@ -25,8 +29,7 @@ class NomicEmbeddings(Embeddings):
         nomic_api_key: Optional[str] = ...,
         dimensionality: Optional[int] = ...,
         inference_mode: Literal["remote"] = ...,
-    ):
-        ...
+    ): ...
 
     @overload
     def __init__(
@@ -37,8 +40,7 @@ class NomicEmbeddings(Embeddings):
         dimensionality: Optional[int] = ...,
         inference_mode: Literal["local", "dynamic"],
         device: Optional[str] = ...,
-    ):
-        ...
+    ): ...
 
     @overload
     def __init__(
@@ -49,8 +51,7 @@ class NomicEmbeddings(Embeddings):
         dimensionality: Optional[int] = ...,
         inference_mode: str,
         device: Optional[str] = ...,
-    ):
-        ...
+    ): ...
 
     def __init__(
         self,
@@ -66,16 +67,18 @@ class NomicEmbeddings(Embeddings):
 
         Args:
             model: model name
-            nomic_api_key: optionally, set the Nomic API key. Uses the NOMIC_API_KEY
+            nomic_api_key: optionally, set the Nomic API key. Uses the ``NOMIC_API_KEY``
                 environment variable by default.
             dimensionality: The embedding dimension, for use with Matryoshka-capable
                 models. Defaults to full-size.
-            inference_mode: How to generate embeddings. One of `remote`, `local`
-                (Embed4All), or `dynamic` (automatic). Defaults to `remote`.
+            inference_mode: How to generate embeddings. One of ``'remote'``, ``'local'``
+                (Embed4All), or ``'dynamic'`` (automatic). Defaults to ``'remote'``.
             device: The device to use for local embeddings. Choices include
-                `cpu`, `gpu`, `nvidia`, `amd`, or a specific device name. See
-                the docstring for `GPT4All.__init__` for more info. Typically
-                defaults to CPU. Do not use on macOS.
+                ``'cpu'``, ``'gpu'``, ``'nvidia'``, ``'amd'``, or a specific device
+                name. See the docstring for ``GPT4All.__init__`` for more info.
+                Typically defaults to ``'cpu'``. Do not use on macOS.
+            vision_model: The vision model to use for image embeddings.
+
         """
         _api_key = nomic_api_key or os.environ.get("NOMIC_API_KEY")
         if _api_key:
@@ -86,15 +89,15 @@ class NomicEmbeddings(Embeddings):
         self.device = device
         self.vision_model = vision_model
 
-    def embed(self, texts: List[str], *, task_type: str) -> List[List[float]]:
+    def embed(self, texts: list[str], *, task_type: str) -> list[list[float]]:
         """Embed texts.
 
         Args:
             texts: list of texts to embed
-            task_type: the task type to use when embedding. One of `search_query`,
-                `search_document`, `classification`, `clustering`
-        """
+            task_type: the task type to use when embedding. One of ``'search_query'``,
+                ``'search_document'``, ``'classification'``, ``'clustering'``
 
+        """
         output = embed.text(
             texts=texts,
             model=self.model,
@@ -105,29 +108,31 @@ class NomicEmbeddings(Embeddings):
         )
         return output["embeddings"]
 
-    def embed_documents(self, texts: List[str]) -> List[List[float]]:
+    def embed_documents(self, texts: list[str]) -> list[list[float]]:
         """Embed search docs.
 
         Args:
             texts: list of texts to embed as documents
+
         """
         return self.embed(
             texts=texts,
             task_type="search_document",
         )
 
-    def embed_query(self, text: str) -> List[float]:
+    def embed_query(self, text: str) -> list[float]:
         """Embed query text.
 
         Args:
             text: query text
+
         """
         return self.embed(
             texts=[text],
             task_type="search_query",
         )[0]
 
-    def embed_image(self, uris: List[str]) -> List[List[float]]:
+    def embed_image(self, uris: list[str]) -> list[list[float]]:
         return embed.image(
             images=uris,
             model=self.vision_model,

langchain_nomic-0.1.5/pyproject.toml

@@ -0,0 +1,143 @@
+[build-system]
+requires = [
+    "pdm-backend",
+]
+build-backend = "pdm.backend"
+
+[project]
+authors = []
+requires-python = ">=3.9"
+dependencies = [
+    "langchain-core>=0.3.76",
+    "nomic<4.0.0,>=3.5.3",
+    "pillow<11.0.0,>=10.3.0",
+]
+name = "langchain-nomic"
+version = "0.1.5"
+description = "An integration package connecting Nomic and LangChain"
+readme = "README.md"
+
+[project.license]
+text = "MIT"
+
+[project.urls]
+"Source Code" = "https://github.com/langchain-ai/langchain/tree/master/libs/partners/nomic"
+"Release Notes" = "https://github.com/langchain-ai/langchain/releases?q=tag%3A%22langchain-nomic%3D%3D0%22&expanded=true"
+repository = "https://github.com/langchain-ai/langchain"
+
+[dependency-groups]
+test = [
+    "pytest<8.0.0,>=7.3.0",
+    "freezegun<2.0.0,>=1.2.2",
+    "pytest-mock<4.0.0,>=3.10.0",
+    "syrupy<5.0.0,>=4.0.2",
+    "pytest-watcher<1.0.0,>=0.3.4",
+    "pytest-asyncio<1.0.0,>=0.21.1",
+    "langchain-core",
+]
+codespell = [
+    "codespell<3.0.0,>=2.2.0",
+]
+test_integration = []
+lint = [
+    "ruff<0.13,>=0.12.2",
+]
+typing = [
+    "mypy<1.0,>=0.991",
+    "langchain-core",
+]
+dev = [
+    "langchain-core",
+]
+
+[tool.uv.sources.langchain-core]
+path = "../../core"
+editable = true
+
+[tool.ruff]
+target-version = "py39"
+
+[tool.ruff.lint]
+select = [
+    "A",
+    "B",
+    "ASYNC",
+    "C4",
+    "COM",
+    "D",
+    "E",
+    "EM",
+    "F",
+    "FA",
+    "FBT",
+    "FLY",
+    "I",
+    "ICN",
+    "INT",
+    "ISC",
+    "PGH",
+    "PIE",
+    "PERF",
+    "PYI",
+    "Q",
+    "RET",
+    "RSE",
+    "RUF",
+    "S",
+    "SLF",
+    "SLOT",
+    "SIM",
+    "T10",
+    "T20",
+    "TID",
+    "UP",
+    "W",
+    "YTT",
+]
+ignore = [
+    "D100",
+    "D101",
+    "D102",
+    "D103",
+    "D104",
+    "D105",
+    "D107",
+    "D203",
+    "D407",
+    "COM812",
+    "ISC001",
+    "PERF203",
+    "S112",
+    "RUF012",
+    "SLF001",
+    "UP007",
+    "UP045",
+]
+unfixable = [
+    "B028",
+]
+
+[tool.ruff.lint.pydocstyle]
+convention = "google"
+
+[tool.ruff.lint.extend-per-file-ignores]
+"tests/**/*.py" = [
+    "S101",
+    "S311",
+]
+
+[tool.mypy]
+disallow_untyped_defs = "True"
+
+[tool.coverage.run]
+omit = [
+    "tests/*",
+]
+
+[tool.pytest.ini_options]
+addopts = "--snapshot-warn-unused --strict-markers --strict-config --durations=5"
+markers = [
+    "requires: mark tests as requiring a specific library",
+    "compile: mark placeholder test used to compile integration tests without running them",
+]
+asyncio_mode = "auto"

langchain_nomic-0.1.5/tests/integration_tests/test_compile.py

@@ -0,0 +1,6 @@
+import pytest
+
+
+@pytest.mark.compile
+def test_placeholder() -> None:
+    """Used for compiling integration tests without running any real tests."""

langchain_nomic-0.1.5/tests/integration_tests/test_embeddings.py

@@ -0,0 +1,29 @@
+"""Test Nomic embeddings."""
+
+from langchain_nomic.embeddings import NomicEmbeddings
+
+
+def test_langchain_nomic_embedding_documents() -> None:
+    """Test nomic embeddings."""
+    documents = ["foo bar"]
+    embedding = NomicEmbeddings(model="nomic-embed-text-v1")
+    output = embedding.embed_documents(documents)
+    assert len(output) == 1
+    assert len(output[0]) > 0
+
+
+def test_langchain_nomic_embedding_query() -> None:
+    """Test nomic embeddings."""
+    document = "foo bar"
+    embedding = NomicEmbeddings(model="nomic-embed-text-v1")
+    output = embedding.embed_query(document)
+    assert len(output) > 0
+
+
+def test_langchain_nomic_embedding_dimensionality() -> None:
+    """Test nomic embeddings."""
+    documents = ["foo bar"]
+    embedding = NomicEmbeddings(model="nomic-embed-text-v1.5", dimensionality=256)
+    output = embedding.embed_documents(documents)
+    assert len(output) == 1
+    assert len(output[0]) == 256

langchain_nomic-0.1.5/tests/unit_tests/test_embeddings.py

@@ -0,0 +1,8 @@
+"""Test embedding model integration."""
+
+from langchain_nomic.embeddings import NomicEmbeddings
+
+
+def test_initialization() -> None:
+    """Test embedding model initialization."""
+    NomicEmbeddings(model="nomic-embed-text-v1")

langchain_nomic-0.1.5/tests/unit_tests/test_imports.py

@@ -0,0 +1,9 @@
+from langchain_nomic import __all__
+
+EXPECTED_ALL = [
+    "NomicEmbeddings",
+]
+
+
+def test_all_imports() -> None:
+    assert sorted(EXPECTED_ALL) == sorted(__all__)

langchain_nomic-0.1.4/pyproject.toml

@@ -1,102 +0,0 @@
-[tool.poetry]
-name = "langchain-nomic"
-version = "0.1.4"
-description = "An integration package connecting Nomic and LangChain"
-authors = []
-readme = "README.md"
-repository = "https://github.com/langchain-ai/langchain"
-license = "MIT"
-
-[tool.poetry.urls]
-"Source Code" = "https://github.com/langchain-ai/langchain/tree/master/libs/partners/nomic"
-"Release Notes" = "https://github.com/langchain-ai/langchain/releases?q=tag%3A%22langchain-nomic%3D%3D0%22&expanded=true"
-
-[tool.poetry.dependencies]
-python = ">=3.9,<4.0"
-langchain-core = ">=0.2.43,<0.4.0,!=0.3.0,!=0.3.1,!=0.3.2,!=0.3.3,!=0.3.4,!=0.3.5,!=0.3.6,!=0.3.7,!=0.3.8,!=0.3.9,!=0.3.10,!=0.3.11,!=0.3.12,!=0.3.13,!=0.3.14"
-nomic = "^3.1.2"
-pillow = "^10.3.0"
-
-[tool.poetry.group.test]
-optional = true
-
-[tool.poetry.group.test.dependencies]
-pytest = "^7.3.0"
-freezegun = "^1.2.2"
-pytest-mock = "^3.10.0"
-syrupy = "^4.0.2"
-pytest-watcher = "^0.3.4"
-pytest-asyncio = "^0.21.1"
-langchain-core = { path = "../../core", develop = true }
-
-[[tool.poetry.group.test.dependencies.numpy]]
-version = "^1.24.0"
-python = "<3.12"
-
-[[tool.poetry.group.test.dependencies.numpy]]
-version = "^1.26.0"
-python = ">=3.12"
-
-[tool.poetry.group.codespell]
-optional = true
-
-[tool.poetry.group.codespell.dependencies]
-codespell = "^2.2.0"
-
-[tool.poetry.group.test_integration]
-optional = true
-
-[tool.poetry.group.test_integration.dependencies]
-
-[tool.poetry.group.lint]
-optional = true
-
-[tool.poetry.group.lint.dependencies]
-ruff = "^0.1.5"
-
-[tool.poetry.group.typing.dependencies]
-mypy = "^0.991"
-langchain-core = { path = "../../core", develop = true }
-
-[tool.poetry.group.dev]
-optional = true
-
-[tool.poetry.group.dev.dependencies]
-langchain-core = { path = "../../core", develop = true }
-
-[tool.ruff.lint]
-select = [
-  "E",    # pycodestyle
-  "F",    # pyflakes
-  "I",    # isort
-  "T201", # print
-]
-
-[tool.mypy]
-disallow_untyped_defs = "True"
-
-[tool.coverage.run]
-omit = ["tests/*"]
-
-[build-system]
-requires = ["poetry-core>=1.0.0"]
-build-backend = "poetry.core.masonry.api"
-
-[tool.pytest.ini_options]
-# --strict-markers will raise errors on unknown marks.
-# https://docs.pytest.org/en/7.1.x/how-to/mark.html#raising-errors-on-unknown-marks
-#
-# https://docs.pytest.org/en/7.1.x/reference/reference.html
-# --strict-config       any warnings encountered while parsing the `pytest`
-#                       section of the configuration file raise errors.
-#
-# https://github.com/tophat/syrupy
-# --snapshot-warn-unused    Prints a warning on unused snapshots rather than fail the test suite.
-addopts = "--snapshot-warn-unused --strict-markers --strict-config --durations=5"
-# Registering custom markers.
-# https://docs.pytest.org/en/7.1.x/example/markers.html#registering-markers
-markers = [
-  "requires: mark tests as requiring a specific library",
-  "compile: mark placeholder test used to compile integration tests without running them",
-]
-asyncio_mode = "auto"