aboutsummaryrefslogtreecommitdiff
path: root/autogpts/autogpt/tests/integration/memory/conftest.py
diff options
context:
space:
mode:
Diffstat (limited to 'autogpts/autogpt/tests/integration/memory/conftest.py')
-rw-r--r--autogpts/autogpt/tests/integration/memory/conftest.py17
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={},
+ )