redis from url (#3024)

fix_agent_callbacks
Harrison Chase 1 year ago committed by GitHub
parent 19c85aa990
commit e3cf00b88b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -328,7 +328,15 @@ class Redis(VectorStore):
},
)
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
def drop_index(
@ -375,6 +383,9 @@ class Redis(VectorStore):
cls,
embedding: Embeddings,
index_name: str,
content_key: str = "content",
metadata_key: str = "metadata",
vector_key: str = "content_vector",
**kwargs: Any,
) -> Redis:
"""Connect to an existing Redis index."""
@ -400,7 +411,15 @@ class Redis(VectorStore):
except Exception as 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:
return RedisVectorStoreRetriever(vectorstore=self, **kwargs)

Loading…
Cancel
Save