Enable InMemoryDocstore to be constructed without providing a dict (#6976)

- Description: Allow `InMemoryDocstore` to be created without passing a
dict to the constructor; the constructor can create a dict at runtime if
one isn't provided.
- Tag maintainer: @dev2049
This commit is contained in:
Mike Salvatore
2023-07-05 16:56:31 -04:00
committed by GitHub
parent 47e7d09dff
commit 265f05b10e
2 changed files with 12 additions and 3 deletions

View File

@@ -54,3 +54,12 @@ def test_adding_document_already_exists() -> None:
bar_output = docstore.search("foo")
assert isinstance(bar_output, Document)
assert bar_output.page_content == "bar"
def test_default_dict_value_in_constructor() -> None:
"""Test proper functioning if no _dict is provided to the constructor."""
docstore = InMemoryDocstore()
docstore.add({"foo": Document(page_content="bar")})
output = docstore.search("foo")
assert isinstance(output, Document)
assert output.page_content == "bar"