diff options
Diffstat (limited to 'autogpts/autogpt/tests/integration/memory/conftest.py')
-rw-r--r-- | autogpts/autogpt/tests/integration/memory/conftest.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/autogpts/autogpt/tests/integration/memory/conftest.py b/autogpts/autogpt/tests/integration/memory/conftest.py new file mode 100644 index 000000000..64ac651de --- /dev/null +++ b/autogpts/autogpt/tests/integration/memory/conftest.py @@ -0,0 +1,17 @@ +import pytest + +from autogpt.memory.vector.memory_item import MemoryItem +from autogpt.memory.vector.utils import Embedding + + +@pytest.fixture +def memory_item(mock_embedding: Embedding): + return MemoryItem( + raw_content="test content", + summary="test content summary", + chunks=["test content"], + chunk_summaries=["test content summary"], + e_summary=mock_embedding, + e_chunks=[mock_embedding], + metadata={}, + ) |