blob: 64ac651deed2846eecd26f93fdb03d54894363c2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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={},
)
|
