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
pull/11789/head
Leonid Kuligin 9 months ago committed by GitHub
parent 5019f59724
commit 38ed55245f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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)

Loading…
Cancel
Save