langchain-core 1.0.0a6__py3-none-any.whl → 1.0.4__py3-none-any.whl

langchain_core/language_models/fake.py

@@ -3,7 +3,7 @@
 import asyncio
 import time
 from collections.abc import AsyncIterator, Iterator, Mapping
-from typing import Any, Optional
+from typing import Any
 
 from typing_extensions import override
 
@@ -23,7 +23,7 @@ class FakeListLLM(LLM):
     """List of responses to return in order."""
     # This parameter should be removed from FakeListLLM since
     # it's only used by sub-classes.
-    sleep: Optional[float] = None
+    sleep: float | None = None
     """Sleep time in seconds between responses.
 
     Ignored by FakeListLLM, but used by sub-classes.
@@ -44,8 +44,8 @@ class FakeListLLM(LLM):
     def _call(
         self,
         prompt: str,
-        stop: Optional[list[str]] = None,
-        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        stop: list[str] | None = None,
+        run_manager: CallbackManagerForLLMRun | None = None,
         **kwargs: Any,
     ) -> str:
         """Return next response."""
@@ -60,8 +60,8 @@ class FakeListLLM(LLM):
     async def _acall(
         self,
         prompt: str,
-        stop: Optional[list[str]] = None,
-        run_manager: Optional[AsyncCallbackManagerForLLMRun] = None,
+        stop: list[str] | None = None,
+        run_manager: AsyncCallbackManagerForLLMRun | None = None,
         **kwargs: Any,
     ) -> str:
         """Return next response."""
@@ -91,16 +91,16 @@ class FakeStreamingListLLM(FakeListLLM):
     chunks in a streaming implementation.
     """
 
-    error_on_chunk_number: Optional[int] = None
+    error_on_chunk_number: int | None = None
     """If set, will raise an exception on the specified chunk number."""
 
     @override
     def stream(
         self,
         input: LanguageModelInput,
-        config: Optional[RunnableConfig] = None,
+        config: RunnableConfig | None = None,
         *,
-        stop: Optional[list[str]] = None,
+        stop: list[str] | None = None,
         **kwargs: Any,
     ) -> Iterator[str]:
         result = self.invoke(input, config)
@@ -119,9 +119,9 @@ class FakeStreamingListLLM(FakeListLLM):
     async def astream(
         self,
         input: LanguageModelInput,
-        config: Optional[RunnableConfig] = None,
+        config: RunnableConfig | None = None,
         *,
-        stop: Optional[list[str]] = None,
+        stop: list[str] | None = None,
         **kwargs: Any,
     ) -> AsyncIterator[str]:
         result = await self.ainvoke(input, config)

langchain_core/language_models/fake_chat_models.py

@@ -1,10 +1,10 @@
-"""Fake ChatModel for testing purposes."""
+"""Fake chat models for testing purposes."""
 
 import asyncio
 import re
 import time
 from collections.abc import AsyncIterator, Iterator
-from typing import Any, Literal, Optional, Union, cast
+from typing import Any, Literal, cast
 
 from typing_extensions import override
 
@@ -19,11 +19,11 @@ from langchain_core.runnables import RunnableConfig
 
 
 class FakeMessagesListChatModel(BaseChatModel):
-    """Fake ``ChatModel`` for testing purposes."""
+    """Fake chat model for testing purposes."""
 
     responses: list[BaseMessage]
     """List of responses to **cycle** through in order."""
-    sleep: Optional[float] = None
+    sleep: float | None = None
     """Sleep time in seconds between responses."""
     i: int = 0
     """Internally incremented after every model invocation."""
@@ -32,8 +32,8 @@ class FakeMessagesListChatModel(BaseChatModel):
     def _generate(
         self,
         messages: list[BaseMessage],
-        stop: Optional[list[str]] = None,
-        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        stop: list[str] | None = None,
+        run_manager: CallbackManagerForLLMRun | None = None,
         **kwargs: Any,
     ) -> ChatResult:
         if self.sleep is not None:
@@ -57,14 +57,14 @@ class FakeListChatModelError(Exception):
 
 
 class FakeListChatModel(SimpleChatModel):
-    """Fake ChatModel for testing purposes."""
+    """Fake chat model for testing purposes."""
 
     responses: list[str]
     """List of responses to **cycle** through in order."""
-    sleep: Optional[float] = None
+    sleep: float | None = None
     i: int = 0
     """Internally incremented after every model invocation."""
-    error_on_chunk_number: Optional[int] = None
+    error_on_chunk_number: int | None = None
     """If set, raise an error on the specified chunk number during streaming."""
 
     @property
@@ -95,8 +95,8 @@ class FakeListChatModel(SimpleChatModel):
     def _stream(
         self,
         messages: list[BaseMessage],
-        stop: Union[list[str], None] = None,
-        run_manager: Union[CallbackManagerForLLMRun, None] = None,
+        stop: list[str] | None = None,
+        run_manager: CallbackManagerForLLMRun | None = None,
         **kwargs: Any,
     ) -> Iterator[ChatGenerationChunk]:
         response = self.responses[self.i]
@@ -113,7 +113,7 @@ class FakeListChatModel(SimpleChatModel):
             ):
                 raise FakeListChatModelError
 
