langchain-nomic 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

langchain_nomic/embeddings.py

@@ -3,6 +3,7 @@ from typing import List, Optional
 
 import nomic  # type: ignore
 from langchain_core.embeddings import Embeddings
+from nomic import embed  # type: ignore
 
 
 class NomicEmbeddings(Embeddings):
@@ -21,6 +22,7 @@ class NomicEmbeddings(Embeddings):
         *,
         model: str,
         nomic_api_key: Optional[str] = None,
+        dimensionality: Optional[int] = None,
     ):
         """Initialize NomicEmbeddings model.
 
@@ -33,6 +35,7 @@ class NomicEmbeddings(Embeddings):
         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.
@@ -42,10 +45,13 @@ class NomicEmbeddings(Embeddings):
             task_type: the task type to use when embedding. One of `search_query`,
                 `search_document`, `classification`, `clustering`
         """
-        # TODO: do this via nomic.embed when fixed in nomic sdk
-        from nomic import embed
 
-        output = embed.text(texts=texts, model=self.model, task_type=task_type)
+        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]]:

{langchain_nomic-0.0.1.dist-info → langchain_nomic-0.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-nomic
-Version: 0.0.1
+Version: 0.0.2
 Summary: An integration package connecting Nomic and LangChain
 Home-page: https://github.com/langchain-ai/langchain
 License: MIT
@@ -11,8 +11,8 @@ 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.0.12)
-Requires-Dist: nomic (>=3.0.7,<4.0.0)
+Requires-Dist: langchain-core (>=0.1,<0.2)
+Requires-Dist: nomic (>=3.0.12,<4.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.0.2.dist-info/RECORD

@@ -0,0 +1,7 @@
+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.0.2.dist-info/LICENSE,sha256=DppmdYJVSc1jd0aio6ptnMUn5tIHrdAhQ12SclEBfBg,1072
+langchain_nomic-0.0.2.dist-info/METADATA,sha256=tdZzynUWUTNBIbC2CKPJ34vZKfQSzKtfNdBiyNA7c_k,1287
+langchain_nomic-0.0.2.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+langchain_nomic-0.0.2.dist-info/RECORD,,

langchain_nomic-0.0.1.dist-info/RECORD

@@ -1,7 +0,0 @@
-langchain_nomic/__init__.py,sha256=MQEfQNlmqQ0aGXXR7fngtWlJJeC6R4zcDhcXucjqvLE,93
-langchain_nomic/embeddings.py,sha256=MA6QpeFoCS8yqE6niTvfXAbgZAOfSdnKPNTg_LsIlt0,1915
-langchain_nomic/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langchain_nomic-0.0.1.dist-info/LICENSE,sha256=DppmdYJVSc1jd0aio6ptnMUn5tIHrdAhQ12SclEBfBg,1072
-langchain_nomic-0.0.1.dist-info/METADATA,sha256=ndoYR2_4PpnNn0HmdAcPSlZMRowR1WgD-FFpfSROo-4,1284
-langchain_nomic-0.0.1.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-langchain_nomic-0.0.1.dist-info/RECORD,,