mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-23 08:54:24 +01:00
* remove unused imports automatically * add linters to pr template * remove useless try statement
18 lines
462 B
Python
18 lines
462 B
Python
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={},
|
|
)
|