From 38ed55245f2fd0f57cb6c4690200f84b10ffda01 Mon Sep 17 00:00:00 2001 From: Leonid Kuligin Date: Tue, 17 Oct 2023 01:55:45 +0200 Subject: [PATCH] added Vertex examples as attributes (#11890) - **Description:** added examples to Vertex chat models as optional class attributes, so that a model with examples can be used inside a chain - **Twitter handle:** lkuligin --- libs/langchain/langchain/chat_models/vertexai.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/libs/langchain/langchain/chat_models/vertexai.py b/libs/langchain/langchain/chat_models/vertexai.py index 42475bfe96..ef23b88e71 100644 --- a/libs/langchain/langchain/chat_models/vertexai.py +++ b/libs/langchain/langchain/chat_models/vertexai.py @@ -120,6 +120,7 @@ class ChatVertexAI(_VertexAICommon, BaseChatModel): model_name: str = "chat-bison" "Underlying model name." + examples: Optional[List[BaseMessage]] = None @classmethod def is_lc_serializable(self) -> bool: @@ -175,7 +176,7 @@ class ChatVertexAI(_VertexAICommon, BaseChatModel): question = _get_question(messages) history = _parse_chat_history(messages[:-1]) params = self._prepare_params(stop=stop, stream=False, **kwargs) - examples = kwargs.get("examples", None) + examples = kwargs.get("examples") or self.examples if examples: params["examples"] = _parse_examples(examples)