langchain-nomic 0.1.0__tar.gz → 0.1.2__tar.gz

{langchain_nomic-0.1.0 → langchain_nomic-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-nomic
-Version: 0.1.0
+Version: 0.1.2
 Summary: An integration package connecting Nomic and LangChain
 Home-page: https://github.com/langchain-ai/langchain
 License: MIT
@@ -12,7 +12,8 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: langchain-core (>=0.1.46,<0.3)
-Requires-Dist: nomic (>=3.0.12,<4.0.0)
+Requires-Dist: nomic (>=3.0.29,<4.0.0)
+Requires-Dist: pillow (>=10.3.0,<11.0.0)
 Project-URL: Repository, https://github.com/langchain-ai/langchain
 Project-URL: Source Code, https://github.com/langchain-ai/langchain/tree/master/libs/partners/nomic
 Description-Content-Type: text/markdown

{langchain_nomic-0.1.0 → langchain_nomic-0.1.2}/langchain_nomic/__init__.py

@@ -1,5 +1,3 @@
 from langchain_nomic.embeddings import NomicEmbeddings
 
-__all__ = [
-    "NomicEmbeddings",
-]
+__all__ = ["NomicEmbeddings"]

langchain_nomic-0.1.2/langchain_nomic/embeddings.py

@@ -0,0 +1,134 @@
+import os
+from typing import List, Literal, Optional, overload
+
+import nomic  # type: ignore[import]
+from langchain_core.embeddings import Embeddings
+from nomic import embed
+
+
+class NomicEmbeddings(Embeddings):
+    """NomicEmbeddings embedding model.
+
+    Example:
+        .. code-block:: python
+
+            from langchain_nomic import NomicEmbeddings
+
+            model = NomicEmbeddings()
+    """
+
+    @overload
+    def __init__(
+        self,
+        *,
+        model: str,
+        nomic_api_key: Optional[str] = ...,
+        dimensionality: Optional[int] = ...,
+        inference_mode: Literal["remote"] = ...,
+    ):
+        ...
+
+    @overload
+    def __init__(
+        self,
+        *,
+        model: str,
+        nomic_api_key: Optional[str] = ...,
+        dimensionality: Optional[int] = ...,
+        inference_mode: Literal["local", "dynamic"],
+        device: Optional[str] = ...,
+    ):
+        ...
+
+    @overload
+    def __init__(
+        self,
+        *,
+        model: str,
+        nomic_api_key: Optional[str] = ...,
+        dimensionality: Optional[int] = ...,
+        inference_mode: str,
+        device: Optional[str] = ...,
+    ):
+        ...
+
+    def __init__(
+        self,
+        *,
+        model: str,
+        nomic_api_key: Optional[str] = None,
+        dimensionality: Optional[int] = None,
+        inference_mode: str = "remote",
+        device: Optional[str] = None,
+        vision_model: Optional[str] = None,
+    ):
+        """Initialize NomicEmbeddings model.
+
+        Args:
+            model: model name
+            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`.
+            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.
+        """
+        _api_key = nomic_api_key or os.environ.get("NOMIC_API_KEY")
+        if _api_key:
+            nomic.login(_api_key)
+        self.model = model
+        self.dimensionality = dimensionality
+        self.inference_mode = inference_mode
+        self.device = device
+        self.vision_model = vision_model
+
+    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`
+        """
+
+        output = embed.text(
+            texts=texts,
+            model=self.model,
+            task_type=task_type,
+            dimensionality=self.dimensionality,
+            inference_mode=self.inference_mode,
+            device=self.device,
+        )
+        return output["embeddings"]
+
+    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]:
+        """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]]:
+        return embed.image(
+            images=uris,
+            model=self.vision_model,
+        )["embeddings"]

{langchain_nomic-0.1.0 → langchain_nomic-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langchain-nomic"
-version = "0.1.0"
+version = "0.1.2"
 description = "An integration package connecting Nomic and LangChain"
 authors = []
 readme = "README.md"
@@ -13,7 +13,8 @@ license = "MIT"
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"
 langchain-core = ">=0.1.46,<0.3"
-nomic = "^3.0.12"
+nomic = "^3.0.29"
+pillow = "^10.3.0"
 
 [tool.poetry.group.test]
 optional = true

langchain_nomic-0.1.0/langchain_nomic/embeddings.py

@@ -1,77 +0,0 @@
-import os
-from typing import List, Optional
-
-import nomic  # type: ignore
-from langchain_core.embeddings import Embeddings
-from nomic import embed  # type: ignore
-
-
-class NomicEmbeddings(Embeddings):
-    """NomicEmbeddings embedding model.
-
-    Example:
-        .. code-block:: python
-
-            from langchain_nomic import NomicEmbeddings
-
-            model = NomicEmbeddings()
-    """
-
-    def __init__(
-        self,
-        *,
-        model: str,
-        nomic_api_key: Optional[str] = None,
-        dimensionality: Optional[int] = None,
-    ):
-        """Initialize NomicEmbeddings model.
-
-        Args:
-            model: model name
-            nomic_api_key: optionally, set the Nomic API key. Uses the NOMIC_API_KEY
-                environment variable by default.
-        """
-        _api_key = nomic_api_key or os.environ.get("NOMIC_API_KEY")
-        if _api_key:
-            nomic.login(_api_key)
-        self.model = model
-        self.dimensionality = dimensionality
-
-    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`
-        """
-
-        output = embed.text(
-            texts=texts,
-            model=self.model,
-            task_type=task_type,
-            dimensionality=self.dimensionality,
-        )
-        return output["embeddings"]
-
-    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]:
-        """Embed query text.
-
-        Args:
-            text: query text
-        """
-        return self.embed(
-            texts=[text],
-            task_type="search_query",
-        )[0]