langchain-ollama 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

langchain_ollama/chat_models.py

@@ -17,7 +17,6 @@ from typing import (
 )
 from uuid import uuid4
 
-import ollama
 from langchain_core.callbacks import (
     CallbackManagerForLLMRun,
 )
@@ -36,11 +35,10 @@ from langchain_core.messages import (
 from langchain_core.messages.ai import UsageMetadata
 from langchain_core.messages.tool import tool_call
 from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, ChatResult
-from langchain_core.pydantic_v1 import BaseModel
 from langchain_core.runnables import Runnable
 from langchain_core.tools import BaseTool
 from langchain_core.utils.function_calling import convert_to_openai_tool
-from ollama import AsyncClient, Message, Options
+from ollama import AsyncClient, Client, Message, Options
 
 
 def _get_usage_metadata_from_generation_info(
@@ -292,6 +290,11 @@ class ChatOllama(BaseChatModel):
     """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."""
 
@@ -316,6 +319,9 @@ class ChatOllama(BaseChatModel):
     keep_alive: Optional[Union[int, str]] = None
     """How long the model will stay loaded into memory."""
 
+    base_url: Optional[str] = None
+    """Base url the model is hosted under."""
+
     @property
     def _default_params(self) -> Dict[str, Any]:
         """Get the default parameters for calling Ollama."""
@@ -333,6 +339,7 @@ class ChatOllama(BaseChatModel):
                 "repeat_last_n": self.repeat_last_n,
                 "repeat_penalty": self.repeat_penalty,
                 "temperature": self.temperature,
+                "seed": self.seed,
                 "stop": self.stop,
                 "tfs_z": self.tfs_z,
                 "top_k": self.top_k,
@@ -346,7 +353,7 @@ class ChatOllama(BaseChatModel):
     ) -> Sequence[Message]:
         ollama_messages: List = []
         for message in messages:
-            role = ""
+            role: Literal["user", "assistant", "system", "tool"]
             tool_call_id: Optional[str] = None
             tool_calls: Optional[List[Dict[str, Any]]] = None
             if isinstance(message, HumanMessage):
@@ -383,11 +390,13 @@ class ChatOllama(BaseChatModel):
                         image_url = None
                         temp_image_url = content_part.get("image_url")
                         if isinstance(temp_image_url, str):
-                            image_url = content_part["image_url"]
+                            image_url = temp_image_url
                         elif (
-                            isinstance(temp_image_url, dict) and "url" in temp_image_url
+                            isinstance(temp_image_url, dict)
+                            and "url" in temp_image_url
+                            and isinstance(temp_image_url["url"], str)
                         ):
-                            image_url = temp_image_url
+                            image_url = temp_image_url["url"]
                         else:
                             raise ValueError(
                                 "Only string image_url or dict with string 'url' "
@@ -408,15 +417,16 @@ class ChatOllama(BaseChatModel):
                             "Must either have type 'text' or type 'image_url' "
                             "with a string 'image_url' field."
                         )
-            msg = {
+            # Should convert to ollama.Message once role includes tool, and tool_call_id is in Message # noqa: E501
+            msg: dict = {
                 "role": role,
                 "content": content,
                 "images": images,
             }
+            if tool_calls:
+                msg["tool_calls"] = tool_calls  # type: ignore
             if tool_call_id:
                 msg["tool_call_id"] = tool_call_id
-            if tool_calls:
-                msg["tool_calls"] = tool_calls
             ollama_messages.append(msg)
 
         return ollama_messages
@@ -439,7 +449,7 @@ class ChatOllama(BaseChatModel):
 
         params["options"]["stop"] = stop
         if "tools" in kwargs:
-            yield await AsyncClient().chat(
+            yield await AsyncClient(host=self.base_url).chat(
                 model=params["model"],
                 messages=ollama_messages,
                 stream=False,
@@ -449,7 +459,7 @@ class ChatOllama(BaseChatModel):
                 tools=kwargs["tools"],
             )  # type:ignore
         else:
-            async for part in await AsyncClient().chat(
+            async for part in await AsyncClient(host=self.base_url).chat(
                 model=params["model"],
                 messages=ollama_messages,
                 stream=True,
@@ -477,7 +487,7 @@ class ChatOllama(BaseChatModel):
 
         params["options"]["stop"] = stop
         if "tools" in kwargs:
-            yield ollama.chat(
+            yield Client(host=self.base_url).chat(
                 model=params["model"],
                 messages=ollama_messages,
                 stream=False,
@@ -487,7 +497,7 @@ class ChatOllama(BaseChatModel):
                 tools=kwargs["tools"],
             )
         else:
-            yield from ollama.chat(
+            yield from Client(host=self.base_url).chat(
                 model=params["model"],
                 messages=ollama_messages,
                 stream=True,
@@ -712,8 +722,19 @@ class ChatOllama(BaseChatModel):
 
     def bind_tools(
         self,
-        tools: Sequence[Union[Dict[str, Any], Type[BaseModel], Callable, BaseTool]],
+        tools: Sequence[Union[Dict[str, Any], Type, Callable, BaseTool]],
         **kwargs: Any,
     ) -> Runnable[LanguageModelInput, BaseMessage]:
+        """Bind tool-like objects to this chat model.
+
+        Assumes model is compatible with OpenAI tool-calling API.
+
+        Args:
+            tools: A list of tool definitions to bind to this chat model.
+                Supports any tool definition handled by
+                :meth:`langchain_core.utils.function_calling.convert_to_openai_tool`.
+            kwargs: Any additional parameters are passed directly to
+                ``self.bind(**kwargs)``.
+        """  # noqa: E501
         formatted_tools = [convert_to_openai_tool(tool) for tool in tools]
         return super().bind(tools=formatted_tools, **kwargs)

langchain_ollama/embeddings.py

@@ -14,7 +14,7 @@ class OllamaEmbeddings(BaseModel, Embeddings):
 
             from langchain_ollama import OllamaEmbeddings
 
-            model = OllamaEmbeddings(model="llama3")
+            embedder = OllamaEmbeddings(model="llama3")
             embedder.embed_query("what is the place that jonathan worked at?")
     """
 
@@ -28,9 +28,7 @@ class OllamaEmbeddings(BaseModel, Embeddings):
 
     def embed_documents(self, texts: List[str]) -> List[List[float]]:
         """Embed search docs."""
-        embedded_docs = []
-        for doc in texts:
-            embedded_docs.append(list(ollama.embeddings(self.model, doc)["embedding"]))
+        embedded_docs = ollama.embed(self.model, texts)["embeddings"]
         return embedded_docs
 
     def embed_query(self, text: str) -> List[float]:
@@ -39,11 +37,7 @@ class OllamaEmbeddings(BaseModel, Embeddings):
 
     async def aembed_documents(self, texts: List[str]) -> List[List[float]]:
         """Embed search docs."""
-        embedded_docs = []
-        for doc in texts:
-            embedded_docs.append(
-                list((await AsyncClient().embeddings(self.model, doc))["embedding"])
-            )
+        embedded_docs = (await AsyncClient().embed(self.model, texts))["embeddings"]
         return embedded_docs
 
     async def aembed_query(self, text: str) -> List[float]:

langchain_ollama/llms.py

@@ -304,11 +304,7 @@ class OllamaLLM(BaseLLM):
         for stream_resp in self._create_generate_stream(prompt, stop, **kwargs):
             if not isinstance(stream_resp, str):
                 chunk = GenerationChunk(
-                    text=(
-                        stream_resp["message"]["content"]
-                        if "message" in stream_resp
-                        else ""
-                    ),
+                    text=(stream_resp.get("response", "")),
                     generation_info=(
                         dict(stream_resp) if stream_resp.get("done") is True else None
                     ),
@@ -330,11 +326,7 @@ class OllamaLLM(BaseLLM):
         async for stream_resp in self._acreate_generate_stream(prompt, stop, **kwargs):
             if not isinstance(stream_resp, str):
                 chunk = GenerationChunk(
-                    text=(
-                        stream_resp["message"]["content"]
-                        if "message" in stream_resp
-                        else ""
-                    ),
+                    text=(stream_resp.get("response", "")),
                     generation_info=(
                         dict(stream_resp) if stream_resp.get("done") is True else None
                     ),

{langchain_ollama-0.1.0.dist-info → langchain_ollama-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-ollama
-Version: 0.1.0
+Version: 0.1.1
 Summary: An integration package connecting Ollama and LangChain
 Home-page: https://github.com/langchain-ai/langchain
 License: MIT
@@ -14,6 +14,7 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: langchain-core (>=0.2.20,<0.3.0)
 Requires-Dist: ollama (>=0.3.0,<1)
 Project-URL: Repository, https://github.com/langchain-ai/langchain
+Project-URL: Release Notes, https://github.com/langchain-ai/langchain/releases?q=tag%3A%22langchain-ollama%3D%3D0%22&expanded=true
 Project-URL: Source Code, https://github.com/langchain-ai/langchain/tree/master/libs/partners/ollama
 Description-Content-Type: text/markdown
 

langchain_ollama-0.1.1.dist-info/RECORD

@@ -0,0 +1,9 @@
+langchain_ollama/__init__.py,sha256=HhQZqbCjhrbr2dC_9Dkw12pg4HPjnDXUoInROMNJKqA,518
+langchain_ollama/chat_models.py,sha256=1lRveHDQxtYz3nNgXNaPQUAuo7WTHTf6jZarenXeCuM,29642
+langchain_ollama/embeddings.py,sha256=8_vd3sZaBCHMGZFTjUtDRUWrmbXNyBZSazrAGtBfFSU,1371
+langchain_ollama/llms.py,sha256=XT8daS4gRx8_w4a3s2pMGP-G7kBMHSYcQBUq4GfAou4,11940
+langchain_ollama/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langchain_ollama-0.1.1.dist-info/LICENSE,sha256=2btS8uNUDWD_UNjw9ba6ZJt_00aUjEw9CGyK-xIHY8c,1072
+langchain_ollama-0.1.1.dist-info/METADATA,sha256=F41rq4WnaphtdkyWuZkKnxEp3gRT70n_OSptvwQtYd4,1827
+langchain_ollama-0.1.1.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+langchain_ollama-0.1.1.dist-info/RECORD,,

langchain_ollama-0.1.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-langchain_ollama/__init__.py,sha256=HhQZqbCjhrbr2dC_9Dkw12pg4HPjnDXUoInROMNJKqA,518
-langchain_ollama/chat_models.py,sha256=oQg-7aZKMueHH4zYJOjqfECBRRJkjy27YO6_kAgBMS8,28552
-langchain_ollama/embeddings.py,sha256=-xEDMMIKgSF3HTrbrllygGklOcD7jUtAam3Rujz2MvY,1540
-langchain_ollama/llms.py,sha256=CDuy7UXvt6JFeVjnzr7gERTLBHiBf8Ft-BHPxUQyVio,12204
-langchain_ollama/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langchain_ollama-0.1.0.dist-info/LICENSE,sha256=2btS8uNUDWD_UNjw9ba6ZJt_00aUjEw9CGyK-xIHY8c,1072
-langchain_ollama-0.1.0.dist-info/METADATA,sha256=_ISdyOMUzWMJ7by7tb8ZoUeuj2xvp9dzdshiVkOJ1f8,1696
-langchain_ollama-0.1.0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-langchain_ollama-0.1.0.dist-info/RECORD,,