langchain-ollama 0.3.0__tar.gz → 0.3.1__tar.gz

{langchain_ollama-0.3.0 → langchain_ollama-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-ollama
-Version: 0.3.0
+Version: 0.3.1
 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: <4.0,>=3.9
 Requires-Dist: ollama<1,>=0.4.4
-Requires-Dist: langchain-core<1.0.0,>=0.3.47
+Requires-Dist: langchain-core<1.0.0,>=0.3.51
 Description-Content-Type: text/markdown
 
 # langchain-ollama

{langchain_ollama-0.3.0 → langchain_ollama-0.3.1}/langchain_ollama/chat_models.py

@@ -743,6 +743,10 @@ class ChatOllama(BaseChatModel):
                         dict(stream_resp) if stream_resp.get("done") is True else None
                     ),
                 )
+                if chunk.generation_info and (
+                    model := chunk.generation_info.get("model")
+                ):
+                    chunk.generation_info["model_name"] = model  # backwards compat
                 if self.extract_reasoning:
                     message, is_thinking = self._extract_reasoning(
                         chunk.message, is_thinking
@@ -791,6 +795,10 @@ class ChatOllama(BaseChatModel):
                         dict(stream_resp) if stream_resp.get("done") is True else None
                     ),
                 )
+                if chunk.generation_info and (
+                    model := chunk.generation_info.get("model")
+                ):
+                    chunk.generation_info["model_name"] = model  # backwards compat
                 if self.extract_reasoning:
                     message, is_thinking = self._extract_reasoning(
                         chunk.message, is_thinking

{langchain_ollama-0.3.0 → langchain_ollama-0.3.1}/langchain_ollama/llms.py

@@ -84,6 +84,11 @@ class OllamaLLM(BaseLLM):
     """The temperature of the model. Increasing the temperature will
     make the model answer more creatively. (Default: 0.8)"""
 
+    seed: Optional[int] = None
+    """Sets the random number seed to use for generation. Setting this
+    to a specific number will make the model generate the same text for
+    the same prompt."""
+
     stop: Optional[List[str]] = None
     """Sets the stop tokens to use."""
 
@@ -150,6 +155,7 @@ class OllamaLLM(BaseLLM):
                 "repeat_last_n": self.repeat_last_n,
                 "repeat_penalty": self.repeat_penalty,
                 "temperature": self.temperature,
+                "seed": self.seed,
                 "stop": self.stop if stop is None else stop,
                 "tfs_z": self.tfs_z,
                 "top_k": self.top_k,

{langchain_ollama-0.3.0 → langchain_ollama-0.3.1}/pyproject.toml

@@ -9,10 +9,10 @@ authors = []
 requires-python = "<4.0,>=3.9"
 dependencies = [
     "ollama<1,>=0.4.4",
-    "langchain-core<1.0.0,>=0.3.47",
+    "langchain-core<1.0.0,>=0.3.51",
 ]
 name = "langchain-ollama"
-version = "0.3.0"
+version = "0.3.1"
 description = "An integration package connecting Ollama and LangChain"
 readme = "README.md"