mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-16 09:48:04 +00:00
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
This commit is contained in:
parent
5019f59724
commit
38ed55245f
@ -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…
Reference in New Issue
Block a user