langchain-nomic 0.0.1rc0__tar.gz → 0.1.0__tar.gz

{langchain_nomic-0.0.1rc0 → langchain_nomic-0.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-nomic
-Version: 0.0.1rc0
+Version: 0.1.0
 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.46,<0.3)
+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.1rc0 → langchain_nomic-0.1.0}/langchain_nomic/embeddings.py

@@ -3,7 +3,7 @@ from typing import List, Optional
 
 import nomic  # type: ignore
 from langchain_core.embeddings import Embeddings
-from nomic import embed
+from nomic import embed  # type: ignore
 
 
 class NomicEmbeddings(Embeddings):
@@ -22,6 +22,7 @@ class NomicEmbeddings(Embeddings):
         *,
         model: str,
         nomic_api_key: Optional[str] = None,
+        dimensionality: Optional[int] = None,
     ):
         """Initialize NomicEmbeddings model.
 
@@ -31,8 +32,10 @@ class NomicEmbeddings(Embeddings):
                 environment variable by default.
         """
         _api_key = nomic_api_key or os.environ.get("NOMIC_API_KEY")
-        nomic.login(_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.
@@ -42,7 +45,13 @@ class NomicEmbeddings(Embeddings):
             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)
+
+        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.1rc0 → langchain_nomic-0.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langchain-nomic"
-version = "0.0.1rc0"
+version = "0.1.0"
 description = "An integration package connecting Nomic and LangChain"
 authors = []
 readme = "README.md"
@@ -12,8 +12,8 @@ license = "MIT"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"
-langchain-core = ">=0.0.12"
-nomic = "^3.0.7"
+langchain-core = ">=0.1.46,<0.3"
+nomic = "^3.0.12"
 
 [tool.poetry.group.test]
 optional = true
@@ -21,11 +21,11 @@ optional = true
 [tool.poetry.group.test.dependencies]
 pytest = "^7.3.0"
 freezegun = "^1.2.2"
-pytest-mock  = "^3.10.0"
+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}
+langchain-core = { path = "../../core", develop = true }
 
 [tool.poetry.group.codespell]
 optional = true
@@ -46,28 +46,27 @@ ruff = "^0.1.5"
 
 [tool.poetry.group.typing.dependencies]
 mypy = "^0.991"
-langchain-core = {path = "../../core", develop = true}
+langchain-core = { path = "../../core", develop = true }
 
 [tool.poetry.group.dev]
 optional = true
 
 [tool.poetry.group.dev.dependencies]
-langchain-core = {path = "../../core", develop = true}
+langchain-core = { path = "../../core", develop = true }
 
-[tool.ruff]
+[tool.ruff.lint]
 select = [
-  "E",  # pycodestyle
-  "F",  # pyflakes
-  "I",  # isort
+  "E",    # pycodestyle
+  "F",    # pyflakes
+  "I",    # isort
+  "T201", # print
 ]
 
 [tool.mypy]
 disallow_untyped_defs = "True"
 
 [tool.coverage.run]
-omit = [
-    "tests/*",
-]
+omit = ["tests/*"]
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]