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

fix: Add _serving_config property to GoogleVertexAISearchRetriever (#12084)

Browse Source 
- 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 :)
This commit is contained in:
Holt Skinner 2023-10-20 14:16:42 -05:00 committed by GitHub
parent 5fee61a207
commit f5be2d525a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libs/langchain/langchain/retrievers/google_vertex_ai_search.py
View File

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