langchain-nomic 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

langchain_nomic/__init__.py

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

langchain_nomic/embeddings.py

@@ -1,9 +1,9 @@
 import os
-from typing import List, Optional
+from typing import List, Literal, Optional, overload
 
-import nomic  # type: ignore
+import nomic  # type: ignore[import]
 from langchain_core.embeddings import Embeddings
-from nomic import embed  # type: ignore
+from nomic import embed
 
 
 class NomicEmbeddings(Embeddings):
@@ -17,12 +17,50 @@ class NomicEmbeddings(Embeddings):
             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.
 
@@ -30,12 +68,23 @@ class NomicEmbeddings(Embeddings):
             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.
@@ -51,6 +100,8 @@ class NomicEmbeddings(Embeddings):
             model=self.model,
             task_type=task_type,
             dimensionality=self.dimensionality,
+            inference_mode=self.inference_mode,
+            device=self.device,
         )
         return output["embeddings"]
 
@@ -75,3 +126,9 @@ class NomicEmbeddings(Embeddings):
             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.dist-info → langchain_nomic-0.1.2.dist-info}/METADATA

@@ -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.2.dist-info/RECORD

@@ -0,0 +1,7 @@
+langchain_nomic/__init__.py,sha256=MYethT7Ss07B2-ZHi0Y61JONOTXUtp_VTFW1ES-ydKM,86
+langchain_nomic/embeddings.py,sha256=9llrozY7UJnwB9hrJxbPamC35Opn1aVbeaoHn1wHYa8,3892
+langchain_nomic/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langchain_nomic-0.1.2.dist-info/LICENSE,sha256=DppmdYJVSc1jd0aio6ptnMUn5tIHrdAhQ12SclEBfBg,1072
+langchain_nomic-0.1.2.dist-info/METADATA,sha256=2vd12iP7vrr4MgFaZdekzjThLUVg1WXMlYEeHPJuyl8,1331
+langchain_nomic-0.1.2.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+langchain_nomic-0.1.2.dist-info/RECORD,,

langchain_nomic-0.1.0.dist-info/RECORD

@@ -1,7 +0,0 @@
-langchain_nomic/__init__.py,sha256=MQEfQNlmqQ0aGXXR7fngtWlJJeC6R4zcDhcXucjqvLE,93
-langchain_nomic/embeddings.py,sha256=S312h3vwoo_zLAjGeU7RWTbDMOFY2GnS5rLY5DC2cN8,2045
-langchain_nomic/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langchain_nomic-0.1.0.dist-info/LICENSE,sha256=DppmdYJVSc1jd0aio6ptnMUn5tIHrdAhQ12SclEBfBg,1072
-langchain_nomic-0.1.0.dist-info/METADATA,sha256=mcsG_BzcXxavKkll8JfW9xKXnxwR7bViaqc6vYamTMQ,1290
-langchain_nomic-0.1.0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-langchain_nomic-0.1.0.dist-info/RECORD,,