wip

libs/partners/nomic/langchain_nomic/embeddings.py

@@ -1,7 +1,7 @@
 import os
 from typing import List, Optional
 
-import nomic
+import nomic  # type: ignore
 from langchain_core.embeddings import Embeddings
 
 
@@ -23,7 +23,7 @@ class NomicEmbeddings(Embeddings):
             model: model name
         """
         _api_key = nomic_api_key or os.environ.get("NOMIC_API_KEY")
-        nomic.login(api_key=_api_key)
+        nomic.login(_api_key)
         self.model = model
 
     def embed_documents(self, texts: List[str]) -> List[List[float]]:

libs/partners/nomic/tests/integration_tests/test_embeddings.py

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

libs/partners/nomic/tests/unit_tests/test_embeddings.py

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

libs/partners/nomic/tests/unit_tests/test_imports.py

@@ -1,9 +1,6 @@
 from langchain_nomic import __all__
 
 EXPECTED_ALL = [
-    "NomicLLM",
-    "ChatNomic",
-    "NomicVectorStore",
     "NomicEmbeddings",
 ]