diff options
Diffstat (limited to 'tests/integration/memory/utils.py')
-rw-r--r-- | tests/integration/memory/utils.py | 44 |
1 files changed, 0 insertions, 44 deletions
diff --git a/tests/integration/memory/utils.py b/tests/integration/memory/utils.py deleted file mode 100644 index 1f791160e..000000000 --- a/tests/integration/memory/utils.py +++ /dev/null @@ -1,44 +0,0 @@ -import numpy -import pytest -from pytest_mock import MockerFixture - -import autogpt.memory.vector.memory_item as vector_memory_item -import autogpt.memory.vector.providers.base as memory_provider_base -from autogpt.config.config import Config -from autogpt.llm.providers.openai import OPEN_AI_EMBEDDING_MODELS -from autogpt.memory.vector import get_memory -from autogpt.memory.vector.utils import Embedding - - -@pytest.fixture -def embedding_dimension(config: Config): - return OPEN_AI_EMBEDDING_MODELS[config.embedding_model].embedding_dimensions - - -@pytest.fixture -def mock_embedding(embedding_dimension: int) -> Embedding: - return numpy.full((1, embedding_dimension), 0.0255, numpy.float32)[0] - - -@pytest.fixture -def mock_get_embedding(mocker: MockerFixture, embedding_dimension: int): - mocker.patch.object( - vector_memory_item, - "get_embedding", - return_value=[0.0255] * embedding_dimension, - ) - mocker.patch.object( - memory_provider_base, - "get_embedding", - return_value=[0.0255] * embedding_dimension, - ) - - -@pytest.fixture -def memory_none(agent_test_config: Config, mock_get_embedding): - was_memory_backend = agent_test_config.memory_backend - - agent_test_config.memory_backend = "no_memory" - yield get_memory(agent_test_config) - - agent_test_config.memory_backend = was_memory_backend |