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

redis from url (#3024)

Browse Source
This commit is contained in:
Harrison Chase 2023-04-17 20:28:12 -07:00 committed by GitHub
parent 19c85aa990
commit e3cf00b88b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
langchain/vectorstores/redis.py
View File

@ -328,7 +328,15 @@ class Redis(VectorStore):
}, },
) )
pipeline.execute() pipeline.execute()
return cls(redis_url, index_name, embedding.embed_query) return cls(
redis_url,
index_name,
embedding.embed_query,
content_key=content_key,
metadata_key=metadata_key,
vector_key=vector_key,
**kwargs,
)
@staticmethod @staticmethod
def drop_index( def drop_index(
@ -375,6 +383,9 @@ class Redis(VectorStore):
cls, cls,
embedding: Embeddings, embedding: Embeddings,
index_name: str, index_name: str,
content_key: str = "content",
metadata_key: str = "metadata",
vector_key: str = "content_vector",
**kwargs: Any, **kwargs: Any,
) -> Redis: ) -> Redis:
"""Connect to an existing Redis index.""" """Connect to an existing Redis index."""
@ -400,7 +411,15 @@ class Redis(VectorStore):
except Exception as e: except Exception as e:
raise ValueError(f"Redis failed to connect: {e}") raise ValueError(f"Redis failed to connect: {e}")
return cls(redis_url, index_name, embedding.embed_query) return cls(
redis_url,
index_name,
embedding.embed_query,
content_key=content_key,
metadata_key=metadata_key,
vector_key=vector_key,
**kwargs,
)
def as_retriever(self, **kwargs: Any) -> BaseRetriever: def as_retriever(self, **kwargs: Any) -> BaseRetriever:
return RedisVectorStoreRetriever(vectorstore=self, **kwargs) return RedisVectorStoreRetriever(vectorstore=self, **kwargs)