Files
Auto-GPT/tests/integration/memory/conftest.py
merwanehamadi ee9f10a8d8 remove unused imports automatically (#4449)
* remove unused imports automatically

* add linters to pr template

* remove useless try statement
2023-05-28 05:50:50 -07:00

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={},
)