langchain-ollama 0.3.4__tar.gz → 0.3.5__tar.gz

{langchain_ollama-0.3.4 → langchain_ollama-0.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-ollama
-Version: 0.3.4
+Version: 0.3.5
 Summary: An integration package connecting Ollama and LangChain
 License: MIT
 Project-URL: Source Code, https://github.com/langchain-ai/langchain/tree/master/libs/partners/ollama
@@ -8,7 +8,7 @@ Project-URL: Release Notes, https://github.com/langchain-ai/langchain/releases?q
 Project-URL: repository, https://github.com/langchain-ai/langchain
 Requires-Python: >=3.9
 Requires-Dist: ollama<1.0.0,>=0.5.1
-Requires-Dist: langchain-core<1.0.0,>=0.3.68
+Requires-Dist: langchain-core<1.0.0,>=0.3.69
 Description-Content-Type: text/markdown
 
 # langchain-ollama

{langchain_ollama-0.3.4 → langchain_ollama-0.3.5}/langchain_ollama/embeddings.py

@@ -296,7 +296,10 @@ class OllamaEmbeddings(BaseModel, Embeddings):
             raise ValueError(msg)
         return (
             await self._async_client.embed(
-                self.model, texts, keep_alive=self.keep_alive
+                self.model,
+                texts,
+                options=self._default_params,
+                keep_alive=self.keep_alive,
             )
         )["embeddings"]
 

{langchain_ollama-0.3.4 → langchain_ollama-0.3.5}/pyproject.toml

@@ -9,10 +9,10 @@ authors = []
 requires-python = ">=3.9"
 dependencies = [
     "ollama>=0.5.1,<1.0.0",
-    "langchain-core<1.0.0,>=0.3.68",
+    "langchain-core<1.0.0,>=0.3.69",
 ]
 name = "langchain-ollama"
-version = "0.3.4"
+version = "0.3.5"
 description = "An integration package connecting Ollama and LangChain"
 readme = "README.md"
 

langchain_ollama-0.3.5/tests/unit_tests/test_embeddings.py

@@ -0,0 +1,63 @@
+"""Test embedding model integration."""
+
+from typing import Any
+from unittest.mock import Mock, patch
+
+from langchain_ollama.embeddings import OllamaEmbeddings
+
+MODEL_NAME = "llama3.1"
+
+
+def test_initialization() -> None:
+    """Test embedding model initialization."""
+    OllamaEmbeddings(model=MODEL_NAME, keep_alive=1)
+
+
+@patch("langchain_ollama.embeddings.validate_model")
+def test_validate_model_on_init(mock_validate_model: Any) -> None:
+    """Test that the model is validated on initialization when requested."""
+    # Test that validate_model is called when validate_model_on_init=True
+    OllamaEmbeddings(model=MODEL_NAME, validate_model_on_init=True)
+    mock_validate_model.assert_called_once()
+    mock_validate_model.reset_mock()
+
+    # Test that validate_model is NOT called when validate_model_on_init=False
+    OllamaEmbeddings(model=MODEL_NAME, validate_model_on_init=False)
+    mock_validate_model.assert_not_called()
+
+    # Test that validate_model is NOT called by default
+    OllamaEmbeddings(model=MODEL_NAME)
+    mock_validate_model.assert_not_called()
+
+
+@patch("langchain_ollama.embeddings.Client")
+def test_embed_documents_passes_options(mock_client_class: Any) -> None:
+    """Test that embed_documents method passes options including num_gpu."""
+    # Create a mock client instance
+    mock_client = Mock()
+    mock_client_class.return_value = mock_client
+
+    # Mock the embed method response
+    mock_client.embed.return_value = {"embeddings": [[0.1, 0.2, 0.3]]}
+
+    # Create embeddings with num_gpu parameter
+    embeddings = OllamaEmbeddings(model=MODEL_NAME, num_gpu=4, temperature=0.5)
+
+    # Call embed_documents
+    result = embeddings.embed_documents(["test text"])
+
+    # Verify the result
+    assert result == [[0.1, 0.2, 0.3]]
+
+    # Check that embed was called with correct arguments
+    mock_client.embed.assert_called_once()
+    call_args = mock_client.embed.call_args
+
+    # Verify the keyword arguments
+    assert "options" in call_args.kwargs
+    assert "keep_alive" in call_args.kwargs
+
+    # Verify options contain num_gpu and temperature
+    options = call_args.kwargs["options"]
+    assert options["num_gpu"] == 4
+    assert options["temperature"] == 0.5

langchain_ollama-0.3.4/tests/unit_tests/test_embeddings.py

@@ -1,30 +0,0 @@
-"""Test embedding model integration."""
-
-from typing import Any
-from unittest.mock import patch
-
-from langchain_ollama.embeddings import OllamaEmbeddings
-
-MODEL_NAME = "llama3.1"
-
-
-def test_initialization() -> None:
-    """Test embedding model initialization."""
-    OllamaEmbeddings(model=MODEL_NAME, keep_alive=1)
-
-
-@patch("langchain_ollama.embeddings.validate_model")
-def test_validate_model_on_init(mock_validate_model: Any) -> None:
-    """Test that the model is validated on initialization when requested."""
-    # Test that validate_model is called when validate_model_on_init=True
-    OllamaEmbeddings(model=MODEL_NAME, validate_model_on_init=True)
-    mock_validate_model.assert_called_once()
-    mock_validate_model.reset_mock()
-
-    # Test that validate_model is NOT called when validate_model_on_init=False
-    OllamaEmbeddings(model=MODEL_NAME, validate_model_on_init=False)
-    mock_validate_model.assert_not_called()
-
-    # Test that validate_model is NOT called by default
-    OllamaEmbeddings(model=MODEL_NAME)
-    mock_validate_model.assert_not_called()