added Vertex examples as attributes

langchain-ai · Oct 16, 2023 · e0774c9 · e0774c9
1 parent e366427
commit e0774c9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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 @@ def _generate(
         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)