mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-13 00:16:01 +00:00
Fixing _InactiveRpcError in MatchingEngine vectorstore (#10056)
- Description: There was an issue with the MatchingEngine VectorStore, preventing from using it with a public endpoint. In the Google Cloud library there are two similar methods for private or public endpoints : `match()` and `find_neighbors()`. - Issue: Fixes #8378 - This uses the `google.cloud.aiplatform` library : https://github.com/googleapis/python-aiplatform/blob/main/google/cloud/aiplatform/matching_engine/matching_engine_index_endpoint.py
This commit is contained in:
parent
4f19ba3065
commit
21b236e5e4
@ -174,11 +174,19 @@ class MatchingEngine(VectorStore):
|
||||
logger.debug(f"Embedding query {query}.")
|
||||
embedding_query = self.embedding.embed_documents([query])
|
||||
|
||||
response = self.endpoint.match(
|
||||
deployed_index_id=self._get_index_id(),
|
||||
queries=embedding_query,
|
||||
num_neighbors=k,
|
||||
)
|
||||
# If the endpoint is public we use the find_neighbors function.
|
||||
if self.endpoint._public_match_client:
|
||||
response = self.endpoint.find_neighbors(
|
||||
deployed_index_id=self._get_index_id(),
|
||||
queries=embedding_query,
|
||||
num_neighbors=k,
|
||||
)
|
||||
else:
|
||||
response = self.endpoint.match(
|
||||
deployed_index_id=self._get_index_id(),
|
||||
queries=embedding_query,
|
||||
num_neighbors=k,
|
||||
)
|
||||
|
||||
if len(response) == 0:
|
||||
return []
|
||||
|
Loading…
Reference in New Issue
Block a user