github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-13 16:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixing _InactiveRpcError in MatchingEngine vectorstore (#10056)

Browse Source 
- 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:
Maxime Bourliatoux 2023-09-20 01:16:04 +02:00 committed by GitHub
parent 4f19ba3065
commit 21b236e5e4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
libs/langchain/langchain/vectorstores/matching_engine.py
View File

@ -174,11 +174,19 @@ class MatchingEngine(VectorStore):
logger.debug(f"Embedding query {query}.") logger.debug(f"Embedding query {query}.")
embedding_query = self.embedding.embed_documents([query]) embedding_query = self.embedding.embed_documents([query])
response = self.endpoint.match( # If the endpoint is public we use the find_neighbors function.
deployed_index_id=self._get_index_id(), if self.endpoint._public_match_client:
queries=embedding_query, response = self.endpoint.find_neighbors(
num_neighbors=k, 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: if len(response) == 0:
return [] return []