aboutsummaryrefslogtreecommitdiff
path: root/tests/integration/memory/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/integration/memory/utils.py')
-rw-r--r--tests/integration/memory/utils.py44
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