github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-27 06:18:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
Eugene Yurtsev
2024-09-11 13:44:54 -04:00
parent 8b7f5e1ec0
commit 6395f02303
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/core/langchain_core/outputs/chat_generation.py
View File

@@ -5,7 +5,7 @@ from typing import List, Literal, Union
from pydantic import model_validator from pydantic import model_validator
from typing_extensions import Self from typing_extensions import Self
from langchain_core.messages import BaseMessage, BaseMessageChunk from langchain_core.messages import AnyMessage, BaseMessageChunk
from langchain_core.outputs.generation import Generation from langchain_core.outputs.generation import Generation
from langchain_core.utils._merge import merge_dicts from langchain_core.utils._merge import merge_dicts
@@ -26,7 +26,7 @@ class ChatGeneration(Generation):
text: str = "" text: str = ""
"""*SHOULD NOT BE SET DIRECTLY* The text contents of the output message.""" """*SHOULD NOT BE SET DIRECTLY* The text contents of the output message."""
message: BaseMessage message: AnyMessage
"""The message output by the chat model.""" """The message output by the chat model."""
# Override type to be ChatGeneration, ignore mypy error as this is intentional # Override type to be ChatGeneration, ignore mypy error as this is intentional
type: Literal["ChatGeneration"] = "ChatGeneration" # type: ignore[assignment] type: Literal["ChatGeneration"] = "ChatGeneration" # type: ignore[assignment]