From f5be2d525ab734b5b5cff1e3c8717ebb5e337611 Mon Sep 17 00:00:00 2001
From: Holt Skinner <13262395+holtskinner@users.noreply.github.com>
Date: Fri, 20 Oct 2023 14:16:42 -0500
Subject: [PATCH] fix: Add `_serving_config` property to
 `GoogleVertexAISearchRetriever` (#12084)

- Fixes error:

```
ValueError: "GoogleVertexAISearchRetriever" object has no field "_serving_config"
```

Introduced in #11736

@baskaryan, @eyurtsev, @hwchase17 if you could review and merge quickly,
that would be appreciated :)
---
 .../langchain/retrievers/google_vertex_ai_search.py          | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/libs/langchain/langchain/retrievers/google_vertex_ai_search.py b/libs/langchain/langchain/retrievers/google_vertex_ai_search.py
index ddb0e5006f..e3b1ebfc7b 100644
--- a/libs/langchain/langchain/retrievers/google_vertex_ai_search.py
+++ b/libs/langchain/langchain/retrievers/google_vertex_ai_search.py
@@ -38,9 +38,6 @@ class _BaseGoogleVertexAISearchRetriever(BaseModel):
     2 - Website data (with Advanced Website Indexing)
     """
 
-    _serving_config: str
-    """Full path of serving config."""
-
     @root_validator(pre=True)
     def validate_environment(cls, values: Dict) -> Dict:
         """Validates the environment."""
@@ -239,6 +236,7 @@ class GoogleVertexAISearchRetriever(BaseRetriever, _BaseGoogleVertexAISearchRetr
     """
 
     _client: SearchServiceClient
+    _serving_config: str
 
     class Config:
         """Configuration for this pydantic object."""
@@ -373,6 +371,7 @@ class GoogleVertexAIMultiTurnSearchRetriever(
     """Vertex AI Search Conversation ID."""
 
     _client: ConversationalSearchServiceClient
+    _serving_config: str
 
     class Config:
         """Configuration for this pydantic object."""