github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-18 02:33:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

added Vertex examples as attributes (#11890)

Browse Source 
- **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:
Leonid Kuligin 2023-10-17 01:55:45 +02:00 committed by GitHub
parent 5019f59724
commit 38ed55245f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
libs/langchain/langchain/chat_models/vertexai.py
View File

@ -120,6 +120,7 @@ class ChatVertexAI(_VertexAICommon, BaseChatModel):
model_name: str = "chat-bison" model_name: str = "chat-bison"
"Underlying model name." "Underlying model name."
examples: Optional[List[BaseMessage]] = None
@classmethod @classmethod
def is_lc_serializable(self) -> bool: def is_lc_serializable(self) -> bool:
@ -175,7 +176,7 @@ class ChatVertexAI(_VertexAICommon, BaseChatModel):
question = _get_question(messages) question = _get_question(messages)
history = _parse_chat_history(messages[:-1]) history = _parse_chat_history(messages[:-1])
params = self._prepare_params(stop=stop, stream=False, **kwargs) params = self._prepare_params(stop=stop, stream=False, **kwargs)
examples = kwargs.get("examples", None) examples = kwargs.get("examples") or self.examples
if examples: if examples:
params["examples"] = _parse_examples(examples) params["examples"] = _parse_examples(examples)