diff --git a/libs/langchain/langchain/llms/vertexai.py b/libs/langchain/langchain/llms/vertexai.py index da85f79edb..35fa573466 100644 --- a/libs/langchain/langchain/llms/vertexai.py +++ b/libs/langchain/langchain/llms/vertexai.py @@ -122,6 +122,11 @@ class _VertexAICommon(BaseModel): return enforce_stop_tokens(text, stop) return text + @property + def _identifying_params(self) -> Dict[str, Any]: + """Get the identifying parameters.""" + return {**{"model_name": self.model_name}, **self._default_params} + @property def _llm_type(self) -> str: return "vertexai"