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 :)
pull/11513/head^2
Holt Skinner 9 months ago committed by GitHub
parent 5fee61a207
commit f5be2d525a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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."""

Loading…
Cancel
Save