-            chunk_position: Optional[Literal["last"]] = (
+            chunk_position: Literal["last"] | None = (
                 "last" if i_c == len(response) - 1 else None
             )
             yield ChatGenerationChunk(
@@ -124,8 +124,8 @@ class FakeListChatModel(SimpleChatModel):
     async def _astream(
         self,
         messages: list[BaseMessage],
-        stop: Union[list[str], None] = None,
-        run_manager: Union[AsyncCallbackManagerForLLMRun, None] = None,
+        stop: list[str] | None = None,
+        run_manager: AsyncCallbackManagerForLLMRun | None = None,
         **kwargs: Any,
     ) -> AsyncIterator[ChatGenerationChunk]:
         response = self.responses[self.i]
@@ -141,7 +141,7 @@ class FakeListChatModel(SimpleChatModel):
                 and i_c == self.error_on_chunk_number
             ):
                 raise FakeListChatModelError
-            chunk_position: Optional[Literal["last"]] = (
+            chunk_position: Literal["last"] | None = (
                 "last" if i_c == len(response) - 1 else None
             )
             yield ChatGenerationChunk(
@@ -158,27 +158,33 @@ class FakeListChatModel(SimpleChatModel):
     def batch(
         self,
         inputs: list[Any],
-        config: Optional[Union[RunnableConfig, list[RunnableConfig]]] = None,
+        config: RunnableConfig | list[RunnableConfig] | None = None,
         *,
         return_exceptions: bool = False,
         **kwargs: Any,
     ) -> list[AIMessage]:
         if isinstance(config, list):
-            return [self.invoke(m, c, **kwargs) for m, c in zip(inputs, config)]
+            return [
+                self.invoke(m, c, **kwargs)
+                for m, c in zip(inputs, config, strict=False)
+            ]
         return [self.invoke(m, config, **kwargs) for m in inputs]
 
     @override
     async def abatch(
         self,
         inputs: list[Any],
-        config: Optional[Union[RunnableConfig, list[RunnableConfig]]] = None,
+        config: RunnableConfig | list[RunnableConfig] | None = None,
         *,
         return_exceptions: bool = False,
         **kwargs: Any,
     ) -> list[AIMessage]:
         if isinstance(config, list):
             # do Not use an async iterator here because need explicit ordering
-            return [await self.ainvoke(m, c, **kwargs) for m, c in zip(inputs, config)]
+            return [
+                await self.ainvoke(m, c, **kwargs)
+                for m, c in zip(inputs, config, strict=False)
+            ]
         # do Not use an async iterator here because need explicit ordering
         return [await self.ainvoke(m, config, **kwargs) for m in inputs]
 
@@ -190,8 +196,8 @@ class FakeChatModel(SimpleChatModel):
     def _call(
         self,
         messages: list[BaseMessage],
-        stop: Optional[list[str]] = None,
-        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        stop: list[str] | None = None,
+        run_manager: CallbackManagerForLLMRun | None = None,
         **kwargs: Any,
     ) -> str:
         return "fake response"
@@ -200,8 +206,8 @@ class FakeChatModel(SimpleChatModel):
     async def _agenerate(
         self,
         messages: list[BaseMessage],
-        stop: Optional[list[str]] = None,
-        run_manager: Optional[AsyncCallbackManagerForLLMRun] = None,
+        stop: list[str] | None = None,
+        run_manager: AsyncCallbackManagerForLLMRun | None = None,
         **kwargs: Any,
     ) -> ChatResult:
         output_str = "fake response"
@@ -222,23 +228,23 @@ class GenericFakeChatModel(BaseChatModel):
     """Generic fake chat model that can be used to test the chat model interface.
 
     * Chat model should be usable in both sync and async tests
-    * Invokes ``on_llm_new_token`` to allow for testing of callback related code for new
-      tokens.
+    * Invokes `on_llm_new_token` to allow for testing of callback related code for new
+        tokens.
     * Includes logic to break messages into message chunk to facilitate testing of
-      streaming.
+        streaming.
 
     """
 
-    messages: Iterator[Union[AIMessage, str]]
+    messages: Iterator[AIMessage | str]
     """Get an iterator over messages.
 
     This can be expanded to accept other types like Callables / dicts / strings
     to make the interface more generic if needed.
 
-    .. note::
-        if you want to pass a list, you can use ``iter`` to convert it to an iterator.
+    !!! note
+        if you want to pass a list, you can use `iter` to convert it to an iterator.
 
-    .. warning::
+    !!! warning
         Streaming is not implemented yet. We should try to implement it in the future by
         delegating to invoke and then breaking the resulting output into message chunks.
 
@@ -248,8 +254,8 @@ class GenericFakeChatModel(BaseChatModel):
     def _generate(
         self,
         messages: list[BaseMessage],
-        stop: Optional[list[str]] = None,
-        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        stop: list[str] | None = None,
+        run_manager: CallbackManagerForLLMRun | None = None,
         **kwargs: Any,
     ) -> ChatResult:
         message = next(self.messages)
@@ -260,8 +266,8 @@ class GenericFakeChatModel(BaseChatModel):
     def _stream(
         self,
         messages: list[BaseMessage],
-        stop: Optional[list[str]] = None,
-        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        stop: list[str] | None = None,
+        run_manager: CallbackManagerForLLMRun | None = None,
         **kwargs: Any,
     ) -> Iterator[ChatGenerationChunk]:
         chat_result = self._generate(
@@ -376,8 +382,8 @@ class ParrotFakeChatModel(BaseChatModel):
     def _generate(
         self,
         messages: list[BaseMessage],
-        stop: Optional[list[str]] = None,
-        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        stop: list[str] | None = None,
+        run_manager: CallbackManagerForLLMRun | None = None,
         **kwargs: Any,
     ) -> ChatResult:
         return ChatResult(generations=[ChatGeneration(message=messages[-1])])