mirror of
https://github.com/cpacker/MemGPT.git
synced 2025-06-03 04:30:22 +00:00
d71f435fea
Co-authored-by: Mindy Long <mindy@letta.com> Co-authored-by: Sarah Wooders <sarahwooders@gmail.com> Co-authored-by: Matt Zhou <mattzh1314@gmail.com>
25 lines
500 B
Python
25 lines
500 B
Python
import logging
|
|
|
|
import pytest
|
|
|
|
from letta.settings import tool_settings
|
|
|
|
|
|
def pytest_configure(config):
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_e2b_api_key_none():
|
|
# Store the original value of e2b_api_key
|
|
original_api_key = tool_settings.e2b_api_key
|
|
|
|
# Set e2b_api_key to None
|
|
tool_settings.e2b_api_key = None
|
|
|
|
# Yield control to the test
|
|
yield
|
|
|
|
# Restore the original value of e2b_api_key
|
|
tool_settings.e2b_api_key = original_api_key
|