github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-07-15 07:00:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

RFC: ChatModel.as_str

Browse Source
This commit is contained in:
Bagatur
2024-02-12 12:45:34 -08:00
parent 42648061ad
commit 4d5af35858
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/core/langchain_core/language_models/chat_models.py
View File

@@ -37,6 +37,7 @@ from langchain_core.messages import (
convert_to_messages,
message_chunk_to_message,
)
from langchain_core.output_parsers import StrOutputParser
from langchain_core.outputs import (
ChatGeneration,
ChatGenerationChunk,
@@ -49,7 +50,7 @@ from langchain_core.pydantic_v1 import Field, root_validator
from langchain_core.runnables.config import ensure_config, run_in_executor
if TYPE_CHECKING:
from langchain_core.runnables import RunnableConfig
from langchain_core.runnables import Runnable, RunnableConfig
def _get_verbosity() -> bool:
@@ -311,6 +312,9 @@ class BaseChatModel(BaseLanguageModel[BaseMessage], ABC):
LLMResult(generations=[[generation]]),
)
def as_str(self) -> Runnable[LanguageModelInput, str]:
return self | StrOutputParser()
# --- Custom methods ---
def _combine_llm_outputs(self, llm_outputs: List[Optional[dict]]) -> dict: