Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
29 commits
Select commit Hold shift + click to select a range
83137b6
feat(vector store): Add usage demo and enhance Redis integration
PingXie Feb 12, 2024
d334e7f
incorporate review feedback
PingXie Feb 12, 2024
400ebfa
added missing license header
PingXie Feb 12, 2024
3263b91
excluded the test file from license check
PingXie Feb 13, 2024
0cf5faa
reformatted source files
PingXie Feb 13, 2024
2e953f4
incorporated review feedback
PingXie Feb 13, 2024
7231b97
fixed lint errors
PingXie Feb 13, 2024
1ac23e2
fixed more lint errors
PingXie Feb 13, 2024
b349c88
removed key_prefix argument from vectorstore
PingXie Feb 13, 2024
ced4961
incorporated review feedback
PingXie Feb 13, 2024
f753433
fixed formatting errors
PingXie Feb 13, 2024
83d3207
fixed a bad merge
PingXie Feb 13, 2024
7ccbca0
suppress mypy errors for setuptools
PingXie Feb 13, 2024
145d153
trying mypy.ini
PingXie Feb 13, 2024
73d5ee3
remove setup.py and requirements
PingXie Feb 13, 2024
794dc0a
add numpy dependency
PingXie Feb 13, 2024
df017e6
Merge remote-tracking branch 'origin/staging' into vectorstore
PingXie Feb 14, 2024
eebe726
refactor(vector-store): enhance validation and initialization processes
PingXie Feb 14, 2024
4c2a303
Merge branch 'staging' into vectorstore
PingXie Feb 14, 2024
a4a2eb2
Incorporated review feedback
PingXie Feb 15, 2024
1b5deab
fixed formatting
PingXie Feb 15, 2024
16e725c
incorporated review feedback
PingXie Feb 15, 2024
e2c79cb
formatting
PingXie Feb 15, 2024
f9a6e2a
fixed bugs
PingXie Feb 15, 2024
987ffc4
fix formatting
PingXie Feb 15, 2024
e952184
added vector store unittests
PingXie Feb 15, 2024
a90d8e9
actualy added vector store tests
PingXie Feb 15, 2024
a0e9d7d
fix formatting
PingXie Feb 15, 2024
fc18664
fix: fix styles and formats
craig08 Feb 15, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion docs/vector_store.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -174,7 +174,7 @@
"outputs": [],
"source": [
"rvs = RedisVectorStore(\n",
" client=redis_client, index_name=\"my_vector_index\", embedding_service=embeddings\n",
" client=redis_client, index_name=\"my_vector_index\", embeddings=embeddings\n",
")\n",
"ids = rvs.add_texts(\n",
" texts=[d.page_content for d in docs], metadatas=[d.metadata for d in docs]\n",
Expand Down
9 changes: 8 additions & 1 deletion src/langchain_google_memorystore_redis/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,12 +13,19 @@
# limitations under the License.
from .chat_message_history import MemorystoreChatMessageHistory
from .doc_saver import MemorystoreDocumentSaver
from .vector_store import DistanceStrategy, FLATConfig, HNSWConfig, RedisVectorStore
from .vector_store import (
DistanceStrategy,
FLATConfig,
HNSWConfig,
RedisVectorStore,
VectorIndexConfig,
)

__all__ = [
"MemorystoreChatMessageHistory",
"MemorystoreDocumentSaver",
"DistanceStrategy",
"VectorIndexConfig",
"FLATConfig",
"HNSWConfig",
"RedisVectorStore",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,9 @@ def messages(self) -> List[BaseMessage]: # type: ignore
"""Retrieve all messages chronologically stored in this session."""
all_elements = self._redis.lrange(self._key, 0, -1)

assert isinstance(all_elements, list)
if not isinstance(all_elements, list):
raise TypeError("Expected a list from `lrange` but got a different type.")

loaded_messages = messages_from_dict(
[json.loads(e.decode(self._encoding)) for e in all_elements]
)
Expand Down
Loading