github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-01 04:29:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve weaviate vectorstore docs (#5201)

Browse Source 
# Improve weaviate vectorstore docs
This commit is contained in:
Shukri 2023-05-25 00:31:48 +08:00 committed by GitHub
parent fd866d1801
commit b00c77dc62
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 175 additions and 94 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

234
docs/modules/indexes/vectorstores/examples/weaviate.ipynb
View File

File diff suppressed because one or more lines are too long

35
langchain/vectorstores/weaviate.py
View File

@ -322,12 +322,24 @@ class Weaviate(VectorStore):
if kwargs.get("search_distance"):
content["certainty"] = kwargs.get("search_distance")
query_obj = self._client.query.get(self._index_name, self._query_attrs)
result = (
query_obj.with_near_text(content)
.with_limit(k)
.with_additional("vector")
.do()
)
if not self._by_text:
embedding = self._embedding.embed_query(query)
vector = {"vector": embedding}
result = (
query_obj.with_near_vector(vector)
.with_limit(k)
.with_additional("vector")
.do()
)
else:
result = (
query_obj.with_near_text(content)
.with_limit(k)
.with_additional("vector")
.do()
)
if "errors" in result:
raise ValueError(f"Error during query: {result['errors']}")
@ -436,6 +448,15 @@ class Weaviate(VectorStore):
batch.flush()
relevance_score_fn = kwargs.get("relevance_score_fn")
by_text: bool = kwargs.get("by_text", False)
return cls(
client, index_name, text_key, embedding=embedding, attributes=attributes
client,
index_name,
text_key,
embedding=embedding,
attributes=attributes,
relevance_score_fn=relevance_score_fn,
by_text=by_text,
)