github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-18 18:53:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

update Weaviate to support multi tenancy (#11842)

Browse Source 
- **Description:** update Weaviate to support multi tenancy
  - **Issue:** 9956
  - **Dependencies:** 
  - **Tag maintainer:** hwchase17
  - **Twitter handle:** dsx1986_
This commit is contained in:
Simon Dai 2023-10-19 00:49:30 -07:00 committed by GitHub
parent f818ec49b8
commit ed62984cb2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/langchain/langchain/vectorstores/weaviate.py
View File

@ -209,6 +209,8 @@ class Weaviate(VectorStore):
query_obj = self._client.query.get(self._index_name, self._query_attrs) query_obj = self._client.query.get(self._index_name, self._query_attrs)
if kwargs.get("where_filter"): if kwargs.get("where_filter"):
query_obj = query_obj.with_where(kwargs.get("where_filter")) query_obj = query_obj.with_where(kwargs.get("where_filter"))
if kwargs.get("tenant"):
query_obj = query_obj.with_tenant(kwargs.get("tenant"))
if kwargs.get("additional"): if kwargs.get("additional"):
query_obj = query_obj.with_additional(kwargs.get("additional")) query_obj = query_obj.with_additional(kwargs.get("additional"))
result = query_obj.with_near_text(content).with_limit(k).do() result = query_obj.with_near_text(content).with_limit(k).do()
@ -228,6 +230,8 @@ class Weaviate(VectorStore):
query_obj = self._client.query.get(self._index_name, self._query_attrs) query_obj = self._client.query.get(self._index_name, self._query_attrs)
if kwargs.get("where_filter"): if kwargs.get("where_filter"):
query_obj = query_obj.with_where(kwargs.get("where_filter")) query_obj = query_obj.with_where(kwargs.get("where_filter"))
if kwargs.get("tenant"):
query_obj = query_obj.with_tenant(kwargs.get("tenant"))
if kwargs.get("additional"): if kwargs.get("additional"):
query_obj = query_obj.with_additional(kwargs.get("additional")) query_obj = query_obj.with_additional(kwargs.get("additional"))
result = query_obj.with_near_vector(vector).with_limit(k).do() result = query_obj.with_near_vector(vector).with_limit(k).do()
@ -304,6 +308,8 @@ class Weaviate(VectorStore):
query_obj = self._client.query.get(self._index_name, self._query_attrs) query_obj = self._client.query.get(self._index_name, self._query_attrs)
if kwargs.get("where_filter"): if kwargs.get("where_filter"):
query_obj = query_obj.with_where(kwargs.get("where_filter")) query_obj = query_obj.with_where(kwargs.get("where_filter"))
if kwargs.get("tenant"):
query_obj = query_obj.with_tenant(kwargs.get("tenant"))
results = ( results = (
query_obj.with_additional("vector") query_obj.with_additional("vector")
.with_near_vector(vector) .with_near_vector(vector)
@ -343,6 +349,8 @@ class Weaviate(VectorStore):
query_obj = self._client.query.get(self._index_name, self._query_attrs) query_obj = self._client.query.get(self._index_name, self._query_attrs)
if kwargs.get("where_filter"): if kwargs.get("where_filter"):
query_obj = query_obj.with_where(kwargs.get("where_filter")) query_obj = query_obj.with_where(kwargs.get("where_filter"))
if kwargs.get("tenant"):
query_obj = query_obj.with_tenant(kwargs.get("tenant"))
embedded_query = self._embedding.embed_query(query) embedded_query = self._embedding.embed_query(query)
if not self._by_text: if not self._by_text: